Inject examples only with new --example flag

The default use case should be to not inject example recipes.

5 files changed, 19 insertions, 6 deletions

diff --git a/Makefile b/Makefile
index 3bfd158..d90b0fb 100644
--- a/Makefile
+++ b/Makefile
@@ -11,7 +11,7 @@ $(BUILD_DIR)/$(TARGET_EXEC):
 
 .PHONY: debug
 debug:
-	go run -ldflags "-X main.version=$(VERSION)" $(MODULE_NAME)
+	go run -ldflags "-X main.version=$(VERSION)" $(MODULE_NAME) -e
 
 .PHONY: tests
 tests:
diff --git a/flags.go b/flags.go
index 708e832..c7d3cff 100644
--- a/flags.go
+++ b/flags.go
@@ -8,13 +8,16 @@ import (
 const help = `Ceres - Recipe server for local networks
 
 Usage: ceres [-h | --help] [-v | --version] [-c | --config]
+      [-e | --example]
 
   -h, --help                 show this help page and exit
   -v, --version              print version information
   -c, --config               file path to configuration file
+  -e, --examples             inject example recipes on startup
 `
 
-var printVersion bool
+var printVersion   bool
+var injectExamples bool
 
 func init() {
 	flag.BoolVar(&printVersion, "version", false, "print version information")
@@ -23,6 +26,9 @@ func init() {
 	flag.StringVar(&config.Path, "config", "", "file path to configuration file")
 	flag.StringVar(&config.Path, "c",      "", "file path to configuration file")
 
+	flag.BoolVar(&injectExamples, "examples", false, "inject example recipes on startup")
+	flag.BoolVar(&injectExamples, "e",        false, "inject example recipes on startup")
+
 	flag.Usage = func() {
 		fmt.Fprintf(flag.CommandLine.Output(), help)
 	}
diff --git a/main.go b/main.go
index 90a29e3..6d6467a 100644
--- a/main.go
+++ b/main.go
@@ -37,7 +37,11 @@ func main() {
 	model.ConnectDatabase(filepath.Join(storage.Path, "ceres.sqlite3"))
 	defer model.DisconnectDatabase()
 	model.MigrateDatabase(version)
-	model.InjectTestRecipes()
+
+	if injectExamples {
+		model.InjectExampleRecipes()
+		log.Println("Added example recipes")
+	}
 
 	server := NewServer(config.HttpAddress)
 	go server.Start()
diff --git a/model/database.go b/model/database.go
index 85d2662..f5fd67a 100644
--- a/model/database.go
+++ b/model/database.go
@@ -149,8 +149,8 @@ func MigrateDatabase(execVersion string) {
 	}
 }
 
-func InjectTestRecipes() {
-	Transaction(func(tx *sql.Tx) error {
+func InjectExampleRecipes() {
+	err := Transaction(func(tx *sql.Tx) error {
 		recipes := RecipeTestData()
 
 		for _, recipe := range recipes {
@@ -162,6 +162,9 @@ func InjectTestRecipes() {
 
 		return nil
 	})
+	if err != nil {
+		log.Fatalf("Failed to inject example recipes: %v", err)
+	}
 }
 
 func DisconnectDatabase() {
diff --git a/model/recipes_test.go b/model/recipes_test.go
index 159422f..13a8b2f 100644
--- a/model/recipes_test.go
+++ b/model/recipes_test.go
@@ -22,7 +22,7 @@ func TestRecipesRead(t *testing.T) {
 	defer DisconnectDatabase()
 	MigrateDatabase("dummy_version")
 
-	InjectTestRecipes()
+	InjectExampleRecipes()
 
 	r := make(Recipes, 0)