pkger/cmd/pkger/pack.go

package main

import (
	"flag"
	"fmt"
	"os"
	"sort"

	"github.com/markbates/pkger"
	"github.com/markbates/pkger/parser"
	"github.com/markbates/pkger/pkging"
	"github.com/markbates/pkger/stuffing"
)

const outName = "pkged.go"

type packCmd struct {
	*flag.FlagSet
	help bool
	list bool
	subs []command
}

func (e *packCmd) Name() string {
	return e.Flags().Name()
}

func (e *packCmd) Exec(args []string) error {
	info, err := pkger.Current()
	if err != nil {
		return err
	}

	res, err := parser.Parse(info)
	if err != nil {
		return err
	}

	if e.list {
		fmt.Println(res.Path)

		for _, p := range res.Paths {
			fmt.Printf("  > %s\n", p)
		}
		return nil
	}

	fp := info.FilePath(outName)
	os.RemoveAll(fp)

	if err := Package(fp, res.Paths); err != nil {
		return err
	}

	return nil
}

func (e *packCmd) Route(args []string) error {
	e.Parse(args)

	if e.help {
		e.Usage()
		return nil
	}

	args = e.Args()

	if len(args) == 0 {
		return e.Exec(args)
	}

	k := args[0]
	for _, c := range e.subs {
		if k == c.Name() {
			args = args[1:]
			for _, a := range args {
				if a == "-h" {
					Usage(os.Stderr, c.Flags())()
					return nil
				}
			}
			return c.Exec(args)
		}
	}

	return e.Exec(args)
}

func New() (*packCmd, error) {
	c := &packCmd{}

	c.subs = []command{
		&readCmd{}, &serveCmd{}, &infoCmd{},
	}
	sort.Slice(c.subs, func(a, b int) bool {
		return c.subs[a].Name() <= c.subs[b].Name()
	})

	c.FlagSet = flag.NewFlagSet("pkger", flag.ExitOnError)
	c.BoolVar(&c.list, "list", false, "prints a list of files/dirs to be packaged")
	c.BoolVar(&c.help, "h", false, "prints help information")
	c.Usage = func() {
		fmt.Fprintf(os.Stderr, "Usage:\n\n")
		Usage(os.Stderr, c.FlagSet)()
		for _, s := range c.subs {
			Usage(os.Stderr, s.Flags())()
		}
	}
	return c, nil
}

func (e *packCmd) Flags() *flag.FlagSet {
	if e.FlagSet == nil {
		e.FlagSet = flag.NewFlagSet("", flag.ExitOnError)
		e.BoolVar(&e.list, "list", false, "prints a list of files/dirs to be packaged")
	}
	e.Usage = Usage(os.Stderr, e.FlagSet)
	return e.FlagSet
}

func Package(out string, paths []pkging.Path) error {
	os.RemoveAll(out)

	f, err := os.Create(out)
	if err != nil {
		return err
	}
	defer f.Close()

	c, err := pkger.Current()
	if err != nil {
		return err
	}
	fmt.Fprintf(f, "package %s\n\n", c.Name)
	fmt.Fprintf(f, "import \"github.com/markbates/pkger\"\n\n")
	fmt.Fprintf(f, "import \"github.com/markbates/pkger/pkging/mem\"\n\n")

	fmt.Fprintf(f, "var _ = pkger.Apply(mem.UnmarshalEmbed([]byte(`")

	if err := stuffing.Stuff(f, c, paths); err != nil {
		return err
	}

	fmt.Fprintf(f, "`)))\n")

	return nil
}