//go:generate npm run build

package main

import (
	"bytes"
	"database/sql"
	"embed"
	"errors"
	"fmt"
	"image"
	"image/jpeg"
	"image/png"
	"io"
	"io/fs"
	"log"
	"log/slog"
	"mime/multipart"
	"net/http"
	"os"
	"os/signal"
	"path/filepath"
	"strconv"
	"strings"
	"syscall"
	"time"

	"github.com/HugoSmits86/nativewebp"
	"github.com/NarmadaWeb/gonify/v3"
	"github.com/caarlos0/env/v11"
	"github.com/disintegration/imaging"
	"github.com/gofiber/fiber/v3"
	"github.com/gofiber/fiber/v3/middleware/compress"
	"github.com/gofiber/fiber/v3/middleware/helmet"
	"github.com/gofiber/fiber/v3/middleware/static"
	"github.com/gofiber/template/handlebars/v2"
	"github.com/google/uuid"
	"github.com/joho/godotenv"
	"github.com/juls0730/passport/src/middleware"
	"github.com/juls0730/passport/src/services"
	"github.com/rwcarlsen/goexif/exif"
	"github.com/rwcarlsen/goexif/tiff"
	"golang.org/x/image/draw"
	_ "modernc.org/sqlite"
)

//go:embed assets/** templates/** schema.sql scripts/**.js
var embeddedAssets embed.FS

var devContent = `<script>
let host = window.location.hostname;
let protocol = window.location.protocol === 'https:' ? 'wss:' : 'ws:';
const socket = new WebSocket(protocol + '//' + host + ':2067/ws');

socket.addEventListener('message', (event) => {
    if (event.data === 'refresh') {
		console.log('Got refresh signal');

		let attempts = 0;
		let delay = 100;
		
        async function testPage() {
            try {
            	let res = await fetch(window.location.href)
            } catch (error) {
				if (attempts > 5) {
					return;
				}
                setTimeout(testPage, delay);

				// exponential backoff
				attempts++;
				delay = 100 * Math.pow(2, attempts);

                return;
            }
            window.location.reload();
        }

		setTimeout(testPage, 150);
	}
});
</script>`

var (
	insertCategoryStmt *sql.Stmt
	insertLinkStmt     *sql.Stmt
)

type Config struct {
	DevMode bool `env:"PASSPORT_DEV_MODE" envDefault:"false"`
	Prefork bool `env:"PASSPORT_ENABLE_PREFORK" envDefault:"false"`

	WeatherAPIKey string `env:"PASSPORT_WEATHER_API_KEY"`
	Weather       *services.WeatherConfig

	UptimeAPIKey string `env:"PASSPORT_UPTIME_API_KEY"`
	Uptime       *services.UptimeConfig

	Admin struct {
		Username string `env:"PASSPORT_ADMIN_USERNAME"`
		Password string `env:"PASSPORT_ADMIN_PASSWORD"`
	}

	SearchProvider struct {
		URL   string `env:"PASSPORT_SEARCH_PROVIDER"`
		Query string `env:"PASSPORT_SEARCH_PROVIDER_QUERY_PARAM" envDefault:"q"`
	}

	Depricated struct {
		WeatherEnabled bool `env:"PASSPORT_ENABLE_WEATHER" envDefault:"false"`
		UptimeEnabled  bool `env:"PASSPORT_ENABLE_UPTIME" envDefault:"false"`
	}
}

func ParseConfig() (*Config, error) {
	config := Config{}

	err := env.Parse(&config)
	if err != nil {
		return nil, err
	}

	if config.WeatherAPIKey != "" {
		config.Weather = &services.WeatherConfig{
			APIKey: config.WeatherAPIKey,
		}
		if err := env.Parse(config.Weather); err != nil {
			return nil, err
		}
	} else if config.Depricated.WeatherEnabled {
		slog.Warn("Your configuration file contains depricated Weather settings. Please update your configuration file!")
		depricatedWeatherConfig := &services.DepricatedWeatherConfig{}
		if err := env.Parse(depricatedWeatherConfig); err != nil {
			return nil, err
		}

		config.Weather = &services.WeatherConfig{}
		config.Weather.Provider = depricatedWeatherConfig.OpenWeather.Provider
		config.Weather.APIKey = depricatedWeatherConfig.OpenWeather.APIKey
		config.Weather.Units = depricatedWeatherConfig.OpenWeather.Units
		config.Weather.Lat = depricatedWeatherConfig.OpenWeather.Lat
		config.Weather.Lon = depricatedWeatherConfig.OpenWeather.Lon
		config.Weather.UpdateInterval = depricatedWeatherConfig.UpdateInterval
	}

	if config.UptimeAPIKey != "" {
		config.Uptime = &services.UptimeConfig{
			APIKey: config.UptimeAPIKey,
		}
		if err := env.Parse(config.Uptime); err != nil {
			return nil, err
		}
	} else if config.Depricated.UptimeEnabled {
		slog.Warn("Your configuration file contains depricated Uptime settings. Please update your configuration file!")
		depricatedUptimeConfig := &services.DepricatedUptimeConfig{}
		if err := env.Parse(depricatedUptimeConfig); err != nil {
			return nil, err
		}

		config.Uptime = &services.UptimeConfig{}
		config.Uptime.APIKey = depricatedUptimeConfig.APIKey
		config.Uptime.UpdateInterval = depricatedUptimeConfig.UpdateInterval
	}

	return &config, nil
}

type App struct {
	*Config
	*CategoryManager
	*services.WeatherManager
	*services.UptimeManager
	db *sql.DB
}

func (app *App) Close() error {
	return app.db.Close()
}

