75 lines
1.9 KiB
Go
75 lines
1.9 KiB
Go
package main // import "github.com/tweag/gomod2nix"
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/tweag/gomod2nix/fetch"
|
|
"github.com/tweag/gomod2nix/formats/buildgopackage"
|
|
"github.com/tweag/gomod2nix/formats/gomod2nix"
|
|
"io/ioutil"
|
|
"path/filepath"
|
|
)
|
|
|
|
func main() {
|
|
|
|
var keepGoing = flag.Bool("keep-going", false, "Whether to panic or not if a rev cannot be resolved (default \"false\")")
|
|
var directory = flag.String("dir", "./", "Go project directory")
|
|
var maxJobs = flag.Int("jobs", 10, "Number of max parallel jobs")
|
|
var outDirFlag = flag.String("outdir", "", "output directory (if different from project directory)")
|
|
var format = flag.String("format", "gomod2nix", "output format (gomod2nix, buildgopackage)")
|
|
flag.Parse()
|
|
|
|
outDir := *outDirFlag
|
|
if outDir == "" {
|
|
outDir = *directory
|
|
}
|
|
|
|
goSumPath := filepath.Join(*directory, "go.sum")
|
|
goModPath := filepath.Join(*directory, "go.mod")
|
|
|
|
wrongFormatError := fmt.Errorf("Format not supported")
|
|
|
|
goMod2NixPath := ""
|
|
depsNixPath := ""
|
|
outFile := ""
|
|
switch *format {
|
|
case "gomod2nix":
|
|
goMod2NixPath = filepath.Join(outDir, "gomod2nix.toml")
|
|
outFile = goMod2NixPath
|
|
case "buildgopackage":
|
|
depsNixPath = filepath.Join(outDir, "deps.nix")
|
|
outFile = depsNixPath
|
|
default:
|
|
panic(wrongFormatError)
|
|
}
|
|
log.Info(fmt.Sprintf("Using output format '%s'", *format))
|
|
|
|
pkgs, err := fetch.FetchPackages(goModPath, goSumPath, goMod2NixPath, depsNixPath, *maxJobs, *keepGoing)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
var output []byte
|
|
switch *format {
|
|
case "gomod2nix":
|
|
output, err = gomod2nix.Marshal(pkgs)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
case "buildgopackage":
|
|
output, err = buildgopackage.Marshal(pkgs)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
default:
|
|
panic(wrongFormatError)
|
|
}
|
|
|
|
err = ioutil.WriteFile(outFile, output, 0644)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
log.Info(fmt.Sprintf("Wrote: %s", outFile))
|
|
|
|
}
|