Example #1
0
// New Config creates a new Config object based on the given URI and other data
func NewConfig(uri string, authFile string, sshAgent bool, createPty bool, bindIoStreams bool) *Config {
	// prefix the uri with ssh:// if invalid
	if !strings.HasPrefix(uri, "ssh://") {
		uri = "ssh://" + uri
	}
	parsed, err := url.Parse(uri)
	if err != nil {
		logger.Fatal("Invalid uri provided: " + uri)
	}

	var (
		user string
		pass string
		port string
		host string
	)

	// load the user and password if provided
	if parsed.User != nil {
		user = parsed.User.Username()
		pass, _ = parsed.User.Password()
	}
	// load host and port if available
	host, port, err = net.SplitHostPort(parsed.Host)
	if err != nil {
		host = parsed.Host
	}

	// add default values if empty
	if host == "" {
		host = "localhost"
	}
	if port == "" {
		port = "22"
	}

	iPort, _ := strconv.Atoi(port)

	// return the filled config object
	return &Config{
		Host:          host,
		User:          user,
		Password:      pass,
		Port:          iPort,
		AuthFile:      authFile,
		SSHAgent:      sshAgent,
		CreatePty:     createPty,
		BindIOStreams: bindIoStreams,
	}
}
Example #2
0
package cmd

import (
	"github.com/Around25/shellbot/logger"
	"github.com/Around25/shellbot/ops"

	"github.com/spf13/cobra"
)

// copyCmd represents the copy command
var copyCmd = &cobra.Command{
	Use:   "copy",
	Short: "Copy a file or directory between the local environment and a specified server",
	Long:  `Copy a file or directory between the local environment and a specified server`,
	Run: func(cmd *cobra.Command, args []string) {
		err := ops.Copy(args[0], args[1], ops.NewConfig(AppConfig))
		if err != nil {
			logger.Fatal(err)
		}
	},
}

func init() {
	RootCmd.AddCommand(copyCmd)
}
Example #3
0
import (
	"github.com/Around25/shellbot/logger"
	"github.com/Around25/shellbot/ops"

	"github.com/spf13/cobra"
)

// shellCmd represents the shell command
var shellCmd = &cobra.Command{
	Use:   "shell",
	Short: "Connect to a single server through ssh",
	Long:  `Using shell you can connect open a ssh session to a server.`,
	Run: func(cmd *cobra.Command, args []string) {
		var name string
		if len(args) == 0 {
			logger.Fatal("Specify the name of the server you want to connect to.")
		}

		// get the name of the server as the first argument
		name = args[0]
		err := ops.OpenTerminalToServer(name, ops.NewConfig(AppConfig))
		if err != nil {
			logger.Fatal(err)
		}
	},
}

func init() {
	RootCmd.AddCommand(shellCmd)
}
Example #4
0
import (
	"github.com/Around25/shellbot/logger"
	"github.com/Around25/shellbot/ops"

	"github.com/spf13/cobra"
)

// setupCmd represents the setup command
var setupCmd = &cobra.Command{
	Use:   "setup",
	Short: "Execute all the tasks associated with each group of servers",
	Long:  `Execute all the tasks associated with each group of servers`,
	Run: func(cmd *cobra.Command, args []string) {
		var name string
		if len(args) == 0 {
			logger.Fatal("No environment specified")
		}
		name = args[0]

		err := ops.ProvisionEnvironment(name, ops.NewConfig(AppConfig))
		if err != nil {
			logger.Fatal(err)
		}
	},
}

func init() {
	RootCmd.AddCommand(setupCmd)
}