package mainimport ("bufio""encoding/json""errors""fmt""io""log""os""strings""github.com/alibaba/git-repo-go/cap""github.com/alibaba/git-repo-go/config""github.com/alibaba/git-repo-go/encode""github.com/alibaba/git-repo-go/helper""github.com/urfave/cli/v2")funcmain() {app:=&cli.App{Name:"git-repo-helper-proto-gitea",Usage:"gitea protocol helper for git repo",Action:cmdMain,Flags: []cli.Flag{&cli.BoolFlag{Name:"upload",Usage:"for upload hook",},&cli.BoolFlag{Name:"download",Usage:"for download hook",},&cli.Int64Flag{Name:"version",Usage:"hook version",},},}iferr:=app.Run(os.Args);err!=nil {log.Fatal(err)}}funcgetGitPushCommand(o*config.UploadOptions) (*helper.GitPushCommand,error) {var (gitPushCmd= helper.GitPushCommand{})cmds:= []string{"git","push"}ifo.RemoteURL=="" {returnnil,errors.New("empty review url for helper")}gitURL:=config.ParseGitURL(o.RemoteURL)ifgitURL==nil|| (gitURL.Proto!="ssh"&&gitURL.Proto!="http"&&gitURL.Proto!="https") {returnnil,fmt.Errorf("bad review URL: %s",o.RemoteURL)}if!cap.GitCanPushOptions() {log.Fatalln("cannot send push options, for your git version is too low")}for_,pushOption:=rangeo.PushOptions {cmds=append(cmds,"-o",pushOption)}refSpec:=""localBranch:=strings.TrimPrefix(o.LocalBranch,config.RefsHeads)destBranch:=strings.TrimPrefix(o.DestBranch,config.RefsHeads)iflocalBranch=="" {refSpec="HEAD"}else {refSpec=config.RefsHeads+localBranch}if!o.CodeReview.Empty() {log.Fatalln("gitea not support for-review now")}else {refSpec+=fmt.Sprintf(":refs/for/%s/%s",destBranch,localBranch)}ifo.Title!="" {cmds=append(cmds,"-o","title="+encode.B64Encode(o.Title))}ifo.Description!="" {cmds=append(cmds,"-o","description="+encode.B64Encode(o.Description))}// TODO:// if o.Issue != "" {// cmds = append(cmds, "-o", "issue="+encode.B64Encode(o.Issue))// }// if o.People != nil && len(o.People) > 0 && len(o.People[0]) > 0 {// reviewers := strings.Join(o.People[0], ",")// cmds = append(cmds, "-o", "reviewers="+encode.B64Encode(reviewers))// }// if o.People != nil && len(o.People) > 1 && len(o.People[1]) > 0 {// cc := strings.Join(o.People[1], ",")// cmds = append(cmds, "-o", "cc="+encode.B64Encode(cc))// }// if o.NoEmails {// cmds = append(cmds, "-o", "notify=no")// }// if o.Private {// cmds = append(cmds, "-o", "private=yes")// }// if o.WIP {// cmds = append(cmds, "-o", "wip=yes")// }// if o.OldOid != "" {// cmds = append(cmds, "-o", "old-oid="+o.OldOid)// }ifo.RemoteName!="" {cmds=append(cmds,o.RemoteName)}else {cmds=append(cmds,o.RemoteURL)}cmds=append(cmds,refSpec)gitPushCmd.Cmd=cmds[0]gitPushCmd.Args=cmds[1:]return&gitPushCmd,nil}funccmdUpload()error {log.Printf("cmdUpload ...\n")input,err:=io.ReadAll(os.Stdin)iferr!=nil {returnerr}cfg:= config.UploadOptions{}err=json.Unmarshal(input,&cfg)iferr!=nil {log.Printf("Error: cmdUpload: %v\n",err)returnerr}log.Printf("Debug: cmdUpload: %#v\n",cfg)result,err:=getGitPushCommand(&cfg)iferr!=nil {returnerr}resultBytes,err:=json.Marshal(result)iferr!=nil {returnerr}_,err=os.Stdout.Write(resultBytes)returnerr}funccmdDownload()error {scan:=bufio.NewScanner(os.Stdin)if!scan.Scan() {returnerrors.New("no input data")}parts:=strings.SplitN(scan.Text()," ",3)iflen(parts)!=2 {returnerrors.New("no enough params")}log.Println("cmdDownload: ",parts[0],parts[1])fmt.Printf("refs/pull/%s/head\n",parts[0])returnnil}varlogFp*os.FiletypeemptyWriterstruct{}func (e*emptyWriter)Write(p []byte) (nint,errerror) {returnlen(p),nil}funcinitLogFile()error {debugFile:=os.Getenv("GIT_REPO_GITEA_DEBUG")iflen(debugFile)==0 {log.SetOutput(&emptyWriter{})returnnil}varerrerrorlogFp,err=os.OpenFile(debugFile,os.O_CREATE|os.O_WRONLY|os.O_TRUNC,0o644)iferr==nil {log.SetOutput(logFp)}returnerr}funccmdMain(ctx*cli.Context)error {iferr:=initLogFile();err!=nil {returnerr}deferfunc() {iflogFp!=nil {logFp.Close()}}()ifctx.Bool("upload") {returncmdUpload()}ifctx.Bool("download") {returncmdDownload()}returnnil}