59 lines
1.2 KiB
Go
59 lines
1.2 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"net/http"
|
||
|
|
)
|
||
|
|
|
||
|
|
type WellKnownGetter interface {
|
||
|
|
GetData(ctx context.Context) (*wkRegistry, error)
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetHealthServer() http.Handler {
|
||
|
|
mux := http.NewServeMux()
|
||
|
|
mux.HandleFunc("/healthz", func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
w.WriteHeader(http.StatusOK)
|
||
|
|
w.Write([]byte("OK"))
|
||
|
|
})
|
||
|
|
return mux
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetServer(wks WellKnownGetter) http.Handler {
|
||
|
|
mux := http.NewServeMux()
|
||
|
|
mux.HandleFunc("/.well-known/{id}", func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
reg, err := wks.GetData(r.Context())
|
||
|
|
if err != nil {
|
||
|
|
w.WriteHeader(http.StatusInternalServerError)
|
||
|
|
w.Write([]byte("Failed to fetch well-known records"))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
if reg == nil {
|
||
|
|
w.WriteHeader(http.StatusNotFound)
|
||
|
|
w.Write([]byte("Not found"))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
items := reg
|
||
|
|
id := r.PathValue("id")
|
||
|
|
if val, ok := (*items)[id]; ok {
|
||
|
|
b, err := json.Marshal(val)
|
||
|
|
if err != nil {
|
||
|
|
w.WriteHeader(http.StatusInternalServerError)
|
||
|
|
w.Write([]byte("Failed to encode"))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
w.WriteHeader(http.StatusOK)
|
||
|
|
w.Write(b)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
w.WriteHeader(http.StatusNotFound)
|
||
|
|
w.Write([]byte("Not found"))
|
||
|
|
})
|
||
|
|
|
||
|
|
return mux
|
||
|
|
}
|