shell.go

package main

import (
	"bufio"
	"fmt"
	"os"
	"strings"
)

//Cmd command structure
type Cmd struct {
	Func func(string) error
}

//Shell interactive interface
type Shell struct {
	CmdList      map[string]Cmd
	Done         *bool
	enableReport *bool

	//show menu
	enableMenu bool
}

//Init initilize interactive shell
func (s *Shell) Init() {
	s.CmdList = make(map[string]Cmd)
	s.enableMenu = false
}

//AddCmd set up command
func (s *Shell) AddCmd(cn string, f func(string) error) {
	s.CmdList[cn] = Cmd{Func: f}
}

//Run invoke shell process
func (s *Shell) Run() {
	var oldEnableReport bool

	for *s.Done == false {
		r := bufio.NewReader(os.Stdin)
		txt, _ := r.ReadString('\n')
		if txt == "\n" {
			if s.enableMenu == false {
				s.CmdList["help"].Func("")
				s.enableMenu = true
				oldEnableReport = *s.enableReport
				*s.enableReport = false

			} else {
				s.enableMenu = false
				*s.enableReport = oldEnableReport
				fmt.Println("\033[H\033[2J")
			}
		} else {
			str := strings.TrimSuffix(txt, "\n")
			clist := strings.SplitN(str, " ", 2)
			cmd, f := s.CmdList[clist[0]]

			var param string

			if len(clist) > 1 {
				param = clist[1]
			}
			if f == true {
				err := cmd.Func(param)

				if err != nil {
					fmt.Println("Err: ", err)
				}

				s.enableMenu = false
			}
		}
	}
}

//Stop stop shell process
func (s *Shell) Stop() {
	*s.Done = true
}