package main

import (
	"log"
	"path/filepath"

	"github.com/fasthttp/router"
	"github.com/fsnotify/fsnotify"
	"github.com/valyala/fasthttp"
	"trle5.xyz/gopkg/trpic/src/site"
)

func main() {
	watcher, err := fsnotify.NewWatcher()
	if err != nil { log.Fatal(err) }
	defer watcher.Close()

	trpic := site.New()

	paths := []string{
		trpic.Config.GalleryDir,
		"config.yaml",
		"web/templates",
		"web/scripts",
		"web/styles",
	}

	for _, path := range paths {
		err = watcher.Add(path)
		if err != nil { log.Fatal(path, err) }
	}

	go func() {
		for {
			select {
			case event, ok := <-watcher.Events:
				if !ok { return }
				switch filepath.ToSlash(filepath.Dir(event.Name)) {
				case "web/templates":
					trpic.LoadTemplates()
					log.Println("Template Reloaded")
				case "web/scripts":
					trpic.LoadScripts()
					log.Println("Script Reloaded")
				case "web/styles":
					trpic.LoadStyles()
					log.Println("Style Reloaded")
				case trpic.Config.GalleryDir:
					trpic.LoadImages()
					log.Println("Image Reloaded")
				default:
					switch filepath.Base(event.Name) {
					case "config.yaml":
						trpic.Reload()
						log.Println("Config Reloaded")
					}
				}
				log.Println("event:", event.Name, event.Op)
			case _, ok := <-watcher.Errors:
				if !ok {
					return
				}
			}
		}
	}()

	r := router.New()
	r.GET("/", trpic.PageHandler)
	r.GET("/noscript", trpic.PageHandler)
	r.GET("/favicon.ico", func(ctx *fasthttp.RequestCtx) {
		fasthttp.ServeFile(ctx, "web/what.png")
	})
	api := r.Group("/api"); {
		api.GET("/", trpic.APIHandler)
		api.GET("/images", trpic.APIImagesHandler)
	}
	r.ServeFiles("/image/{filepath:*}", trpic.Config.GalleryDir)
	r.ServeFiles("/thumbnail/{filepath:*}", trpic.Config.ThumbnailDir)
	log.Println("服务器启动，访问 http://localhost:8080 进行预览")
	log.Fatal(fasthttp.ListenAndServe(":8080", r.Handler))
}