mirror of
https://github.com/yusing/godoxy.git
synced 2025-05-19 20:32:35 +02:00
455a85e6a0
- Introduced Docker socket proxy handling in the agent. - Added environment variables for Docker socket configuration. - Implemented new Docker handler with endpoint permissions based on environment settings. - Added tests for Docker handler functionality. - Updated go.mod to include gorilla/mux for routing.
74 lines
1.9 KiB
Go
74 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/yusing/go-proxy/agent/pkg/agent"
|
|
"github.com/yusing/go-proxy/agent/pkg/env"
|
|
"github.com/yusing/go-proxy/agent/pkg/handler"
|
|
"github.com/yusing/go-proxy/agent/pkg/server"
|
|
"github.com/yusing/go-proxy/internal/gperr"
|
|
"github.com/yusing/go-proxy/internal/logging"
|
|
"github.com/yusing/go-proxy/internal/logging/memlogger"
|
|
"github.com/yusing/go-proxy/internal/metrics/systeminfo"
|
|
httpServer "github.com/yusing/go-proxy/internal/net/gphttp/server"
|
|
"github.com/yusing/go-proxy/internal/task"
|
|
"github.com/yusing/go-proxy/pkg"
|
|
)
|
|
|
|
func main() {
|
|
logging.InitLogger(os.Stderr, memlogger.GetMemLogger())
|
|
|
|
ca := &agent.PEMPair{}
|
|
err := ca.Load(env.AgentCACert)
|
|
if err != nil {
|
|
gperr.LogFatal("init CA error", err)
|
|
}
|
|
caCert, err := ca.ToTLSCert()
|
|
if err != nil {
|
|
gperr.LogFatal("init CA error", err)
|
|
}
|
|
|
|
srv := &agent.PEMPair{}
|
|
srv.Load(env.AgentSSLCert)
|
|
if err != nil {
|
|
gperr.LogFatal("init SSL error", err)
|
|
}
|
|
srvCert, err := srv.ToTLSCert()
|
|
if err != nil {
|
|
gperr.LogFatal("init SSL error", err)
|
|
}
|
|
|
|
logging.Info().Msgf("GoDoxy Agent version %s", pkg.GetVersion())
|
|
logging.Info().Msgf("Agent name: %s", env.AgentName)
|
|
logging.Info().Msgf("Agent port: %d", env.AgentPort)
|
|
|
|
logging.Info().Msg(`
|
|
Tips:
|
|
1. To change the agent name, you can set the AGENT_NAME environment variable.
|
|
2. To change the agent port, you can set the AGENT_PORT environment variable.
|
|
`)
|
|
|
|
t := task.RootTask("agent", false)
|
|
opts := server.Options{
|
|
CACert: caCert,
|
|
ServerCert: srvCert,
|
|
Port: env.AgentPort,
|
|
}
|
|
|
|
server.StartAgentServer(t, opts)
|
|
|
|
if env.DockerSocketAddr != "" {
|
|
logging.Info().Msgf("Docker socket listening on: %s", env.DockerSocketAddr)
|
|
opts := httpServer.Options{
|
|
Name: "docker",
|
|
HTTPAddr: env.DockerSocketAddr,
|
|
Handler: handler.NewDockerHandler(),
|
|
}
|
|
httpServer.StartServer(t, opts)
|
|
}
|
|
|
|
systeminfo.Poller.Start()
|
|
|
|
task.WaitExit(3)
|
|
}
|