mirror of
https://github.com/amir20/dozzle.git
synced 2025-12-21 21:33:18 +01:00
101 lines
3.1 KiB
Go
101 lines
3.1 KiB
Go
package web
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"time"
|
|
|
|
"io"
|
|
"io/fs"
|
|
|
|
"github.com/amir20/dozzle/internal/container"
|
|
docker_support "github.com/amir20/dozzle/internal/support/docker"
|
|
"github.com/docker/docker/api/types/system"
|
|
"github.com/go-chi/chi/v5"
|
|
|
|
"github.com/stretchr/testify/mock"
|
|
|
|
"github.com/spf13/afero"
|
|
)
|
|
|
|
type MockedClient struct {
|
|
mock.Mock
|
|
container.Client
|
|
}
|
|
|
|
func (m *MockedClient) FindContainer(ctx context.Context, id string) (container.Container, error) {
|
|
args := m.Called(ctx, id)
|
|
return args.Get(0).(container.Container), args.Error(1)
|
|
}
|
|
|
|
func (m *MockedClient) ContainerActions(ctx context.Context, action container.ContainerAction, containerID string) error {
|
|
args := m.Called(ctx, action, containerID)
|
|
return args.Error(0)
|
|
}
|
|
|
|
func (m *MockedClient) ContainerEvents(ctx context.Context, events chan<- container.ContainerEvent) error {
|
|
args := m.Called(ctx, events)
|
|
return args.Error(0)
|
|
}
|
|
|
|
func (m *MockedClient) ListContainers(ctx context.Context, labels container.ContainerLabels) ([]container.Container, error) {
|
|
args := m.Called(ctx, labels)
|
|
return args.Get(0).([]container.Container), args.Error(1)
|
|
}
|
|
|
|
func (m *MockedClient) ContainerLogs(ctx context.Context, id string, since time.Time, stdType container.StdType) (io.ReadCloser, error) {
|
|
args := m.Called(ctx, id, since, stdType)
|
|
return args.Get(0).(io.ReadCloser), args.Error(1)
|
|
}
|
|
|
|
func (m *MockedClient) ContainerStats(context.Context, string, chan<- container.ContainerStat) error {
|
|
return nil
|
|
}
|
|
|
|
func (m *MockedClient) ContainerLogsBetweenDates(ctx context.Context, id string, from time.Time, to time.Time, stdType container.StdType) (io.ReadCloser, error) {
|
|
args := m.Called(ctx, id, from, to, stdType)
|
|
return args.Get(0).(io.ReadCloser), args.Error(1)
|
|
}
|
|
|
|
func (m *MockedClient) Host() container.Host {
|
|
args := m.Called()
|
|
return args.Get(0).(container.Host)
|
|
}
|
|
|
|
func (m *MockedClient) IsSwarmMode() bool {
|
|
return false
|
|
}
|
|
|
|
func (m *MockedClient) SystemInfo() system.Info {
|
|
return system.Info{ID: "123"}
|
|
}
|
|
|
|
func createHandler(client container.Client, content fs.FS, config Config) *chi.Mux {
|
|
if client == nil {
|
|
client = new(MockedClient)
|
|
client.(*MockedClient).On("ListContainers", mock.Anything, mock.Anything).Return([]container.Container{}, nil)
|
|
client.(*MockedClient).On("Host").Return(container.Host{
|
|
ID: "localhost",
|
|
})
|
|
client.(*MockedClient).On("ContainerEvents", mock.Anything, mock.AnythingOfType("chan<- container.ContainerEvent")).Return(nil)
|
|
}
|
|
|
|
if content == nil {
|
|
fs := afero.NewMemMapFs()
|
|
afero.WriteFile(fs, "index.html", []byte("index page"), 0644)
|
|
content = afero.NewIOFS(fs)
|
|
}
|
|
|
|
manager := docker_support.NewRetriableClientManager(nil, 3*time.Second, tls.Certificate{}, docker_support.NewDockerClientService(client, container.ContainerLabels{}))
|
|
multiHostService := docker_support.NewMultiHostService(manager, 3*time.Second)
|
|
return createRouter(&handler{
|
|
hostService: multiHostService,
|
|
content: content,
|
|
config: &config,
|
|
})
|
|
}
|
|
|
|
func createDefaultHandler(client container.Client) *chi.Mux {
|
|
return createHandler(client, nil, Config{Base: "/", Authorization: Authorization{Provider: NONE}})
|
|
}
|