func NewApp(dbPath string, options map[string]any) (*App, error) {
	config, err := ParseConfig()
	if err != nil {
		return nil, err
	}

	file, err := os.OpenFile(dbPath, os.O_RDWR|os.O_CREATE, 0644)
	if err != nil {
		if os.IsPermission(err) {
			return nil, fmt.Errorf("file %s is not readable and writable: %v", dbPath, err)
		}
		return nil, fmt.Errorf("failed to open file %s for read/write: %v", dbPath, err)
	}
	defer file.Close()

	var connectionOpts string
	for k, v := range options {
		if connectionOpts != "" {
			connectionOpts += "&"
		}

		connectionOpts += fmt.Sprintf("%s=%v", k, v)
	}

	db, err := sql.Open("sqlite", fmt.Sprintf("%s?%s", dbPath, connectionOpts))
	if err != nil {
		return nil, err
	}

	err = db.Ping()
	if err != nil {
		return nil, err
	}

	schema, err := embeddedAssets.ReadFile("schema.sql")
	if err != nil {
		return nil, err
	}

	_, err = db.Exec(string(schema))
	if err != nil {
		return nil, err
	}

	categoryManager, err := NewCategoryManager(db)
	if err != nil {
		return nil, err
	}

	var weatherCache *services.WeatherManager
	if config.WeatherAPIKey != "" {
		weatherCache = services.NewWeatherManager(config.Weather)
	}

	var uptimeManager *services.UptimeManager
	if config.UptimeAPIKey != "" {
		uptimeManager = services.NewUptimeManager(config.Uptime)
	}

	return &App{
		Config:          config,
		WeatherManager:  weatherCache,
		CategoryManager: categoryManager,
		UptimeManager:   uptimeManager,
		db:              db,
	}, nil
}

func CropToCenter(img image.Image, outputSize int) (image.Image, error) {
	if img == nil {
		return nil, fmt.Errorf("input image is nil")
	}
	if outputSize <= 0 {
		return nil, fmt.Errorf("output size must be positive")
	}

	srcBounds := img.Bounds()
	srcWidth := srcBounds.Dx()
	srcHeight := srcBounds.Dy()

	squareSide := min(srcWidth, srcHeight)

	cropX := (srcWidth - squareSide) / 2
	cropY := (srcHeight - squareSide) / 2

	srcCropRect := image.Rect(cropX, cropY, cropX+squareSide, cropY+squareSide)

	croppedSquareImg := image.NewRGBA(image.Rect(0, 0, squareSide, squareSide))
	draw.Draw(croppedSquareImg, croppedSquareImg.Rect, img, srcCropRect.Min, draw.Src)

	if squareSide == outputSize {
		return croppedSquareImg, nil
	}

	outputImg := image.NewRGBA(image.Rect(0, 0, outputSize, outputSize))

	draw.CatmullRom.Scale(outputImg, outputImg.Rect, croppedSquareImg, croppedSquareImg.Bounds(), draw.Src, nil)

	return outputImg, nil
}

func UploadFile(file *multipart.FileHeader, contentType string, c fiber.Ctx) (string, error) {
	fileId, err := uuid.NewV7()
	if err != nil {
		return "", err
	}

	var fileName string
	if filepath.Ext(file.Filename) != ".svg" {
		fileName = fmt.Sprintf("%s.webp", fileId.String())
	} else {
		fileName = fmt.Sprintf("%s.svg", fileId.String())
	}

	srcFile, err := file.Open()
	if err != nil {
		return "", err
	}
	defer srcFile.Close()

	var img image.Image
	switch contentType {
	case "image/jpeg":
		img, err = jpeg.Decode(srcFile)
	case "image/png":
		img, err = png.Decode(srcFile)
	case "image/webp":
		img, err = nativewebp.Decode(srcFile)
	case "image/svg+xml":
		// does not fall through (my C brain was tripping over this)
	default:
		return "", errors.New("unsupported file type")
	}

	if err != nil {
		return "", err
	}

	if contentType != "image/svg+xml" {
		off, err := srcFile.Seek(0, io.SeekStart)
		if err != nil {
			return "", fmt.Errorf("failed to seek to start of file: %v", err)
		}

		if off != 0 {
			return "", fmt.Errorf("failed to seek to start of file: %v", err)
		}

		x, err := exif.Decode(srcFile)
		// if there *is* exif, parse it
		if err == nil {
			tag, err := x.Get(exif.Orientation)
			if err == nil {
				if tag.Count == 1 && tag.Format() == tiff.IntVal {
					orientation, err := tag.Int(0)
					if err != nil {
						return "", fmt.Errorf("failed to get orientation: %v", err)
					}

					slog.Debug("Orientation tag found", "orientation", orientation)

					switch orientation {
					case 3:
						img = imaging.Rotate180(img)
					case 6:
						img = imaging.Rotate270(img)
					case 8:
						img = imaging.Rotate90(img)
					}
				}
			}
		}

		img, err = CropToCenter(img, 96)
		if err != nil {
			return "", err
		}
	}

	assetsDir := "public/uploads"

	iconPath := filepath.Join(assetsDir, fileName)

	if contentType == "image/svg+xml" {
		// replace currentColor with a text color
		outFile, err := os.Create(iconPath)
		if err != nil {
			return "", err
		}
		defer outFile.Close()

		svgText, err := io.ReadAll(srcFile)
		if err != nil {
			return "", err
		}

		svgText = bytes.ReplaceAll(svgText, []byte("currentColor"), []byte(`oklch(87% 0.015 286)`))
		_, err = outFile.Write(svgText)
		if err != nil {
			return "", err
		}
	} else {
		outFile, err := os.Create(iconPath)
		if err != nil {
			return "", err
		}
		defer outFile.Close()

		var buf bytes.Buffer
		options := &nativewebp.Options{}
		if err := nativewebp.Encode(&buf, img, options); err != nil {
			return "", err
		}

		if _, err := io.Copy(outFile, &buf); err != nil {
			return "", err
		}
	}

	iconPath = "/uploads/" + fileName

	return iconPath, nil
}

