package main

import (
	"context"
	"flag"
	"fmt"
	"github.com/tomwright/lifetime"
	"github.com/tomwright/mermaid-server/internal"
	"os"
)

func main() {
	mermaid := flag.String("mermaid", "", "The full path to the mermaidcli executable.")
	in := flag.String("in", "", "Directory to store input files.")
	out := flag.String("out", "", "Directory to store output files.")
	puppeteer := flag.String("puppeteer", "", "Full path to optional puppeteer config.")
	flag.Parse()

	if *mermaid == "" {
		_, _ = fmt.Fprintf(os.Stderr, "Missing required argument `mermaid`")
		os.Exit(1)
	}

	if *in == "" {
		_, _ = fmt.Fprintf(os.Stderr, "Missing required argument `in`")
		os.Exit(1)
	}

	if *out == "" {
		_, _ = fmt.Fprintf(os.Stderr, "Missing required argument `out`")
		os.Exit(1)
	}

	cache := internal.NewDiagramCache()
	generator := internal.NewGenerator(cache, *mermaid, *in, *out, *puppeteer)

	httpService := internal.NewHTTPService(generator)
	cleanupService := internal.NewCleanupService(generator)

	lt := lifetime.New(context.Background()).Init()

	// Start the http service.
	lt.Start(httpService)
	// Start the cleanup service.
	lt.Start(cleanupService)

	// Wait for all routines to stop running.
	lt.Wait()
}