ssmsh.go

package main

import (
	"flag"
	"fmt"
	"io/ioutil"
	"os"
	"strings"

	"github.com/abiosoft/ishell"
	"github.com/bwhaley/ssmsh/commands"
	"github.com/bwhaley/ssmsh/config"
	"github.com/bwhaley/ssmsh/parameterstore"
	"github.com/mattn/go-shellwords"
)

var Version string

func main() {
	cfgFile := flag.String("config", "", "Load configuration from the specified file")
	file := flag.String("file", "", "Read commands from file (use - for stdin)")
	version := flag.Bool("version", false, "Display the current version")
	flag.Parse()

	if *version {
		fmt.Println("Version", Version)
		os.Exit(0)
	}

	cfg, err := config.ReadConfig(*cfgFile)
	if err != nil {
		fmt.Printf("Error reading configuration file %s: %s\n", *cfgFile, err)
		os.Exit(1)
	}

	shell := ishell.New()
	var ps parameterstore.ParameterStore
	ps.SetDefaults(cfg)
	err = ps.NewParameterStore(true)
	if err != nil {
		shell.Println("Error initializing session. Is your authentication correct?", err)
		os.Exit(1)
	}
	commands.Init(shell, &ps, &cfg)

	if *file == "-" {
		processStdin(shell)
	} else if *file != "" {
		processFile(shell, *file)
	} else if len(flag.Args()) > 1 {
		err := shell.Process(flag.Args()...)
		if err != nil {
			shell.Println("Error executing shell process:", err)
			shell.Println("This might be a bug. Please open an issue at github.com/bwhaley/ssmsh.\n")
			os.Exit(1)
		}
	} else {
		shell.Run()
		shell.Close()
	}
}

func processStdin(shell *ishell.Shell) {
	data, err := ioutil.ReadAll(os.Stdin)
	if err != nil {
		shell.Println("Error reading from stdin:", err)
		os.Exit(1)
	}
	processData(shell, string(data))
}

func processFile(shell *ishell.Shell, fn string) {
	data, err := ioutil.ReadFile(fn)
	if err != nil {
		shell.Println("Error reading from file:", err)
	}
	processData(shell, string(data))
}

func processData(shell *ishell.Shell, data string) {
	lines := strings.Split(data, "\n")
	for _, line := range lines {
		if line == "" || string(line[0]) == "#" {
			continue
		}
		args, err := shellwords.Parse(line)
		if err != nil {
			msg := fmt.Errorf("Error parsing %s: %v", line, err)
			shell.Println(msg)
			os.Exit(1)
		}
		err = shell.Process(args...)
		if err != nil {
			msg := fmt.Errorf("Error executing %s: %v", line, err)
			shell.Println(msg)
			os.Exit(1)
		}
	}
}