[Contrib] Checkout a PR (#6021)
Antoine GIRARD
6 years ago
committed by
techknowlogick
parent
9fd8b2621f
commit
9d3732dfd5
@ -0,0 +1,256 @@ |
|||||||
|
package main |
||||||
|
|
||||||
|
/* |
||||||
|
Checkout a PR and load the tests data into sqlite database |
||||||
|
*/ |
||||||
|
|
||||||
|
import ( |
||||||
|
"flag" |
||||||
|
"fmt" |
||||||
|
"io/ioutil" |
||||||
|
"log" |
||||||
|
"net/http" |
||||||
|
"net/url" |
||||||
|
"os" |
||||||
|
"os/exec" |
||||||
|
"os/user" |
||||||
|
"path" |
||||||
|
"path/filepath" |
||||||
|
"runtime" |
||||||
|
"time" |
||||||
|
|
||||||
|
"code.gitea.io/gitea/modules/markup/external" |
||||||
|
"code.gitea.io/gitea/routers" |
||||||
|
"code.gitea.io/gitea/routers/routes" |
||||||
|
"github.com/Unknwon/com" |
||||||
|
"github.com/go-xorm/xorm" |
||||||
|
context2 "github.com/gorilla/context" |
||||||
|
"gopkg.in/src-d/go-git.v4" |
||||||
|
"gopkg.in/src-d/go-git.v4/config" |
||||||
|
"gopkg.in/src-d/go-git.v4/plumbing" |
||||||
|
"gopkg.in/testfixtures.v2" |
||||||
|
|
||||||
|
"code.gitea.io/gitea/models" |
||||||
|
"code.gitea.io/gitea/modules/setting" |
||||||
|
) |
||||||
|
|
||||||
|
var codeFilePath = "contrib/pr/checkout.go" |
||||||
|
|
||||||
|
func runPR() { |
||||||
|
log.Printf("[PR] Starting gitea ...\n") |
||||||
|
curDir, err := os.Getwd() |
||||||
|
if err != nil { |
||||||
|
log.Fatal(err) |
||||||
|
} |
||||||
|
setting.NewContext() |
||||||
|
|
||||||
|
setting.RepoRootPath, err = ioutil.TempDir(os.TempDir(), "repos") |
||||||
|
if err != nil { |
||||||
|
log.Fatalf("TempDir: %v\n", err) |
||||||
|
} |
||||||
|
setting.AppDataPath, err = ioutil.TempDir(os.TempDir(), "appdata") |
||||||
|
if err != nil { |
||||||
|
log.Fatalf("TempDir: %v\n", err) |
||||||
|
} |
||||||
|
setting.AppWorkPath = curDir |
||||||
|
setting.StaticRootPath = curDir |
||||||
|
setting.GravatarSourceURL, err = url.Parse("https://secure.gravatar.com/avatar/") |
||||||
|
if err != nil { |
||||||
|
log.Fatalf("url.Parse: %v\n", err) |
||||||
|
} |
||||||
|
|
||||||
|
setting.AppURL = "http://localhost:8080/" |
||||||
|
setting.HTTPPort = "8080" |
||||||
|
setting.SSH.Domain = "localhost" |
||||||
|
setting.SSH.Port = 3000 |
||||||
|
setting.InstallLock = true |
||||||
|
setting.SecretKey = "9pCviYTWSb" |
||||||
|
setting.InternalToken = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJuYmYiOjE0OTI3OTU5ODN9.OQkH5UmzID2XBdwQ9TAI6Jj2t1X-wElVTjbE7aoN4I8" |
||||||
|
curUser, err := user.Current() |
||||||
|
if err != nil { |
||||||
|
log.Fatal(err) |
||||||
|
} |
||||||
|
setting.RunUser = curUser.Username |
||||||
|
|
||||||
|
log.Printf("[PR] Loading fixtures data ...\n") |
||||||
|
setting.CheckLFSVersion() |
||||||
|
//models.LoadConfigs()
|
||||||
|
/* |
||||||
|
models.DbCfg.Type = "sqlite3" |
||||||
|
models.DbCfg.Path = ":memory:" |
||||||
|
models.DbCfg.Timeout = 500 |
||||||
|
*/ |
||||||
|
db := setting.Cfg.Section("database") |
||||||
|
db.NewKey("DB_TYPE", "sqlite3") |
||||||
|
db.NewKey("PATH", ":memory:") |
||||||
|
setting.LogSQL = true |
||||||
|
models.LoadConfigs() |
||||||
|
routers.NewServices() |
||||||
|
//x, err = xorm.NewEngine("sqlite3", "file::memory:?cache=shared")
|
||||||
|
|
||||||
|
var helper testfixtures.Helper |
||||||
|
helper = &testfixtures.SQLite{} |
||||||
|
models.NewEngine(func(_ *xorm.Engine) error { |
||||||
|
return nil |
||||||
|
}) |
||||||
|
models.HasEngine = true |
||||||
|
//x.ShowSQL(true)
|
||||||
|
err = models.InitFixtures( |
||||||
|
helper, |
||||||
|
path.Join(curDir, "models/fixtures/"), |
||||||
|
) |
||||||
|
if err != nil { |
||||||
|
fmt.Printf("Error initializing test database: %v\n", err) |
||||||
|
os.Exit(1) |
||||||
|
} |
||||||
|
models.LoadFixtures() |
||||||
|
os.RemoveAll(setting.RepoRootPath) |
||||||
|
os.RemoveAll(models.LocalCopyPath()) |
||||||
|
os.RemoveAll(models.LocalWikiPath()) |
||||||
|
com.CopyDir(path.Join(curDir, "integrations/gitea-repositories-meta"), setting.RepoRootPath) |
||||||
|
|
||||||
|
log.Printf("[PR] Setting up router\n") |
||||||
|
//routers.GlobalInit()
|
||||||
|
external.RegisterParsers() |
||||||
|
m := routes.NewMacaron() |
||||||
|
routes.RegisterRoutes(m) |
||||||
|
|
||||||
|
log.Printf("[PR] Ready for testing !\n") |
||||||
|
log.Printf("[PR] Login with user1, user2, user3, ... with pass: password\n") |
||||||
|
/* |
||||||
|
log.Info("Listen: %v://%s%s", setting.Protocol, listenAddr, setting.AppSubURL) |
||||||
|
|
||||||
|
if setting.LFS.StartServer { |
||||||
|
log.Info("LFS server enabled") |
||||||
|
} |
||||||
|
|
||||||
|
if setting.EnablePprof { |
||||||
|
go func() { |
||||||
|
log.Info("Starting pprof server on localhost:6060") |
||||||
|
log.Info("%v", http.ListenAndServe("localhost:6060", nil)) |
||||||
|
}() |
||||||
|
} |
||||||
|
*/ |
||||||
|
|
||||||
|
//Start the server
|
||||||
|
http.ListenAndServe(":8080", context2.ClearHandler(m)) |
||||||
|
|
||||||
|
log.Printf("[PR] Cleaning up ...\n") |
||||||
|
/* |
||||||
|
if err = os.RemoveAll(setting.Indexer.IssuePath); err != nil { |
||||||
|
fmt.Printf("os.RemoveAll: %v\n", err) |
||||||
|
os.Exit(1) |
||||||
|
} |
||||||
|
if err = os.RemoveAll(setting.Indexer.RepoPath); err != nil { |
||||||
|
fmt.Printf("Unable to remove repo indexer: %v\n", err) |
||||||
|
os.Exit(1) |
||||||
|
} |
||||||
|
*/ |
||||||
|
if err = os.RemoveAll(setting.RepoRootPath); err != nil { |
||||||
|
log.Fatalf("os.RemoveAll: %v\n", err) |
||||||
|
} |
||||||
|
if err = os.RemoveAll(setting.AppDataPath); err != nil { |
||||||
|
log.Fatalf("os.RemoveAll: %v\n", err) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func main() { |
||||||
|
var runPRFlag = flag.Bool("run", false, "Run the PR code") |
||||||
|
flag.Parse() |
||||||
|
if *runPRFlag { |
||||||
|
runPR() |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
//Otherwise checkout PR
|
||||||
|
if len(os.Args) != 2 { |
||||||
|
log.Fatal("Need only one arg: the PR number") |
||||||
|
} |
||||||
|
pr := os.Args[1] |
||||||
|
|
||||||
|
codeFilePath = filepath.FromSlash(codeFilePath) //Convert to running OS
|
||||||
|
|
||||||
|
//Copy this file if it will not exist in the PR branch
|
||||||
|
dat, err := ioutil.ReadFile(codeFilePath) |
||||||
|
if err != nil { |
||||||
|
log.Fatalf("Failed to cache this code file : %v", err) |
||||||
|
} |
||||||
|
|
||||||
|
repo, err := git.PlainOpen(".") |
||||||
|
if err != nil { |
||||||
|
log.Fatalf("Failed to open the repo : %v", err) |
||||||
|
} |
||||||
|
|
||||||
|
//Find remote upstream
|
||||||
|
remotes, err := repo.Remotes() |
||||||
|
if err != nil { |
||||||
|
log.Fatalf("Failed to list remotes of repo : %v", err) |
||||||
|
} |
||||||
|
remoteUpstream := "origin" //Default
|
||||||
|
for _, r := range remotes { |
||||||
|
if r.Config().URLs[0] == "https://github.com/go-gitea/gitea" || r.Config().URLs[0] == "git@github.com:go-gitea/gitea.git" { //fetch at index 0
|
||||||
|
remoteUpstream = r.Config().Name |
||||||
|
break |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
branch := fmt.Sprintf("pr-%s-%d", pr, time.Now().Unix()) |
||||||
|
branchRef := plumbing.NewBranchReferenceName(branch) |
||||||
|
|
||||||
|
log.Printf("Fetching PR #%s in %s\n", pr, branch) |
||||||
|
if runtime.GOOS == "windows" { |
||||||
|
//Use git cli command for windows
|
||||||
|
runCmd("git", "fetch", "origin", fmt.Sprintf("pull/%s/head:%s", pr, branch)) |
||||||
|
} else { |
||||||
|
ref := fmt.Sprintf("refs/pull/%s/head:%s", pr, branchRef) |
||||||
|
err = repo.Fetch(&git.FetchOptions{ |
||||||
|
RemoteName: remoteUpstream, |
||||||
|
RefSpecs: []config.RefSpec{ |
||||||
|
config.RefSpec(ref), |
||||||
|
}, |
||||||
|
}) |
||||||
|
if err != nil { |
||||||
|
log.Fatalf("Failed to fetch %s from %s : %v", ref, remoteUpstream, err) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
tree, err := repo.Worktree() |
||||||
|
if err != nil { |
||||||
|
log.Fatalf("Failed to parse git tree : %v", err) |
||||||
|
} |
||||||
|
log.Printf("Checkout PR #%s in %s\n", pr, branch) |
||||||
|
err = tree.Checkout(&git.CheckoutOptions{ |
||||||
|
Branch: branchRef, |
||||||
|
//Force: runtime.GOOS == "windows",
|
||||||
|
}) |
||||||
|
if err != nil { |
||||||
|
log.Fatalf("Failed to checkout %s : %v", branch, err) |
||||||
|
} |
||||||
|
|
||||||
|
//Copy this file if not exist
|
||||||
|
if _, err := os.Stat(codeFilePath); os.IsNotExist(err) { |
||||||
|
err = os.MkdirAll(filepath.Dir(codeFilePath), 0755) |
||||||
|
if err != nil { |
||||||
|
log.Fatalf("Failed to duplicate this code file in PR : %v", err) |
||||||
|
} |
||||||
|
err = ioutil.WriteFile(codeFilePath, dat, 0644) |
||||||
|
if err != nil { |
||||||
|
log.Fatalf("Failed to duplicate this code file in PR : %v", err) |
||||||
|
} |
||||||
|
} |
||||||
|
time.Sleep(5 * time.Second) |
||||||
|
//Start with integration test
|
||||||
|
runCmd("go", "run", "-tags", "sqlite sqlite_unlock_notify", codeFilePath, "-run") |
||||||
|
} |
||||||
|
func runCmd(cmd ...string) { |
||||||
|
log.Printf("Executing : %s ...\n", cmd) |
||||||
|
c := exec.Command(cmd[0], cmd[1:]...) |
||||||
|
c.Stdout = os.Stdout |
||||||
|
c.Stderr = os.Stderr |
||||||
|
if err := c.Start(); err != nil { |
||||||
|
log.Panicln(err) |
||||||
|
} |
||||||
|
if err := c.Wait(); err != nil { |
||||||
|
log.Panicln(err) |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue