initial

2025-03-21 19:49:41 +11:00
commit 2ee078c992
60 changed files with 5410 additions and 0 deletions
--- a/server/handler/encryptData.go
+++ b/server/handler/encryptData.go
@@ -0,0 +1,61 @@
+package handler
+
+import (
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+)
+
+func (h *Handler) EncryptData(w http.ResponseWriter, r *http.Request) {
+	//ctx := context.Background()
+	var cipherText string
+
+	reqBody, err := io.ReadAll(r.Body)
+	if err != nil {
+		h.Logger.Error("Invalid data received", "error", err)
+		fmt.Fprintf(w, "Invalid data received")
+		w.WriteHeader(http.StatusInternalServerError)
+		return
+	} else {
+		h.Logger.Debug("received input data", "length", len(reqBody))
+	}
+
+	// get the json input
+	var input map[string]string
+	if err := json.Unmarshal(reqBody, &input); err != nil {
+		h.Logger.Error("unable to unmarshal json", "error", err)
+		prettyPrint(reqBody)
+		w.Header().Set("Content-Type", "application/json")
+		w.WriteHeader(http.StatusInternalServerError)
+		json.NewEncoder(w).Encode(map[string]string{
+			"status":  "ERROR",
+			"message": fmt.Sprintf("Unable to unmarshal JSON in request body: '%s'", err),
+		})
+		return
+	} else {
+		h.Logger.Debug("successfully decoded JSON")
+		//prettyPrint(input)
+	}
+
+	//cipher, err := h.Secret.Encrypt()
+	for k := range input {
+		//h.Logger.Debug("foo", "key", k, "value", input[k])
+		cipherText, err = h.Secret.Encrypt([]byte(input[k]))
+		if err != nil {
+			h.Logger.Error("Unable to encrypt", "error", err)
+		} else {
+			h.Logger.Debug("Encrypted plaintext", "length", len(input[k]), "ciphertext", cipherText)
+			w.WriteHeader(http.StatusOK)
+			json.NewEncoder(w).Encode(map[string]string{
+				"status":  "OK",
+				"message": cipherText,
+			})
+			return
+		}
+
+	}
+
+	// return the result
+
+}
--- a/server/handler/handler.go
+++ b/server/handler/handler.go
@@ -0,0 +1,33 @@
+package handler
+
+import (
+	"context"
+	"log/slog"
+	"net/http"
+	"wnzl-snow/db"
+	"wnzl-snow/internal/secrets"
+	"wnzl-snow/internal/settings"
+
+	"github.com/a-h/templ"
+)
+
+// Handler handles requests.
+type Handler struct {
+	Logger    *slog.Logger
+	Database  db.Database
+	BuildTime string
+	SHA1Ver   string
+	GoVersion string
+	//VcCreds   *vcenter.VcenterLogin
+	Secret   *secrets.Secrets
+	Settings *settings.Settings
+}
+
+func (h *Handler) html(ctx context.Context, w http.ResponseWriter, status int, t templ.Component) {
+	w.Header().Set("Content-Type", "text/html; charset=utf-8")
+	w.WriteHeader(status)
+
+	if err := t.Render(ctx, w); err != nil {
+		h.Logger.Error("Failed to render component", "error", err)
+	}
+}
--- a/server/handler/home.go
+++ b/server/handler/home.go
@@ -0,0 +1,62 @@
+package handler
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"runtime"
+	"time"
+	"wnzl-snow/components/views"
+)
+
+// Home handles the home page.
+func (h *Handler) Home(w http.ResponseWriter, r *http.Request) {
+	//h.html(r.Context(), w, http.StatusOK, core.HTML("Example Site", home.Home()))
+
+	// Render the template
+	/*
+		err := home.Home(h.BuildTime, h.SHA1Ver, h.GoVersion).Render(r.Context(), w)
+		if err != nil {
+			http.Error(w, "Failed to render template", http.StatusInternalServerError)
+		}
+	*/
+
+	info := views.BuildInfo{
+		BuildTime: h.BuildTime,
+		SHA1Ver:   h.SHA1Ver,
+		GoVersion: h.GoVersion,
+	}
+
+	err := views.Index(info).Render(r.Context(), w)
+	if err != nil {
+		http.Error(w, "Failed to render template", http.StatusInternalServerError)
+	}
+}
+
+// prettyPrint comes from https://gist.github.com/sfate/9d45f6c5405dc4c9bf63bf95fe6d1a7c
+func prettyPrint(args ...interface{}) {
+	var caller string
+
+	timeNow := time.Now().Format("01-02-2006 15:04:05")
+	prefix := fmt.Sprintf("[%s] %s -- ", "PrettyPrint", timeNow)
+	_, fileName, fileLine, ok := runtime.Caller(1)
+
+	if ok {
+		caller = fmt.Sprintf("%s:%d", fileName, fileLine)
+	} else {
+		caller = ""
+	}
+
+	fmt.Printf("\n%s%s\n", prefix, caller)
+
+	if len(args) == 2 {
+		label := args[0]
+		value := args[1]
+
+		s, _ := json.MarshalIndent(value, "", "\t")
+		fmt.Printf("%s%s: %s\n", prefix, label, string(s))
+	} else {
+		s, _ := json.MarshalIndent(args, "", "\t")
+		fmt.Printf("%s%s\n", prefix, string(s))
+	}
+}
--- a/server/handler/newSnow.go
+++ b/server/handler/newSnow.go
@@ -0,0 +1,22 @@
+package handler
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+)
+
+// NewSnow receives data from the DMSP Snow New() function
+func (h *Handler) NewSnow(w http.ResponseWriter, r *http.Request) {
+	var ()
+
+	reqBody, err := io.ReadAll(r.Body)
+	if err != nil {
+		h.Logger.Error("Invalid data received", "error", err)
+		fmt.Fprintf(w, "Invalid data received")
+		w.WriteHeader(http.StatusInternalServerError)
+		return
+	} else {
+		h.Logger.Debug("received input data", "length", len(reqBody))
+	}
+}
--- a/server/handler/reportDownload.go
+++ b/server/handler/reportDownload.go
@@ -0,0 +1,61 @@
+package handler
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"wnzl-snow/internal/report"
+)
+
+func (h *Handler) InventoryReportDownload(w http.ResponseWriter, r *http.Request) {
+
+	ctx := context.Background()
+
+	// Generate the XLSX report
+	reportData, err := report.CreateInventoryReport(h.Logger, h.Database, ctx)
+	if err != nil {
+		h.Logger.Error("Failed to create report", "error", err)
+		w.Header().Set("Content-Type", "application/json")
+		w.WriteHeader(http.StatusInternalServerError)
+		json.NewEncoder(w).Encode(map[string]string{
+			"status":  "ERROR",
+			"message": fmt.Sprintf("Unable to create xlsx report: '%s'", err),
+		})
+		return
+	}
+
+	// Set HTTP headers to indicate file download
+	w.Header().Set("Content-Type", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
+	w.Header().Set("Content-Disposition", `attachment; filename="inventory_report.xlsx"`)
+	w.Header().Set("File-Name", "inventory_report.xlsx")
+
+	// Write the XLSX file to the HTTP response
+	w.Write(reportData)
+}
+
+func (h *Handler) UpdateReportDownload(w http.ResponseWriter, r *http.Request) {
+
+	ctx := context.Background()
+
+	// Generate the XLSX report
+	reportData, err := report.CreateUpdatesReport(h.Logger, h.Database, ctx)
+	if err != nil {
+		h.Logger.Error("Failed to create report", "error", err)
+		w.Header().Set("Content-Type", "application/json")
+		w.WriteHeader(http.StatusInternalServerError)
+		json.NewEncoder(w).Encode(map[string]string{
+			"status":  "ERROR",
+			"message": fmt.Sprintf("Unable to create xlsx report: '%s'", err),
+		})
+		return
+	}
+
+	// Set HTTP headers to indicate file download
+	w.Header().Set("Content-Type", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
+	w.Header().Set("Content-Disposition", `attachment; filename="updates_report.xlsx"`)
+	w.Header().Set("File-Name", "updates_report.xlsx")
+
+	// Write the XLSX file to the HTTP response
+	w.Write(reportData)
+}
--- a/server/handler/updateCleanup.go
+++ b/server/handler/updateCleanup.go
@@ -0,0 +1,42 @@
+package handler
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+)
+
+// VmUpdate receives the CloudEvent for a VM modification or move
+func (h *Handler) UpdateCleanup(w http.ResponseWriter, r *http.Request) {
+	/*
+		// Get the current time
+		now := time.Now()
+		// Get the start of the current day (midnight today)
+		midnightToday := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location())
+		// Convert to Unix time
+		unixTime := midnightToday.Unix()
+
+		// create the database parameters
+		params := queries.CleanupUpdatesParams{
+			UpdateType: "diskchange",
+			UpdateTime: sql.NullInt64{Int64: unixTime, Valid: unixTime > 0},
+		}
+
+		h.Logger.Debug("database params", "params", params)
+		err := h.Database.Queries().CleanupUpdates(context.Background(), params)
+	*/
+
+	//err := h.Database.Queries().InventoryCleanupTemplates(context.Background())
+	err := h.Database.Queries().CleanupUpdatesNullVm(context.Background())
+
+	if err != nil {
+		h.Logger.Error("Error received cleaning updates table", "error", err)
+		w.WriteHeader(http.StatusInternalServerError)
+		fmt.Fprintf(w, "Delete Request unsuccessful %s\n", err)
+	} else {
+		h.Logger.Debug("Processed update cleanup successfully")
+		w.WriteHeader(http.StatusOK)
+		// TODO - return some JSON
+		fmt.Fprintf(w, "Processed update cleanup successfully")
+	}
+}