mirror of
https://github.com/yusing/godoxy.git
synced 2025-06-01 01:22:34 +02:00
78 lines
2 KiB
Go
78 lines
2 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/yusing/go-proxy/agent/pkg/agent"
|
|
"github.com/yusing/go-proxy/agent/pkg/env"
|
|
"github.com/yusing/go-proxy/agent/pkg/server"
|
|
"github.com/yusing/go-proxy/internal/gperr"
|
|
"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"
|
|
socketproxy "github.com/yusing/go-proxy/socketproxy/pkg"
|
|
)
|
|
|
|
func main() {
|
|
writer := zerolog.ConsoleWriter{
|
|
Out: os.Stderr,
|
|
TimeFormat: "01-02 15:04",
|
|
}
|
|
zerolog.TimeFieldFormat = writer.TimeFormat
|
|
log.Logger = zerolog.New(writer).Level(zerolog.InfoLevel).With().Timestamp().Logger()
|
|
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)
|
|
}
|
|
|
|
log.Info().Msgf("GoDoxy Agent version %s", pkg.GetVersion())
|
|
log.Info().Msgf("Agent name: %s", env.AgentName)
|
|
log.Info().Msgf("Agent port: %d", env.AgentPort)
|
|
|
|
log.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 socketproxy.ListenAddr != "" {
|
|
log.Info().Msgf("Docker socket listening on: %s", socketproxy.ListenAddr)
|
|
opts := httpServer.Options{
|
|
Name: "docker",
|
|
HTTPAddr: socketproxy.ListenAddr,
|
|
Handler: socketproxy.NewHandler(),
|
|
}
|
|
httpServer.StartServer(t, opts)
|
|
}
|
|
|
|
systeminfo.Poller.Start()
|
|
|
|
task.WaitExit(3)
|
|
}
|