Add CLI option -mocks-directory

main.go

@@ -31,6 +31,7 @@ func parseConfig() (c config.Config) {
 	flag.StringVar(&c.StaticFiles, "static-files", ".", "The location of the static files to serve (index.html, etc.)")
 	flag.IntVar(&c.HistoryMaxRetention, "history-retention", 0, "The maximum number of calls to keep in the history by sessions (0 = infinity)")
 	flag.StringVar(&c.PersistenceDirectory, "persistence-directory", "", "If defined, the directory where the sessions will be synchronized")
+	flag.StringVar(&c.MocksDirectory, "mocks-directory", "", "If defined, the directory use to load the mocks at server start")
 	flag.Parse()
 	return
 }

server/admin_server.go

@@ -35,6 +35,15 @@ func Serve(config config.Config) {
 	adminServerEngine.Use(recoverMiddleware(), loggerMiddleware(), middleware.Gzip())
 
 	mockServerEngine, mockServices := NewMockServer(config)
+
+	// Autoload mocks at start
+	if config.MocksDirectory != "" {
+		log.WithField("port", config.ConfigListenPort).Infof("Loading mocks from %s", config.MocksDirectory)
+		if err := mockServices.Load(config.MocksDirectory); err != nil {
+			log.WithField("port", config.ConfigListenPort).Fatal(err)
+		}
+	}
+
 	graphServices := services.NewGraph()
 	handler := handlers.NewAdmin(mockServices, graphServices)
 

server/config/config.go

@@ -8,6 +8,7 @@ type Config struct {
 	StaticFiles          string
 	HistoryMaxRetention  int
 	PersistenceDirectory string
+	MocksDirectory       string
 	Build                Build
 }
 

server/services/mocks.go

@@ -1,14 +1,18 @@
 package services
 
 import (
+	"encoding/json"
 	"fmt"
+	"io/ioutil"
+	"path/filepath"
 	"regexp"
 	"strings"
 	"sync"
 	"time"
 
 	"github.com/Thiht/smocker/server/types"
 	"github.com/teris-io/shortid"
+	"gopkg.in/yaml.v3"
 )
 
 var (
@@ -32,6 +36,7 @@ type Mocks interface {
 	GetSessions() types.Sessions
 	SetSessions(sessions types.Sessions)
 	Reset(force bool)
+	Load(path string) error
 }
 
 type mocks struct {
@@ -315,3 +320,59 @@ func (s *mocks) Reset(force bool) {
 
 	go s.persistence.StoreSessions(s.sessions.Clone())
 }
+
+func (s *mocks) Load(path string) error {
+	mockslist := make(map[string][]types.Mock)
+
+	files, err := ioutil.ReadDir(path)
+	if err != nil {
+		return fmt.Errorf("unable to find mocks files: %w ", err)
+	}
+
+	for _, file := range files {
+		ext := filepath.Ext(file.Name())
+
+		if ext != ".yml" && ext != ".yaml" && ext != ".json" {
+			continue
+		}
+
+		content, err := ioutil.ReadFile(filepath.Join(path, file.Name()))
+		if err != nil {
+			return fmt.Errorf("unable to read mock file %s: %w ", file.Name(), err)
+		}
+
+		var Ms []types.Mock
+
+		switch ext {
+		case ".yml", ".yaml":
+			err = yaml.Unmarshal(content, &Ms)
+		case ".json":
+			err = json.Unmarshal(content, &Ms)
+		}
+
+		if err != nil {
+			return fmt.Errorf("unable to parse mock file %s: %w ", file.Name(), err)
+		}
+
+		for _, mock := range Ms {
+			if err = mock.Validate(); err != nil {
+				return fmt.Errorf("unvalid mock file %s: %w ", file.Name(), err)
+			}
+		}
+
+		mockslist[file.Name()] = Ms
+	}
+
+	sessionID := s.GetLastSession().ID
+
+	for fileName, mocks := range mockslist {
+		for _, mock := range mocks {
+			m := mock
+			if _, err = s.AddMock(sessionID, &m); err != nil {
+				return fmt.Errorf("unable to add mock file %s: %w ", fileName, err)
+			}
+		}
+	}
+
+	return nil
+}