From 01d6746c30f8da24274486b837e30f764f44fd30 Mon Sep 17 00:00:00 2001
From: Hayden <64056131+hay-kot@users.noreply.github.com>
Date: Wed, 12 Oct 2022 12:35:21 -0800
Subject: [PATCH] setup initialization when demo

---
 backend/app/api/demo.go | 62 +++++++++++++++++++++++++++++++++++++++++
 backend/app/api/main.go |  6 ++++
 2 files changed, 68 insertions(+)
 create mode 100644 backend/app/api/demo.go

diff --git a/backend/app/api/demo.go b/backend/app/api/demo.go
new file mode 100644
index 0000000..2bde834
--- /dev/null
+++ b/backend/app/api/demo.go
@@ -0,0 +1,62 @@
+package main
+
+import (
+	"context"
+	"encoding/csv"
+	"strings"
+
+	"github.com/hay-kot/homebox/backend/internal/services"
+	"github.com/rs/zerolog/log"
+)
+
+func (a *app) SetupDemo() {
+	csvText := `Import Ref,Location,Labels,Quantity,Name,Description,Insured,Serial Number,Model Number,Manufacturer,Notes,Purchase From,Purchased Price,Purchased Time,Lifetime Warranty,Warranty Expires,Warranty Details,Sold To,Sold Price,Sold Time,Sold Notes
+,Garage,IOT;Home Assistant; Z-Wave,1,Zooz Universal Relay ZEN17,"Zooz 700 Series Z-Wave Universal Relay ZEN17 for Awnings, Garage Doors, Sprinklers, and More | 2 NO-C-NC Relays (20A, 10A) | Signal Repeater | Hub Required (Compatible with SmartThings and Hubitat)",,,ZEN17,Zooz,,Amazon,39.95,10/13/2021,,,,,,,
+,Living Room,IOT;Home Assistant; Z-Wave,1,Zooz Motion Sensor,"Zooz Z-Wave Plus S2 Motion Sensor ZSE18 with Magnetic Mount, Works with Vera and SmartThings",,,ZSE18,Zooz,,Amazon,29.95,10/15/2021,,,,,,,
+,Office,IOT;Home Assistant; Z-Wave,1,Zooz 110v Power Switch,"Zooz Z-Wave Plus Power Switch ZEN15 for 110V AC Units, Sump Pumps, Humidifiers, and More",,,ZEN15,Zooz,,Amazon,39.95,10/13/2021,,,,,,,
+,Downstairs,IOT;Home Assistant; Z-Wave,1,Ecolink Z-Wave PIR Motion Sensor,"Ecolink Z-Wave PIR Motion Detector Pet Immune, White (PIRZWAVE2.5-ECO)",,,PIRZWAVE2.5-ECO,Ecolink,,Amazon,35.58,10/21/2020,,,,,,,
+,Entry,IOT;Home Assistant; Z-Wave,1,Yale Security Touchscreen Deadbolt,"Yale Security YRD226-ZW2-619 YRD226ZW2619 Touchscreen Deadbolt, Satin Nickel",,,YRD226ZW2619,Yale,,Amazon,120.39,10/14/2020,,,,,,,
+,Kitchen,IOT;Home Assistant; Z-Wave,1,Smart Rocker Light Dimmer,"UltraPro Z-Wave Smart Rocker Light Dimmer with QuickFit and SimpleWire, 3-Way Ready, Compatible with Alexa, Google Assistant, ZWave Hub Required, Repeater/Range Extender, White Paddle Only, 39351",,,‎39351,Honeywell,,Amazon,65.98,09/30/0202,,,,,,,
+`
+
+	var (
+		registration = services.UserRegistration{
+			Email:    "demo@email.com",
+			Name:     "Demo",
+			Password: "demo",
+		}
+	)
+
+	// First check if we've already setup a demo user and skip if so
+	_, err := a.services.User.Login(context.Background(), registration.Email, registration.Password)
+	if err == nil {
+		return
+	}
+
+	_, err = a.services.User.RegisterUser(context.Background(), registration)
+	if err != nil {
+		log.Err(err).Msg("Failed to register demo user")
+		log.Fatal().Msg("Failed to setup demo")
+	}
+
+	token, _ := a.services.User.Login(context.Background(), registration.Email, registration.Password)
+	self, _ := a.services.User.GetSelf(context.Background(), token.Raw)
+
+	// Read CSV Text
+	reader := csv.NewReader(strings.NewReader(csvText))
+	reader.Comma = ','
+
+	records, err := reader.ReadAll()
+	if err != nil {
+		log.Err(err).Msg("Failed to read CSV")
+		log.Fatal().Msg("Failed to setup demo")
+	}
+
+	err = a.services.Items.CsvImport(context.Background(), self.GroupID, records)
+	if err != nil {
+		log.Err(err).Msg("Failed to import CSV")
+		log.Fatal().Msg("Failed to setup demo")
+	}
+
+	log.Info().Msg("Demo setup complete")
+}
diff --git a/backend/app/api/main.go b/backend/app/api/main.go
index 0309784..350c2f4 100644
--- a/backend/app/api/main.go
+++ b/backend/app/api/main.go
@@ -144,5 +144,11 @@ func run(cfg *config.Config) error {
 		}
 	})
 
+	// TODO: Remove through external API that does setup
+	if cfg.Demo {
+		log.Info().Msg("Running in demo mode, creating demo data")
+		app.SetupDemo()
+	}
+
 	return app.server.Start(routes)
 }