mirror of
https://github.com/amir20/dozzle.git
synced 2025-12-21 13:23:07 +01:00
129 lines
3.1 KiB
Go
129 lines
3.1 KiB
Go
package web
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"io/fs"
|
|
"time"
|
|
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/amir20/dozzle/internal/auth"
|
|
"github.com/amir20/dozzle/internal/docker"
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/go-chi/chi/v5"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type AuthProvider string
|
|
|
|
const (
|
|
SIMPLE AuthProvider = "simple"
|
|
FORWARD_PROXY AuthProvider = "forward-proxy"
|
|
)
|
|
|
|
// Config is a struct for configuring the web service
|
|
type Config struct {
|
|
Base string
|
|
Addr string
|
|
Version string
|
|
Username string
|
|
Password string
|
|
Hostname string
|
|
NoAnalytics bool
|
|
Dev bool
|
|
AuthProvider AuthProvider
|
|
}
|
|
|
|
type handler struct {
|
|
clients map[string]DockerClient
|
|
content fs.FS
|
|
config *Config
|
|
}
|
|
|
|
// Client is a proxy around the docker client
|
|
type DockerClient interface {
|
|
ListContainers() ([]docker.Container, error)
|
|
FindContainer(string) (docker.Container, error)
|
|
ContainerLogs(context.Context, string, string, docker.StdType) (io.ReadCloser, error)
|
|
Events(context.Context, chan<- docker.ContainerEvent) <-chan error
|
|
ContainerLogsBetweenDates(context.Context, string, time.Time, time.Time, docker.StdType) (io.ReadCloser, error)
|
|
ContainerStats(context.Context, string, chan<- docker.ContainerStat) error
|
|
Ping(context.Context) (types.Ping, error)
|
|
Host() *docker.Host
|
|
}
|
|
|
|
func CreateServer(clients map[string]DockerClient, content fs.FS, config Config) *http.Server {
|
|
handler := &handler{
|
|
clients: clients,
|
|
content: content,
|
|
config: &config,
|
|
}
|
|
return &http.Server{Addr: config.Addr, Handler: createRouter(handler)}
|
|
}
|
|
|
|
var fileServer http.Handler
|
|
|
|
func createRouter(h *handler) *chi.Mux {
|
|
initializeAuth(h)
|
|
|
|
base := h.config.Base
|
|
r := chi.NewRouter()
|
|
|
|
if !h.config.Dev {
|
|
r.Use(cspHeaders)
|
|
}
|
|
|
|
r.Route(base, func(r chi.Router) {
|
|
r.Group(func(r chi.Router) {
|
|
if h.config.AuthProvider == FORWARD_PROXY {
|
|
r.Use(auth.ForwardProxyAuthorizationRequired)
|
|
}
|
|
r.Group(func(r chi.Router) {
|
|
r.Use(authorizationRequired)
|
|
r.Get("/api/logs/stream/{host}/{id}", h.streamLogs)
|
|
r.Get("/api/logs/download/{host}/{id}", h.downloadLogs)
|
|
r.Get("/api/logs/{host}/{id}", h.fetchLogsBetweenDates)
|
|
r.Get("/api/events/stream", h.streamEvents)
|
|
r.Get("/logout", h.clearSession)
|
|
r.Get("/version", h.version)
|
|
r.Put("/api/profile/settings", h.saveSettings)
|
|
})
|
|
|
|
defaultHandler := http.StripPrefix(strings.Replace(base+"/", "//", "/", 1), http.HandlerFunc(h.index))
|
|
r.Get("/*", func(w http.ResponseWriter, req *http.Request) {
|
|
defaultHandler.ServeHTTP(w, req)
|
|
})
|
|
})
|
|
|
|
r.Post("/api/validateCredentials", h.validateCredentials)
|
|
r.Get("/healthcheck", h.healthcheck)
|
|
})
|
|
|
|
if base != "/" {
|
|
r.Get(base, func(w http.ResponseWriter, req *http.Request) {
|
|
http.Redirect(w, req, base+"/", http.StatusMovedPermanently)
|
|
})
|
|
}
|
|
|
|
fileServer = http.FileServer(http.FS(h.content))
|
|
|
|
return r
|
|
}
|
|
|
|
func (h *handler) clientFromRequest(r *http.Request) DockerClient {
|
|
host := chi.URLParam(r, "host")
|
|
|
|
if host == "" {
|
|
log.Fatalf("No host found for url %v", r.URL)
|
|
}
|
|
|
|
if client, ok := h.clients[host]; ok {
|
|
return client
|
|
}
|
|
|
|
log.Fatalf("No client found for host %v and url %v", host, r.URL)
|
|
return nil
|
|
}
|