mirror of
https://github.com/amir20/dozzle.git
synced 2025-12-21 21:33:18 +01:00
68 lines
1.9 KiB
Go
68 lines
1.9 KiB
Go
package cli
|
|
|
|
import (
|
|
"context"
|
|
"embed"
|
|
"fmt"
|
|
"io"
|
|
"net"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"github.com/amir20/dozzle/internal/agent"
|
|
"github.com/amir20/dozzle/internal/docker"
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
type AgentCmd struct {
|
|
Addr string `arg:"--agent-addr,env:DOZZLE_AGENT_ADDR" default:":7007" help:"sets the host:port to bind for the agent"`
|
|
}
|
|
|
|
func (a *AgentCmd) Run(args Args, embeddedCerts embed.FS) error {
|
|
if args.Mode != "server" {
|
|
return fmt.Errorf("agent command is only available in server mode")
|
|
}
|
|
client, err := docker.NewLocalClient(args.Hostname)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create docker client: %w", err)
|
|
}
|
|
certs, err := ReadCertificates(embeddedCerts)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to read certificates: %w", err)
|
|
}
|
|
|
|
listener, err := net.Listen("tcp", args.Agent.Addr)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to listen: %w", err)
|
|
}
|
|
tempFile, err := os.CreateTemp("", "agent-*.addr")
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create temp file: %w", err)
|
|
}
|
|
io.WriteString(tempFile, listener.Addr().String())
|
|
log.Debug().Str("file", tempFile.Name()).Msg("Created temp file")
|
|
go StartEvent(args, "", client, "agent")
|
|
server, err := agent.NewServer(client, certs, args.Version(), args.Filter)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create agent server: %w", err)
|
|
}
|
|
ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
|
|
defer stop()
|
|
go func() {
|
|
log.Info().Msgf("Dozzle agent version %s", args.Version())
|
|
log.Info().Msgf("Agent listening on %s", listener.Addr().String())
|
|
|
|
if err := server.Serve(listener); err != nil {
|
|
log.Error().Err(err).Msg("failed to serve")
|
|
}
|
|
}()
|
|
<-ctx.Done()
|
|
stop()
|
|
log.Info().Msg("Shutting down agent")
|
|
server.Stop()
|
|
log.Debug().Str("file", tempFile.Name()).Msg("Removing temp file")
|
|
os.Remove(tempFile.Name())
|
|
return nil
|
|
}
|