* issue #1250, replace {pre,post}-receive and update hooks with a single shell script that does not require custom hooks to be a sh-script * issue #1250, make script posix compilant * v23, add migration script to update {pre,post}-receive and update hooks * migration: use a more common name and rename v23 to v26 to avoid conflicts * gofmt'ed and added copyright header * fix SyncRepositoryHooks to also sync wiki repostags/v1.2.0-rc1
NewMigration("change the key_id and primary_key_id type", changeGPGKeysColumns), | NewMigration("change the key_id and primary_key_id type", changeGPGKeysColumns), | ||||
// v25 -> v26 | // v25 -> v26 | ||||
NewMigration("add show field in user openid table", addUserOpenIDShow), | NewMigration("add show field in user openid table", addUserOpenIDShow), | ||||
// v26 -> v27 | |||||
NewMigration("generate and migrate repo and wiki Git hooks", generateAndMigrateGitHookChains), | |||||
} | } | ||||
// Migrate database to current version | // Migrate database to current version |
// Copyright 2017 The Gitea Authors. All rights reserved. | |||||
// Use of this source code is governed by a MIT-style | |||||
// license that can be found in the LICENSE file. | |||||
package migrations | |||||
import ( | |||||
"crypto/md5" | |||||
"encoding/hex" | |||||
"fmt" | |||||
"io" | |||||
"io/ioutil" | |||||
"os" | |||||
"path/filepath" | |||||
"strings" | |||||
"code.gitea.io/gitea/modules/setting" | |||||
"github.com/Unknwon/com" | |||||
"github.com/go-xorm/xorm" | |||||
) | |||||
func generateAndMigrateGitHookChains(x *xorm.Engine) (err error) { | |||||
type Repository struct { | |||||
ID int64 | |||||
OwnerID int64 | |||||
Name string | |||||
} | |||||
type User struct { | |||||
ID int64 | |||||
Name string | |||||
} | |||||
var ( | |||||
hookNames = []string{"pre-receive", "update", "post-receive"} | |||||
hookTpl = fmt.Sprintf("#!/usr/bin/env %s\ndata=$(cat)\nexitcodes=\"\"\nhookname=$(basename $0)\nGIT_DIR=${GIT_DIR:-$(dirname $0)}\n\nfor hook in ${GIT_DIR}/hooks/${hookname}.d/*; do\ntest -x \"${hook}\" || continue\necho \"${data}\" | \"${hook}\"\nexitcodes=\"${exitcodes} $?\"\ndone\n\nfor i in ${exitcodes}; do\n[ ${i} -eq 0 ] || exit ${i}\ndone\n", setting.ScriptType) | |||||
) | |||||
return x.Where("id > 0").Iterate(new(Repository), | |||||
func(idx int, bean interface{}) error { | |||||
repo := bean.(*Repository) | |||||
user := new(User) | |||||
has, err := x.Where("id = ?", repo.OwnerID).Get(user) | |||||
if err != nil { | |||||
return fmt.Errorf("query owner of repository [repo_id: %d, owner_id: %d]: %v", repo.ID, repo.OwnerID, err) | |||||
} else if !has { | |||||
return nil | |||||
} | |||||
repoPaths := []string{ | |||||
filepath.Join(setting.RepoRootPath, strings.ToLower(user.Name), strings.ToLower(repo.Name)) + ".git", | |||||
filepath.Join(setting.RepoRootPath, strings.ToLower(user.Name), strings.ToLower(repo.Name)) + ".wiki.git", | |||||
} | |||||
for _, repoPath := range repoPaths { | |||||
if com.IsExist(repoPath) { | |||||
hookDir := filepath.Join(repoPath, "hooks") | |||||
for _, hookName := range hookNames { | |||||
oldHookPath := filepath.Join(hookDir, hookName) | |||||
// compare md5sums of hooks | |||||
if com.IsExist(oldHookPath) { | |||||
f, err := os.Open(oldHookPath) | |||||
if err != nil { | |||||
return fmt.Errorf("cannot open old hook file '%s': %v", oldHookPath, err) | |||||
} | |||||
defer f.Close() | |||||
h := md5.New() | |||||
if _, err := io.Copy(h, f); err != nil { | |||||
return fmt.Errorf("cannot read old hook file '%s': %v", oldHookPath, err) | |||||
} | |||||
if hex.EncodeToString(h.Sum(nil)) == "6718ef67d0834e0a7908259acd566e3f" { | |||||
return nil | |||||
} | |||||
} | |||||
if err = ioutil.WriteFile(oldHookPath, []byte(hookTpl), 0777); err != nil { | |||||
return fmt.Errorf("write old hook file '%s': %v", oldHookPath, err) | |||||
} | |||||
} | |||||
} | |||||
} | |||||
return nil | |||||
}) | |||||
} |
func createDelegateHooks(repoPath string) (err error) { | func createDelegateHooks(repoPath string) (err error) { | ||||
var ( | var ( | ||||
hookNames = []string{"pre-receive", "update", "post-receive"} | hookNames = []string{"pre-receive", "update", "post-receive"} | ||||
hookTpls = []string{ | |||||
fmt.Sprintf("#!/usr/bin/env %s\nORI_DIR=`pwd`\nSHELL_FOLDER=$(cd \"$(dirname \"$0\")\";pwd)\ncd \"$ORI_DIR\"\nfor i in `ls \"$SHELL_FOLDER/pre-receive.d\"`; do\n sh \"$SHELL_FOLDER/pre-receive.d/$i\"\ndone", setting.ScriptType), | |||||
fmt.Sprintf("#!/usr/bin/env %s\nORI_DIR=`pwd`\nSHELL_FOLDER=$(cd \"$(dirname \"$0\")\";pwd)\ncd \"$ORI_DIR\"\nfor i in `ls \"$SHELL_FOLDER/update.d\"`; do\n sh \"$SHELL_FOLDER/update.d/$i\" $1 $2 $3\ndone", setting.ScriptType), | |||||
fmt.Sprintf("#!/usr/bin/env %s\nORI_DIR=`pwd`\nSHELL_FOLDER=$(cd \"$(dirname \"$0\")\";pwd)\ncd \"$ORI_DIR\"\nfor i in `ls \"$SHELL_FOLDER/post-receive.d\"`; do\n sh \"$SHELL_FOLDER/post-receive.d/$i\"\ndone", setting.ScriptType), | |||||
} | |||||
hookTpl = fmt.Sprintf("#!/usr/bin/env %s\ndata=$(cat)\nexitcodes=\"\"\nhookname=$(basename $0)\nGIT_DIR=${GIT_DIR:-$(dirname $0)}\n\nfor hook in ${GIT_DIR}/hooks/${hookname}.d/*; do\ntest -x \"${hook}\" || continue\necho \"${data}\" | \"${hook}\"\nexitcodes=\"${exitcodes} $?\"\ndone\n\nfor i in ${exitcodes}; do\n[ ${i} -eq 0 ] || exit ${i}\ndone\n", setting.ScriptType) | |||||
giteaHookTpls = []string{ | giteaHookTpls = []string{ | ||||
fmt.Sprintf("#!/usr/bin/env %s\n\"%s\" hook --config='%s' pre-receive\n", setting.ScriptType, setting.AppPath, setting.CustomConf), | fmt.Sprintf("#!/usr/bin/env %s\n\"%s\" hook --config='%s' pre-receive\n", setting.ScriptType, setting.AppPath, setting.CustomConf), | ||||
fmt.Sprintf("#!/usr/bin/env %s\n\"%s\" hook --config='%s' update $1 $2 $3\n", setting.ScriptType, setting.AppPath, setting.CustomConf), | fmt.Sprintf("#!/usr/bin/env %s\n\"%s\" hook --config='%s' update $1 $2 $3\n", setting.ScriptType, setting.AppPath, setting.CustomConf), | ||||
} | } | ||||
// WARNING: This will override all old server-side hooks | // WARNING: This will override all old server-side hooks | ||||
if err = ioutil.WriteFile(oldHookPath, []byte(hookTpls[i]), 0777); err != nil { | |||||
if err = ioutil.WriteFile(oldHookPath, []byte(hookTpl), 0777); err != nil { | |||||
return fmt.Errorf("write old hook file '%s': %v", oldHookPath, err) | return fmt.Errorf("write old hook file '%s': %v", oldHookPath, err) | ||||
} | } | ||||
if err := createDelegateHooks(bean.(*Repository).RepoPath()); err != nil { | if err := createDelegateHooks(bean.(*Repository).RepoPath()); err != nil { | ||||
return fmt.Errorf("SyncRepositoryHook: %v", err) | return fmt.Errorf("SyncRepositoryHook: %v", err) | ||||
} | } | ||||
if bean.(*Repository).HasWiki() { | |||||
if err := createDelegateHooks(bean.(*Repository).WikiPath()); err != nil { | |||||
return fmt.Errorf("SyncRepositoryHook: %v", err) | |||||
} | |||||
} | |||||
return nil | return nil | ||||
}) | }) | ||||
} | } |