package commands

import (
	"fmt"
	"os"
)

type RunCommandArgs struct {
	Command string `json:"command"`
	RemoteCommand string `json:"remote_command"`
	PrivateKeyFile string `json:"private_key_file"`
	Username string `json:"username"`
	Host string `json:"host"`
}

func (rca *RunCommandArgs) ReadArgs() {

	if len(os.Args) != 6 {
		fmt.Fprintf(os.Stderr, "Usage: %s run_command <remote-command> <private-key> <username> <host>\n", os.Args[0])
		os.Exit(1)
	}

	rca.Command = os.Args[1]
	rca.RemoteCommand = os.Args[2]
	rca.PrivateKeyFile = os.Args[3]
	rca.Username = os.Args[4]
	rca.Host = os.Args[5]

}

func RunCommand(args RunCommandArgs) {

	//args := RunCommandArgs{} 
	//args.ReadArgs()

	client := GetClient(args.PrivateKeyFile, args.Username, args.Host)
	defer client.Close()

	session, err := client.NewSession()
	if err != nil {
		fmt.Fprintf(os.Stderr, "Error creating SSH session: %v\n", err)
		os.Exit(1)
	}
	defer session.Close()

	session.Stdout = os.Stdout
	session.Stderr = os.Stderr
	if err := session.Run(args.RemoteCommand); err != nil {
		fmt.Fprintf(os.Stderr, "Error running command: %v\n", err)
		os.Exit(1)
	}
	return

}