Create and save tmuxp config

Author: arunvelsriram

pkg/kubetmuxp/kubetmuxp.go

@@ -7,6 +7,7 @@ import (
 
 	"github.com/arunvelsriram/kube-tmuxp/pkg/filesystem"
 	"github.com/arunvelsriram/kube-tmuxp/pkg/kubeconfig"
+	"github.com/arunvelsriram/kube-tmuxp/pkg/tmuxp"
 	yaml "gopkg.in/yaml.v2"
 )
 
@@ -113,6 +114,23 @@ func (c *Config) Process() error {
 			}
 			c.kubeCfg.RenameContext(defaultCtxName, cluster.Context, kubeCfgFile)
 
+			fmt.Println("Creating tmuxp config...")
+			windows := tmuxp.Windows{{Name: "default"}}
+			env := tmuxp.Environment{"KUBECONFIG": kubeCfgFile}
+			for k, v := range cluster.Envs {
+				env[k] = v
+			}
+
+			tmuxpCfg, err := tmuxp.NewConfig(cluster.Context, windows, env, c.filesystem)
+			if err != nil {
+				return err
+			}
+
+			tmuxpCfgFile := path.Join(tmuxpCfg.TmuxpConfigsDir(), fmt.Sprintf("%s.yaml", cluster.Context))
+			if err := tmuxpCfg.Save(tmuxpCfgFile); err != nil {
+				return err
+			}
+
 			fmt.Println("")
 		}
 	}

pkg/tmuxp/tmuxp.go

@@ -4,12 +4,13 @@ import (
 	"path"
 
 	"github.com/arunvelsriram/kube-tmuxp/pkg/filesystem"
+	yaml "gopkg.in/yaml.v2"
 )
 
 // Window represents a window in a tmux session
 type Window struct {
-	Name  string
-	Panes []struct{}
+	Name  string     `yaml:"window_name"`
+	Panes []struct{} `yaml:"panes"`
 }
 
 // Windows represents a list of windows in a tmux session
@@ -20,9 +21,9 @@ type Environment map[string]string
 
 // Config represents tmuxp config
 type Config struct {
-	SessionName string
-	Windows
-	Environment
+	SessionName  string `yaml:"session_name"`
+	Windows      `yaml:"windows"`
+	Environment  `yaml:"environment"`
 	filesystem   filesystem.FileSystem
 	tmuxpCfgsDir string
 }
@@ -33,6 +34,24 @@ func (c *Config) TmuxpConfigsDir() string {
 	return c.tmuxpCfgsDir
 }
 
+// Save saves the tmuxp config as file
+func (c *Config) Save(file string) error {
+	writer, err := c.filesystem.Create(file)
+	if err != nil {
+		return err
+	}
+
+	data, err := yaml.Marshal(c)
+	if err != nil {
+		return err
+	}
+	if _, err := writer.Write(data); err != nil {
+		return err
+	}
+
+	return nil
+}
+
 // NewConfig returns a new tmuxp config
 func NewConfig(sessionName string, windows Windows, environment Environment, fs filesystem.FileSystem) (*Config, error) {
 	home, err := fs.HomeDir()

pkg/tmuxp/tmuxp_test.go

@@ -1,6 +1,7 @@
 package tmuxp_test
 
 import (
+	"bytes"
 	"fmt"
 	"testing"
 
@@ -45,3 +46,44 @@ func TestTmuxpConfigsDir(t *testing.T) {
 
 	assert.Equal(t, "/Users/test/.tmuxp", tmuxpCfg.TmuxpConfigsDir())
 }
+
+func TestSave(t *testing.T) {
+	t.Run("should save the tmuxp config as file", func(t *testing.T) {
+		ctrl := gomock.NewController(t)
+		defer ctrl.Finish()
+
+		mockFS := mock.NewFileSystem(ctrl)
+		mockFS.EXPECT().HomeDir().Return("/Users/test", nil)
+		var writer bytes.Buffer
+		mockFS.EXPECT().Create("tmuxp-config.yaml").Return(&writer, nil)
+		tmuxpCfg, _ := tmuxp.NewConfig("session", tmuxp.Windows{{Name: "window"}}, tmuxp.Environment{"TEST_ENV": "value", "ANOTHER_TEST_ENV": "another-value"}, mockFS)
+
+		err := tmuxpCfg.Save("tmuxp-config.yaml")
+
+		actualContent := writer.String()
+		expectedContent := `session_name: session
+windows:
+- window_name: window
+  panes: []
+environment:
+  ANOTHER_TEST_ENV: another-value
+  TEST_ENV: value
+`
+		assert.Nil(t, err)
+		assert.Equal(t, expectedContent, actualContent)
+	})
+
+	t.Run("should return error if tmuxp config cannot be saved", func(t *testing.T) {
+		ctrl := gomock.NewController(t)
+		defer ctrl.Finish()
+
+		mockFS := mock.NewFileSystem(ctrl)
+		mockFS.EXPECT().HomeDir().Return("/Users/test", nil)
+		mockFS.EXPECT().Create("tmuxp-config.yaml").Return(nil, fmt.Errorf("some error"))
+		tmuxpCfg, _ := tmuxp.NewConfig("session", tmuxp.Windows{{Name: "window"}}, tmuxp.Environment{"TEST_ENV": "value", "ANOTHER_TEST_ENV": "another-value"}, mockFS)
+
+		err := tmuxpCfg.Save("tmuxp-config.yaml")
+
+		assert.EqualError(t, err, "some error")
+	})
+}