cli.go

package pi

import (
	"fmt"
	"io"
	"log"

	flags "github.com/jessevdk/go-flags"
)

const (
	exitCodeOK = iota
	exitCodeErr
)

// CLI is struct for command line tool
type CLI struct {
	OutStream, ErrStream io.Writer
}

// Run the pi
func (cli *CLI) Run(argv []string) int {
	log.SetOutput(cli.ErrStream)
	log.SetFlags(0)
	err := parseArgs(argv)
	if err != nil {
		if ferr, ok := err.(*flags.Error); ok {
			if ferr.Type == flags.ErrHelp {
				return exitCodeOK
			}
			return exitCodeErr
		}
		return exitCodeErr
	}
	return exitCodeOK
}

type piOpts struct {
	Users         usersCommand         `description:"operate Users" command:"users" subcommands-optional:"true"`
	Channels      channelsCommand      `description:"operate Channels" command:"channels" subcommands-optional:"true"`
	Graphs        graphsCommand        `description:"operate Graphs" command:"graphs" subcommands-optional:"true"`
	Pixel         pixelCommand         `description:"operate Pixel in Graph" command:"pixel" subcommands-optional:"true"`
	Webhooks      webhooksCommand      `description:"operate Webhooks" command:"webhooks" subcommands-optional:"true"`
	Ver           verCommand           `description:"display version" command:"version" subcommands-optional:"true"`
	Notifications notificationsCommand `description:"operate Notifications" command:"ntf" subcommands-optional:"true"`
}

type verCommand struct{}

func (b *verCommand) Execute(args []string) error {
	fmt.Printf("pi version: %s (rev: %s)\n", version, revision)
	return nil
}

func parseArgs(args []string) error {
	opts := &piOpts{}
	_, err := flags.ParseArgs(opts, args)
	return err
}