github.com/zhiqiangxu/go-ethereum@v1.9.16-0.20210824055606-be91cfdebc48/internal/build/util.go (about) 1 // Copyright 2016 The go-ethereum Authors 2 // This file is part of the go-ethereum library. 3 // 4 // The go-ethereum library is free software: you can redistribute it and/or modify 5 // it under the terms of the GNU Lesser General Public License as published by 6 // the Free Software Foundation, either version 3 of the License, or 7 // (at your option) any later version. 8 // 9 // The go-ethereum library is distributed in the hope that it will be useful, 10 // but WITHOUT ANY WARRANTY; without even the implied warranty of 11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 12 // GNU Lesser General Public License for more details. 13 // 14 // You should have received a copy of the GNU Lesser General Public License 15 // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>. 16 17 package build 18 19 import ( 20 "bytes" 21 "flag" 22 "fmt" 23 "io" 24 "io/ioutil" 25 "log" 26 "os" 27 "os/exec" 28 "path" 29 "path/filepath" 30 "runtime" 31 "strings" 32 "text/template" 33 ) 34 35 var DryRunFlag = flag.Bool("n", false, "dry run, don't execute commands") 36 37 // MustRun executes the given command and exits the host process for 38 // any error. 39 func MustRun(cmd *exec.Cmd) { 40 fmt.Println(">>>", strings.Join(cmd.Args, " ")) 41 if !*DryRunFlag { 42 cmd.Stderr = os.Stderr 43 cmd.Stdout = os.Stdout 44 if err := cmd.Run(); err != nil { 45 log.Fatal(err) 46 } 47 } 48 } 49 50 func MustRunCommand(cmd string, args ...string) { 51 MustRun(exec.Command(cmd, args...)) 52 } 53 54 var warnedAboutGit bool 55 56 // RunGit runs a git subcommand and returns its output. 57 // The command must complete successfully. 58 func RunGit(args ...string) string { 59 cmd := exec.Command("git", args...) 60 var stdout, stderr bytes.Buffer 61 cmd.Stdout, cmd.Stderr = &stdout, &stderr 62 if err := cmd.Run(); err != nil { 63 if e, ok := err.(*exec.Error); ok && e.Err == exec.ErrNotFound { 64 if !warnedAboutGit { 65 log.Println("Warning: can't find 'git' in PATH") 66 warnedAboutGit = true 67 } 68 return "" 69 } 70 log.Fatal(strings.Join(cmd.Args, " "), ": ", err, "\n", stderr.String()) 71 } 72 return strings.TrimSpace(stdout.String()) 73 } 74 75 // readGitFile returns content of file in .git directory. 76 func readGitFile(file string) string { 77 content, err := ioutil.ReadFile(path.Join(".git", file)) 78 if err != nil { 79 return "" 80 } 81 return strings.TrimSpace(string(content)) 82 } 83 84 // Render renders the given template file into outputFile. 85 func Render(templateFile, outputFile string, outputPerm os.FileMode, x interface{}) { 86 tpl := template.Must(template.ParseFiles(templateFile)) 87 render(tpl, outputFile, outputPerm, x) 88 } 89 90 // RenderString renders the given template string into outputFile. 91 func RenderString(templateContent, outputFile string, outputPerm os.FileMode, x interface{}) { 92 tpl := template.Must(template.New("").Parse(templateContent)) 93 render(tpl, outputFile, outputPerm, x) 94 } 95 96 func render(tpl *template.Template, outputFile string, outputPerm os.FileMode, x interface{}) { 97 if err := os.MkdirAll(filepath.Dir(outputFile), 0755); err != nil { 98 log.Fatal(err) 99 } 100 out, err := os.OpenFile(outputFile, os.O_CREATE|os.O_WRONLY|os.O_EXCL, outputPerm) 101 if err != nil { 102 log.Fatal(err) 103 } 104 if err := tpl.Execute(out, x); err != nil { 105 log.Fatal(err) 106 } 107 if err := out.Close(); err != nil { 108 log.Fatal(err) 109 } 110 } 111 112 // GoTool returns the command that runs a go tool. This uses go from GOROOT instead of PATH 113 // so that go commands executed by build use the same version of Go as the 'host' that runs 114 // build code. e.g. 115 // 116 // /usr/lib/go-1.12.1/bin/go run build/ci.go ... 117 // 118 // runs using go 1.12.1 and invokes go 1.12.1 tools from the same GOROOT. This is also important 119 // because runtime.Version checks on the host should match the tools that are run. 120 func GoTool(tool string, args ...string) *exec.Cmd { 121 args = append([]string{tool}, args...) 122 return exec.Command(filepath.Join(runtime.GOROOT(), "bin", "go"), args...) 123 } 124 125 // UploadSFTP uploads files to a remote host using the sftp command line tool. 126 // The destination host may be specified either as [user@]host[: or as a URI in 127 // the form sftp://[user@]host[:port]. 128 func UploadSFTP(identityFile, host, dir string, files []string) error { 129 sftp := exec.Command("sftp") 130 sftp.Stdout = nil 131 sftp.Stderr = os.Stderr 132 if identityFile != "" { 133 sftp.Args = append(sftp.Args, "-i", identityFile) 134 } 135 sftp.Args = append(sftp.Args, host) 136 fmt.Println(">>>", strings.Join(sftp.Args, " ")) 137 if *DryRunFlag { 138 return nil 139 } 140 141 stdin, err := sftp.StdinPipe() 142 if err != nil { 143 return fmt.Errorf("can't create stdin pipe for sftp: %v", err) 144 } 145 if err := sftp.Start(); err != nil { 146 return err 147 } 148 in := io.MultiWriter(stdin, os.Stdout) 149 for _, f := range files { 150 fmt.Fprintln(in, "put", f, path.Join(dir, filepath.Base(f))) 151 } 152 stdin.Close() 153 return sftp.Wait() 154 }