Merge pull request #19 from Kilgaloon/development

When new recipe is created, add it to queue

Author: kilgaloon

.gitignore

@@ -1,2 +1,4 @@
 .vscode
-debug
+/bin/debug
+/bin/leprechaun
+

configs/client.ini

@@ -1,9 +1,9 @@
 # we want to log our errors, and specify where
-error_log = ./logs/error-client.log
+error_log = ../logs/error-client.log
 # we want to log our info about app to know whats going on
-info_log = ./logs/info-client.log
+info_log = ../logs/info-client.log
 # path to recipes directory
-recipes_path = ./recipes
+recipes_path = ../recipes
 
 
 [variables]

src/client/main.go

@@ -1,9 +1,11 @@
 package client
 
 import (
-	"../log"
+	"fmt"
 	"io/ioutil"
 	"time"
+	"../log"
+	"github.com/fsnotify/fsnotify"
 )
 
 // Client settings and configurations
@@ -25,7 +27,34 @@ func Start(iniPath *string) {
 
 	q := BuildQueue(client.Config.recipesPath, files)
 
+	// watch for new recipes
+	watcher, err := fsnotify.NewWatcher()
+	if err != nil {
+		client.Logs.Error("Failed to create watcher")
+	}
+
+	defer watcher.Close()
+
+	go func() {
+		for {
+			select {
+			case event := <-watcher.Events:
+				if event.Op&fsnotify.Create == fsnotify.Create {
+					AddToQueue(&q.Stack, event.Name)
+				}
+			case err := <-watcher.Errors:
+				fmt.Println("error:", err)
+			}
+		}
+	}()
+
+	err = watcher.Add(client.Config.recipesPath)
+	if err != nil {
+		fmt.Println(err)
+	}
+
 	for {
+
 		go ProcessQueue(&q, client)
 
 		time.Sleep(60 * time.Second)

src/client/queue.go

@@ -7,6 +7,7 @@ import (
 	"os"
 	"os/exec"
 	"strings"
+	"path/filepath"
 )
 
 // Queue stack for pulling out recipes
@@ -28,6 +29,13 @@ func BuildQueue(path string, files []os.FileInfo) Queue {
 	return q
 }
 
+// AddToQueue takes freshly created recipes and add them to queue
+func AddToQueue(stack *[]recipe.Recipe, path string) {
+	if filepath.Ext(path) == ".yml" {
+		*stack = append(*stack, recipe.Build(path))
+	}
+}
+
 // ProcessQueue queue
 func ProcessQueue(queue *Queue, client *Client) {
 	for index, r := range queue.Stack {