func getDocument(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "driver", "/"+c.GetPathVar("slug")) { return } var driver Driver if ok := db.GetAs("driver", c.GetFromSession("id").(string), &driver); !ok { c.SetFlash("alertError", "Error finding driver") c.Logout() http.Redirect(w, r, "/"+c.GetPathVar("slug"), 303) return } var company Company if ok := db.GetAs("company", c.GetFromSession("companyId").(string), &company); !ok { c.SetFlash("alertError", "Error finding company") c.Logout() http.Redirect(w, r, "/"+c.GetPathVar("slug"), 303) return } var document Document if ok := db.GetAs("document", c.GetPathVar("id"), &document); !ok { c.SetFlash("alertError", "Error finding document") //c.Logout() http.Redirect(w, r, "/"+c.GetPathVar("slug")+"/driver", 303) return } ts.Render(w, document.Name+".tmpl", tmpl.Model{ "driver": driver, "company": company, "document": document, }) }
func AdminDriverDocumentDelete(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") { return } service.DeleteDocument(c.GetPathVar("documentId")) c.SetFlash("alertSuccess", "Successfully deleted document") http.Redirect(w, r, "/admin/driver/"+c.GetPathVar("driverId")+"/document", 303) }
func AdminVehicleDelete(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") { return } service.DeleteVehicle(c.GetPathVar("vehicleId")) c.SetFlash("alertSuccess", "Successfuly deleted vehicle") http.Redirect(w, r, "/admin/vehicle", 303) return }
// POST admin delete company func AdminCompanyDelete(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") { return } service.DeleteCompany(c.GetPathVar("id")) c.SetFlash("alertSuccess", "Successfully deleted company") http.Redirect(w, r, "/admin/company", 303) return }
func DevCommentSave(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "developer") { return } comment := service.FindOneComment(c.GetPathVar("id")) closed, _ := strconv.ParseBool(r.FormValue("closed")) comment.Closed = closed service.SaveComment(comment) http.Redirect(w, r, "/dev/comment", 303) }
func AdminDriverDocumentSave(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") { return } document := service.FindOneDocument(r.FormValue("id")) document.Data = r.FormValue("data") service.SaveDocument(document) c.SetFlash("alertSuccess", "Successfully save document") fmt.Fprintf(w, "/admin/driver/%s/document", c.GetPathVar("driverId")) }
// POSt driver save document func saveDocument(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/"+c.GetPathVar("slug"), "driver") { return } document := service.FindOneDocument(r.FormValue("id")) document.Data = r.FormValue("data") service.SaveDocument(document) c.SetFlash("alertSuccess", "Successfully save document") fmt.Fprintf(w, "/%s/driver", c.GetPathVar("slug")) }
// GET admin get new vehicle page func AdminCompanyVehicleNew(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") { return } msgK, msgV := c.GetFlash() ts.Render(w, "admin-company-vehicle-form.tmpl", tmpl.Model{ msgK: msgV, "vehicles": service.FindAllVehicleByCompany(c.GetPathVar("companyId")), "company": service.FindOneCompany(c.GetPathVar("companyId")), }) }
// GET admin get all company's drivers func AdminCompanyDriverGetAll(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") { return } msgK, msgV := c.GetFlash() ts.Render(w, "admin-company-driver.tmpl", tmpl.Model{ msgK: msgV, "drivers": service.FindAllDriverByCompany(c.GetPathVar("companyId")), "company": service.FindOneCompany(c.GetPathVar("companyId")), }) return }
func AdminVehicleGetOne(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") { return } msgK, msgV := c.GetFlash() ts.Render(w, "admin-vehicle-form.tmpl", tmpl.Model{ msgK: msgV, "vehicle": service.FindOneVehicle(c.GetPathVar("id")), "vehicles": service.FindAllVehicle(), "companies": service.CompanyNames(), }) return }
func AdminDriverDocumentGetAll(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") { return } msgK, msgV := c.GetFlash() ts.Render(w, "admin-driver-document.tmpl", tmpl.Model{ msgK: msgV, "driver": service.FindOneDriver(c.GetPathVar("driverId")), "documents": service.FindAllDocumentByDriver(c.GetPathVar("driverId")), "dqfs": service.DQFS, }) return }
// GET admin get company func AdminCompanyGetOne(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") { return } msgK, msgV := c.GetFlash() ts.Render(w, "admin-company.tmpl", tmpl.Model{ msgK: msgV, "company": service.FindOneCompany(c.GetPathVar("id")), "driverCount": service.GetDriverCount(c.GetPathVar("id")), "vehicleCount": service.GetVehicleCount(c.GetPathVar("id")), }) return }
func companyRegister(w http.ResponseWriter, r *http.Request, c *web.Context) { var company Company if ok := GetCompBySlug(c.GetPathVar("slug"), &company); !ok || !company.Feature { fmt.Fprintf(w, "404 Not Page Found") return } msgK, msgV := c.GetFlash() ts.Render(w, "company-register.tmpl", tmpl.Model{ msgK: msgV, "company": company, }) return }
func AdminDriverDocumentGetOne(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") { return } driver := service.FindOneDriver(c.GetPathVar("driverId")) document := service.FindOneDocument(c.GetPathVar("documentId")) ts.Render(w, document.Name+".tmpl", tmpl.Model{ "driver": driver, "company": service.FindOneCompany(driver.CompanyId), "document": document, "admin": true, }) }
// GET admin get one employee func AdminEmployeeGetOne(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "developer") { return } msgK, msgV := c.GetFlash() ts.Render(w, "admin-employee-form.tmpl", tmpl.Model{ msgK: msgV, "employees": service.FindAllEmployee(), "employee": service.FindOneEmployee(c.GetPathVar("id")), "users": service.UserRoles(), }) return }
// GET get company register page func companyRegister(w http.ResponseWriter, r *http.Request, c *web.Context) { company, ok := service.FindOneCompanyBySlug(c.GetPathVar("slug")) if !ok || !company.Feature { fmt.Fprintf(w, "404 Not Page Found") return } msgK, msgV := c.GetFlash() ts.Render(w, "company-register.tmpl", tmpl.Model{ msgK: msgV, "company": company, }) return }
// POST admin save vehilce to company func AdminCompanyVehicleSave(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") { return } r.ParseForm() vehicle := service.FindOneVehicle(r.FormValue("id")) util.FormToStruct(&vehicle, r.Form, "") if vehicle.Id == "" { vehicle.Id = util.UUID4() } service.SaveVehicle(vehicle) c.SetFlash("alertSuccess", "Successfully saved vehicle") http.Redirect(w, r, "/admin/company/"+c.GetPathVar("companyId")+"/vehicle", 303) return }
// GET admin get driver func AdminDriverGetOne(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") { return } msgK, msgV := c.GetFlash() driver := service.FindOneDriver(c.GetPathVar("id")) ts.Render(w, "admin-driver-form.tmpl", tmpl.Model{ msgK: msgV, "driver": driver, "drivers": service.FindAllDriver(), "companies": service.CompanyNames(), "user": service.FindOneUser(driver.UserId), }) return }
// POST admin add driver to company func AdminCompanyDriverAdd(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") { return } if !service.CanUpdateUser("", r.FormValue("email")) { c.SetFlash("alertError", "Email already registered") http.Redirect(w, r, "/admin/company/"+c.GetPathVar("companyId")+"/driver/new", 303) return } r.ParseForm() driver, user := service.NewDriver(r.Form) service.SaveDriver(driver) service.SaveUser(user) c.SetFlash("alertSuccess", "Successfully added driver") http.Redirect(w, r, "/admin/company/"+c.GetPathVar("companyId")+"/driver", 303) return }
func postCompanyRegister(w http.ResponseWriter, r *http.Request, c *web.Context) { company, ok := service.FindOneCompanyBySlug(c.GetPathVar("slug")) if !ok || !company.Feature { fmt.Fprintf(w, "404 Not Page Found") return } if !service.CanUpdate("", r.FormValue("email")) { c.SetFlash("alertError", "Email already registered") http.Redirect(w, r, "/"+c.GetPathVar("slug")+"/register", 303) return } driver, user := service.NewDriver(r.Form) fmt.Println(driver.DOB) dobTS, err := time.Parse("2006-01-02", driver.DOB) if err != nil { c.SetFlash("alertError", "Error invalid date of birth") http.Redirect(w, r, "/"+c.GetPathVar("slug")+"/register", 303) return } if age := service.GetDriverAge(dobTS.UnixNano()); age < company.AgeRequirement { c.SetFlash("alertError", "Not old enough sucker") user.Active = false driver.Status = 0 driver.CompanyId = "" service.SaveUser(user) service.SaveDriver(driver) http.Redirect(w, r, "/"+c.GetPathVar("slug")+"/register", 303) return } document := service.Document{ Id: util.UUID4(), DocumentId: "100", CompanyId: company.Id, DriverId: driver.Id, Name: "dqf-100", Complete: false, } service.SaveUser(user) service.SaveDriver(driver) service.SaveDocument(document) c.SetFlash("alertSuccess", "Successfully Registered for "+company.Name+". Please Login") http.Redirect(w, r, "/"+c.GetPathVar("slug")+"/login", 303) return }
func AdminDriverDocumentAdd(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") { return } driver := service.FindOneDriver(c.GetPathVar("driverId")) docIds := util.SliceString(r.FormValue("docIds"), ',') for _, id := range docIds { document := service.Document{ Id: util.UUID4(), Name: "dqf-" + id, DocumentId: id, Complete: false, CompanyId: driver.CompanyId, DriverId: driver.Id, } service.SaveDocument(document) } c.SetFlash("alertSuccess", "Successfully added documents") http.Redirect(w, r, "/admin/driver/"+c.GetPathVar("driverId")+"/document", 303) }
// POST admin save driver func AdminDriverSave(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") { return } if !service.CanUpdateUser(r.FormValue("userId"), r.FormValue("email")) { c.SetFlash("alertError", "Email already registered") http.Redirect(w, r, "/admin/driver/"+c.GetPathVar("id"), 303) return } r.ParseForm() driver := service.FindOneDriver(r.FormValue("id")) util.FormToStruct(&driver, r.Form, "") user := service.FindOneUser(driver.UserId) util.FormToStruct(&user, r.Form, "") user.Id = driver.UserId service.SaveDriver(driver) service.SaveUser(user) c.SetFlash("alertSuccess", "Successfully saved driver") http.Redirect(w, r, "/admin/driver", 303) return }
// POST post to cmopany login func postCompanyLogin(w http.ResponseWriter, r *http.Request, c *web.Context) { company, ok := service.FindOneCompanyBySlug(c.GetPathVar("slug")) if !ok || !company.Feature { fmt.Fprintf(w, "404 Not Page Found") return } driver, ok2 := service.FindOneDriverByCompanyLogin(r.FormValue("email"), r.FormValue("password"), company.Id) if !ok2 { c.SetFlash("alertError", "Invalid email or password") http.Redirect(w, r, "/"+company.Slug+"/login", 303) return } c.Login("driver") c.SetSession(map[string]interface{}{ "id": driver.Id, "companyId": driver.CompanyId, "userId": driver.UserId, "email": driver.Email, "slug": company.Slug, }) c.SetFlash("alertSuccess", "Welcome "+driver.FirstName+" "+driver.LastName) http.Redirect(w, r, "/"+c.GetPathVar("slug")+"/driver", 303) return }
func postCompanyLogin(w http.ResponseWriter, r *http.Request, c *web.Context) { var company Company if ok := db.GetAs("company", r.FormValue("companyId"), &company); !ok || !company.Feature { fmt.Fprintf(w, "404 Not Page Found") return } var driver Driver if ok := GetDriverFromLogin(r.FormValue("email"), r.FormValue("password"), company.Id, &driver); !ok { c.SetFlash("alertError", "Invalid email or password") http.Redirect(w, r, "/"+company.Slug+"/login", 303) return } c.Login("driver") c.SetSession(map[string]interface{}{ "id": driver.Id, "companyId": driver.CompanyId, "userId": driver.UserId, "email": driver.Email, "slug": company.Slug, }) c.SetFlash("alertSuccess", "Welcome "+driver.FirstName+" "+driver.LastName) http.Redirect(w, r, "/"+c.GetPathVar("slug")+"/driver", 303) return }
// GET driver get document func getDocument(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/"+c.GetPathVar("slug"), "driver") { return } if c.GetFromSession("slug") != c.GetPathVar("slug") { http.Redirect(w, r, "/"+c.GetFromSession("slug").(string)+"/document/"+c.GetPathVar("id"), 303) } document := service.FindOneDocument(c.GetPathVar("id")) driver := service.FindOneDriver(c.GetFromSession("id").(string)) if document.DriverId != driver.Id { http.Redirect(w, r, "/"+c.GetFromSession("slug").(string)+"/driver/", 303) return } ts.Render(w, document.Name+".tmpl", tmpl.Model{ "driver": driver, "company": service.FindOneCompany(c.GetFromSession("companyId").(string)), "document": document, }) }
// POST admin upload company logo func AdminUploadCompanyLogo(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") { return } path := "static/logo/" if err := os.MkdirAll(path, 0755); err != nil { c.SetFlash("alertError", "Error uploading file") http.Redirect(w, r, "/admin/company/"+c.GetPathVar("id"), 303) return } r.ParseMultipartForm(32 << 20) // 32 MB file, handler, err := r.FormFile("logo") if err != nil || len(handler.Header["Content-Type"]) < 1 { fmt.Println(err) c.SetFlash("alertError", "Error uploading file") http.Redirect(w, r, "/admin/company/"+c.GetPathVar("id"), 303) return } defer file.Close() if handler.Header["Content-Type"][0] != "image/png" && handler.Header["Content-Type"][0] != "image/jpeg" { fmt.Println(err) c.SetFlash("alertError", "Error uploading file") http.Redirect(w, r, "/admin/company/"+c.GetPathVar("id"), 303) return } company := service.FindOneCompany(c.GetPathVar("id")) f, err := os.OpenFile(path+company.Id+".png", os.O_WRONLY|os.O_CREATE, 0666) if err != nil { fmt.Println(err) c.SetFlash("alertError", "Error uploading file") http.Redirect(w, r, "/admin/company/"+company.Id, 303) return } defer f.Close() io.Copy(f, file) company.Logo = true service.SaveCompany(company) c.SetFlash("alertSuccess", "Successfully uploaded file") http.Redirect(w, r, "/admin/company/"+company.Id, 303) return }
func saveDocument(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "driver", "/"+c.GetPathVar("slug")) { return } var document Document if ok := db.GetAs("document", r.FormValue("id"), &document); !ok { c.SetFlash("alertError", "Error finding document") http.Redirect(w, r, "/"+c.GetPathVar("slug")+"/driver", 303) return } document.Data = r.FormValue("data") db.Set("document", r.FormValue("id"), document) c.SetFlash("alertSuccess", "Successfully save document") fmt.Fprintf(w, "/%s/driver", c.GetPathVar("slug")) }
func getDriverHome(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "/"+c.GetPathVar("slug"), "driver") { return } if c.GetFromSession("slug") != c.GetPathVar("slug") { http.Redirect(w, r, "/"+c.GetFromSession("slug").(string)+"/driver", 303) } msgK, msgV := c.GetFlash() ts.Render(w, "driver-home.tmpl", tmpl.Model{ msgK: msgV, "driver": service.FindOneDriver(c.GetFromSession("id").(string)), "slug": c.GetPathVar("slug"), "documents": service.FindAllIncompleteDocumentByDriver(c.GetFromSession("id").(string)), }) return }
func getDriverHome(w http.ResponseWriter, r *http.Request, c *web.Context) { if !c.CheckAuth(w, r, "driver", "/"+c.GetPathVar("slug")) { return } if c.GetFromSession("slug") != c.GetPathVar("slug") { http.Redirect(w, r, "/"+c.GetFromSession("slug").(string), 303) } var driver Driver if ok := db.GetAs("driver", c.GetFromSession("id").(string), &driver); !ok { c.SetFlash("alertError", "Error finding driver") c.Logout() http.Redirect(w, r, "/"+c.GetPathVar("slug"), 303) return } msgK, msgV := c.GetFlash() ts.Render(w, "driver-home.tmpl", tmpl.Model{ msgK: msgV, "driver": driver, "slug": c.GetPathVar("slug"), "documents": GetDriverDocuments(driver.Id), }) return }
// POST company logout func companyLogout(w http.ResponseWriter, r *http.Request, c *web.Context) { c.Logout() http.Redirect(w, r, "/"+c.GetPathVar("slug"), 303) }