Company now works

application/controllers/index.go

@@ -0,0 +1,44 @@
+package controllers
+
+import (
+	"html/template"
+	"log"
+	"net/http"
+	"os"
+	"path/filepath"
+)
+
+func Index(w http.ResponseWriter, r *http.Request) {
+	lp := filepath.Join("application", "layouts", "main.html")
+	fp := filepath.Join("application", "views", "index.html")
+
+	// Return a 404 if the template doesn't exist
+	info, err := os.Stat(fp)
+	if err != nil {
+		if os.IsNotExist(err) {
+			http.NotFound(w, r)
+			return
+		}
+	}
+
+	// Return a 404 if the request is for a directory
+	if info.IsDir() {
+		http.NotFound(w, r)
+		return
+	}
+
+	tmpl, err := template.ParseFiles(lp, fp)
+	if err != nil {
+		// Log the detailed error
+		log.Print(err.Error())
+		// Return a generic "Internal Server Error" message
+		http.Error(w, http.StatusText(500), 500)
+		return
+	}
+
+	err = tmpl.ExecuteTemplate(w, "main.html", nil)
+	if err != nil {
+		log.Print(err.Error())
+		http.Error(w, http.StatusText(500), 500)
+	}
+}

application/controllers/signup.go

@@ -0,0 +1,115 @@
+package controllers
+
+import (
+	"html/template"
+	"log"
+	"net/http"
+	"net/url"
+	"os"
+	"path/filepath"
+	"risklet/db"
+)
+
+func Signup(w http.ResponseWriter, r *http.Request) {
+	if r.Method == "GET" {
+		handleGet(w, r)
+	} else if r.Method == "POST" {
+		handlePost(w, r)
+	} else {
+		http.Error(w, "Method not allowed.", http.StatusMethodNotAllowed)
+		return
+	}
+}
+
+func handlePost(w http.ResponseWriter, r *http.Request) {
+	if err := r.ParseForm(); err != nil {
+		log.Println("Error processing form: ", err)
+		handleGet(w, r)
+	}
+	company := createCompany(r.PostForm)
+	companyId, err := db.InsertCompany(company)
+	if err != nil {
+		log.Println("Error inserting company into database ", err)
+		handleGet(w, r)
+	}
+
+	basicProfile := createBasicProfile(companyId, r.PostForm)
+
+	_, err = db.InsertBasicProfile(basicProfile)
+	if err != nil {
+		log.Println("Error inserting into database ", err)
+		handleGet(w, r)
+	}
+
+}
+
+func handleGet(w http.ResponseWriter, r *http.Request) {
+	lp := filepath.Join("application", "layouts", "main.html")
+	fp := filepath.Join("application", "views", "signup.html")
+
+	log.Println("Hitting Signup")
+
+	// Return a 404 if the template doesn't exist
+	info, err := os.Stat(fp)
+	if err != nil {
+		if os.IsNotExist(err) {
+			http.NotFound(w, r)
+			return
+		}
+	}
+
+	// Return a 404 if the request is for a directory
+	if info.IsDir() {
+		http.NotFound(w, r)
+		return
+	}
+
+	tmpl, err := template.ParseFiles(lp, fp)
+	if err != nil {
+		// Log the detailed error
+		log.Print(err.Error())
+		// Return a generic "Internal Server Error" message
+		http.Error(w, http.StatusText(500), 500)
+		return
+	}
+
+	err = tmpl.ExecuteTemplate(w, "main.html", nil)
+	if err != nil {
+		log.Print(err.Error())
+		http.Error(w, http.StatusText(500), 500)
+	}
+}
+
+func createBasicProfile(companyId int, f url.Values) db.BasicProfile {
+	return db.BasicProfile{
+		CompanyId:           companyId,
+		Employees:           f.Get("Employees"),
+		Revenue:             f.Get("Revenue"),
+		Applications:        f.Get("Applications"),
+		Compliance:          f.Get("Compliance"),
+		Industry:            f.Get("Industry"),
+		ITDependency:        f.Get("ITDependency"),
+		DataSensitivity:     f.Get("DataSensitivity"),
+		DataVolume:          f.Get("DataVolume"),
+		NetworkSegmentation: f.Get("NetworkSegmentation"),
+		LegacySystems:       f.Get("LegacySystems"),
+		IoTIntegration:      f.Get("IoTIntegration"),
+		RemoteWork:          f.Get("RemoteWork"),
+		BYOD:                f.Get("BYOD"),
+		VPN:                 f.Get("VPN"),
+		API:                 f.Get("API"),
+		VendorAccess:        f.Get("VendorAccess"),
+		InternalDev:         f.Get("InternalDev"),
+	}
+}
+
+func createCompany(f url.Values) db.Company {
+	return db.Company{
+		UUID:     db.GenerateRandomString(),
+		Name:     f.Get("Name"),
+		Email:    f.Get("Email"),
+		TaxId:    f.Get("TaxId"),
+		Password: db.GenerateRandomString(),
+	}
+
+}