type Category struct {
	ID    int64  `json:"id"`
	Name  string `json:"name"`
	Icon  string `json:"icon"`
	Links []Link `json:"links"`
}

type Link struct {
	ID          int64  `json:"id"`
	CategoryID  int64  `json:"category_id"`
	Name        string `json:"name"`
	Description string `json:"description"`
	Icon        string `json:"icon"`
	URL         string `json:"url"`
}

type CategoryManager struct {
	db *sql.DB
}

func NewCategoryManager(db *sql.DB) (*CategoryManager, error) {
	return &CategoryManager{
		db: db,
	}, nil
}

func (manager *CategoryManager) GetCategories() []Category {
	rows, err := manager.db.Query(`
		SELECT id, name, icon
		FROM categories 
		ORDER BY id ASC
	`)

	if err != nil {
		return nil
	}
	defer rows.Close()

	var categories []Category
	for rows.Next() {
		var cat Category

		if err := rows.Scan(&cat.ID, &cat.Name, &cat.Icon); err != nil {
			return nil
		}

		categories = append(categories, cat)
	}

	for i, cat := range categories {
		categories[i].Links = manager.GetLinks(cat.ID)
	}

	return categories
}

// Get Category by ID, returns nil if not found
func (manager *CategoryManager) GetCategory(id int64) *Category {
	row := manager.db.QueryRow(`SELECT id, name, icon FROM categories WHERE id = ?`, id)

	var cat Category
	if err := row.Scan(&cat.ID, &cat.Name, &cat.Icon); err != nil {
		return nil
	}

	return &cat
}

func (manager *CategoryManager) CreateCategory(category Category) (*Category, error) {
	var err error

	insertCategoryStmt, err = manager.db.Prepare(`
		INSERT INTO categories (name, icon) 
		VALUES (?, ?) RETURNING id`)

	if err != nil {
		return nil, err
	}

	defer insertCategoryStmt.Close()

	var categoryID int64

	if err := insertCategoryStmt.QueryRow(category.Name, category.Icon).Scan(&categoryID); err != nil {
		return nil, err
	}

	category.ID = categoryID

	return &category, nil
}

func (manager *CategoryManager) DeleteCategory(id int64) error {
	rows, err := manager.db.Query(`
				SELECT icon FROM categories WHERE id = ?
				UNION
				SELECT icon FROM links WHERE category_id = ?
			`, id, id)

	if err != nil {
		return err
	}

	defer rows.Close()

	var icons []string
	for rows.Next() {
		var icon string
		if err := rows.Scan(&icon); err != nil {
			return err
		}
		icons = append(icons, icon)
	}

	tx, err := manager.db.Begin()
	if err != nil {
		return err
	}
	defer tx.Rollback()

	_, err = tx.Exec("DELETE FROM categories WHERE id = ?", id)
	if err != nil {
		return err
	}

	_, err = tx.Exec("DELETE FROM links WHERE category_id = ?", id)
	if err != nil {
		return err
	}

	if err := tx.Commit(); err != nil {
		return err
	}

	for _, icon := range icons {
		if icon == "" {
			slog.Debug("blank icon")
			continue
		}

		if err := os.Remove(filepath.Join("public/", icon)); err != nil {
			// dont error to the user if the icon doesnt exist, just log it
			slog.Error("Failed to delete icon", "icon", icon, "error", err)
		}
	}

	return nil
}

func (manager *CategoryManager) GetLink(id int64) *Link {
	row := manager.db.QueryRow(`SELECT id, category_id, name, description, icon, url FROM links WHERE id = ?`, id)

	var link Link
	if err := row.Scan(&link.ID, &link.CategoryID, &link.Name, &link.Description, &link.Icon, &link.URL); err != nil {
		return nil
	}

	return &link
}

func (manager *CategoryManager) GetLinks(categoryID int64) []Link {
	rows, err := manager.db.Query(`
		SELECT id, category_id, name, description, icon, url 
		FROM links 
		WHERE category_id = ? 
		ORDER BY id ASC
	`, categoryID)

	if err != nil {
		return nil
	}
	defer rows.Close()

	var links []Link
	for rows.Next() {
		var link Link
		if err := rows.Scan(&link.ID, &link.CategoryID, &link.Name, &link.Description,
			&link.Icon, &link.URL); err != nil {
			return nil
		}
		links = append(links, link)
	}

	return links
}

func (manager *CategoryManager) CreateLink(db *sql.DB, link Link) (*Link, error) {
	var err error
	insertLinkStmt, err = db.Prepare(`
		INSERT INTO links (category_id, name, description, icon, url) 
		VALUES (?, ?, ?, ?, ?) RETURNING id`)
	if err != nil {
		return nil, err
	}

	defer insertLinkStmt.Close()

	var linkID int64
	if err := insertLinkStmt.QueryRow(link.CategoryID, link.Name, link.Description, link.Icon, link.URL).Scan(&linkID); err != nil {
		return nil, err
	}

	link.ID = linkID

	return &link, nil
}

func (manager *CategoryManager) DeleteLink(id any) error {
	var icon string
	if err := manager.db.QueryRow("SELECT icon FROM links WHERE id = ?", id).Scan(&icon); err != nil {
		return err
	}

	_, err := manager.db.Exec("DELETE FROM links WHERE id = ?", id)
	if err != nil {
		return err
	}

	if icon != "" {
		if err := os.Remove(filepath.Join("public/", icon)); err != nil {
			slog.Error("Failed to delete icon", "icon", icon, "error", err)
		}
	}

	return nil
}

