-y parameter to execute script in -s mode

[vroby65]

I think that's good idea to add -y parameter to automatize shell request

[vroby65]

yes_support_patch.txt
this is the code:
in main.go
...
var temperature *string
var top_p *string
var max_length *string
var preprompt *string
var url *string
var yes = false
....
isShell := flag.Bool("s", false, "Generate and Execute shell commands.")
flag.BoolVar(isShell, "shell", false, "Generate and Execute shell commands.")

isYes := flag.Bool("y", false, "")
flag.BoolVar(isYes, "yes", false, "Deactivate confirm request.")

.....
case *isShell:
if len(prompt) > 1 {
go loading(&stopSpin)
trimmedPrompt := strings.TrimSpace(prompt)
if len(trimmedPrompt) < 1 {
fmt.Fprintln(os.Stderr, "You need to provide some text")
fmt.Fprintln(os.Stderr, Example: tgpt -s "How to update system")
os.Exit(1)
}
if *isYes {
yes=true
}
shellCommand(*preprompt + trimmedPrompt + contextText + pipedInput, yes)
.....

in helper.go

....

func shellCommand(input string, yes bool) {
// Get OS
operatingSystem := ""
if runtime.GOOS == "windows" {
operatingSystem = "Windows"
} else if runtime.GOOS == "darwin" {
operatingSystem = "MacOS"
} else if runtime.GOOS == "linux" {
result, err := exec.Command("lsb_release", "-si").Output()
distro := strings.TrimSpace(string(result))
if err != nil {
distro = ""
}
operatingSystem = "Linux" + "/" + distro
} else {
operatingSystem = runtime.GOOS
}

// Get Shell

shellName := "/bin/sh"

if runtime.GOOS == "windows" {
	shellName = "cmd.exe"

	if len(os.Getenv("PSModulePath")) > 0 {
		shellName = "powershell.exe"
	}
} else {
	shellEnv := os.Getenv("SHELL")
	if len(shellEnv) > 0 {
		shellName = shellEnv
	}
}

shellPrompt := fmt.Sprintf(
	"Your role: Provide only plain text without Markdown formatting. Do not show any warnings or information regarding your capabilities. Do not provide any description. If you need to store any data, assume it will be stored in the chat. Provide only %s command for %s without any description. If there is a lack of details, provide most logical solution. Ensure the output is a valid shell command. If multiple steps required try to combine them together. Prompt: %s\n\nCommand:", shellName, operatingSystem, input)

getCommand(shellPrompt)

}

// Get a command in response
func getCommand(shellPrompt string) {
checkInputLength(shellPrompt)

resp, err := providers.NewRequest(shellPrompt, structs.Params{ApiKey: *apiKey, ApiModel: *apiModel, Provider: *provider, Max_length: *max_length, Temperature: *temperature, Top_p: *top_p, Preprompt: *preprompt, Url: *url}, structs.ExtraOptions{})

if err != nil {
	stopSpin = true
	printConnectionErrorMsg(err)
}

defer resp.Body.Close()

stopSpin = true

code := resp.StatusCode

if code >= 400 {
	handleStatus400(resp)
}

fmt.Print("\r          \r")

scanner := bufio.NewScanner(resp.Body)

// Variables
fullLine := ""
previousText := ""

// Handling each part
for scanner.Scan() {
	newText := providers.GetMainText(scanner.Text(), *provider, shellPrompt)
	if len(newText) < 1 {
		continue
	}
	mainText := strings.Replace(newText, previousText, "", -1)
	previousText = newText
	fullLine += mainText

	bold.Print(mainText)
}
lineCount := strings.Count(fullLine, "\n") + 1
if lineCount == 1 {
	var userInput string
	if yes {
		userInput ="Y"
                    bold.Print("\n");
	} else {
		bold.Print("\n\nExecute shell command? [Y/n]: ")
		reader := bufio.NewReader(os.Stdin)
		userInput, _ = reader.ReadString('\n')
		userInput = strings.TrimSpace(userInput)
	}
	if userInput != "n" {			     
		var cmd *exec.Cmd
		if runtime.GOOS == "windows" {
			shellName := "cmd"

			if len(os.Getenv("PSModulePath")) > 0 {
				shellName = "powershell"
			}
			if shellName == "cmd" {
				cmd = exec.Command("cmd", "/C", fullLine)

			} else {
				cmd = exec.Command("powershell", fullLine)
			}

		} else {
			cmd = exec.Command("bash", "-c", fullLine)

		}

		cmd.Stdin = os.Stdin
		cmd.Stdout = os.Stdout
		cmd.Stderr = os.Stderr
		err = cmd.Run()

		if err != nil {
			fmt.Fprintln(os.Stderr, err)
		}
	}
	if err := scanner.Err(); err != nil {
		fmt.Fprintln(os.Stderr, "Some error has occurred. Error:", err)
		os.Exit(1)
	}
}

}

.......