moq/main.go

100 lines
2.2 KiB
Go
Raw Normal View History

2016-08-29 14:00:18 +02:00
package main
2016-09-21 22:49:12 +02:00
import (
2016-09-21 23:02:49 +02:00
"bytes"
"errors"
2016-09-21 22:49:12 +02:00
"flag"
"fmt"
2016-09-21 23:02:49 +02:00
"io"
"io/ioutil"
2016-09-21 22:49:12 +02:00
"os"
"path/filepath"
2016-09-21 22:49:12 +02:00
2017-07-11 21:02:46 +02:00
"github.com/matryer/moq/pkg/moq"
2016-09-21 22:49:12 +02:00
)
2020-09-02 11:30:40 +02:00
// version is the command version, injected at build time.
var version string = "dev"
2020-09-02 11:30:40 +02:00
type userFlags struct {
outFile string
pkgName string
formatter string
stubImpl bool
skipEnsure bool
args []string
}
2016-08-29 14:00:18 +02:00
func main() {
var flags userFlags
flag.StringVar(&flags.outFile, "out", "", "output file (default stdout)")
flag.StringVar(&flags.pkgName, "pkg", "", "package name (default will infer)")
flag.StringVar(&flags.formatter, "fmt", "", "go pretty-printer: gofmt, goimports or noop (default gofmt)")
2020-08-16 08:59:28 +02:00
flag.BoolVar(&flags.stubImpl, "stub", false,
"return zero values when no mock implementation is provided, do not panic")
printVersion := flag.Bool("version", false, "show the version for moq")
flag.BoolVar(&flags.skipEnsure, "skip-ensure", false,
"suppress mock implementation check, avoid import cycle if mocks generated outside of the tested package")
2016-10-14 11:57:57 +02:00
2016-09-21 22:49:12 +02:00
flag.Usage = func() {
fmt.Println(`moq [flags] source-dir interface [interface2 [interface3 [...]]]`)
2016-09-21 22:49:12 +02:00
flag.PrintDefaults()
fmt.Println(`Specifying an alias for the mock is also supported with the format 'interface:alias'`)
fmt.Println(`Ex: moq -pkg different . MyInterface:MyMock`)
2016-09-21 22:49:12 +02:00
}
2016-09-21 22:49:12 +02:00
flag.Parse()
flags.args = flag.Args()
if *printVersion {
fmt.Printf("moq version %s\n", version)
os.Exit(0)
}
if err := run(flags); err != nil {
fmt.Fprintln(os.Stderr, err)
flag.Usage()
os.Exit(1)
2016-09-21 22:49:12 +02:00
}
}
func run(flags userFlags) error {
if len(flags.args) < 2 {
return errors.New("not enough arguments")
}
2016-09-21 23:02:49 +02:00
var buf bytes.Buffer
var out io.Writer = os.Stdout
if flags.outFile != "" {
2016-09-21 23:02:49 +02:00
out = &buf
2016-09-21 22:49:12 +02:00
}
srcDir, args := flags.args[0], flags.args[1:]
m, err := moq.New(moq.Config{
SrcDir: srcDir,
PkgName: flags.pkgName,
Formatter: flags.formatter,
StubImpl: flags.stubImpl,
SkipEnsure: flags.skipEnsure,
})
2016-09-21 22:49:12 +02:00
if err != nil {
return err
2016-09-21 22:49:12 +02:00
}
if err = m.Mock(out, args...); err != nil {
return err
}
if flags.outFile == "" {
return nil
2016-09-21 23:02:49 +02:00
}
2016-09-21 23:02:49 +02:00
// create the file
err = os.MkdirAll(filepath.Dir(flags.outFile), 0755)
if err != nil {
return err
2016-09-21 23:02:49 +02:00
}
return ioutil.WriteFile(flags.outFile, buf.Bytes(), 0644)
2016-08-29 14:00:18 +02:00
}