go-urlsh/internal/web/index.go

package web

import (
	"context"
	"database/sql"
	"log"

	"code.lila.network/adoralaura/go-urlsh/internal/constants"
	"code.lila.network/adoralaura/go-urlsh/internal/db"
	"code.lila.network/adoralaura/go-urlsh/models"
	"github.com/gofiber/fiber/v2"
)

func HandleIndexGet(c *fiber.Ctx) error {
	if c.Params("id", "") == "" {
		if db.IsCookieValid(c.Cookies(constants.LoginCookieName, "")) {
			c.Location("/admin/")
			c.Status(fiber.StatusSeeOther)
			return nil
		} else {
			err := c.SendString("This is the index page of go-urlsh.\n " +
				"See https://code.lila.network/adoralaura/go-urlsh for more info.")
			if err != nil {
				log.Println(err)
				return err
			}
			c.Status(fiber.StatusOK)
			return nil
		}
	} else {
		link := new(models.Link)

		err := models.DB.NewSelect().Model(link).Where("id = ?", c.Params("id")).Scan(context.Background())
		if err != nil {
			if err == sql.ErrNoRows {
				log.Printf("Shortlink %v not found\n", c.Params("id"))
				return fiber.NewError(fiber.StatusNotFound, "404 Not Found")
			} else {
				log.Printf("Error querying Link %v from database: %v\n", c.Params("id"), err)
				return fiber.NewError(fiber.StatusInternalServerError, "500 Internal Server Error")
			}
		}

		c.Location(link.URL)
		c.Status(fiber.StatusSeeOther)
		return nil
	}
}