package main

import (
	"flag"
	"fmt"
	"os"
	"path/filepath"
	"plugin"
	"runtime"

	"github.com/gofiber/fiber/v3"
	bf2 "gopkg.in/russross/blackfriday.v2"

	"fountain/article"
	"fountain/utils"
)

const VERSION = "0.6.1"

var (
	serve   bool   // 运行WEB服务
	port    uint   // 服务端口
	root    string // 博客根目录
	theme   string // 皮肤主题
	clean   bool   // 清理旧输出
	verbose bool   // 输出详情
)

func init() {
	flag.BoolVar(&serve, "s", false, "运行WEB服务")
	flag.UintVar(&port, "p", 0, "服务端口")
	flag.StringVar(&root, "r", "", "博客根目录")
	flag.StringVar(&theme, "t", "", "皮肤主题")
	flag.BoolVar(&clean, "c", false, "清理旧输出")
	flag.BoolVar(&verbose, "v", false, "输出详情")
	flag.Usage = usage
	flag.Parse()
}

func usage() {
	desc := `fountain version: v%s
Usage: fountain [-r root] [-s] [-p port] [-t theme] [-c] [-v]

Options:
`
	fmt.Fprintf(os.Stderr, desc, VERSION)
	flag.PrintDefaults()
}

func main() {
	if runtime.GOOS == "windows" {
		name := "Fountain"
		desc := "Fountain Static Blog Server"
		utils.WinMain(name, desc, run)
	} else {
		run()
	}
}

func run() {
	site := article.NewWebsite(root)
	site.LoadConfig("config.toml")
	if theme != "" {
		site.Conf.Theme = theme
	}
	site.Convert = func(source []byte, format string) []byte {
		if format == "" {
			return source
		} else if format == "markdown" {
			flags := bf2.CommonHTMLFlags
			if site.Conf.Theme != "night" {
				flags = flags | bf2.TOC
			}
			return bf2.Run(source, WithOptions(flags))
		}
		return PluginConvert(source, format)
	}
	site.Debug = func(data ...interface{}) {
		if verbose {
			fmt.Println(data...)
		}
	}

	if clean {
		fmt.Println("Clean ...")
		if len(site.Conf.Public) >= 3 {
			pubDir := site.Root + site.Conf.Public
			utils.CleanDir(pubDir)
		}
	}
	fmt.Println("Build ...")
	site.InitTheme()
	site.BuildFiles()

	if serve || port >= 0 {
		if port == 0 || port > 65535 {
			port = site.Conf.Port
		}
		fmt.Printf("Server at :%d\n", port)

		app := fiber.New()
		app.Static("/", filepath.Join(root, "public"))
		app.Listen(fmt.Sprintf("0.0.0.0:%d", port))
	}
}

func WithOptions(flags bf2.HTMLFlags) bf2.Option {
	params := bf2.HTMLRendererParameters{Flags: flags}
	renderer := bf2.NewHTMLRenderer(params)
	return bf2.WithRenderer(renderer)
}

func PluginConvert(source []byte, format string) []byte {
	plug, err := plugin.Open("./" + format + ".so")
	if err != nil {
		panic(err)
	}
	symb, err := plug.Lookup("Convert")
	if err != nil {
		panic(err)
	}
	conv := symb.(func(source []byte) []byte)
	return conv(source)
}