100 lines
2.2 KiB
Go
100 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/matryer/moq/pkg/moq"
|
|
)
|
|
|
|
// Version is the command version, injected at build time.
|
|
var Version string = "dev"
|
|
|
|
type userFlags struct {
|
|
outFile string
|
|
pkgName string
|
|
formatter string
|
|
stubImpl bool
|
|
skipEnsure bool
|
|
args []string
|
|
}
|
|
|
|
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)")
|
|
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")
|
|
|
|
flag.Usage = func() {
|
|
fmt.Println(`moq [flags] source-dir interface [interface2 [interface3 [...]]]`)
|
|
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`)
|
|
}
|
|
|
|
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)
|
|
}
|
|
}
|
|
|
|
func run(flags userFlags) error {
|
|
if len(flags.args) < 2 {
|
|
return errors.New("not enough arguments")
|
|
}
|
|
|
|
var buf bytes.Buffer
|
|
var out io.Writer = os.Stdout
|
|
if flags.outFile != "" {
|
|
out = &buf
|
|
}
|
|
|
|
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,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err = m.Mock(out, args...); err != nil {
|
|
return err
|
|
}
|
|
|
|
if flags.outFile == "" {
|
|
return nil
|
|
}
|
|
|
|
// create the file
|
|
err = os.MkdirAll(filepath.Dir(flags.outFile), 0750)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return ioutil.WriteFile(flags.outFile, buf.Bytes(), 0600)
|
|
}
|