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 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"))
}
// GET admin home
func AdminHome(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-home.tmpl", tmpl.Model{
msgK: msgV,
})
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 DevComments(w http.ResponseWriter, r *http.Request, c *web.Context) {
if !c.CheckAuth(w, r, "/login", "developer") {
return
}
msgK, msgV := c.GetFlash()
ts.Render(w, "dev-comments.tmpl", tmpl.Model{
msgK: msgV,
"comments": service.FindAllComment(),
})
}
// 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 new company page
func AdminCompanyNew(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-companies-form.tmpl", tmpl.Model{
msgK: msgV,
"companies": service.FindAllCompany(),
})
return
}
// GET admin get all vehicles
func AdminVehicleGetAll(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.tmpl", tmpl.Model{
msgK: msgV,
"vehicles": service.FindAllVehicle(),
"companies": service.CompanyNames(),
})
return
}
// POST admin edit company
func AdminCompanyEdit(w http.ResponseWriter, r *http.Request, c *web.Context) {
if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") {
return
}
r.ParseForm()
company := service.FindOneCompany(r.FormValue("id"))
util.FormToStruct(&company, r.Form, "")
service.SaveCompany(company)
c.SetFlash("alertSuccess", "Successfully saved company")
http.Redirect(w, r, "/admin/company/"+r.FormValue("id"), 303)
return
}
// 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 AdminVehicleSave(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, "")
service.SaveVehicle(vehicle)
c.SetFlash("alertSuccess", "Successfully saved vehicle")
http.Redirect(w, r, "/admin/vehicle", 303)
return
}
// GET admin new employee page
func AdminEmployeeNew(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(),
"users": service.UserRoles(),
})
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 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
}
// POST admin add company
func AdminCompanyAdd(w http.ResponseWriter, r *http.Request, c *web.Context) {
if !c.CheckAuth(w, r, "/login", "admin", "employee", "developer") {
return
}
r.ParseForm()
var company service.Company
util.FormToStruct(&company, r.Form, "")
company.Id = util.UUID4()
service.SaveCompany(company)
c.SetFlash("alertSuccess", "Successfully saved company")
http.Redirect(w, r, "/admin/company", 303)
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
}
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"))
}
// 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
}
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
}
// 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
}
// POST add employee
func AdminEmployeeAdd(w http.ResponseWriter, r *http.Request, c *web.Context) {
if !c.CheckAuth(w, r, "/login", "admin", "developer") {
return
}
if !service.CanUpdateUser("", r.FormValue("email")) {
c.SetFlash("alertError", "Email already registered")
http.Redirect(w, r, "/admin/employee/new", 303)
return
}
r.ParseForm()
employee, user := service.NewEmployee(r.Form)
service.SaveUser(user)
service.SaveEmployee(employee)
c.SetFlash("alertSuccess", "Successfully added employee")
http.Redirect(w, r, "/admin/employee", 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
}
// POST edit employee
func AdminEmployeeEdit(w http.ResponseWriter, r *http.Request, c *web.Context) {
if !c.CheckAuth(w, r, "/login", "admin", "developer") {
return
}
if !service.CanUpdateUser(r.FormValue("userId"), r.FormValue("email")) {
c.SetFlash("alertError", "Email already registered")
http.Redirect(w, r, "/admin/employee/"+r.FormValue("id"), 303)
return
}
r.ParseForm()
employee := service.FindOneEmployee(r.FormValue("id"))
util.FormToStruct(&employee, r.Form, "")
user := service.FindOneUser(employee.UserId)
util.FormToStruct(&user, r.Form, "")
user.Id = employee.UserId
service.SaveEmployee(employee)
service.SaveUser(user)
c.SetFlash("alertSuccess", "Successfully saved employee")
http.Redirect(w, r, "/admin/employee", 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)
}