zdravko/internal/handlers/settingsmonitors.go

package handlers

import (
	"context"
	"database/sql"
	"fmt"
	"net/http"
	"slices"
	"strings"

	"code.tjo.space/mentos1386/zdravko/database/models"
	"code.tjo.space/mentos1386/zdravko/internal/script"
	"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 CreateMonitor struct {
	Name         string `validate:"required"`
	Group        string `validate:"required"`
	WorkerGroups string `validate:"required"`
	Schedule     string `validate:"required,cron"`
	Script       string `validate:"required"`
}

type UpdateMonitor struct {
	Group        string `validate:"required"`
	WorkerGroups string `validate:"required"`
	Schedule     string `validate:"required,cron"`
	Script       string `validate:"required"`
}

type MonitorWithWorkerGroupsAndStatus struct {
	*models.MonitorWithWorkerGroups
	Status services.MonitorStatus
}

type SettingsMonitors struct {
	*Settings
	Monitors      map[string][]*MonitorWithWorkerGroupsAndStatus
	MonitorGroups []string
}

type SettingsMonitor struct {
	*Settings
	Monitor *MonitorWithWorkerGroupsAndStatus
	History []*models.MonitorHistory
}

type SettingsMonitorCreate struct {
	*Settings
	Example string
}

func (h *BaseHandler) SettingsMonitorsGET(c echo.Context) error {
	cc := c.(AuthenticatedContext)

	monitors, err := services.GetMonitorsWithWorkerGroups(context.Background(), h.db)
	if err != nil {
		return err
	}

	monitorsWithStatus := make([]*MonitorWithWorkerGroupsAndStatus, len(monitors))
	for i, monitor := range monitors {
		status, err := services.GetMonitorStatus(context.Background(), h.temporal, monitor.Id)
		if err != nil {
			return err
		}
		monitorsWithStatus[i] = &MonitorWithWorkerGroupsAndStatus{
			MonitorWithWorkerGroups: monitor,
			Status:                  status,
		}
	}

	monitorGroups := []string{}
	monitorsByGroup := map[string][]*MonitorWithWorkerGroupsAndStatus{}
	for _, monitor := range monitorsWithStatus {
		monitorsByGroup[monitor.Group] = append(monitorsByGroup[monitor.Group], monitor)
		if slices.Contains(monitorGroups, monitor.Group) == false {
			monitorGroups = append(monitorGroups, monitor.Group)
		}
	}

	return c.Render(http.StatusOK, "settings_monitors.tmpl", &SettingsMonitors{
		Settings: NewSettings(
			cc.Principal.User,
			GetPageByTitle(SettingsPages, "Checks"),
			[]*components.Page{GetPageByTitle(SettingsPages, "Checks")},
		),
		Monitors:      monitorsByGroup,
		MonitorGroups: monitorGroups,
	})
}

func (h *BaseHandler) SettingsMonitorsDescribeGET(c echo.Context) error {
	cc := c.(AuthenticatedContext)

	slug := c.Param("id")

	monitor, err := services.GetMonitorWithWorkerGroups(context.Background(), h.db, slug)
	if err != nil {
		return err
	}

	status, err := services.GetMonitorStatus(context.Background(), h.temporal, monitor.Id)
	if err != nil {
		return err
	}

	monitorWithStatus := &MonitorWithWorkerGroupsAndStatus{
		MonitorWithWorkerGroups: monitor,
		Status:                  status,
	}

	history, err := services.GetMonitorHistoryForMonitor(context.Background(), h.db, slug)
	if err != nil {
		return err
	}

	maxElements := 10
	if len(history) < maxElements {
		maxElements = len(history)
	}

	return c.Render(http.StatusOK, "settings_monitors_describe.tmpl", &SettingsMonitor{
		Settings: NewSettings(
			cc.Principal.User,
			GetPageByTitle(SettingsPages, "Checks"),
			[]*components.Page{
				GetPageByTitle(SettingsPages, "Checks"),
				{
					Path:       fmt.Sprintf("/settings/monitors/%s", slug),
					Title:      "Describe",
					Breadcrumb: monitor.Name,
				},
			}),
		Monitor: monitorWithStatus,
		History: history[:maxElements],
	})
}

func (h *BaseHandler) SettingsMonitorsDescribeDELETE(c echo.Context) error {
	slug := c.Param("id")

	err := services.DeleteMonitor(context.Background(), h.db, slug)
	if err != nil {
		return err
	}

	err = services.DeleteMonitorSchedule(context.Background(), h.temporal, slug)
	if err != nil {
		return err
	}

	return c.Redirect(http.StatusSeeOther, "/settings/monitors")
}

func (h *BaseHandler) SettingsMonitorsDisableGET(c echo.Context) error {
	slug := c.Param("id")

	monitor, err := services.GetMonitor(context.Background(), h.db, slug)
	if err != nil {
		return err
	}

	err = services.SetMonitorStatus(context.Background(), h.temporal, monitor.Id, services.MonitorStatusPaused)
	if err != nil {
		return err
	}

	return c.Redirect(http.StatusSeeOther, fmt.Sprintf("/settings/monitors/%s", slug))
}

