Added sqlite database

2026-04-02 21:31:19 +02:00
parent f0eeaecbb1
commit 18a030a0a8
16 changed files with 1036 additions and 79 deletions
@@ -0,0 +1,176 @@
+package handlers
+
+import (
+	"database/sql"
+	"errors"
+	"log"
+	"quay/app/models"
+	"quay/app/repository"
+
+	"github.com/gofiber/fiber/v3"
+)
+
+type SiteHandler struct {
+	Repo repository.SiteRepository
+}
+
+func NewSiteHandler(repo repository.SiteRepository) *SiteHandler {
+	return &SiteHandler{Repo: repo}
+}
+
+// GetSites godoc
+// @Summary Get all sites
+// @Description Get a list of all sites
+// @Tags Sites
+// @Accept json
+// @Produce json
+// @Success 200 {object} models.GetAllSitesResponse
+// @Failure 500 {object} models.APIError
+// @Router /sites [get]
+func (h *SiteHandler) GetSites(c fiber.Ctx) error {
+	sites, err := h.Repo.ListSites()
+
+	if err != nil {
+		log.Println("Error listing sites: ", err)
+		return c.Status(fiber.StatusInternalServerError).JSON(&models.APIError{
+			Message: "Unexpected error while listing sites",
+		})
+	}
+
+	if sites == nil {
+		sites = []models.Site{}
+	}
+
+	return c.JSON(models.GetAllSitesResponse{
+		Sites: sites,
+		Total: len(sites),
+	})
+}
+
+// GetSite godoc
+// @Summary Get site by ID
+// @Description Get a single site by its ID
+// @Tags Sites
+// @Accept json
+// @Produce json
+// @Param id path string true "Site ID"
+// @Success 200 {object} models.Site
+// @Failure 404 {object} models.APIError
+// @Failure 500 {object} models.APIError
+// @Router /sites/{id} [get]
+func (h *SiteHandler) GetSite(c fiber.Ctx) error {
+	id := c.Params("id")
+
+	site, err := h.Repo.GetSite(id)
+
+	if err != nil {
+		if errors.Is(err, sql.ErrNoRows) {
+			return c.Status(fiber.StatusNotFound).JSON(&models.APIError{
+				Message: "Site not found",
+			})
+		}
+		log.Println("Message getting site: ", err)
+		return c.Status(fiber.StatusInternalServerError).JSON(&models.APIError{
+			Message: "Unexpected error while getting site",
+		})
+	}
+
+	if site == nil {
+		return c.Status(fiber.StatusNotFound).JSON(&models.APIError{
+			Message: "Site not found",
+		})
+	}
+
+	return c.JSON(site)
+}
+
+func validateIncomingSite(site *models.Site) error {
+	if site == nil {
+		return errors.New("site is required")
+	}
+	if site.GitServer == "" {
+		return errors.New("git server required")
+	}
+	if site.Owner == "" {
+		return errors.New("owner required")
+	}
+	if site.Repository == "" {
+		return errors.New("repository required")
+	}
+	if site.Branch == "" {
+		return errors.New("branch required")
+	}
+	if site.Domain == "" {
+		return errors.New("domain required")
+	}
+	if site.NotFoundFile == "" {
+		site.NotFoundFile = "404.html"
+	}
+	if site.ForwardRules == nil {
+		site.ForwardRules = []models.ForwardRule{}
+	}
+	if site.ForwardRules != nil {
+		for _, r := range site.ForwardRules {
+			if r.Source == "" {
+				return errors.New("forward rule source is required")
+			}
+			if r.Destination == "" {
+				return errors.New("forward rule destination is required")
+			}
+			if r.StatusCode < 300 || r.StatusCode > 399 {
+				return errors.New("forward rule status code must be between 300 and 399")
+			}
+		}
+	}
+	if site.CustomHeaders == nil {
+		site.CustomHeaders = []models.CustomHeaders{}
+	}
+	if site.CustomHeaders != nil {
+		for _, h := range site.CustomHeaders {
+			if h.Source == "" {
+				return errors.New("custom header source required")
+			}
+			if h.Headers == nil {
+				return errors.New("custom header is required")
+			}
+		}
+	}
+	return nil
+}
+
+// PostSite godoc
+// @Summary Create a new site
+// @Description Create a new site with the provided details
+// @Tags Sites
+// @Accept json
+// @Produce json
+// @Param site body models.Site true "Site details"
+// @Success 200 {object} models.Site
+// @Failure 400 {object} models.APIError
+// @Failure 500 {object} models.APIError
+// @Router /sites [post]
+func (h *SiteHandler) PostSite(c fiber.Ctx) error {
+	var site models.Site
+
+	if err := c.Bind().Body(&site); err != nil {
+		log.Println("Error parsing body: ", err)
+		return c.Status(fiber.StatusBadRequest).JSON(&models.APIError{
+			Message: "Invalid request body",
+		})
+	}
+
+	if err := validateIncomingSite(&site); err != nil {
+		return c.Status(fiber.StatusBadRequest).JSON(&models.APIError{
+			Message: "Invalid request body: " + err.Error(),
+		})
+	}
+
+	if err := h.Repo.CreateSite(&site); err != nil {
+		log.Println("Error creating site: ", err)
+		return c.Status(fiber.StatusInternalServerError).JSON(&models.APIError{
+			Message: "Unexpected error while creating site",
+		})
+	}
+
+	return c.JSON(site)
+}
@@ -1,6 +1,7 @@
 package handlers

 import (
+	"crypto/subtle"
 	"path/filepath"
 	"quay/app/github"
 	"quay/app/models"
@@ -8,78 +9,83 @@ import (
 	"quay/internal/envconfig"
 	"strings"

-	"crypto/subtle"
-
 	"github.com/gofiber/fiber/v3"
 )

-func NewUpdateSiteHandler(cfg *config.Config, envCfg *envconfig.EnvConfig) fiber.Handler {
-	return func(c fiber.Ctx) error {
-		sitename := c.Query("site")
-		if sitename == "" {
-			return c.Status(400).JSON(models.APIError{
-				Error: "Missing 'site' query parameter",
-			})
-		}
-
-		var siteConfig *config.SiteConfig
-		for _, site := range cfg.Sites {
-			if site.Name == sitename {
-				siteConfig = &site
-				break
-			}
-		}
-
-		if siteConfig == nil {
-			return c.Status(404).JSON(models.APIError{
-				Error: "Site not found",
-			})
-		}
-
-		deployToken := siteConfig.DeployToken
-		if deployToken == "" {
-			return c.Status(500).JSON(models.APIError{
-				Error: "Deploy token not configured for this site",
-			})
-		}
-
-		providedToken := c.Get("Authorization")
-		if strings.HasPrefix(providedToken, "Bearer ") {
-			providedToken = strings.TrimPrefix(providedToken, "Bearer ")
-		}
-		if providedToken == "" {
-			return c.Status(401).JSON(models.APIError{
-				Error: "Missing Authorization header",
-			})
-		}
-
-		if subtle.ConstantTimeCompare([]byte(providedToken), []byte(deployToken)) != 1 {
-			return c.Status(403).JSON(models.APIError{
-				Error: "Invalid deploy token",
-			})
-		}
-
-		sitePath := filepath.Join(envCfg.StoragePath, siteConfig.Name)
-		if _, err := filepath.Abs(sitePath); err != nil {
-			return c.Status(500).JSON(models.APIError{
-				Error: "Failed to resolve site path",
-			})
-		}
-
-		err := github.FetchAndDeployBranch(
-			siteConfig.Owner,
-			siteConfig.Repo,
-			siteConfig.Branch,
-			envCfg.GithubPat,
-			sitePath,
-		)
-
-		if err != nil {
-			return c.Status(500).JSON(models.APIError{
-				Error: "Failed to deploy site: " + err.Error(),
-			})
-		}
-
-		return c.SendStatus(201)
-	}
+type UpdateSiteHandler struct {
+	Cfg    *config.Config
+	EnvCfg *envconfig.EnvConfig
+}
+
+func NewUpdateSiteHandler(cfg *config.Config, envCfg *envconfig.EnvConfig) *UpdateSiteHandler {
+	return &UpdateSiteHandler{Cfg: cfg, EnvCfg: envCfg}
+}
+
+func (h *UpdateSiteHandler) PostUpdate(c fiber.Ctx) error {
+	sitename := c.Query("site")
+	if sitename == "" {
+		return c.Status(400).JSON(models.APIError{
+			Message: "Missing 'site' query parameter",
+		})
+	}
+
+	var siteConfig *config.SiteConfig
+	for _, site := range h.Cfg.Sites {
+		if site.Name == sitename {
+			siteConfig = &site
+			break
+		}
+	}
+
+	if siteConfig == nil {
+		return c.Status(404).JSON(models.APIError{
+			Message: "Site not found",
+		})
+	}
+
+	deployToken := siteConfig.DeployToken
+	if deployToken == "" {
+		return c.Status(500).JSON(models.APIError{
+			Message: "Deploy token not configured for this site",
+		})
+	}
+
+	providedToken := c.Get("Authorization")
+	if strings.HasPrefix(providedToken, "Bearer ") {
+		providedToken = strings.TrimPrefix(providedToken, "Bearer ")
+	}
+	if providedToken == "" {
+		return c.Status(401).JSON(models.APIError{
+			Message: "Missing Authorization header",
+		})
+	}
+
+	if subtle.ConstantTimeCompare([]byte(providedToken), []byte(deployToken)) != 1 {
+		return c.Status(403).JSON(models.APIError{
+			Message: "Invalid deploy token",
+		})
+	}
+
+	sitePath := filepath.Join(h.EnvCfg.StoragePath, siteConfig.Name)
+	if _, err := filepath.Abs(sitePath); err != nil {
+		return c.Status(500).JSON(models.APIError{
+			Message: "Failed to resolve site path",
+		})
+	}
+
+	err := github.FetchAndDeployBranch(
+		siteConfig.Owner,
+		siteConfig.Repo,
+		siteConfig.Branch,
+		h.EnvCfg.GithubPat,
+		sitePath,
+	)
+
+	if err != nil {
+		return c.Status(500).JSON(models.APIError{
+			Message: "Failed to deploy site: " + err.Error(),
+		})
+	}
+
+	return c.SendStatus(201)
 }