package whois

import (
	"flag"
	"strings"
)

func shellSplit(s string) []string {
	split := strings.Split(s, " ")

	var result []string
	var inquote string
	var block string
	for _, i := range split {
		if inquote == "" {
			if strings.HasPrefix(i, "'") || strings.HasPrefix(i, "\"") {
				inquote = string(i[0])
				block = strings.TrimPrefix(i, inquote) + " "
			} else {
				result = append(result, i)
			}
		} else {
			if !strings.HasSuffix(i, inquote) {
				block += i + " "
			} else {
				block += strings.TrimSuffix(i, inquote)
				inquote = ""
				result = append(result, block)
				block = ""
			}
		}
	}

	return result
}

type Flags struct {
	ServerInfo string
	Args       []string
}

func parseFlags(request string) (*Flags, *flag.FlagSet, error) {
	args := shellSplit(request)
	set := flag.NewFlagSet("whois42d", flag.ContinueOnError)
	f := Flags{}
	set.StringVar(&f.ServerInfo, "q", "", "[version|sources|types] query specified server info")
	if err := set.Parse(args); err != nil {
		return nil, set, err
	}
	f.Args = set.Args()
	return &f, set, nil
}