mirror of
https://github.com/yusing/godoxy.git
synced 2025-05-20 12:42:34 +02:00
57292f0fe8
* feat: idle sleep for proxmox LXCs * refactor: replace deprecated docker api types * chore(api): remove debug task list endpoint * refactor: move servemux to gphttp/servemux; favicon.go to v1/favicon * refactor: introduce Pool interface, move agent_pool to agent module * refactor: simplify api code * feat: introduce debug api * refactor: remove net.URL and net.CIDR types, improved unmarshal handling * chore: update Makefile for debug build tag, update README * chore: add gperr.Unwrap method * feat: relative time and duration formatting * chore: add ROOT_DIR environment variable, refactor * migration: move homepage override and icon cache to $BASE_DIR/data, add migration code * fix: nil dereference on marshalling service health * fix: wait for route deletion * chore: enhance tasks debuggability * feat: stdout access logger and MultiWriter * fix(agent): remove agent properly on verify error * fix(metrics): disk exclusion logic and added corresponding tests * chore: update schema and prettify, fix package.json and Makefile * fix: I/O buffer not being shrunk before putting back to pool * feat: enhanced error handling module * chore: deps upgrade * feat: better value formatting and handling --------- Co-authored-by: yusing <yusing@6uo.me>
86 lines
2.3 KiB
Go
86 lines
2.3 KiB
Go
package config
|
|
|
|
import (
|
|
"slices"
|
|
|
|
"github.com/yusing/go-proxy/agent/pkg/agent"
|
|
"github.com/yusing/go-proxy/internal/gperr"
|
|
"github.com/yusing/go-proxy/internal/route"
|
|
"github.com/yusing/go-proxy/internal/route/provider"
|
|
)
|
|
|
|
func (cfg *Config) DumpRoutes() map[string]*route.Route {
|
|
entries := make(map[string]*route.Route)
|
|
cfg.providers.RangeAll(func(_ string, p *provider.Provider) {
|
|
p.RangeRoutes(func(alias string, r *route.Route) {
|
|
entries[alias] = r
|
|
})
|
|
})
|
|
return entries
|
|
}
|
|
|
|
func (cfg *Config) DumpRouteProviders() map[string]*provider.Provider {
|
|
entries := make(map[string]*provider.Provider)
|
|
cfg.providers.RangeAll(func(_ string, p *provider.Provider) {
|
|
entries[p.ShortName()] = p
|
|
})
|
|
return entries
|
|
}
|
|
|
|
func (cfg *Config) RouteProviderList() []string {
|
|
var list []string
|
|
cfg.providers.RangeAll(func(_ string, p *provider.Provider) {
|
|
list = append(list, p.ShortName())
|
|
})
|
|
return list
|
|
}
|
|
|
|
func (cfg *Config) Statistics() map[string]any {
|
|
var rps, streams provider.RouteStats
|
|
var total uint16
|
|
providerStats := make(map[string]provider.ProviderStats)
|
|
|
|
cfg.providers.RangeAll(func(_ string, p *provider.Provider) {
|
|
stats := p.Statistics()
|
|
providerStats[p.ShortName()] = stats
|
|
rps.AddOther(stats.RPs)
|
|
streams.AddOther(stats.Streams)
|
|
total += stats.RPs.Total + stats.Streams.Total
|
|
})
|
|
|
|
return map[string]any{
|
|
"total": total,
|
|
"reverse_proxies": rps,
|
|
"streams": streams,
|
|
"providers": providerStats,
|
|
}
|
|
}
|
|
|
|
func (cfg *Config) VerifyNewAgent(host string, ca agent.PEMPair, client agent.PEMPair) (int, gperr.Error) {
|
|
if slices.ContainsFunc(cfg.value.Providers.Agents, func(a *agent.AgentConfig) bool {
|
|
return a.Addr == host
|
|
}) {
|
|
return 0, gperr.New("agent already exists")
|
|
}
|
|
|
|
agentCfg := new(agent.AgentConfig)
|
|
agentCfg.Addr = host
|
|
err := agentCfg.InitWithCerts(cfg.task.Context(), ca.Cert, client.Cert, client.Key)
|
|
if err != nil {
|
|
return 0, gperr.Wrap(err, "failed to start agent")
|
|
}
|
|
// must add it first to let LoadRoutes() reference from it
|
|
agent.Agents.Add(agentCfg)
|
|
|
|
provider := provider.NewAgentProvider(agentCfg)
|
|
if err := cfg.errIfExists(provider); err != nil {
|
|
agent.Agents.Del(agentCfg)
|
|
return 0, err
|
|
}
|
|
err = provider.LoadRoutes()
|
|
if err != nil {
|
|
agent.Agents.Del(agentCfg)
|
|
return 0, gperr.Wrap(err, "failed to load routes")
|
|
}
|
|
return provider.NumRoutes(), nil
|
|
}
|