2
2
mirror of https://github.com/octoleo/restic.git synced 2024-11-27 15:26:37 +00:00
restic/build.go

411 lines
9.3 KiB
Go
Raw Normal View History

2015-06-24 15:59:31 +00:00
// +build ignore
package main
import (
"fmt"
"io"
"io/ioutil"
"os"
"os/exec"
"path"
"path/filepath"
2015-08-14 13:08:07 +00:00
"runtime"
2015-06-24 15:59:31 +00:00
"strings"
"time"
)
var (
verbose bool
keepGopath bool
runTests bool
2015-06-24 15:59:31 +00:00
)
const timeFormat = "2006-01-02 15:04:05"
// specialDir returns true if the file begins with a special character ('.' or '_').
func specialDir(name string) bool {
if name == "." {
return false
}
base := filepath.Base(name)
if base == "vendor" || base[0] == '_' || base[0] == '.' {
return true
}
return false
2015-06-24 15:59:31 +00:00
}
// excludePath returns true if the file should not be copied to the new GOPATH.
func excludePath(name string) bool {
ext := path.Ext(name)
if ext == ".go" || ext == ".s" {
return false
}
parentDir := filepath.Base(filepath.Dir(name))
if parentDir == "testdata" {
return false
}
return true
}
2015-06-24 15:59:31 +00:00
// updateGopath builds a valid GOPATH at dst, with all Go files in src/ copied
// to dst/prefix/, so calling
//
// updateGopath("/tmp/gopath", "/home/u/restic", "github.com/restic/restic")
//
// with "/home/u/restic" containing the file "foo.go" yields the following tree
// at "/tmp/gopath":
//
// /tmp/gopath
// └── src
// └── github.com
// └── restic
// └── restic
// └── foo.go
func updateGopath(dst, src, prefix string) error {
return filepath.Walk(src, func(name string, fi os.FileInfo, err error) error {
if specialDir(name) {
if fi.IsDir() {
return filepath.SkipDir
}
return nil
}
if fi.IsDir() {
return nil
}
if excludePath(name) {
2015-06-24 15:59:31 +00:00
return nil
}
intermediatePath, err := filepath.Rel(src, name)
if err != nil {
return err
}
fileSrc := filepath.Join(src, intermediatePath)
fileDst := filepath.Join(dst, "src", prefix, intermediatePath)
return copyFile(fileDst, fileSrc)
})
}
// copyFile creates dst from src, preserving file attributes and timestamps.
func copyFile(dst, src string) error {
fi, err := os.Stat(src)
if err != nil {
return err
}
fsrc, err := os.Open(src)
if err != nil {
return err
}
if err = os.MkdirAll(filepath.Dir(dst), 0755); err != nil {
fmt.Printf("MkdirAll(%v)\n", filepath.Dir(dst))
return err
}
fdst, err := os.Create(dst)
if err != nil {
return err
}
if _, err = io.Copy(fdst, fsrc); err != nil {
return err
}
if err == nil {
err = fsrc.Close()
}
if err == nil {
err = fdst.Close()
}
if err == nil {
err = os.Chmod(dst, fi.Mode())
}
if err == nil {
err = os.Chtimes(dst, fi.ModTime(), fi.ModTime())
}
return nil
}
// die prints the message with fmt.Fprintf() to stderr and exits with an error
// code.
func die(message string, args ...interface{}) {
fmt.Fprintf(os.Stderr, message, args...)
os.Exit(1)
}
func showUsage(output io.Writer) {
fmt.Fprintf(output, "USAGE: go run build.go OPTIONS\n")
fmt.Fprintf(output, "\n")
fmt.Fprintf(output, "OPTIONS:\n")
fmt.Fprintf(output, " -v --verbose output more messages\n")
fmt.Fprintf(output, " -t --tags specify additional build tags\n")
fmt.Fprintf(output, " -k --keep-gopath do not remove the GOPATH after build\n")
fmt.Fprintf(output, " -T --test run tests\n")
fmt.Fprintf(output, " --goos value set GOOS for cross-compilation\n")
fmt.Fprintf(output, " --goarch value set GOARCH for cross-compilation\n")
2015-06-24 15:59:31 +00:00
}
func verbosePrintf(message string, args ...interface{}) {
if !verbose {
return
}
fmt.Printf("build: "+message, args...)
2015-06-24 15:59:31 +00:00
}
// cleanEnv returns a clean environment with GOPATH and GOBIN removed (if
// present).
func cleanEnv() (env []string) {
for _, v := range os.Environ() {
if strings.HasPrefix(v, "GOPATH=") || strings.HasPrefix(v, "GOBIN=") {
continue
}
env = append(env, v)
}
return env
}
// build runs "go build args..." with GOPATH set to gopath.
func build(cwd, goos, goarch, gopath string, args ...string) error {
2015-06-24 15:59:31 +00:00
args = append([]string{"build"}, args...)
cmd := exec.Command("go", args...)
cmd.Env = append(cleanEnv(), "GOPATH="+gopath, "GOARCH="+goarch, "GOOS="+goos)
cmd.Dir = cwd
2015-06-24 15:59:31 +00:00
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
verbosePrintf("go %s\n", args)
return cmd.Run()
}
// test runs "go test args..." with GOPATH set to gopath.
func test(cwd, gopath string, args ...string) error {
args = append([]string{"test"}, args...)
cmd := exec.Command("go", args...)
cmd.Env = append(cleanEnv(), "GOPATH="+gopath)
cmd.Dir = cwd
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
verbosePrintf("go %s\n", args)
return cmd.Run()
}
// getVersion returns the version string from the file VERSION in the current
// directory.
func getVersionFromFile() string {
buf, err := ioutil.ReadFile("VERSION")
if err != nil {
verbosePrintf("error reading file VERSION: %v\n", err)
return ""
}
return strings.TrimSpace(string(buf))
}
// getVersion returns a version string which is a combination of the contents
// of the file VERSION in the current directory and the version from git (if
// available).
2015-06-24 15:59:31 +00:00
func getVersion() string {
versionFile := getVersionFromFile()
versionGit := getVersionFromGit()
verbosePrintf("version from file 'VERSION' is %q, version from git %q\n",
versionFile, versionGit)
switch {
case versionFile == "":
return versionGit
case versionGit == "":
return versionFile
2015-06-24 15:59:31 +00:00
}
return fmt.Sprintf("%s (%s)", versionFile, versionGit)
2015-06-24 15:59:31 +00:00
}
// getVersionFromGit returns a version string that identifies the currently
// checked out git commit.
func getVersionFromGit() string {
2015-06-24 15:59:31 +00:00
cmd := exec.Command("git", "describe",
"--long", "--tags", "--dirty", "--always")
out, err := cmd.Output()
if err != nil {
2015-08-19 04:31:14 +00:00
verbosePrintf("git describe returned error: %v\n", err)
return ""
2015-06-24 15:59:31 +00:00
}
version := strings.TrimSpace(string(out))
verbosePrintf("git version is %s\n", version)
return version
}
2015-08-26 18:03:16 +00:00
// Constants represents a set of constants that are set in the final binary to
// the given value via compiler flags.
type Constants map[string]string
// LDFlags returns the string that can be passed to go build's `-ldflags`.
func (cs Constants) LDFlags() string {
l := make([]string, 0, len(cs))
2016-08-03 18:58:59 +00:00
if runtime.Version() < "go1.5" {
for k, v := range cs {
2016-08-03 18:58:59 +00:00
l = append(l, fmt.Sprintf(`-X %q %q`, k, v))
}
} else {
for k, v := range cs {
2016-08-03 18:58:59 +00:00
l = append(l, fmt.Sprintf(`-X "%s=%s"`, k, v))
}
}
return strings.Join(l, " ")
}
2015-06-24 15:59:31 +00:00
func main() {
2016-08-16 18:32:18 +00:00
if runtime.Version() < "go1.6" {
fmt.Fprintf(os.Stderr, "old version of Go detected (%v), I'll try but no guarantees\n", runtime.Version())
}
2015-06-24 17:57:08 +00:00
buildTags := []string{}
skipNext := false
params := os.Args[1:]
2016-07-15 22:14:41 +00:00
targetGOOS := runtime.GOOS
targetGOARCH := runtime.GOARCH
2016-07-15 22:14:41 +00:00
2015-06-24 17:57:08 +00:00
for i, arg := range params {
if skipNext {
skipNext = false
continue
}
2015-06-24 15:59:31 +00:00
switch arg {
case "-v", "--verbose":
verbose = true
case "-k", "--keep-gopath":
keepGopath = true
2015-06-24 17:57:08 +00:00
case "-t", "-tags", "--tags":
skipNext = true
buildTags = strings.Split(params[i+1], " ")
case "-T", "--test":
runTests = true
case "--goos":
skipNext = true
targetGOOS = params[i+1]
case "--goarch":
skipNext = true
targetGOARCH = params[i+1]
2015-06-24 15:59:31 +00:00
case "-h":
showUsage(os.Stdout)
return
2015-06-24 15:59:31 +00:00
default:
fmt.Fprintf(os.Stderr, "Error: unknown option %q\n\n", arg)
showUsage(os.Stderr)
os.Exit(1)
}
}
2015-06-24 17:57:08 +00:00
if len(buildTags) == 0 {
verbosePrintf("adding build-tag release\n")
buildTags = []string{"release"}
}
for i := range buildTags {
buildTags[i] = strings.TrimSpace(buildTags[i])
}
verbosePrintf("build tags: %s\n", buildTags)
2015-06-24 15:59:31 +00:00
root, err := os.Getwd()
if err != nil {
die("Getwd(): %v\n", err)
}
gopath, err := ioutil.TempDir("", "restic-build-")
if err != nil {
die("TempDir(): %v\n", err)
}
verbosePrintf("create GOPATH at %v\n", gopath)
if err = updateGopath(gopath, filepath.Join(root, "src", "restic"), "restic"); err != nil {
die("copying files from %v/src/restic to %v/src/restic failed: %v\n", root, gopath, err)
}
if err = updateGopath(gopath, filepath.Join(root, "src", "cmds"), "cmds"); err != nil {
die("copying files from %v/src/cmds to %v/src/restic/cmds failed: %v\n", root, gopath, err)
2015-06-24 15:59:31 +00:00
}
vendor := filepath.Join(root, "vendor", "src")
2015-06-24 15:59:31 +00:00
if err = updateGopath(gopath, vendor, ""); err != nil {
die("copying files from %v to %v/src failed: %v\n", vendor, gopath, err)
2015-06-24 15:59:31 +00:00
}
defer func() {
if !keepGopath {
verbosePrintf("remove %v\n", gopath)
if err = os.RemoveAll(gopath); err != nil {
die("remove GOPATH at %s failed: %v\n", err)
}
} else {
2015-08-26 18:03:16 +00:00
verbosePrintf("leaving temporary GOPATH at %v\n", gopath)
}
}()
outputFilename := "restic"
if targetGOOS == "windows" {
outputFilename = "restic.exe"
}
cwd, err := os.Getwd()
if err != nil {
die("Getwd() returned %v\n", err)
2015-08-16 12:20:37 +00:00
}
output := filepath.Join(cwd, outputFilename)
2015-08-19 04:31:14 +00:00
version := getVersion()
compileTime := time.Now().Format(timeFormat)
constants := Constants{`main.compiledAt`: compileTime}
2015-08-19 04:31:14 +00:00
if version != "" {
constants["main.version"] = version
2015-08-19 04:31:14 +00:00
}
2016-07-15 21:42:22 +00:00
ldflags := "-s -w " + constants.LDFlags()
2015-08-26 18:03:16 +00:00
verbosePrintf("ldflags: %s\n", ldflags)
2015-06-24 15:59:31 +00:00
args := []string{
2015-06-24 17:57:08 +00:00
"-tags", strings.Join(buildTags, " "),
2015-08-19 04:31:14 +00:00
"-ldflags", ldflags,
"-o", output, "cmds/restic",
2015-08-14 13:08:07 +00:00
}
err = build(filepath.Join(gopath, "src"), targetGOOS, targetGOARCH, gopath, args...)
2015-06-24 15:59:31 +00:00
if err != nil {
2015-08-19 19:40:36 +00:00
die("build failed: %v\n", err)
2015-06-24 15:59:31 +00:00
}
if runTests {
verbosePrintf("running tests\n")
err = test(filepath.Join(gopath, "src"), gopath, "restic/...")
if err != nil {
2015-08-19 19:40:36 +00:00
die("running tests failed: %v\n", err)
2015-06-24 15:59:31 +00:00
}
}
}