headscale/swagger.go

95 lines
2.2 KiB
Go
Raw Permalink Normal View History

2021-10-30 14:29:53 +00:00
package headscale
import (
"bytes"
_ "embed"
2021-12-23 02:43:53 +00:00
"html/template"
2021-10-30 14:29:53 +00:00
"net/http"
2021-11-13 08:39:04 +00:00
"github.com/rs/zerolog/log"
2021-10-30 14:29:53 +00:00
)
//go:embed gen/openapiv2/headscale/v1/headscale.swagger.json
var apiV1JSON []byte
func SwaggerUI(
2022-06-26 09:43:17 +00:00
writer http.ResponseWriter,
req *http.Request,
) {
swaggerTemplate := template.Must(template.New("swagger").Parse(`
2021-10-30 14:29:53 +00:00
<html>
<head>
<link rel="stylesheet" type="text/css" href="https://unpkg.com/swagger-ui-dist@3/swagger-ui.css">
<script src="https://unpkg.com/swagger-ui-dist@3/swagger-ui-standalone-preset.js"></script>
<script src="https://unpkg.com/swagger-ui-dist@3/swagger-ui-bundle.js" charset="UTF-8"></script>
</head>
<body>
<div id="swagger-ui"></div>
<script>
window.addEventListener('load', (event) => {
const ui = SwaggerUIBundle({
url: "/swagger/v1/openapiv2.json",
dom_id: '#swagger-ui',
presets: [
SwaggerUIBundle.presets.apis,
SwaggerUIBundle.SwaggerUIStandalonePreset
],
plugins: [
SwaggerUIBundle.plugins.DownloadUrl
],
deepLinking: true,
// TODO(kradalby): Figure out why this does not work
// layout: "StandaloneLayout",
})
window.ui = ui
});
</script>
</body>
</html>`))
var payload bytes.Buffer
if err := swaggerTemplate.Execute(&payload, struct{}{}); err != nil {
2021-10-30 14:29:53 +00:00
log.Error().
Caller().
Err(err).
Msg("Could not render Swagger")
2022-06-26 09:43:17 +00:00
writer.Header().Set("Content-Type", "text/plain; charset=utf-8")
writer.WriteHeader(http.StatusInternalServerError)
2022-06-26 10:21:35 +00:00
_, err := writer.Write([]byte("Could not render Swagger"))
if err != nil {
log.Error().
Caller().
Err(err).
Msg("Failed to write response")
}
2021-11-14 15:46:09 +00:00
2021-10-30 14:29:53 +00:00
return
}
2022-06-26 09:43:17 +00:00
writer.Header().Set("Content-Type", "text/html; charset=utf-8")
writer.WriteHeader(http.StatusOK)
2022-06-26 10:21:35 +00:00
_, err := writer.Write(payload.Bytes())
if err != nil {
log.Error().
Caller().
Err(err).
Msg("Failed to write response")
}
2021-10-30 14:29:53 +00:00
}
func SwaggerAPIv1(
2022-06-26 09:43:17 +00:00
writer http.ResponseWriter,
req *http.Request,
) {
2022-06-26 09:43:17 +00:00
writer.Header().Set("Content-Type", "application/json; charset=utf-88")
writer.WriteHeader(http.StatusOK)
2022-06-26 10:21:35 +00:00
if _, err := writer.Write(apiV1JSON); err != nil {
log.Error().
Caller().
Err(err).
Msg("Failed to write response")
}
2021-10-30 14:29:53 +00:00
}