60 lines
1016 B
Go
60 lines
1016 B
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
|
|
"github.com/matryer/moq/pkg/moq"
|
|
)
|
|
|
|
func main() {
|
|
var err error
|
|
defer func() {
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
flag.Usage()
|
|
os.Exit(1)
|
|
|
|
}
|
|
}()
|
|
var (
|
|
outFile = flag.String("out", "", "output file (default stdout)")
|
|
pkgName = flag.String("pkg", "", "package name (default will infer)")
|
|
)
|
|
flag.Usage = func() {
|
|
fmt.Println(`moq [flags] destination interface [interface2 [interface3 [...]]]`)
|
|
flag.PrintDefaults()
|
|
}
|
|
flag.Parse()
|
|
args := flag.Args()
|
|
if len(args) < 2 {
|
|
err = errors.New("not enough arguments")
|
|
return
|
|
}
|
|
destination := args[0]
|
|
args = args[1:]
|
|
var buf bytes.Buffer
|
|
var out io.Writer
|
|
out = os.Stdout
|
|
if len(*outFile) > 0 {
|
|
out = &buf
|
|
}
|
|
m, err := moq.New(destination, *pkgName)
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = m.Mock(out, args...)
|
|
if err != nil {
|
|
return
|
|
}
|
|
// create the file
|
|
if len(*outFile) > 0 {
|
|
err = ioutil.WriteFile(*outFile, buf.Bytes(), 0777)
|
|
}
|
|
}
|