package handlers import ( "context" "fmt" "net/http" "code.tjo.space/mentos1386/zdravko/database/models" "code.tjo.space/mentos1386/zdravko/internal/jwt" "code.tjo.space/mentos1386/zdravko/internal/services" "code.tjo.space/mentos1386/zdravko/web/templates/components" "github.com/go-playground/validator/v10" "github.com/gosimple/slug" "github.com/labstack/echo/v4" ) type WorkerWithToken struct { *models.WorkerGroup Token string } type SettingsWorkerGroups struct { *Settings WorkerGroups []*models.WorkerGroupWithMonitors WorkerGroupsLength int } type SettingsWorker struct { *Settings Worker *WorkerWithToken } func (h *BaseHandler) SettingsWorkerGroupsGET(c echo.Context) error { cc := c.(AuthenticatedContext) workerGroups, err := services.GetWorkerGroupsWithMonitors(context.Background(), h.db) if err != nil { return err } return c.Render(http.StatusOK, "settings_worker_groups.tmpl", &SettingsWorkerGroups{ Settings: NewSettings( cc.Principal.User, GetPageByTitle(SettingsPages, "Worker Groups"), []*components.Page{GetPageByTitle(SettingsPages, "Worker Groups")}, ), WorkerGroups: workerGroups, WorkerGroupsLength: len(workerGroups), }) } func (h *BaseHandler) SettingsWorkerGroupsDescribeGET(c echo.Context) error { cc := c.(AuthenticatedContext) slug := c.Param("slug") worker, err := services.GetWorkerGroup(context.Background(), h.db, slug) if err != nil { return err } // Allow write access to default namespace token, err := jwt.NewTokenForWorker(h.config.Jwt.PrivateKey, h.config.Jwt.PublicKey, worker) if err != nil { return err } return c.Render(http.StatusOK, "settings_worker_groups_describe.tmpl", &SettingsWorker{ Settings: NewSettings( cc.Principal.User, GetPageByTitle(SettingsPages, "Worker Groups"), []*components.Page{ GetPageByTitle(SettingsPages, "Worker Groups"), { Path: fmt.Sprintf("/settings/worker-groups/%s", slug), Title: "Describe", Breadcrumb: worker.Name, }, }), Worker: &WorkerWithToken{ WorkerGroup: worker, Token: token, }, }) } func (h *BaseHandler) SettingsWorkerGroupsCreateGET(c echo.Context) error { cc := c.(AuthenticatedContext) return c.Render(http.StatusOK, "settings_worker_groups_create.tmpl", NewSettings( cc.Principal.User, GetPageByTitle(SettingsPages, "Worker Groups"), []*components.Page{ GetPageByTitle(SettingsPages, "Worker Groups"), GetPageByTitle(SettingsPages, "Worker Groups Create"), }, )) } func (h *BaseHandler) SettingsWorkerGroupsCreatePOST(c echo.Context) error { ctx := context.Background() slug := slug.Make(c.FormValue("name")) workerGroup := &models.WorkerGroup{ Name: c.FormValue("name"), Slug: slug, } err := validator.New(validator.WithRequiredStructEnabled()).Struct(workerGroup) if err != nil { return err } err = services.CreateWorkerGroup( ctx, h.db, workerGroup, ) if err != nil { return err } return c.Redirect(http.StatusSeeOther, fmt.Sprintf("/settings/worker-groups/%s", slug)) }