jhot
/
hats
1
0
Fork 0
Code Actions

HATS home and status pages

main
Jordan Hotmann 2023-11-17 11:43:50 -07:00
parent 2076619576
commit 68b7443de9
No known key found for this signature in database
GPG Key ID: 01B504170C2A2EA3
4 changed files with 39 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
go.mod
View File

@ -25,6 +25,7 @@ require (
github.com/go-co-op/gocron v1.35.2
github.com/go-resty/resty/v2 v2.9.1
github.com/golang/protobuf v1.5.3 // indirect
github.com/gomarkdown/markdown v0.0.0-20231115200524-a660076da3fd
github.com/klauspost/compress v1.17.0 // indirect
github.com/nats-io/nats-server/v2 v2.10.2 // indirect
github.com/nats-io/nkeys v0.4.5 // indirect

2
go.sum
View File

@ -19,6 +19,8 @@ github.com/go-resty/resty/v2 v2.9.1/go.mod h1:4/GYJVjh9nhkhGR6AUNW3XhpDYNUr+Uvy9
github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
github.com/golang/protobuf v1.5.3 h1:KhyjKVUg7Usr/dYsdSqoFveMYd5ko72D+zANwlG1mmg=
github.com/golang/protobuf v1.5.3/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=
github.com/gomarkdown/markdown v0.0.0-20231115200524-a660076da3fd h1:PppHBegd3uPZ3Y/Iax/2mlCFJm1w4Qf/zP1MdW4ju2o=
github.com/gomarkdown/markdown v0.0.0-20231115200524-a660076da3fd/go.mod h1:JDGcbDT52eL4fju3sZ4TeHGsQwhG9nbDV21aMyhwPoA=
github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
github.com/google/uuid v1.3.1 h1:KjJaJ9iWZ3jOFZIf1Lqf4laDRCasjl0BCmnEGxkdLb4=
github.com/google/uuid v1.3.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=

43
internal/api/api.go
View File

@ -17,6 +17,9 @@ import (
"github.com/go-chi/chi/middleware"
"github.com/go-chi/chi/v5"
"github.com/go-chi/render"
"github.com/gomarkdown/markdown"
"github.com/gomarkdown/markdown/html"
"github.com/gomarkdown/markdown/parser"
)
var (
@ -24,14 +27,21 @@ var (
logger *slog.Logger
server http.Server
haClient *homeassistant.RestClient
homepage string
)
const (
HA_STATE_PREFIX = "homeassistant.states"
)
func Listen(parentLogger *slog.Logger) {
func Listen(parentLogger *slog.Logger, readme []byte) {
logger = parentLogger
// render readme to HTML
p := parser.NewWithExtensions(parser.CommonExtensions | parser.AutoHeadingIDs | parser.NoEmptyLineBeforeBlock)
doc := p.Parse(readme)
homepage = string(markdown.Render(doc, html.NewRenderer(html.RendererOptions{Flags: html.CommonFlags | html.HrefTargetBlank})))
cfg = config.FromEnvironment()
haClient = homeassistant.NewRestClient(cfg.GetHomeAssistantBaseUrl(), cfg.HomeAssistantToken)
router := chi.NewRouter()
@ -41,22 +51,31 @@ func Listen(parentLogger *slog.Logger) {
router.Use(loggerMiddleware)
router.Use(middleware.Recoverer)
router.Use(middleware.Timeout(60 * time.Second))
router.Use(tokenAuthMiddleware)
router.Get(`/api/state/{entityId}`, getEntityStateHandler)
router.Post("/api/state/{entityId}/{service}", setEntityStateHandler)
router.Get("/", func(w http.ResponseWriter, r *http.Request) {
render.HTML(w, r, homepage)
})
router.Get("/status", func(w http.ResponseWriter, r *http.Request) {
render.PlainText(w, r, "OK")
})
router.Get("/api/timer/{timerName}", getTimerHandler)
router.Post("/api/timer/{timerName}", startTimerHandler)
router.Delete("/api/timer/{timerName}", deleteTimerHandler)
router.Route("/api", func(r chi.Router) {
r.Use(tokenAuthMiddleware)
r.Get(`/api/state/{entityId}`, getEntityStateHandler)
r.Post("/api/state/{entityId}/{service}", setEntityStateHandler)
router.Get("/api/schedule/{scheduleName}", getScheduleHandler)
router.Post("/api/schedule/{scheduleName}", createScheduleHandler)
router.Delete("/api/schedule/{scheduleName}", deleteScheduleHandler)
r.Get("/api/timer/{timerName}", getTimerHandler)
r.Post("/api/timer/{timerName}", startTimerHandler)
r.Delete("/api/timer/{timerName}", deleteTimerHandler)
router.Post("/api/ntfy", postNtfyHandler)
r.Get("/api/schedule/{scheduleName}", getScheduleHandler)
r.Post("/api/schedule/{scheduleName}", createScheduleHandler)
r.Delete("/api/schedule/{scheduleName}", deleteScheduleHandler)
router.Post("/api/command/{commandName}", postCommandHandler)
r.Post("/api/ntfy", postNtfyHandler)
r.Post("/api/command/{commandName}", postCommandHandler)
})
server = http.Server{
Addr: ":8888",

6
main.go
View File

@ -2,6 +2,7 @@ package main
import (
"context"
_ "embed"
"log/slog"
"os"
"os/signal"
@ -13,6 +14,9 @@ import (
"code.jhot.me/jhot/hats/pkg/config"
)
//go:embed README.md
var readme []byte
var (
interrupt chan os.Signal
cfg *config.HatsConfig
@ -56,7 +60,7 @@ func main() {
ntfy.InitClient(cfg)
api.Listen(logger)
api.Listen(logger, readme)
defer api.Close()
for sig := range interrupt {