added some extra commands

cmd/add/add.go

@@ -0,0 +1,53 @@
+package add
+
+import (
+	"errors"
+	"git.denysoft.de/CubeBit/ssh-hub/util"
+	"github.com/jedib0t/go-pretty/v6/table"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+	"os"
+	"regexp"
+)
+
+var addCmd = &cobra.Command{
+	Use:   "add",
+	Short: "Add new command with alias",
+	Args: func(cmd *cobra.Command, args []string) error {
+		if err := cobra.ExactArgs(2)(cmd, args); err != nil {
+			return err
+		}
+
+		conf := viper.GetStringMapString("data")
+		if _, ok := conf[args[0]]; ok {
+			return errors.New("such alias already exists")
+		}
+
+		if matched, err := regexp.MatchString("/[a-zA-Z|1-9-]{1,}", args[0]); !matched {
+			cobra.CheckErr(err)
+			return errors.New("alias is in false format. Example of correct one: /work/dev. Regex: /[a-zA-Z|1-9-]{1,}")
+		}
+
+		return nil
+	},
+	Run: func(cmd *cobra.Command, args []string) {
+		cfg := viper.GetStringMapString("data")
+		cfg[args[0]] = args[1]
+
+		err := util.SaveConfig(&cfg)
+		cobra.CheckErr(err)
+
+		t := table.NewWriter()
+		t.SetOutputMirror(os.Stdout)
+		t.AppendHeader(table.Row{"Alias", "Command"})
+
+		for alias, command := range cfg {
+			t.AppendRow(table.Row{alias, command})
+		}
+		t.Render()
+	},
+}
+
+func Cmd() *cobra.Command {
+	return addCmd
+}