var WeatherIcons = map[string]string{
	"clear-day":           `<svg aria-label="Clear day" xmlns="http://www.w3.org/2000/svg" width="32" height="32" viewBox="0 0 32 32"><path fill="currentColor" d="M16 12.005a4 4 0 1 1-4 4a4.005 4.005 0 0 1 4-4m0-2a6 6 0 1 0 6 6a6 6 0 0 0-6-6M5.394 6.813L6.81 5.399l3.505 3.506L8.9 10.319zM2 15.005h5v2H2zm3.394 10.193L8.9 21.692l1.414 1.414l-3.505 3.506zM15 25.005h2v5h-2zm6.687-1.9l1.414-1.414l3.506 3.506l-1.414 1.414zm3.313-8.1h5v2h-5zm-3.313-6.101l3.506-3.506l1.414 1.414l-3.506 3.506zM15 2.005h2v5h-2z"/></svg>`,
	"clear-night":         `<svg aria-label="Clear night" xmlns="http://www.w3.org/2000/svg" width="32" height="32" viewBox="0 0 32 32"><path fill="currentColor" d="M13.503 5.414a15.076 15.076 0 0 0 11.593 18.194a11.1 11.1 0 0 1-7.975 3.39c-.138 0-.278.005-.418 0a11.094 11.094 0 0 1-3.2-21.584M14.98 3a1 1 0 0 0-.175.016a13.096 13.096 0 0 0 1.825 25.981c.164.006.328 0 .49 0a13.07 13.07 0 0 0 10.703-5.555a1.01 1.01 0 0 0-.783-1.565A13.08 13.08 0 0 1 15.89 4.38A1.015 1.015 0 0 0 14.98 3"/></svg>`,
	"partly-cloudy-day":   `<svg aria-label="Partly cloudy day" xmlns="http://www.w3.org/2000/svg" width="32" height="32" viewBox="0 0 32 32"><path fill="currentColor" d="M27 15h4v2h-4zm-4-7.413l3-3L27.415 6l-3 3zM15 1h2v4h-2zM4.586 26l3-3l1.415 1.415l-3 3zM4.585 6L6 4.587l3 3l-1.414 1.415z"/><path fill="currentColor" d="M1 15h4v2H1zm25.794 5.342a6.96 6.96 0 0 0-1.868-3.267A9 9 0 0 0 25 16a9 9 0 1 0-14.585 7.033A4.977 4.977 0 0 0 15 30h10a4.995 4.995 0 0 0 1.794-9.658M9 16a6.996 6.996 0 0 1 13.985-.297A6.9 6.9 0 0 0 20 15a7.04 7.04 0 0 0-6.794 5.342a5 5 0 0 0-1.644 1.048A6.97 6.97 0 0 1 9 16m16 12H15a2.995 2.995 0 0 1-.696-5.908l.658-.157l.099-.67a4.992 4.992 0 0 1 9.878 0l.099.67l.658.156A2.995 2.995 0 0 1 25 28"/></svg>`,
	"partly-cloudy-night": `<svg aria-label="Partly cloudy night" xmlns="http://www.w3.org/2000/svg" width="32" height="32" viewBox="0 0 32 32"><path fill="currentColor" d="M30 19a4.97 4.97 0 0 0-3.206-4.658A6.971 6.971 0 0 0 13.758 12.9a13.14 13.14 0 0 1 .131-8.52A1.015 1.015 0 0 0 12.98 3a1 1 0 0 0-.175.016a13.096 13.096 0 0 0 1.825 25.981c.164.006.328 0 .49 0a13.04 13.04 0 0 0 10.29-5.038A4.99 4.99 0 0 0 30 19m-15.297 7.998a11.095 11.095 0 0 1-3.2-21.584a15.2 15.2 0 0 0 .844 9.367A4.988 4.988 0 0 0 15 24h7.677a11.1 11.1 0 0 1-7.556 2.998c-.138 0-.278.004-.418 0M25 22H15a2.995 2.995 0 0 1-.696-5.908l.658-.157l.099-.67a4.992 4.992 0 0 1 9.878 0l.099.67l.658.157A2.995 2.995 0 0 1 25 22"/></svg>`,
	"mostly-cloudy-day":   `<svg aria-label="Mostly cloudy day" xmlns="http://www.w3.org/2000/svg" width="32" height="32" viewBox="0 0 32 32"><path fill="currentColor" d="M21.743 18.692a6 6 0 0 0 1.057-1.086a5.998 5.998 0 1 0-10.733-4.445A7.56 7.56 0 0 0 6.35 18.25A5.993 5.993 0 0 0 8 30.005h11a5.985 5.985 0 0 0 2.743-11.313M18 10.005a4.004 4.004 0 0 1 4 4a3.96 3.96 0 0 1-.8 2.4a4 4 0 0 1-.94.891a7.54 7.54 0 0 0-6.134-4.24A4 4 0 0 1 18 10.006m1 18H8a3.993 3.993 0 0 1-.673-7.93l.663-.112l.146-.656a5.496 5.496 0 0 1 10.729 0l.146.656l.662.112a3.993 3.993 0 0 1-.673 7.93m7-15.001h4v2h-4zM22.95 7.64l2.828-2.827l1.415 1.414l-2.829 2.828zM17 2.005h2v4h-2zM8.808 6.227l1.414-1.414l2.829 2.828l-1.415 1.414z"/></svg>`,
	"mostly-cloudy-night": `<svg aria-label="Mostly cloudy night" xmlns="http://www.w3.org/2000/svg" width="32" height="32" viewBox="0 0 32 32"><path fill="currentColor" d="M29.844 15.035a1.52 1.52 0 0 0-1.23-.866a5.36 5.36 0 0 1-3.41-1.716a6.47 6.47 0 0 1-1.286-6.392a1.6 1.6 0 0 0-.299-1.546a1.45 1.45 0 0 0-1.36-.493l-.019.003a7.93 7.93 0 0 0-6.22 7.431A7.4 7.4 0 0 0 13.5 11a7.55 7.55 0 0 0-7.15 5.244A5.993 5.993 0 0 0 8 28h11a5.977 5.977 0 0 0 5.615-8.088a7.5 7.5 0 0 0 5.132-3.357a1.54 1.54 0 0 0 .097-1.52M19 26H8a3.993 3.993 0 0 1-.673-7.93l.663-.112l.145-.656a5.496 5.496 0 0 1 10.73 0l.145.656l.663.113A3.993 3.993 0 0 1 19 26m4.465-8.001h-.021a5.96 5.96 0 0 0-2.795-1.755a7.5 7.5 0 0 0-2.6-3.677c-.01-.101-.036-.197-.041-.3a6.08 6.08 0 0 1 3.79-6.05a8.46 8.46 0 0 0 1.94 7.596a7.4 7.4 0 0 0 3.902 2.228a5.43 5.43 0 0 1-4.175 1.958"/></svg>`,
	"light-rain":          `<svg aria-label="Light rain" xmlns="http://www.w3.org/2000/svg" width="32" height="32" viewBox="0 0 32 32"><path fill="currentColor" d="M11 30a1 1 0 0 1-.894-1.447l2-4a1 1 0 1 1 1.788.894l-2 4A1 1 0 0 1 11 30"/><path fill="currentColor" d="M24.8 9.136a8.994 8.994 0 0 0-17.6 0A6.497 6.497 0 0 0 8.5 22h10.881l-1.276 2.553a1 1 0 0 0 1.789.894L21.618 22H23.5a6.497 6.497 0 0 0 1.3-12.864M23.5 20h-15a4.498 4.498 0 0 1-.356-8.981l.816-.064l.099-.812a6.994 6.994 0 0 1 13.883 0l.099.812l.815.064A4.498 4.498 0 0 1 23.5 20"/></svg>`,
	"rain":                `<svg aria-label="Rain" xmlns="http://www.w3.org/2000/svg" width="32" height="32" viewBox="0 0 32 32"><path fill="currentColor" d="M23.5 22h-15A6.5 6.5 0 0 1 7.2 9.14a9 9 0 0 1 17.6 0A6.5 6.5 0 0 1 23.5 22M16 4a7 7 0 0 0-6.94 6.14L9 11h-.86a4.5 4.5 0 0 0 .36 9h15a4.5 4.5 0 0 0 .36-9H23l-.1-.82A7 7 0 0 0 16 4m-2 26a.93.93 0 0 1-.45-.11a1 1 0 0 1-.44-1.34l2-4a1 1 0 1 1 1.78.9l-2 4A1 1 0 0 1 14 30m6 0a.93.93 0 0 1-.45-.11a1 1 0 0 1-.44-1.34l2-4a1 1 0 1 1 1.78.9l-2 4A1 1 0 0 1 20 30M8 30a.93.93 0 0 1-.45-.11a1 1 0 0 1-.44-1.34l2-4a1 1 0 1 1 1.78.9l-2 4A1 1 0 0 1 8 30"/></svg>`,
	"thunder":             `<svg aria-label="Thunder" xmlns="http://www.w3.org/2000/svg" width="32" height="32" viewBox="0 0 32 32"><path fill="currentColor" d="M21 30a1 1 0 0 1-.894-1.447l2-4a1 1 0 1 1 1.788.894l-2 4A1 1 0 0 1 21 30M9 32a1 1 0 0 1-.894-1.447l2-4a1 1 0 1 1 1.788.894l-2 4A1 1 0 0 1 9 32m6.901-1.504l-1.736-.992L17.31 24h-6l4.855-8.496l1.736.992L14.756 22h6.001z"/><path fill="currentColor" d="M24.8 9.136a8.994 8.994 0 0 0-17.6 0a6.493 6.493 0 0 0 .23 12.768l-1.324 2.649a1 1 0 1 0 1.789.894l2-4a1 1 0 0 0-.447-1.341A1 1 0 0 0 9 20.01V20h-.5a4.498 4.498 0 0 1-.356-8.981l.816-.064l.099-.812a6.994 6.994 0 0 1 13.883 0l.099.812l.815.064A4.498 4.498 0 0 1 23.5 20H23v2h.5a6.497 6.497 0 0 0 1.3-12.864"/></svg>`,
	"snow":                `<svg aria-label="Snow" xmlns="http://www.w3.org/2000/svg" width="32" height="32" viewBox="0 0 32 32"><path fill="currentColor" d="M23.5 22h-15A6.5 6.5 0 0 1 7.2 9.14a9 9 0 0 1 17.6 0A6.5 6.5 0 0 1 23.5 22M16 4a7 7 0 0 0-6.94 6.14L9 11h-.86a4.5 4.5 0 0 0 .36 9h15a4.5 4.5 0 0 0 .36-9H23l-.1-.82A7 7 0 0 0 16 4m-4 21.05L10.95 24L9.5 25.45L8.05 24L7 25.05l1.45 1.45L7 27.95L8.05 29l1.45-1.45L10.95 29L12 27.95l-1.45-1.45zm14 0L24.95 24l-1.45 1.45L22.05 24L21 25.05l1.45 1.45L21 27.95L22.05 29l1.45-1.45L24.95 29L26 27.95l-1.45-1.45zm-7 2L17.95 26l-1.45 1.45L15.05 26L14 27.05l1.45 1.45L14 29.95L15.05 31l1.45-1.45L17.95 31L19 29.95l-1.45-1.45z"/></svg>`,
	"mist":                `<svg aria-label="Mist" xmlns="http://www.w3.org/2000/svg" width="32" height="32" viewBox="0 0 32 32"><path fill="currentColor" d="M24.8 11.138a8.994 8.994 0 0 0-17.6 0A6.53 6.53 0 0 0 2 17.5V19a1 1 0 0 0 1 1h12a1 1 0 0 0 0-2H4v-.497a4.52 4.52 0 0 1 4.144-4.482l.816-.064l.099-.812a6.994 6.994 0 0 1 13.883 0l.099.813l.815.063A4.496 4.496 0 0 1 23.5 22H7a1 1 0 0 0 0 2h16.5a6.496 6.496 0 0 0 1.3-12.862"/><rect width="18" height="2" x="2" y="26" fill="currentColor" rx="1"/></svg>`,
}

