Add support for multiple workspaces

config/config.go

@@ -5,6 +5,7 @@ import (
 	"errors"
 	"fmt"
 	"os"
+	"sort"
 
 	"github.com/erroneousboat/termui"
 )
@@ -17,6 +18,7 @@ const (
 // Config is the definition of a Config struct
 type Config struct {
 	SlackToken   string                `json:"slack_token"`
+	Workspaces   map[string]Config     `json:"workspaces"`
 	Notify       string                `json:"notify"`
 	Emoji        bool                  `json:"emoji"`
 	SidebarWidth int                   `json:"sidebar_width"`
@@ -28,7 +30,7 @@ type Config struct {
 type keyMapping map[string]string
 
 // NewConfig loads the config file and returns a Config struct
-func NewConfig(filepath string) (*Config, error) {
+func NewConfig(filepath string, workspaceName string) (*Config, error) {
 	cfg := getDefaultConfig()
 
 	file, err := os.Open(filepath)
@@ -40,6 +42,43 @@ func NewConfig(filepath string) (*Config, error) {
 		return &cfg, fmt.Errorf("the slack-term config file isn't valid json: %v", err)
 	}
 
+	// If no workspace is specified, select the first (ABC-order).
+	if workspaceName == "" {
+		keys := make([]string, len(cfg.Workspaces))
+		i := 0
+		for k := range cfg.Workspaces {
+			keys[i] = k
+			i++
+		}
+		sort.Strings(keys)
+		workspaceName = keys[0]
+	}
+	workspaceConfig := cfg.Workspaces[workspaceName]
+
+	// Overwrite all options if they're specified in the
+	// workspace-specific config:
+	cfg.SlackToken = workspaceConfig.SlackToken
+
+	if workspaceConfig.Notify != "" {
+		cfg.Notify = workspaceConfig.Notify
+	}
+
+	// TODO: There's no way to distinguish between falsy and unset.
+	// cfg.Emoji = workspaceConfig.Emoji
+
+	if workspaceConfig.SidebarWidth != 0 {
+		cfg.SidebarWidth = workspaceConfig.SidebarWidth
+	}
+	if workspaceConfig.MainWidth != 0 {
+		cfg.MainWidth = workspaceConfig.MainWidth
+	}
+	if workspaceConfig.KeyMap != nil {
+		cfg.KeyMap = workspaceConfig.KeyMap
+	}
+	if workspaceConfig.Theme != *new(Theme) {
+		cfg.Theme = workspaceConfig.Theme
+	}
+
 	if cfg.SidebarWidth < 1 || cfg.SidebarWidth > 11 {
 		return &cfg, errors.New("please specify the 'sidebar_width' between 1 and 11")
 	}

context/context.go

@@ -36,7 +36,7 @@ type AppContext struct {
 
 // CreateAppContext creates an application context which can be passed
 // and referenced througout the application
-func CreateAppContext(flgConfig string, flgToken string, flgDebug bool, version string, usage string) (*AppContext, error) {
+func CreateAppContext(flgConfig string, flgWorkspace string, flgDebug bool, version string, usage string) (*AppContext, error) {
 	if flgDebug {
 		go func() {
 			http.ListenAndServe(":6060", nil)
@@ -47,19 +47,15 @@ func CreateAppContext(flgConfig string, flgToken string, flgDebug bool, version
 	views.Loading()
 
 	// Load config
-	config, err := config.NewConfig(flgConfig)
+	config, err := config.NewConfig(flgConfig, flgWorkspace)
 	if err != nil {
 		return nil, err
 	}
 
-	// When slack token isn't set in the config file, we'll check
+	// When slack token still isn't set in the config file, we'll check
 	// the command-line flag or the environment variable
 	if config.SlackToken == "" {
-		if flgToken != "" {
-			config.SlackToken = flgToken
-		} else {
-			config.SlackToken = os.Getenv("SLACK_TOKEN")
-		}
+		config.SlackToken = os.Getenv("SLACK_TOKEN")
 	}
 
 	// Create desktop notifier

main.go

@@ -31,17 +31,17 @@ WEBSITE:
 
 GLOBAL OPTIONS:
    -config [path-to-config-file]
-   -token [slack-token]
+   -workspace [slack-workspace]
    -debug
    -help, -h
 `
 )
 
 var (
-	flgConfig string
-	flgToken  string
-	flgDebug  bool
-	flgUsage  bool
+	flgConfig    string
+	flgWorkspace string
+	flgDebug     bool
+	flgUsage     bool
 )
 
 func init() {
@@ -51,21 +51,28 @@ func init() {
 		log.Fatal(err)
 	}
 
-	// Parse flags
+	// Parse flags.
+
+	// Path to the config file to use.
+	// Defaults to ~/.slack-term
 	flag.StringVar(
 		&flgConfig,
 		"config",
 		path.Join(usr.HomeDir, ".slack-term"),
 		"location of config file",
 	)
 
+	// The name of the workspace to use.
+	// If none is provided, will pick the alphabetical first (in other words,
+	// will pick the only workspace in a list of only one workspace).
 	flag.StringVar(
-		&flgToken,
-		"token",
+		&flgWorkspace,
+		"workspace",
 		"",
-		"the slack token",
+		"the slack workspace to use",
 	)
 
+	// Toggle debug mode.
 	flag.BoolVar(
 		&flgDebug,
 		"debug",
@@ -100,7 +107,7 @@ func main() {
 	// Create context
 	usage := fmt.Sprintf(USAGE, VERSION)
 	ctx, err := context.CreateAppContext(
-		flgConfig, flgToken, flgDebug, VERSION, usage,
+		flgConfig, flgWorkspace, flgDebug, VERSION, usage,
 	)
 	if err != nil {
 		termbox.Close()