ifacemaker.go

package main

import (
	"fmt"
	"log"
	"os"
	"path/filepath"

	"github.com/jessevdk/go-flags"
	"github.com/vburenin/ifacemaker/maker"
)

type cmdlineArgs struct {
	Files           []string `short:"f" long:"file" description:"Go source file to read, either filename or glob" required:"true"`
	StructType      string   `short:"s" long:"struct" description:"Generate an interface for this structure name" required:"true"`
	IfaceName       string   `short:"i" long:"iface" description:"Name of the generated interface" required:"true"`
	PkgName         string   `short:"p" long:"pkg" description:"Package name for the generated interface" required:"true"`
	IfaceComment    string   `short:"y" long:"iface-comment" description:"Comment for the interface, default is '// <iface> ...'"`
	ImportModule    string   `short:"m" long:"import-module" description:"Fully qualified module import for packages with a different target package '// <iface> ...'"`
	ExcludeMethods  []string `short:"e" long:"exclude-method" description:"Name of method that will be excluded from output interface"`
	WithNotExported bool     `short:"x" long:"not-exported" description:"Include not exported methods"`

	// jessevdk/go-flags doesn't support default values for boolean flags,
	// so we use a string for backwards-compatibility and then convert it to a bool later.
	CopyDocs string `short:"d" long:"doc" description:"Copy docs from methods" option:"true" option:"false" default:"true"`
	copyDocs bool

	CopyTypeDoc bool   `short:"D" long:"type-doc" description:"Copy type doc from struct"`
	Comment     string `short:"c" long:"comment" description:"Append comment to top, default is '// Code generated by ifacemaker; DO NOT EDIT.'"`
	Output      string `short:"o" long:"output" description:"Output file name. If not provided, result will be printed to stdout."`
}

func main() {
	var args cmdlineArgs
	_, err := flags.ParseArgs(&args, os.Args)
	if err != nil {
		if flags.WroteHelp(err) {
			return
		}
		// No need to log the error, flags.ParseArgs() already does this
		os.Exit(1)
	}

	// Workaround because jessevdk/go-flags doesn't support default values for boolean flags
	args.copyDocs = args.CopyDocs == "true"

	if args.IfaceComment == "" {
		args.IfaceComment = fmt.Sprintf("%s ...", args.IfaceName)
	}

	if args.Comment == "" {
		args.Comment = "Code generated by ifacemaker; DO NOT EDIT."
	}

	var files []string
	for _, filePattern := range args.Files {
		matches, err := filepath.Glob(filePattern)
		if err != nil {
			log.Fatal(err)
		}
		files = append(files, matches...)
	}
	result, err := maker.Make(maker.MakeOptions{
		Files:           files,
		StructType:      args.StructType,
		Comment:         args.Comment,
		PkgName:         args.PkgName,
		IfaceName:       args.IfaceName,
		IfaceComment:    args.IfaceComment,
		CopyDocs:        args.copyDocs,
		CopyTypeDoc:     args.CopyTypeDoc,
		ImportModule:    args.ImportModule,
		ExcludeMethods:  args.ExcludeMethods,
		WithNotExported: args.WithNotExported,
	})
	if err != nil {
		log.Fatal(err.Error())
	}

	if args.Output == "" {
		fmt.Println(string(result))
	} else {
		f, err := os.Create(args.Output)
		if err != nil {
			log.Fatal(err)
		}
		if _, err := f.Write(result); err != nil {
			log.Fatal(err)
		}
	}
}