func getWeatherIcon(iconId string) string {
	switch iconId {
	case "01d":
		return WeatherIcons["clear-day"]
	case "01n":
		return WeatherIcons["clear-night"]
	case "02d", "03d":
		return WeatherIcons["partly-cloudy-day"]
	case "02n", "03n":
		return WeatherIcons["partly-cloudy-night"]
	case "04d":
		return WeatherIcons["mostly-cloudy-day"]
	case "04n":
		return WeatherIcons["mostly-cloudy-night"]
	case "09d", "09n":
		return WeatherIcons["light-rain"]
	case "10d", "10n":
		return WeatherIcons["rain"]
	case "11d", "11n":
		return WeatherIcons["thunder"]
	case "13d", "13n":
		return WeatherIcons["snow"]
	case "50d", "50n":
		return WeatherIcons["mist"]
	default:
		return ""
	}
}

func init() {
	godotenv.Load()
}

func main() {
	if err := os.MkdirAll("public/uploads", 0755); err != nil {
		log.Fatal(err)
	}

	dbPath, err := filepath.Abs("passport.db")
	if err != nil {
		log.Fatal(err)
	}

	app, err := NewApp(dbPath, map[string]any{
		"_time_format":  "sqlite",
		"cache":         "shared",
		"mode":          "rwc",
		"_journal_mode": "WAL",
	})
	if err != nil {
		log.Fatal(err)
	}
	defer app.Close()

	go func() {
		c := make(chan os.Signal, 1)
		signal.Notify(c, syscall.SIGINT, syscall.SIGTERM)
		<-c
		app.Close()
		os.Exit(0)
	}()

	templatesDir, err := fs.Sub(embeddedAssets, "templates")
	if err != nil {
		log.Fatal(err)
	}

	assetsDir, err := fs.Sub(embeddedAssets, "assets")
	if err != nil {
		log.Fatal(err)
	}

	engine := handlebars.NewFileSystem(http.FS(templatesDir), ".hbs")

	engine.AddFunc("eq", func(a, b any) bool {
		return a == b
	})

	engine.AddFunc("embedFile", func(fileToEmbed string) string {
		content, err := fs.ReadFile(embeddedAssets, fileToEmbed)
		if err != nil {
			return ""
		}

		fileExtension := filepath.Ext(fileToEmbed)

		switch fileExtension {
		case ".js":
			return fmt.Sprintf("<script>%s</script>", content)
		case ".css":
			return fmt.Sprintf("<style>%s</style>", content)
		default:
			return string(content)
		}
	})

	engine.AddFunc("devContent", func() string {
		if app.Config.DevMode {
			return devContent
		}
		return ""
	})

	router := fiber.New(fiber.Config{
		Views: engine,
	})

	router.Use(helmet.New(helmet.ConfigDefault))

	// redirect /favicon.ico to /assets/favicon.ico
	router.Get("/favicon.ico", func(c fiber.Ctx) error {
		return c.Redirect().To("/assets/favicon.ico")
	})

	router.Use(compress.New(compress.Config{
		Level: compress.LevelBestSpeed,
	}))

	router.Use(gonify.New(gonify.Config{
		MinifySVG:  !app.DevMode,
		MinifyCSS:  !app.DevMode,
		MinifyJS:   !app.DevMode,
		MinifyHTML: !app.DevMode,
	}))

	router.Use("/", static.New("./public", static.Config{
		Browse: false,
		MaxAge: 31536000,
	}))

	router.Use("/assets", static.New("", static.Config{
		FS:     assetsDir,
		Browse: false,
		MaxAge: 31536000,
	}))

	router.Get("/", func(c fiber.Ctx) error {
		c.Response().Header.Set("Link", "</assets/fonts/InstrumentSans-VariableFont_wdth,wght.woff2>; rel=preload; as=font; type=font/woff2; crossorigin")

		renderData := fiber.Map{
			"SearchProviderURL": app.Config.SearchProvider.URL,
			"SearchParam":       app.Config.SearchProvider.Query,
			"Categories":        app.CategoryManager.GetCategories(),
		}

		if app.Config.WeatherAPIKey != "" {
			weather := app.WeatherManager.GetWeather()

			renderData["WeatherData"] = fiber.Map{
				"Temp": weather.Temperature,
				"Desc": weather.WeatherText,
				"Icon": getWeatherIcon(weather.Icon),
			}
		}

		if app.Config.UptimeAPIKey != "" {
			renderData["UptimeData"] = app.UptimeManager.GetUptime()
		}

		return c.Render("views/index", renderData)
	})

	router.Use(middleware.AdminMiddleware(app.db))

	router.Get("/admin/login", func(c fiber.Ctx) error {
		if c.Locals("IsAdmin") != nil {
			return c.Redirect().To("/admin")
		}

		return c.Render("views/admin/login", fiber.Map{})
	})

	router.Post("/admin/login", func(c fiber.Ctx) error {
		if c.Locals("IsAdmin") != nil {
			return c.Redirect().To("/admin")
		}

		var loginData struct {
			Username string `json:"username"`
			Password string `json:"password"`
		}
		if err := c.Bind().JSON(&loginData); err != nil {
			return err
		}

		// possible vulnerable to timing attacks
		if loginData.Username != app.Config.Admin.Username || loginData.Password != app.Config.Admin.Password {
			return c.Status(http.StatusUnauthorized).JSON(fiber.Map{"message": "Invalid username or password"})
		}

		// Create new session
		sessionID := uuid.NewString()
		expiresAt := time.Now().Add(time.Hour * 24 * 7)
		_, err := app.db.Exec(`
			INSERT INTO sessions (session_id, expires_at) 
			VALUES (?, ?)
		`, sessionID, expiresAt)
		if err != nil {
			return err
		}

		// Set cookie
		c.Cookie(&fiber.Cookie{
			Name:    "SessionToken",
			Value:   sessionID,
			Expires: expiresAt,
		})

		return c.Status(http.StatusOK).JSON(fiber.Map{"message": "Logged in successfully"})
	})

	router.Get("/admin", func(c fiber.Ctx) error {
		if c.Locals("IsAdmin") == nil {
			return c.Redirect().To("/admin/login")
		}

		return c.Render("views/admin/index", fiber.Map{
			"Categories": app.CategoryManager.GetCategories(),
			"IsAdmin":    true,
		})
	})

	api := router.Group("/api")
	{
		// all API routes require admin auth. No user needs to make api requests since the site is SSR
		api.Use(func(c fiber.Ctx) error {
			if c.Locals("IsAdmin") == nil {
				return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{"message": "Unauthorized"})
			}
			return c.Next()
		})

		api.Post("/category", func(c fiber.Ctx) error {
			var req struct {
				Name string `form:"name"`
			}
			if err := c.Bind().Form(&req); err != nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Failed to parse request",
				})
			}

			if req.Name == "" {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Name is required",
				})
			}

			req.Name = strings.TrimSpace(req.Name)

			if len(req.Name) > 50 {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Name is too long. Maximum length is 50 characters",
				})
			}

			file, err := c.FormFile("icon")
			if err != nil || file == nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Icon is required",
				})
			}

			if file.Size > 5*1024*1024 {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "File size too large. Maximum size is 5MB",
				})
			}

			contentType := file.Header.Get("Content-Type")
			if contentType != "image/svg+xml" {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Only SVGs are supported for category icons!",
				})
			}

			iconPath, err := UploadFile(file, contentType, c)
			if err != nil {
				return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
					"message": "Failed to upload file, please try again!",
				})
			}

			category, err := app.CategoryManager.CreateCategory(Category{
				Name:  req.Name,
				Icon:  iconPath,
				Links: []Link{},
			})

			if err != nil {
				return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
					"message": fmt.Sprintf("Failed to create category: %v", err),
				})
			}

			return c.Status(fiber.StatusCreated).JSON(fiber.Map{
				"message":  "Category created successfully",
				"category": category,
			})
		})

		api.Post("/category/:id/link", func(c fiber.Ctx) error {
			var req struct {
				Name        string `form:"name"`
				Description string `form:"description"`
				URL         string `form:"url"`
			}
			if err := c.Bind().Form(&req); err != nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Failed to parse request",
				})
			}

			if req.Name == "" || req.URL == "" {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Name and URL are required",
				})
			}

			req.Name = strings.TrimSpace(req.Name)
			if req.Description != "" {
				req.Description = strings.TrimSpace(req.Description)
			}

			if len(req.Name) > 50 {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Name is too long. Maximum length is 50 characters",
				})
			}

			if len(req.Description) > 150 {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Description is too long. Maximum length is 150 characters",
				})
			}

			categoryID, err := strconv.ParseInt(c.Params("id"), 10, 64)
			if err != nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": fmt.Sprintf("Failed to parse category ID: %v", err),
				})
			}

			if app.CategoryManager.GetCategory(categoryID) == nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Category not found",
				})
			}

			file, err := c.FormFile("icon")
			if err != nil || file == nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Icon is required",
				})
			}

			if file.Size > 5*1024*1024 {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "File size too large. Maximum size is 5MB",
				})
			}

			contentType := file.Header.Get("Content-Type")
			if !strings.HasPrefix(contentType, "image/") {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Only image files are allowed",
				})
			}

			iconPath, err := UploadFile(file, contentType, c)
			if err != nil {
				slog.Error("Failed to upload file", "error", err)
				return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
					"message": "Failed to upload file, please try again!",
				})
			}

			link, err := app.CategoryManager.CreateLink(app.CategoryManager.db, Link{
				CategoryID:  categoryID,
				Name:        req.Name,
				Description: req.Description,
				Icon:        iconPath,
				URL:         req.URL,
			})
			if err != nil {
				slog.Error("Failed to create link", "error", err)
				return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
					"message": "Failed to create link",
				})
			}

			return c.Status(fiber.StatusCreated).JSON(fiber.Map{
				"message": "Link created successfully",
				"link":    link,
			})
		})

		api.Patch("/category/:id", func(c fiber.Ctx) error {
			var req struct {
				Name string `form:"name"`
			}

			if c.Params("id") == "" {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "ID is required",
				})
			}

			id, err := strconv.ParseInt(c.Params("id"), 10, 64)
			if err != nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": fmt.Sprintf("Failed to parse category ID: %v", err),
				})
			}

			if err := c.Bind().Form(&req); err != nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Failed to parse request",
				})
			}

			if req.Name != "" {
				if len(req.Name) > 50 {
					return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
						"message": "Name is too long. Maximum length is 50 characters",
					})
				}
			}

			category := app.CategoryManager.GetCategory(id)
			if category == nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Category not found",
				})
			}

			tx, err := app.db.Begin()
			if err != nil {
				return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
					"message": "Failed to start transaction",
				})
			}
			defer tx.Rollback()

			file, err := c.FormFile("icon")
			if err == nil {
				if file.Size > 5*1024*1024 {
					return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
						"message": "File size too large. Maximum size is 5MB",
					})
				}

				contentType := file.Header.Get("Content-Type")
				if contentType != "image/svg+xml" {
					return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
						"message": "Only svg files are allowed",
					})
				}

				oldIconPath := category.Icon

				iconPath, err := UploadFile(file, contentType, c)
				if err != nil {
					slog.Error("Failed to upload file", "error", err)
					return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
						"message": "Failed to upload file, please try again!",
					})
				}

				_, err = tx.Exec("UPDATE categories SET icon = ? WHERE id = ?", iconPath, id)
				if err != nil {
					return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
						"message": "Failed to update category",
					})
				}

				err = os.Remove(filepath.Join("public/", oldIconPath))
				if err != nil {
					slog.Error("Failed to delete icon", "error", err)
				}
			}

			if req.Name != "" {
				_, err = tx.Exec("UPDATE categories SET name = ? WHERE id = ?", req.Name, category.ID)
				if err != nil {
					return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
						"message": "Failed to update category",
					})
				}
			}

			err = tx.Commit()
			if err != nil {
				return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
					"message": "Failed to commit transaction",
				})
			}

			return c.Status(fiber.StatusOK).JSON(fiber.Map{
				"message": "Category updated successfully",
			})
		})

		api.Patch("/category/:categoryID/link/:linkID", func(c fiber.Ctx) error {
			var req struct {
				Name        string `form:"name"`
				Description string `form:"description"`
				Icon        string `form:"icon"`
			}
			if err := c.Bind().Form(&req); err != nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Failed to parse request",
				})
			}

			if len(req.Name) > 50 {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Name is too long. Maximum length is 50 characters",
				})
			}

			if len(req.Description) > 150 {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Description is too long. Maximum length is 150 characters",
				})
			}

			linkID, err := strconv.ParseInt(c.Params("linkID"), 10, 64)
			if err != nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": fmt.Sprintf("Failed to parse link ID: %v", err),
				})
			}

			categoryID, err := strconv.ParseInt(c.Params("categoryID"), 10, 64)
			if err != nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": fmt.Sprintf("Failed to parse category ID: %v", err),
				})
			}

			if app.CategoryManager.GetCategory(categoryID) == nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Category not found",
				})
			}

			link := app.CategoryManager.GetLink(linkID)
			if link == nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Link not found",
				})
			}

			tx, err := app.db.Begin()
			if err != nil {
				return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
					"message": "Failed to start transaction",
				})
			}
			defer tx.Rollback()

			file, err := c.FormFile("icon")
			if err == nil {
				if file.Size > 5*1024*1024 {
					return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
						"message": "File size too large. Maximum size is 5MB",
					})
				}

				contentType := file.Header.Get("Content-Type")
				if !strings.HasPrefix(contentType, "image/") {
					return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
						"message": "Only image files are allowed",
					})
				}

				oldIconPath := link.Icon

				iconPath, err := UploadFile(file, contentType, c)
				if err != nil {
					slog.Error("Failed to upload file", "error", err)
					return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
						"message": "Failed to upload file, please try again!",
					})
				}

				_, err = tx.Exec("UPDATE links SET icon = ? WHERE id = ?", iconPath, linkID)
				if err != nil {
					return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
						"message": "Failed to update link",
					})
				}

				err = os.Remove(filepath.Join("public/", oldIconPath))
				if err != nil {
					slog.Error("Failed to delete icon", "error", err)
				}
			}

			if req.Name != "" {
				_, err = tx.Exec("UPDATE links SET name = ? WHERE id = ?", req.Name, linkID)
				if err != nil {
					return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
						"message": "Failed to update link",
					})
				}
			}

			if req.Description != "" {
				_, err = tx.Exec("UPDATE links SET description = ? WHERE id = ?", req.Description, linkID)
				if err != nil {
					return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
						"message": "Failed to update link",
					})
				}
			}

			err = tx.Commit()
			if err != nil {
				return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
					"message": "Failed to commit transaction",
				})
			}

			slog.Info("Link updated successfully", "id", linkID, "name", req.Name)

			return c.Status(fiber.StatusOK).JSON(fiber.Map{
				"message": "Link updated successfully",
			})
		})

		api.Delete("/category/:categoryID/link/:linkID", func(c fiber.Ctx) error {
			linkID, err := strconv.ParseInt(c.Params("linkID"), 10, 64)
			if err != nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": fmt.Sprintf("Failed to parse link ID: %v", err),
				})
			}

			categoryID, err := strconv.ParseInt(c.Params("categoryID"), 10, 64)
			if err != nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": fmt.Sprintf("Failed to parse category ID: %v", err),
				})
			}

			if app.CategoryManager.GetCategory(categoryID) == nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Category not found",
				})
			}

			link := app.CategoryManager.GetLink(linkID)
			if link == nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Link not found",
				})
			}

			if link.CategoryID != categoryID {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Invalid category ID",
				})
			}

			err = app.CategoryManager.DeleteLink(linkID)
			if err != nil {
				return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
					"message": fmt.Sprintf("Failed to delete link: %v", err),
				})
			}

			return c.SendStatus(fiber.StatusOK)
		})

		api.Delete("/category/:id", func(c fiber.Ctx) error {
			// id = parseInt(c.Params("id"))
			id, err := strconv.ParseInt(c.Params("id"), 10, 64)
			if err != nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": fmt.Sprintf("Failed to parse category ID: %v", err),
				})
			}

			if app.CategoryManager.GetCategory(id) == nil {
				return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{
					"message": "Category not found",
				})
			}

			err = app.CategoryManager.DeleteCategory(id)
			if err != nil {
				return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
					"message": fmt.Sprintf("Failed to delete category: %v", err),
				})
			}

			return c.SendStatus(fiber.StatusOK)
		})
	}

	router.Listen(":3000", fiber.ListenConfig{
		EnablePrefork: app.Config.Prefork,
	})
}