add mode

cmd/root.go

@@ -5,6 +5,7 @@
 package cmd
 
 import (
+	"github.com/ingbyr/gohost/conf"
 	"github.com/ingbyr/gohost/display"
 	"github.com/ingbyr/gohost/host"
 	"github.com/spf13/cobra"
@@ -16,7 +17,7 @@ var rootCmd = &cobra.Command{
 }
 
 func Execute() {
-	host.M.SetCmdMode()
+	host.M.Init(conf.C.Mode)
 	rootCmd.AddCommand(listCmd)
 	rootCmd.AddCommand(editCmd)
 	rootCmd.AddCommand(useCmd)

conf/conf.go

@@ -1,6 +1,7 @@
 package conf
 
 import (
+	"fmt"
 	"github.com/ingbyr/gohost/display"
 	"github.com/ingbyr/gohost/editor"
 	"gopkg.in/ini.v1"
@@ -14,36 +15,50 @@ const (
 	TmpCombinedHost  = ".tmp_combined"
 	BaseHostFileName = "base"
 	HostFileExt      = ".txt"
+	ModeStorage      = "storage"
+	ModeMemory       = "memory"
 )
 
 type conf struct {
-	Editor string `ini:"editor"`
+	Mode   string
+	Editor string
 }
 
 var (
 	currUser, _  = user.Current()
 	BaseDir      = path.Join(currUser.HomeDir, ".gohost")
 	BaseHostFile = path.Join(BaseDir, "."+BaseHostFileName)
 	ConfigFile   = path.Join(BaseDir, "config.ini")
-	Conf         = &conf{
+	C            = &conf{
+		Mode:   ModeStorage,
 		Editor: editor.Default,
 	}
 )
 
 func init() {
 	// init config file
-	err := ini.MapTo(Conf, ConfigFile)
-	if err != nil {
-		Sync()
+	_ = ini.MapTo(C, ConfigFile)
+	checkMode()
+}
+
+func checkMode() {
+	switch C.Mode {
+	case ModeStorage, ModeMemory:
+	default:
+		display.ErrExit(fmt.Errorf("not valid mode %s", C.Mode))
 	}
 }
 
 func Sync() {
-	cfg := ini.Empty()
-	if err := ini.ReflectFrom(cfg, Conf); err != nil {
-		display.ErrExit(err)
-	}
-	if err := cfg.SaveTo(ConfigFile); err != nil {
-		display.ErrExit(err)
+	switch C.Mode {
+	case ModeStorage:
+		cfg := ini.Empty()
+		if err := ini.ReflectFrom(cfg, C); err != nil {
+			display.ErrExit(err)
+		}
+		if err := cfg.SaveTo(ConfigFile); err != nil {
+			display.ErrExit(err)
+		}
+	case ModeMemory:
 	}
 }

host/host_test.go

@@ -7,11 +7,12 @@ package host
 import (
 	"github.com/google/go-cmp/cmp"
 	"github.com/google/go-cmp/cmp/cmpopts"
+	"github.com/ingbyr/gohost/conf"
 	"testing"
 )
 
 func TestNewHostByFileName(t *testing.T) {
-	M.SetMockMode()
+	M.Init(conf.ModeMemory)
 	var tests = []struct {
 		name string
 		want *Host

host/manager.go

@@ -42,16 +42,17 @@ func init() {
 			FilePath: conf.BaseHostFile,
 			Groups:   nil,
 		},
+		editor: editor.New(conf.C.Editor),
 	}
 }
 
-func (m *manager) SetCmdMode() {
-	m.SetFs(myfs.NewOsFs())
-	m.editor = editor.New(conf.Conf.Editor)
-}
-
-func (m *manager) SetMockMode() {
-	m.SetFs(myfs.NewMemFs())
+func (m *manager) Init(mode string) {
+	switch mode {
+	case conf.ModeStorage:
+		M.SetFs(myfs.NewOsFs())
+	case conf.ModeMemory:
+		M.SetFs(myfs.NewMemFs())
+	}
 }
 
 func (m *manager) SetFs(newFs myfs.HostFs) {

host/manager_test.go

@@ -5,6 +5,7 @@
 package host
 
 import (
+	"github.com/ingbyr/gohost/conf"
 	"testing"
 )
 
@@ -23,7 +24,7 @@ type hostTestOp struct {
 }
 
 func TestManager_CreateRemoveNewHost(t *testing.T) {
-	M.SetMockMode()
+	M.Init(conf.ModeMemory)
 	var tests = []hostTestOp{
 		{
 			initHosts: []hosts{