Delete kubeconfig file for the given kubecontext

Author: arunvelsriram

cmd/generate.go

@@ -5,6 +5,8 @@ import (
 	"fmt"
 	"os"
 
+	"github.com/arunvelsriram/kube-tmuxp/pkg/filesystem"
+	"github.com/arunvelsriram/kube-tmuxp/pkg/kubeconfig"
 	"github.com/arunvelsriram/kube-tmuxp/pkg/kubetmuxp"
 	"github.com/spf13/cobra"
 )
@@ -20,14 +22,19 @@ var generateCmd = &cobra.Command{
 			os.Exit(1)
 		}
 
-		kubetmuxpCfg := kubetmuxp.NewConfig(bufio.NewReader(reader))
+		bufioReader := bufio.NewReader(reader)
+		kubeCfg := kubeconfig.New(filesystem.Default{})
+		kubetmuxpCfg := kubetmuxp.New(bufioReader, kubeCfg)
 		err = kubetmuxpCfg.Load()
 		if err != nil {
 			fmt.Println(err)
 			os.Exit(1)
 		}
 
-		kubetmuxpCfg.Process()
+		if err = kubetmuxpCfg.Process(); err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
 	},
 }
 

pkg/filesystem/filesystem.go

@@ -2,15 +2,16 @@ package filesystem
 
 import "os"
 
-// FileSystem abcd
+// FileSystem represents a filesystem
 type FileSystem interface {
 	Remove(file string) error
 }
 
-type defaultFS struct{}
+// Default represents the Operating System's filesystem
+type Default struct{}
 
