package main

import (
	"bufio"
	"fmt"
	"log"
	"os"
	"path/filepath"
	"time"

	"github.com/adrg/xdg"
	"github.com/gofiber/fiber/v2"
	"github.com/google/uuid"
	"github.com/valyala/fasthttp"

	"embed"
	"net/http"

	"github.com/gofiber/template/html"

	"git.bascht.space/bascht/scanberry/scan"
)

//go:embed views/*
var viewsfs embed.FS

var documents = make(map[string]scan.Document)
var basedir string

func main() {
	// Create a new engine by passing the template folder
	// and template extension using <engine>.New(dir, ext string)
	// engine := html.New("./views", ".html")

	// // We also support the http.FileSystem interface
	// // See examples below to load templates from embedded files
	// engine := html.NewFileSystem(http.Dir("./views"), ".html")
	engine := html.NewFileSystem(http.FS(viewsfs), ".html")

	basedir = xdg.RuntimeDir + "/himbeerscan"
	os.MkdirAll(filepath.Join(basedir, "downloads"), os.ModePerm)
	os.Chdir(basedir)

	engine.Reload(true) // Optional. Default: false
	engine.Debug(false) // Optional. Default: false
	engine.Layout("embed") // Optional. Default: "embed"
	engine.Delims("{{", "}}") // Optional. Default: engine delimiters

	// engine.AddFunc("greet", func(name string) string {
	// 	return "Hallouhu, " + name + "!"
	// })

	// After you created your engine, you can pass it to Fiber's Views Engine
	app := fiber.New(fiber.Config{
		Views: engine,
	})

	app.Static("/downloads", basedir+"/downloads")

	app.Get("/", func(c *fiber.Ctx) error {
		return c.Redirect("/scan")
	})

	app.Get("/scan", func(c *fiber.Ctx) error {
		return c.Render("views/scan", fiber.Map{
			"Title": "Hello, World!",
		}, "views/layouts/main")
	})

	app.Post("/scan", func(c *fiber.Ctx) error {

		document := scan.Document{
			Id:     uuid.NewString(),
			Name:   c.FormValue("name"),
			Date:   time.Now(),
			Duplex: c.FormValue("duplex") == "on",
			Events: make(chan string),
		}

		documents[document.Id] = document

		go scan.Process(basedir, &document)

		return c.Render("views/status", fiber.Map{
			"id":                    document.Id,
			"name":                  document.Name,
			"FullName":              document.FullName(),
			"FullNameWithExtension": document.FullNameWithExtension(),
		}, "views/layouts/main")
	})

	app.Get("/show/:id", func(c *fiber.Ctx) error {
		id := c.Params("id")
		document := documents[id]

		return c.Render("views/show", fiber.Map{
			"id":                    document.Id,
			"name":                  document.Name,
			"FullName":              document.FullName(),
			"FullNameWithExtension": document.FullNameWithExtension(),
		}, "views/layouts/main")
	})

	app.Get("/events/:id", func(c *fiber.Ctx) error {
		c.Set("Content-Type", "text/event-stream")
		c.Set("Cache-Control", "no-cache")
		c.Set("Connection", "keep-alive")
		c.Set("Transfer-Encoding", "chunked")

		id := c.Params("id")
		fmt.Println("Searching for " + id)
		document := documents[id]

		c.Context().SetBodyStreamWriter(fasthttp.StreamWriter(func(w *bufio.Writer) {
			fmt.Fprintf(w, "event: status\ndata: started\n\n")
			w.Flush()
			fmt.Println("WRITER")

			for event := range document.Events {
				fmt.Fprintf(w, "data: Message: %s\n\n", event)
				w.Flush()
			}

			fmt.Fprintf(w, "event: status\ndata: stopped\n\n")
			w.Flush()

		}))

		return nil
	})

	log.Fatal(app.Listen(":3000"))
}