package helper

import (
	"github.com/pkg/errors"
	"github.com/spf13/pflag"
)

// PathFlag will use path flag to specify config file and dot path
type PathFlag struct {
	// path flag mode
	RemainArgs []string
	flags      *pflag.FlagSet
}

// NewPathFlag new PathFlag
func NewPathFlag(args []string) *PathFlag {
	pathFlag := &PathFlag{
		RemainArgs: args,
		flags:      pflag.NewFlagSet("", pflag.ContinueOnError),
	}
	return pathFlag
}

// Flags return flags of PathFlag
func (f *PathFlag) Flags() *pflag.FlagSet {
	return f.flags
}

// Parse do flag parse
func (f *PathFlag) Parse() error {
	err := f.flags.Parse(f.RemainArgs)
	if err != nil {
		return errors.WithStack(err)
	}
	f.RemainArgs = f.flags.Args()
	return nil
}