-// Remove abcd
-func (d *defaultFS) Remove(file string) error {
+// Remove removes a file from Default filesystem
+func (Default) Remove(file string) error {
 	if err := os.Remove(file); err != nil {
 		return err
 	}

pkg/kubeconfig/kubeconfig.go

@@ -1,16 +1,24 @@
 package kubeconfig
 
 import (
+	"fmt"
+	"os"
+	"path"
+
 	"github.com/arunvelsriram/kube-tmuxp/pkg/filesystem"
+	homedir "github.com/mitchellh/go-homedir"
 )
 
 // KubeConfig exposes methods to perform actions on kubeconfig
 type KubeConfig struct {
 	filesystem filesystem.FileSystem
+	dir        string
 }
 
-// Delete deletes the given kubeconfig file
-func (k *KubeConfig) Delete(file string) error {
+// Delete deletes the kubeconfig file for the given context
+func (k *KubeConfig) Delete(context string) error {
+	file := path.Join(k.dir, context)
+
 	if err := k.filesystem.Remove(file); err != nil {
 		return err
 	}
@@ -20,7 +28,15 @@ func (k *KubeConfig) Delete(file string) error {
 
 // New returns a new KubeConfig
 func New(fs filesystem.FileSystem) KubeConfig {
+	home, err := homedir.Dir()
+	if err != nil {
+		fmt.Println(err)
+		os.Exit(1)
+	}
+	kubeConfigsDir := path.Join(home, ".kube/configs")
+
 	return KubeConfig{
 		filesystem: fs,
+		dir:        kubeConfigsDir,
 	}
 }

pkg/kubeconfig/kubeconfig_test.go

@@ -23,10 +23,10 @@ func TestDelete(t *testing.T) {
 		ctrl := gomock.NewController(t)
 		defer ctrl.Finish()
 		mockFS := mock.NewFileSystem(ctrl)
-		mockFS.EXPECT().Remove("kubeconfig.yaml").Return(nil)
+		mockFS.EXPECT().Remove("/Users/arunvelsriram/.kube/configs/context-name").Return(nil)
 
 		cfg := kubeconfig.New(mockFS)
-		err := cfg.Delete("kubeconfig.yaml")
+		err := cfg.Delete("context-name")
 
 		assert.Nil(t, err)
 	})
@@ -35,10 +35,10 @@ func TestDelete(t *testing.T) {
 		ctrl := gomock.NewController(t)
 		defer ctrl.Finish()
 		mockFS := mock.NewFileSystem(ctrl)
-		mockFS.EXPECT().Remove("kubeconfig.yaml").Return(fmt.Errorf("some error"))
+		mockFS.EXPECT().Remove("/Users/arunvelsriram/.kube/configs/context-name").Return(fmt.Errorf("some error"))
 
 		cfg := kubeconfig.New(mockFS)
-		err := cfg.Delete("kubeconfig.yaml")
+		err := cfg.Delete("context-name")
 
 		assert.EqualError(t, err, "some error")
 	})

pkg/kubetmuxp/kubetmuxp.go

@@ -1,14 +1,14 @@
 package kubetmuxp
 
 import (
-	"fmt"
 	"io"
 	"io/ioutil"
 
+	"github.com/arunvelsriram/kube-tmuxp/pkg/kubeconfig"
 	yaml "gopkg.in/yaml.v2"
 )
 
-// Envs abcd
+// Envs reprensents environemnt variables
 type Envs map[string]string
 
 //Cluster represents a Kubernetes cluster
@@ -36,6 +36,7 @@ type Projects []Project
 type Config struct {
 	Projects `yaml:"projects"`
 	reader   io.Reader
+	kubeCfg  kubeconfig.KubeConfig
 }
 
 // Load constructs kube-tmuxp config from given file
@@ -57,16 +58,19 @@ func (c *Config) Load() error {
 func (c *Config) Process() error {
 	for _, project := range c.Projects {
 		for _, cluster := range project.Clusters {
-			fmt.Println(cluster)
+			if err := c.kubeCfg.Delete(cluster.Context); err != nil {
+				return err
+			}
 		}
 	}
 
 	return nil
 }
 
-// NewConfig creates a new Config
-func NewConfig(reader io.Reader) Config {
+// New creates a new kube-tmuxp Config
+func New(reader io.Reader, kubeCfg kubeconfig.KubeConfig) Config {
 	return Config{
-		reader: reader,
+		reader:  reader,
+		kubeCfg: kubeCfg,
 	}
 }

pkg/kubetmuxp/kubetmuxp_test.go

@@ -5,15 +5,20 @@ import (
 	"strings"
 	"testing"
 
+	"github.com/arunvelsriram/kube-tmuxp/pkg/filesystem"
+	"github.com/arunvelsriram/kube-tmuxp/pkg/kubeconfig"
 	"github.com/arunvelsriram/kube-tmuxp/pkg/kubetmuxp"
 	"github.com/stretchr/testify/assert"
 )
 
-func TestNewConfig(t *testing.T) {
+func TestNew(t *testing.T) {
 	var reader io.Reader
-	cfg := kubetmuxp.NewConfig(reader)
+	var fs filesystem.FileSystem
+	kubeCfg := kubeconfig.New(fs)
 
-	assert.NotNil(t, cfg)
+	kubetmuxpCfg := kubetmuxp.New(reader, kubeCfg)
+
+	assert.NotNil(t, kubetmuxpCfg)
 }
 
 func TestLoadConfig(t *testing.T) {
@@ -29,9 +34,11 @@ projects:
         envs:
           TEST_ENV: test-value`
 		reader := strings.NewReader(content)
+		var fs filesystem.FileSystem
+		kubeCfg := kubeconfig.New(fs)
+		kubetmuxpCfg := kubetmuxp.New(reader, kubeCfg)
 
-		cfg := kubetmuxp.NewConfig(reader)
-		err := cfg.Load()
+		err := kubetmuxpCfg.Load()
 
 		expectedProjects := kubetmuxp.Projects{
 			{
@@ -51,16 +58,18 @@ projects:
 		}
 
 		assert.Nil(t, err)
-		assert.Equal(t, cfg.Projects, expectedProjects)
+		assert.Equal(t, kubetmuxpCfg.Projects, expectedProjects)
 	})
 
 	t.Run("should return error if loading fails", func(t *testing.T) {
 		reader := strings.NewReader("invalid yaml")
+		var fs filesystem.FileSystem
+		kubeCfg := kubeconfig.New(fs)
+		kubetmuxpCfg := kubetmuxp.New(reader, kubeCfg)
 
-		cfg := kubetmuxp.NewConfig(reader)
-		err := cfg.Load()
+		err := kubetmuxpCfg.Load()
 
 		assert.NotNil(t, err)
-		assert.Equal(t, cfg.Projects, kubetmuxp.Projects(nil))
+		assert.Equal(t, kubetmuxpCfg.Projects, kubetmuxp.Projects(nil))
 	})
 }