132 lines
2.2 KiB
Go
132 lines
2.2 KiB
Go
package h
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/labstack/echo/v4"
|
|
"github.com/maddalax/htmgo/framework/htmgo/service"
|
|
"github.com/maddalax/htmgo/framework/util/process"
|
|
"log/slog"
|
|
"time"
|
|
)
|
|
|
|
type RequestContext struct {
|
|
echo.Context
|
|
locator *service.Locator
|
|
}
|
|
|
|
func (c *RequestContext) ServiceLocator() *service.Locator {
|
|
return c.locator
|
|
}
|
|
|
|
type AppOpts struct {
|
|
LiveReload bool
|
|
ServiceLocator *service.Locator
|
|
Register func(echo *echo.Echo)
|
|
}
|
|
|
|
type App struct {
|
|
Opts AppOpts
|
|
Echo *echo.Echo
|
|
}
|
|
|
|
var instance *App
|
|
|
|
func GetApp() *App {
|
|
if instance == nil {
|
|
panic("App instance not initialized")
|
|
}
|
|
return instance
|
|
}
|
|
|
|
func Start(opts AppOpts) {
|
|
e := echo.New()
|
|
instance := App{
|
|
Opts: opts,
|
|
Echo: e,
|
|
}
|
|
instance.start()
|
|
}
|
|
|
|
func (a App) start() {
|
|
|
|
a.Echo.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
|
|
return func(c echo.Context) error {
|
|
cc := &RequestContext{
|
|
c,
|
|
a.Opts.ServiceLocator,
|
|
}
|
|
return next(cc)
|
|
}
|
|
})
|
|
|
|
if a.Opts.Register != nil {
|
|
a.Opts.Register(a.Echo)
|
|
}
|
|
|
|
if a.Opts.LiveReload {
|
|
AddLiveReloadHandler("/dev/livereload", a.Echo)
|
|
}
|
|
|
|
port := ":3000"
|
|
err := a.Echo.Start(port)
|
|
|
|
if err != nil {
|
|
// If we are in watch mode, just try to kill any processes holding that port
|
|
// and try again
|
|
if IsDevelopment() && IsWatchMode() {
|
|
slog.Info("Port already in use, trying to kill the process and start again")
|
|
process.RunOrExit(fmt.Sprintf("kill -9 $(lsof -t -i%s)", port))
|
|
time.Sleep(time.Millisecond * 50)
|
|
err = a.Echo.Start(port)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
} else {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func HtmlView(c echo.Context, page *Page) error {
|
|
root := page.Root
|
|
return c.HTML(200,
|
|
Render(
|
|
root,
|
|
),
|
|
)
|
|
}
|
|
|
|
func PartialViewWithHeaders(c echo.Context, headers *Headers, partial *Partial) error {
|
|
if partial.Headers != nil {
|
|
for s, a := range *partial.Headers {
|
|
c.Set(s, a)
|
|
}
|
|
}
|
|
|
|
if headers != nil {
|
|
for s, a := range *headers {
|
|
c.Response().Header().Set(s, a)
|
|
}
|
|
}
|
|
|
|
return c.HTML(200,
|
|
Render(
|
|
partial,
|
|
),
|
|
)
|
|
}
|
|
|
|
func PartialView(c echo.Context, partial *Partial) error {
|
|
c.Set(echo.HeaderContentType, echo.MIMETextHTML)
|
|
if partial.Headers != nil {
|
|
for s, a := range *partial.Headers {
|
|
c.Response().Header().Set(s, a)
|
|
}
|
|
}
|
|
|
|
return c.HTML(200,
|
|
Render(
|
|
partial,
|
|
),
|
|
)
|
|
}
|