blob: 19850af0f37b29b22186c80a25d9bce6f7f696d1 [file] [log] [blame]
// Copyright 2018 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
// +build ignore
// Addmod adds a module as a txtar archive to the testdata/mod directory.
//
// Usage:
//
// go run addmod.go path@version...
//
// It should only be used for very small modules - we do not want to check
// very large files into testdata/mod.
//
// It is acceptable to edit the archive afterward to remove or shorten files.
// See mod/README for more information.
//
package main
import (
"bytes"
"flag"
"fmt"
"io/ioutil"
"log"
"os"
"os/exec"
"path/filepath"
"strings"
"../internal/txtar"
)
func usage() {
fmt.Fprintf(os.Stderr, "usage: go run addmod.go path@version...\n")
os.Exit(2)
}
var tmpdir string
func fatalf(format string, args ...interface{}) {
os.RemoveAll(tmpdir)
log.Fatalf(format, args...)
}
const goCmd = "vgo"
func main() {
flag.Usage = usage
flag.Parse()
if flag.NArg() == 0 {
usage()
}
log.SetPrefix("addmod: ")
log.SetFlags(0)
var err error
tmpdir, err = ioutil.TempDir("", "addmod-")
if err != nil {
log.Fatal(err)
}
run := func(command string, args ...string) string {
cmd := exec.Command(command, args...)
cmd.Dir = tmpdir
var stderr bytes.Buffer
cmd.Stderr = &stderr
out, err := cmd.Output()
if err != nil {
fatalf("%s %s: %v\n%s", command, strings.Join(args, " "), err, stderr.Bytes())
}
return string(out)
}
gopath := strings.TrimSpace(run("go", "env", "GOPATH"))
if gopath == "" {
fatalf("cannot find GOPATH")
}
exitCode := 0
for _, arg := range flag.Args() {
if err := ioutil.WriteFile(filepath.Join(tmpdir, "go.mod"), []byte("module m\n"), 0666); err != nil {
fatalf("%v", err)
}
run(goCmd, "get", "-d", arg)
path := arg
if i := strings.Index(path, "@"); i >= 0 {
path = path[:i]
}
out := run(goCmd, "list", "-m", "-f={{.Path}} {{.Version}} {{.Dir}}", path)
f := strings.Fields(out)
if len(f) != 3 {
log.Printf("go list -m %s: unexpected output %q", arg, out)
exitCode = 1
continue
}
path, vers, dir := f[0], f[1], f[2]
mod, err := ioutil.ReadFile(filepath.Join(gopath, "pkg/mod/cache/download", path, "@v", vers+".mod"))
if err != nil {
log.Printf("%s: %v", arg, err)
exitCode = 1
continue
}
info, err := ioutil.ReadFile(filepath.Join(gopath, "pkg/mod/cache/download", path, "@v", vers+".info"))
if err != nil {
log.Printf("%s: %v", arg, err)
exitCode = 1
continue
}
a := new(txtar.Archive)
title := arg
if !strings.Contains(arg, "@") {
title += "@" + vers
}
a.Comment = []byte(fmt.Sprintf("module %s\n\n", title))
a.Files = []txtar.File{
{Name: ".mod", Data: mod},
{Name: ".info", Data: info},
}
dir = filepath.Clean(dir)
err = filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
if !info.Mode().IsRegular() {
return nil
}
name := info.Name()
if name == "go.mod" || strings.HasSuffix(name, ".go") {
data, err := ioutil.ReadFile(path)
if err != nil {
return err
}
a.Files = append(a.Files, txtar.File{Name: strings.TrimPrefix(path, dir+string(filepath.Separator)), Data: data})
}
return nil
})
if err != nil {
log.Printf("%s: %v", arg, err)
exitCode = 1
continue
}
data := txtar.Format(a)
target := filepath.Join("mod", strings.Replace(path, "/", "_", -1)+"_"+vers+".txt")
if err := ioutil.WriteFile(target, data, 0666); err != nil {
log.Printf("%s: %v", arg, err)
exitCode = 1
continue
}
}
os.RemoveAll(tmpdir)
os.Exit(exitCode)
}