package main

import (
	"context"
	"encoding/json"
	"flag"
	"fmt"
	"log"
	"os"
	"strings"
	"time"

	"git.sr.ht/~adnano/go-gemini"
	"git.sr.ht/~adnano/go-gemini/certificate"
)

var (
	hostname        string
	port            string
	fileSrvHost     string
	fileSrvPort     string
	certificatePath string
)

func init() {
	flag.StringVar(&hostname, "hostname", "", "capsule hostname")
	flag.StringVar(&port, "port", "1965", "capsule port")
	flag.StringVar(&certificatePath, "certificatePath", "", "capsule certificate path")
	flag.StringVar(&fileSrvHost, "fileSrvHost", "", "file push server host")
	flag.StringVar(&fileSrvPort, "fileSrvPort", "", "file push server port")
	flag.Parse()
}

func main() {
	certificates := &certificate.Store{}
	certificates.Register(hostname)
	if err := certificates.Load(certificatePath); err != nil {
		panic(err)
	}

	mux := &gemini.Mux{}
	mux.HandleFunc("/favicon.txt", processFavicon)
	mux.HandleFunc("/", process)

	server := &gemini.Server{
		Handler:        mux,
		ReadTimeout:    30 * time.Second,
		WriteTimeout:   1 * time.Minute,
		GetCertificate: certificates.Get,
	}

	ctx := context.Background()
	if err := server.ListenAndServe(ctx); err != nil {
		log.Fatal(err)
	}
}

func processFavicon(ctx context.Context, w gemini.ResponseWriter, r *gemini.Request) {
	fmt.Fprintf(w, "\u263d")
}

func process(ctx context.Context, w gemini.ResponseWriter, r *gemini.Request) {
	fmt.Println(ctx, r, r.URL, r.Host)

	renderIndex("en", w)
}

func renderIndex(lang string, w gemini.ResponseWriter) {
	langPath := lang + "/"
	if lang == "sgs" {
		langPath = ""
	}
	w.SetMediaType("text/gemini")
	content, err := os.ReadFile(fmt.Sprintf("templates/%s/index.gmi", lang))
	if err != nil {
		w.WriteHeader(gemini.StatusTemporaryFailure, "Internal server error")
		return
	}
	page := string(content)

	summery := GetSummery()
	fmt.Println(summery)
	tree := TSummery{}
	err = json.Unmarshal([]byte(summery), &tree)
	if err != nil {
		w.WriteHeader(gemini.StatusTemporaryFailure, "Internal server error")
		return
	}
	fmt.Println(tree)

	categories := ""
	for c, count := range tree.Categories {
		fmt.Println(c, count)
		categories = categories + fmt.Sprintf("=> /%s%s/%s %s (%d)\n", langPath, "tree/path", c, c, count)
	}
	page = strings.Replace(page, "{{categories}}", categories, 1)

	tags := ""
	for t, count := range tree.Tags {
		tags = tags + fmt.Sprintf("=> /t/%s %s (%d)\n", t, t, count)
	}
	page = strings.Replace(page, "{{tags}}", tags, 1)

	lastFiles := ""
	for _, f := range GetLastFiles(tree.Files) {
		lastFiles = lastFiles + fmt.Sprintf("=> /%s/%s %s (%s)\n", strings.Join(f.Category, "/"), f.Id, f.Description, f.Created)
	}
	page = strings.Replace(page, "{{last_posts}}", lastFiles, 1)

	fmt.Fprintf(w, page)
}