Add the fix command

The fix command recreate the missing folders and files to make the wo
configuration works

internal/cmd/fix.go

@@ -0,0 +1,21 @@
+package cmd
+
+import (
+	"github.com/spf13/cobra"
+)
+
+func newFixCmd(workspaceManager workspaceManager) *cobra.Command {
+	return &cobra.Command{
+		Use:   "fix",
+		Short: "Fix the possible failure in the config folder",
+		Args:  cobra.NoArgs,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			err := workspaceManager.Fix()
+			if err != nil {
+				return err
+			}
+			cmd.Print(regularStyle.Render("Config folder fixed"))
+			return nil
+		},
+	}
+}

internal/cmd/fix_test.go

@@ -0,0 +1,57 @@
+package cmd
+
+import (
+	"bytes"
+	"errors"
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestNewFixCmd(t *testing.T) {
+	type scenario struct {
+		name  string
+		setup func(*testing.T) workspaceManager
+		test  func(*testing.T, *bytes.Buffer, *bytes.Buffer, error)
+	}
+	scenarios := []scenario{
+		{
+			"An error occurred when fixing the config",
+			func(t *testing.T) workspaceManager {
+				w := newMockWorkspaceManager(t)
+				w.Mock.On("Fix").Return(errors.New("an error occurred"))
+				return w
+			},
+			func(t *testing.T, outBuf *bytes.Buffer, errBuf *bytes.Buffer, err error) {
+				assert.Error(t, err)
+			},
+		},
+		{
+			"Creating a workspace env successfully",
+			func(t *testing.T) workspaceManager {
+				w := newMockWorkspaceManager(t)
+				w.Mock.On("Fix").Return(nil)
+				return w
+			},
+			func(t *testing.T, outBuf *bytes.Buffer, errBuf *bytes.Buffer, err error) {
+				assert.NoError(t, err)
+				assert.Equal(t, "Config folder fixed", outBuf.String())
+			},
+		},
+	}
+	for _, s := range scenarios {
+		t.Run(s.name, func(t *testing.T) {
+			os.Setenv("EDITOR", "emacs")
+			os.Setenv("SHELL", "/bin/sh")
+			errBuf := &bytes.Buffer{}
+			outBuf := &bytes.Buffer{}
+			w := s.setup(t)
+			cmd := newFixCmd(w)
+			cmd.SetArgs([]string{})
+			cmd.SetErr(errBuf)
+			cmd.SetOut(outBuf)
+			s.test(t, outBuf, errBuf, cmd.Execute())
+		})
+	}
+}

internal/cmd/interface.go

@@ -13,6 +13,7 @@ type workspaceManager interface {
 	CreateEnv(string, string) error
 	Edit(string) error
 	EditEnv(string, string) error
+	Fix() error
 	List() ([]workspace.Workspace, error)
 	RunFunction(string, string, []string) error
 	Remove(string) error

internal/cmd/root.go

@@ -94,6 +94,7 @@ func newRootCmd() *cobra.Command {
 	rootCmd.AddCommand(envCmd)
 	rootCmd.AddCommand(globalCmd)
 	rootCmd.AddCommand(newSetupCmd(w))
+	rootCmd.AddCommand(newFixCmd(w))
 	rootCmd.AddCommand(newCreateCmd(w, dirCompMgr))
 	rootCmd.AddCommand(newEditCmd(w, wksCompMgr))
 	rootCmd.AddCommand(newListCmd(w))

internal/workspace/workspace.go

@@ -325,6 +325,35 @@ func (s WorkspaceManager) Migrate() error {
 	return nil
 }
 
+func (s WorkspaceManager) Fix() error {
+	entries, err := os.ReadDir(s.getWorkspacesDir())
+	if os.IsNotExist(err) {
+		return err
+	}
+	err = os.MkdirAll(s.getWorkspacesDir(), 0o777)
+	if err != nil {
+		return err
+	}
+	for _, e := range entries {
+		if !e.IsDir() {
+			continue
+		}
+		err = s.createWorkspaceFolder(e.Name())
+		if err != nil {
+			return err
+		}
+		err = s.createFile(s.resolveFunctionFile(e.Name()))
+		if err != nil {
+			return err
+		}
+		err = s.createFile(s.resolveEnvFile(e.Name(), defaultEnv))
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
 func (s WorkspaceManager) appendLoadStatement(name string, env string, functionAndArgs []string) []string {
 	data := []string{}
 	data = append(data, s.CreateEnvVariableStatement(fmt.Sprintf("%s_NAME", envVariablePrefix), name))

internal/workspace/workspace_test.go

@@ -586,6 +586,63 @@ func TestRemove(t *testing.T) {
 	}
 }
 
+func TestFix(t *testing.T) {
+	config := &config{}
+	project := &project{}
+	type scenario struct {
+		name  string
+		setup func(*testing.T, WorkspaceManager)
+		test  func(*testing.T, error)
+	}
+	scenarios := []scenario{
+		{
+			"Fix a failing workspace",
+			func(*testing.T, WorkspaceManager) {
+				path := config.getPath(t)
+				err := os.RemoveAll(path + "/workspaces/test/envs")
+				assert.NoError(t, err)
+				err = os.Remove(path + "/workspaces/test/functions/functions.bash")
+				assert.NoError(t, err)
+				err = os.RemoveAll(path + "/workspaces/front/envs")
+				assert.NoError(t, err)
+				err = os.Remove(path + "/workspaces/front/functions/functions.bash")
+				assert.NoError(t, err)
+			},
+			func(t *testing.T, e error) {
+				assert.NoError(t, e)
+				path := config.getPath(t)
+				_, err := os.Stat(path + "/workspaces/test/functions/functions.bash")
+				assert.NoError(t, err)
+				_, err = os.Stat(path + "/workspaces/test/envs/default.bash")
+				assert.NoError(t, err)
+				_, err = os.Stat(path + "/workspaces/front/functions/functions.bash")
+				assert.NoError(t, err)
+				_, err = os.Stat(path + "/workspaces/front/envs/default.bash")
+				assert.NoError(t, err)
+			},
+		},
+	}
+	for _, s := range scenarios {
+		t.Run(s.name, func(t *testing.T) {
+			os.RemoveAll(config.getPath(t))
+			w, err := NewWorkspaceManager(WithEditor("emacs", "emacs"), WithShellPath("/bin/bash"), WithConfigPath(config.getPath(t)))
+			assert.NoError(t, err)
+			err = w.Create("test", project.getPath(t))
+			assert.NoError(t, err)
+			err = w.CreateEnv("test", "prod")
+			assert.NoError(t, err)
+			err = w.CreateEnv("test", "dev")
+			assert.NoError(t, err)
+			err = w.Create("front", project.getPath(t))
+			assert.NoError(t, err)
+			err = w.CreateEnv("front", "dev")
+			assert.NoError(t, err)
+			s.setup(t, w)
+			s.test(t, w.Fix())
+		})
+	}
+}
+
 func TestSetConfig(t *testing.T) {
 	config := &config{}
 	project := &project{}