/***********************************************************************************************************************
 *
 *   MIT License
 *
 *   Copyright (c) 2021 Nicolas JUHEL
 *
 *   Permission is hereby granted, free of charge, to any person obtaining a copy
 *   of this software and associated documentation files (the "Software"), to deal
 *   in the Software without restriction, including without limitation the rights
 *   to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 *   copies of the Software, and to permit persons to whom the Software is
 *   furnished to do so, subject to the following conditions:
 *
 *   The above copyright notice and this permission notice shall be included in all
 *   copies or substantial portions of the Software.
 *
 *   THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 *   IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 *   FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 *   AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 *   LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 *   OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 *   SOFTWARE.
 *
 *
 **********************************************************************************************************************/

package shell

import (
	"fmt"
	"io"
	"os"
	"strings"

	libshl "github.com/c-bata/go-prompt"
	liberr "github.com/nabbar/golib/errors"
	shlcmd "github.com/nabbar/golib/shell/command"
)

func (s *shell) RunPrompt(out, err io.Writer, opt ...libshl.Option) {
	var (
		cmp = make([]libshl.Suggest, 0)
	)

	_ = s.Walk(func(name string, item shlcmd.Command) (shlcmd.Command, liberr.Error) {
		cmp = append(cmp, libshl.Suggest{
			Text:        name,
			Description: item.Describe(),
		})

		return nil, nil
	})

	p := libshl.New(
		func(inputLine string) {
			if out == nil {
				out = os.Stdout
			}

			if err == nil {
				err = os.Stderr
			}

			inputLine = strings.TrimSpace(inputLine)
			if inputLine == "" {
				return
			} else if inputLine == "quit" || inputLine == "exit" {
				_, _ = fmt.Fprintf(out, "Bye !\n")
				os.Exit(0)
				return
			}

			s.Run(out, err, strings.Fields(inputLine))
		},
		func(document libshl.Document) []libshl.Suggest {
			return libshl.FilterHasPrefix(cmp, document.GetWordBeforeCursor(), true)
		},
		opt...,
	)
	p.Run()
}
