package cmd
import (
"errors"
"fmt"
"io/fs"
"os"
"strings"
"<your pkg path>/ui"
"code.icod.de/dalu/ginpongo2/v5"
"github.com/flosch/pongo2/v5"
"github.com/gin-gonic/gin"
"github.com/spf13/cobra"
)
const (
prefixTCP = "tcp:"
prefixUNIX = "unix:"
)
var (
uiDebug bool
uiAddr string
)
// uiCmd represents the ui command
var uiCmd = &cobra.Command{
Use: "ui",
Short: "serves the management ui over http",
RunE: func(cmd *cobra.Command, args []string) error {
if !uiDebug {
gin.SetMode(gin.ReleaseMode)
}
r := gin.Default()
r.SecureJsonPrefix(")]}',\n")
r.MaxMultipartMemory = 8 << 20
if uiDebug {
fmt.Println("debug mode")
fl, e := pongo2.NewLocalFileSystemLoader("ui/templates/")
if e != nil {
return e
}
hr := ginpongo2.New(true, fl)
r.HTMLRender = hr
} else {
fmt.Println("release mode")
subFS, e := fs.Sub(ui.Templates, "templates")
if e != nil {
return e
}
fl := pongo2.NewFSLoader(subFS)
hr := ginpongo2.New(false, fl)
r.HTMLRender = hr
}
// Static
r.Static("/assets/", "./assets/")
r.GET("/", func(cx *gin.Context) {
ctx := make(pongo2.Context)
type Data struct {
Target string
Message string
}
ctx["data"] = &Data{
Target: "World",
Message: "It's a great day to be alive",
}
cx.HTML(200, "index", ctx)
})
// serve
if strings.HasPrefix(uiAddr, prefixTCP) {
addr := strings.TrimPrefix(uiAddr, prefixTCP)
fmt.Println("listening on", addr)
return r.Run(addr)
} else if strings.HasPrefix(uiAddr, prefixUNIX) {
addr := strings.TrimPrefix(uiAddr, prefixUNIX)
if _, e := os.Stat(addr); errors.Is(e, os.ErrNotExist) {
fmt.Println("listening on", addr)
return r.RunUnix(addr)
} else {
if e := os.Remove(addr); e != nil {
return e
} else {
fmt.Println("listening on", addr)
return r.RunUnix(addr)
}
}
} else {
return nil
}
},
}
func init() {
rootCmd.AddCommand(uiCmd)
uiCmd.Flags().BoolVar(&uiDebug, "debug", false, "enable dev mode")
uiCmd.Flags().StringVar(
&uiAddr,
"addr",
"tcp:localhost:8080",
"either tcp: or unix: e.g. tcp:localhost:3030 unix:/tmp/listen.sock",
)
}