func (h *BaseHandler) SettingsMonitorsEnableGET(c echo.Context) error {
	slug := c.Param("id")

	monitor, err := services.GetMonitor(context.Background(), h.db, slug)
	if err != nil {
		return err
	}

	err = services.SetMonitorStatus(context.Background(), h.temporal, monitor.Id, services.MonitorStatusActive)
	if err != nil {
		return err
	}

	return c.Redirect(http.StatusSeeOther, fmt.Sprintf("/settings/monitors/%s", slug))
}

func (h *BaseHandler) SettingsMonitorsDescribePOST(c echo.Context) error {
	ctx := context.Background()
	monitorId := c.Param("id")

	update := UpdateMonitor{
		Group:        strings.ToLower(c.FormValue("group")),
		WorkerGroups: strings.ToLower(strings.TrimSpace(c.FormValue("workergroups"))),
		Schedule:     c.FormValue("schedule"),
		Script:       script.EscapeString(c.FormValue("script")),
	}
	err := validator.New(validator.WithRequiredStructEnabled()).Struct(update)
	if err != nil {
		return err
	}

	monitor, err := services.GetMonitor(ctx, h.db, monitorId)
	if err != nil {
		return err
	}
	monitor.Group = update.Group
	monitor.Schedule = update.Schedule
	monitor.Script = update.Script

	err = services.UpdateMonitor(
		ctx,
		h.db,
		monitor,
	)
	if err != nil {
		return err
	}

	workerGroups := []*models.WorkerGroup{}
	for _, group := range strings.Split(update.WorkerGroups, " ") {
		if group == "" {
			continue
		}
		workerGroup, err := services.GetWorkerGroup(ctx, h.db, slug.Make(group))
		if err != nil {
			if err == sql.ErrNoRows {
				workerGroup = &models.WorkerGroup{Name: group, Id: slug.Make(group)}
				err = services.CreateWorkerGroup(ctx, h.db, workerGroup)
				if err != nil {
					return err
				}
			} else {
				return err
			}
		}
		workerGroups = append(workerGroups, workerGroup)
	}

	err = services.UpdateMonitorWorkerGroups(ctx, h.db, monitor, workerGroups)
	if err != nil {
		return err
	}

	err = services.CreateOrUpdateMonitorSchedule(ctx, h.temporal, monitor, workerGroups)
	if err != nil {
		return err
	}

	return c.Redirect(http.StatusSeeOther, fmt.Sprintf("/settings/monitors/%s", monitorId))
}

func (h *BaseHandler) SettingsMonitorsCreateGET(c echo.Context) error {
	cc := c.(AuthenticatedContext)

	return c.Render(http.StatusOK, "settings_monitors_create.tmpl", &SettingsMonitorCreate{
		Settings: NewSettings(
			cc.Principal.User,
			GetPageByTitle(SettingsPages, "Checks"),
			[]*components.Page{
				GetPageByTitle(SettingsPages, "Checks"),
				GetPageByTitle(SettingsPages, "Checks Create"),
			},
		),
		Example: h.examples.Monitor,
	})
}

func (h *BaseHandler) SettingsMonitorsCreatePOST(c echo.Context) error {
	ctx := context.Background()
	monitorId := slug.Make(c.FormValue("name"))

	create := CreateMonitor{
		Name:         c.FormValue("name"),
		Group:        strings.ToLower(c.FormValue("group")),
		WorkerGroups: strings.ToLower(strings.TrimSpace(c.FormValue("workergroups"))),
		Schedule:     c.FormValue("schedule"),
		Script:       script.EscapeString(c.FormValue("script")),
	}
	err := validator.New(validator.WithRequiredStructEnabled()).Struct(create)
	if err != nil {
		return err
	}

	workerGroups := []*models.WorkerGroup{}
	for _, group := range strings.Split(create.WorkerGroups, " ") {
		if group == "" {
			continue
		}
		workerGroup, err := services.GetWorkerGroup(ctx, h.db, slug.Make(group))
		if err != nil {
			if err == sql.ErrNoRows {
				workerGroup = &models.WorkerGroup{Name: group, Id: slug.Make(group)}
				err = services.CreateWorkerGroup(ctx, h.db, workerGroup)
				if err != nil {
					return err
				}
			} else {
				return err
			}
		}
		workerGroups = append(workerGroups, workerGroup)
	}

	monitor := &models.Monitor{
		Name:     create.Name,
		Group:    create.Group,
		Id:       monitorId,
		Schedule: create.Schedule,
		Script:   create.Script,
	}

	err = services.CreateMonitor(
		ctx,
		h.db,
		monitor,
	)
	if err != nil {
		return err
	}

	err = services.UpdateMonitorWorkerGroups(ctx, h.db, monitor, workerGroups)
	if err != nil {
		return err
	}

	err = services.CreateOrUpdateMonitorSchedule(ctx, h.temporal, monitor, workerGroups)
	if err != nil {
		return err
	}

	return c.Redirect(http.StatusSeeOther, "/settings/monitors")
}