htmgo/cli/htmgo/runner.go

package main

import (
	"bufio"
	"flag"
	"fmt"
	"github.com/maddalax/htmgo/cli/htmgo/internal"
	"github.com/maddalax/htmgo/cli/htmgo/tasks/astgen"
	"github.com/maddalax/htmgo/cli/htmgo/tasks/copyassets"
	"github.com/maddalax/htmgo/cli/htmgo/tasks/css"
	"github.com/maddalax/htmgo/cli/htmgo/tasks/downloadtemplate"
	"github.com/maddalax/htmgo/cli/htmgo/tasks/formatter"
	"github.com/maddalax/htmgo/cli/htmgo/tasks/process"
	"github.com/maddalax/htmgo/cli/htmgo/tasks/reloader"
	"github.com/maddalax/htmgo/cli/htmgo/tasks/run"
	"log/slog"
	"os"
	"strings"
	"sync"
)

func main() {
	needsSignals := true

	commandMap := make(map[string]*flag.FlagSet)
	commands := []string{"template", "run", "watch", "build", "setup", "css", "schema", "generate", "format"}

	for _, command := range commands {
		commandMap[command] = flag.NewFlagSet(command, flag.ExitOnError)
	}

	if len(os.Args) < 2 {
		fmt.Println(fmt.Sprintf("Usage: htmgo [%s]", strings.Join(commands, " | ")))
		os.Exit(1)
	}

	c := commandMap[os.Args[1]]

	if c == nil {
		fmt.Println(fmt.Sprintf("Usage: htmgo [%s]", strings.Join(commands, " | ")))
		os.Exit(1)
		return
	}

	err := c.Parse(os.Args[2:])

	if err != nil {
		fmt.Println(err.Error())
		os.Exit(1)
		return
	}

	slog.SetLogLoggerLevel(internal.GetLogLevel())

	taskName := os.Args[1]

	slog.Debug("Running task:", slog.String("task", taskName))
	slog.Debug("working dir:", slog.String("dir", process.GetWorkingDir()))

	if taskName == "format" {
		needsSignals = false
	}

	done := make(chan bool, 1)
	if needsSignals {
		done = RegisterSignals()
	}

	if taskName == "watch" {
		fmt.Printf("Running in watch mode\n")
		os.Setenv("ENV", "development")
		os.Setenv("WATCH_MODE", "true")
		fmt.Printf("Starting processes...\n")

		copyassets.CopyAssets()

		fmt.Printf("Generating CSS...\n")
		css.GenerateCss(process.ExitOnError)

		wg := sync.WaitGroup{}

		wg.Add(1)
		go func() {
			defer wg.Done()
			astgen.GenAst(process.ExitOnError)
		}()

		wg.Add(1)
		go func() {
			defer wg.Done()
			run.EntGenerate()
		}()

		wg.Wait()

		fmt.Printf("Starting server...\n")
		process.KillAll()
		go func() {
			_ = run.Server()
		}()
		startWatcher(reloader.OnFileChange)
	} else {
		if taskName == "format" {
			if len(os.Args) < 3 {
				fmt.Println(fmt.Sprintf("Usage: htmgo format <file>"))
				os.Exit(1)
			}
			file := os.Args[2]
			if file == "." {
				formatter.FormatDir(process.GetWorkingDir())
			} else {
				formatter.FormatFile(os.Args[2])
			}
		} else if taskName == "schema" {
			reader := bufio.NewReader(os.Stdin)
			fmt.Print("Enter entity name:")
			text, _ := reader.ReadString('\n')
			text = strings.TrimSuffix(text, "\n")
			run.EntNewSchema(text)
		} else if taskName == "generate" {
			run.EntGenerate()
			astgen.GenAst(process.ExitOnError)
		} else if taskName == "setup" {
			run.Setup()
		} else if taskName == "css" {
			_ = css.GenerateCss(process.ExitOnError)
		} else if taskName == "ast" {
			_ = astgen.GenAst(process.ExitOnError)
		} else if taskName == "run" {
			run.MakeBuildable()
			_ = run.Server(process.ExitOnError)
		} else if taskName == "template" {
			name := ""
			if len(os.Args) > 2 {
				name = os.Args[2]
			} else {
				reader := bufio.NewReader(os.Stdin)
				fmt.Print("What would you like to call your new app?: ")
				name, _ = reader.ReadString('\n')
			}
			name = strings.TrimSuffix(name, "\n")
			name = strings.ReplaceAll(name, " ", "-")
			name = strings.ToLower(name)
			downloadtemplate.DownloadTemplate(fmt.Sprintf("./%s", name))
		} else if taskName == "build" {
			run.Build()
		} else if taskName == "generate" {
			astgen.GenAst(process.ExitOnError)
		} else {
			fmt.Println(fmt.Sprintf("Usage: htmgo [%s]", strings.Join(commands, " | ")))
		}
		os.Exit(0)
	}

	<-done
}