/*
Copyright © 2025 Ronmi Ren <ronmi.ren@gmail.com
*/
package cmd

import (
	"context"
	"fmt"
	"net/url"
	"os"
	"os/signal"
	"syscall"
	"time"

	"git.ronmi.tw/ronmi/forgejo-pages/lib"
	"github.com/spf13/cobra"
)

// serveCmd represents the serve command
var serveCmd = &cobra.Command{
	Use:   "serve",
	Short: "Start the static page server.",
	Run: func(cmd *cobra.Command, args []string) {
		// check flags
		bind, _ := cmd.Flags().GetString("bind")
		server, _ := cmd.Flags().GetString("server")
		token, _ := cmd.Flags().GetString("token")
		branch, _ := cmd.Flags().GetString("branch")
		if bind == "" || server == "" || token == "" || branch == "" {
			fmt.Println("bind, server token and branch are required")
			return
		}
		serverUrl, err := url.Parse(server)
		if err != nil {
			fmt.Println("invalid server url: ", err)
			return
		}
		f := &lib.Forgejo{
			Server: *serverUrl,
			Token:  token,
			Branch: branch,
		}

		s, err := lib.Run(bind, f)
		if err != nil {
			fmt.Println("cannot create server: ", err)
			return
		}

		ctx, stop := signal.NotifyContext(
			context.TODO(),
			os.Interrupt,
			syscall.SIGTERM,
			os.Kill,
		)
		defer stop()

		fmt.Println("starting server")
		go func() {
			<-ctx.Done()
			stop()
			ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
			defer cancel()
			s.Shutdown(ctx)
		}()
		s.ListenAndServe()
	},
}

func init() {
	rootCmd.AddCommand(serveCmd)

	f := serveCmd.Flags()
	f.StringP("bind", "a", ":8080", "bind address")
	f.StringP("server", "s", "", "Forgejo server address")
	f.StringP("token", "k", "", "Forgejo api token")
	f.StringP("branch", "b", "static-pages", "branch to use")
}