package main

import (
	"context"
	"flag"
	"log"
	"log/slog"
	"net"
	"net/http"
	"os"
	"path/filepath"

	"github.com/donatj/hmacsig"
	"github.com/facebookgo/flagenv"
	_ "github.com/joho/godotenv/autoload"
	"tailscale.com/hostinfo"
	"tailscale.com/tsnet"
	"tailscale.com/tsweb"
	"xeiaso.net/v4/internal"
	"xeiaso.net/v4/internal/lume"
)

var (
	bind                = flag.String("bind", ":3000", "Port to listen on")
	devel               = flag.Bool("devel", false, "Enable development mode")
	dataDir             = flag.String("data-dir", "./var", "Directory to store data in")
	gitBranch           = flag.String("git-branch", "main", "Git branch to clone")
	gitRepo             = flag.String("git-repo", "https://github.com/Xe/site", "Git repository to clone")
	githubSecret        = flag.String("github-secret", "", "GitHub secret to use for webhooks")
	miToken             = flag.String("mi-token", "", "Token to use for the mi API")
	patreonSaasProxyURL = flag.String("patreon-saasproxy-url", "http://patreon-saasproxy/give-token", "URL to use for the patreon saasproxy")
	siteURL             = flag.String("site-url", "https://xeiaso.net/", "URL to use for the site")
	tsnetHostname       = flag.String("tailscale-hostname", "xesite", "Tailscale hostname to use")
)

func main() {
	flagenv.Parse()
	flag.Parse()
	internal.Slog()

	hostinfo.SetApp("xeiaso.net/v4/cmd/xesite")

	ctx := context.Background()

	ln, err := net.Listen("tcp", *bind)
	if err != nil {
		log.Fatal(err)
	}

	os.MkdirAll(*dataDir, 0700)
	os.MkdirAll(filepath.Join(*dataDir, "tsnet"), 0700)

	srv := &tsnet.Server{
		Hostname: *tsnetHostname + "-" + os.Getenv("FLY_REGION"),
		Logf:     func(string, ...any) {},
		Dir:      filepath.Join(*dataDir, "tsnet"),
	}

	if err := srv.Start(); err != nil {
		log.Fatal(err)
	}

	if _, err := srv.Up(context.Background()); err != nil {
		log.Fatal(err)
	}

	hc := srv.HTTPClient()

	pc, err := NewPatreonClient(hc)
	if err != nil {
		slog.Error("can't create patreon client", "err", err)
	}

	fs, err := lume.New(ctx, &lume.Options{
		Branch:        *gitBranch,
		Repo:          *gitRepo,
		StaticSiteDir: "lume",
		URL:           *siteURL,
		Development:   *devel,
		PatreonClient: pc,
		DataDir:       *dataDir,
		MiToken:       *miToken,
	})
	if err != nil {
		log.Fatal(err)
	}

	defer fs.Close()

	if *devel {
		go rebuildOnChange(fs)
	}

	go internalAPI(srv, fs)

	if err != nil {
		log.Fatal(err)
	}

	mux := http.NewServeMux()
	mux.Handle("/", http.FileServer(http.FS(fs)))
	mux.HandleFunc("/metrics", tsweb.VarzHandler)

	mux.HandleFunc("/blog.atom", func(w http.ResponseWriter, r *http.Request) {
		http.Redirect(w, r, "/blog.rss", http.StatusMovedPermanently)
	})

	// NOTE(Xe): Had to rename this page because of a Lume/Go embed bug.
	mux.HandleFunc(`/blog/%F0%9F%A5%BA`, func(w http.ResponseWriter, r *http.Request) {
		http.Redirect(w, r, "/blog/xn--ts9h/", http.StatusMovedPermanently)
	})
	mux.HandleFunc(`/blog/🥺`, func(w http.ResponseWriter, r *http.Request) {
		http.Redirect(w, r, "/blog/xn--ts9h/", http.StatusMovedPermanently)
	})

	mux.HandleFunc("/static/manifest.json", func(w http.ResponseWriter, r *http.Request) {
		http.Redirect(w, r, "/static/site.webmanifest", http.StatusMovedPermanently)
	})

	gh := &GitHubWebhook{fs: fs}
	s := hmacsig.Handler256(gh, *githubSecret)
	mux.Handle("/.within/hook/github", s)

	mux.Handle("/.within/hook/patreon", &PatreonWebhook{fs: fs})

	var h http.Handler = mux
	h = internal.ClackSet(fs.Clacks()).Middleware(h)
	h = internal.CacheHeader(h)
	h = internal.AcceptEncodingMiddleware(h)
	h = internal.RefererMiddleware(h)

	slog.Info("starting server", "bind", *bind)
	log.Fatal(http.Serve(ln, h))
}