package main

import (
	"html/template"
	"log"
	"net/http"
	"os"
	"strconv"
	"strings"
)

type Index struct {
	WebpageTitle       string
	WebpageIcon        string
	WebpageKeywords    string
	WebpageDescription string
	WebpageURL         string
	GalleryItems       []GalleryItems
	SocialItems        []SocialItems
	VideoItems         []VideoItems
	GalleryPhoto       *GalleryItems
	VideoContent       *VideoItems
	Nav                bool
}

type GalleryItems struct {
	URL   string
	ID    int
	Title string
}

type SocialItems struct {
	URL      string
	Name     string
	Username string
	Icon     string
	Color    string
}

type VideoItems struct {
	URL         string
	ID          int
	Title       string
	Description string
	Thumbnail   string
}

var data = &Index{
	WebpageTitle:       "Keyemail",
	WebpageIcon:        "/src/assets/main/favicon.ico",
	WebpageKeywords:    "keyemail",
	WebpageDescription: "Welcome to keyemail.dev! A profile based on Keyemail.",
	WebpageURL:         "https://keyemail.dev",
	GalleryItems:       galleryItems,
	SocialItems:        socialItems,
	VideoItems:         videoItems,
}

var currentData = &Index{}

var galleryItems = []GalleryItems{
	{"/src/assets/gallery/2_cats.jpg", 1, "Resting Day"},
	{"/src/assets/gallery/friend_cat.jpg", 2, "Orange Furr"},
	{"/src/assets/gallery/miku_poster.jpg", 3, "Miku Spotted"},
	{"/src/assets/gallery/networking.jpg", 4, "Wires of Networking"},
	{"/src/assets/gallery/another_friend_cat.jpg", 5, "Night Cat"},
	{"/src/assets/gallery/pc_case_stickers.jpg", 6, "I use arch BTW"},
	{"/src/assets/gallery/movie_picture_1.jpg", 7, "Prime Cut Cafe"},
	{"/src/assets/gallery/movie_picture_2.jpg", 8, "Big Fountain"},
	{"/src/assets/gallery/bird.jpg", 9, "Birds Eye"},
	{"/src/assets/gallery/japanese_garden_1.jpg", 10, "Pond of Fishes"},
	{"/src/assets/gallery/japanese_garden_2.jpg", 11, "Waterfall Pond"},
	{"/src/assets/gallery/japanese_garden_3.jpg", 12, "Bridge of Faith"},
	{"/src/assets/gallery/japanese_garden_4.jpg", 13, "Little Rocks"},
	{"/src/assets/gallery/japanese_garden_5.jpg", 14, "Arc over Pond"},
	{"/src/assets/gallery/japanese_garden_6.jpg", 15, "Beautiful Pond"},
}

var socialItems = []SocialItems{
	{"https://discord.gg/VTEn4zuh", "Discord", "keyemail", "/src/assets/icons/discord.svg", "5865F2"},
	{"https://github.com/Keyemail", "GitHub", "Keyemail", "/src/assets/icons/github.svg", "171515"},
	{"https://steamcommunity.com/id/keyemail/", "Steam", "Keyemail", "/src/assets/icons/steam.svg", "000000"},
	{"https://open.spotify.com/user/316yuurxrw3zcprxsnrvgamxktp4", "Spotify", "Keyemail", "/src/assets/icons/spotify.svg", "1DB954"},
	{"https://www.youtube.com/channel/UCCNkKG8XoZCh52vLCbXYy7g", "YouTube", "Keyemail", "/src/assets/icons/youtube.svg", "CD201F"},
	{"https://www.instagram.com/keyemail1?igsh=OGQ5ZDc2ODk2ZA%3D%3D&utm_source=qr", "Instagram", "keyemail1", "/src/assets/icons/instagram.svg", "C13584"},
}

var videoItems = []VideoItems{
	{"/src/assets/videos/orange-cat.mp4", 1, "Orange Cat", "Cute little orange cat that my friend has!! Hes so cute lol.", "/src/assets/video_thumbnails/orange-cat.png"},
}

func renderTemplate(w http.ResponseWriter, tmpl string, data *Index, nav bool) {
	var templateFiles []string

	templateFiles = append(templateFiles, "src/pages/"+tmpl+".tmpl", "src/index.tmpl", "src/components/navbar.tmpl")

	templates, err := template.ParseFiles(templateFiles...)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}

	data.Nav = nav

	err = templates.ExecuteTemplate(w, "base", data)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
	}

	log.Println("GET /" + tmpl)
}

func galleryContentHandler(w http.ResponseWriter, req *http.Request) {
	parts := strings.Split(req.URL.Path, "/")

	if len(parts) == 3 && parts[1] == "gallery" {
		idStr := parts[2]
		id, err := strconv.Atoi(idStr)
		if err != nil || id < 1 || id > len(galleryItems) {
			renderTemplate(w, "404", data, true)
			return
		}

		var item GalleryItems
		for _, gi := range galleryItems {
			if gi.ID == id {
				item = gi
				break
			}
		}

		*currentData = *data
		currentData.GalleryPhoto = &item
		currentData.GalleryItems = nil

		renderTemplate(w, "gallery_content", currentData, true)
		return
	}

	renderTemplate(w, "404", data, true)
}

func videoContentHandler(w http.ResponseWriter, req *http.Request) {
	parts := strings.Split(req.URL.Path, "/")

	if len(parts) == 3 && parts[1] == "videos" {
		idStr := parts[2]
		id, err := strconv.Atoi(idStr)
		if err != nil || id < 1 || id > len(videoItems) {
			renderTemplate(w, "404", data, true)
			return
		}

		var item VideoItems
		for _, gi := range videoItems {
			if gi.ID == id {
				item = gi
				break
			}
		}

		*currentData = *data
		currentData.VideoContent = &item
		currentData.VideoItems = nil

		renderTemplate(w, "video_content", currentData, true)
		return
	}

	renderTemplate(w, "404", data, true)
}

func videoHandler(w http.ResponseWriter, req *http.Request) {
	renderTemplate(w, "video", data, true)
}

func socialsHandler(w http.ResponseWriter, req *http.Request) {
	renderTemplate(w, "socials", data, true)
}

func galleryHandler(w http.ResponseWriter, req *http.Request) {
	renderTemplate(w, "gallery", data, true)
}

func indexHandler(w http.ResponseWriter, req *http.Request) {
	if req.URL.Path != "/" {
		renderTemplate(w, "404", data, true)
		return
	}

	renderTemplate(w, "home", data, true)
}

func main() {
	if _, err := os.Stat("src/index.tmpl"); os.IsNotExist(err) {
		log.Fatal("Template file does not exist:\n", err)
	}

	fs := http.FileServer(http.Dir("./src"))
	http.Handle("/src/", http.StripPrefix("/src/", fs))

	port := ":3939"

	http.HandleFunc("/", indexHandler)
	http.HandleFunc("/socials", socialsHandler)
	http.HandleFunc("/gallery", galleryHandler)
	http.HandleFunc("/videos", videoHandler)
	http.HandleFunc("/gallery/", galleryContentHandler)
	http.HandleFunc("/videos/", videoContentHandler)

	log.Println("Server running at http://localhost" + port)
	http.ListenAndServe(port, nil)
}