cli.go

package cli

import (
	"fmt"
	"io"

	"github.com/oinume/lekcije/backend/errors"
)

const (
	ExitOK    = 0
	ExitError = 1
)

type Main struct {
	command Commander
}

type Commander interface {
	SetOutStream(io.Writer)
	SetErrStream(io.Writer)
	//SetFlagSet(*flag.FlagSet)
	//Name() string
	Run([]string) error
}

func NewMain(command Commander, out, err io.Writer) *Main {
	command.SetOutStream(out)
	command.SetErrStream(err)
	return &Main{
		command: command,
	}
}

func (m *Main) Run(args []string) error {
	return m.command.Run(args)
}

func WriteError(w io.Writer, err error) {
	fmt.Fprintf(w, "%v", err.Error())
	fmt.Fprint(w, "\n--- stacktrace ---")
	switch e := err.(type) {
	case *errors.AnnotatedError:
		if e.OutputStackTrace() {
			fmt.Fprintf(w, "%+v\n", e.StackTrace())
		}
	default:
		fmt.Fprintf(w, "%+v", err)
	}
}