make fetching args a member function

master
Joshua Herring 1 month ago
parent 013c0abd00
commit c55bcf60c3

@ -78,26 +78,25 @@ type RunCommandArgs struct {
Host string `json:"host"` Host string `json:"host"`
} }
func run_command_args() RunCommandArgs { func (rca *RunCommandArgs) ReadArgs() {
if len(os.Args) != 6 { if len(os.Args) != 6 {
fmt.Fprintf(os.Stderr, "Usage: %s run_command <remote-command> <private-key> <username> <host>\n", os.Args[0]) fmt.Fprintf(os.Stderr, "Usage: %s run_command <remote-command> <private-key> <username> <host>\n", os.Args[0])
os.Exit(1) os.Exit(1)
} }
return RunCommandArgs{ rca.Command = os.Args[1]
os.Args[1], rca.RemoteCommand = os.Args[2]
os.Args[2], rca.PrivateKeyFile = os.Args[3]
os.Args[3], rca.Username = os.Args[4]
os.Args[4], rca.Host = os.Args[5]
os.Args[5],
}
} }
func RunCommand() { func RunCommand() {
args := run_command_args() args := RunCommandArgs{}
args.ReadArgs()
client := GetClient(args.PrivateKeyFile, args.Username, args.Host) client := GetClient(args.PrivateKeyFile, args.Username, args.Host)
defer client.Close() defer client.Close()

Loading…
Cancel
Save