proto-fleet/server/cmd/fleetd/config.go at 1b1f7fc9d780e5e5715b1e169cacad1a1b7c4ac6 · block/proto-fleet · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
package main

import (
	"time"

	"github.com/block/proto-fleet/server/internal/domain/command"
	curtailmentReconciler "github.com/block/proto-fleet/server/internal/domain/curtailment/reconciler"
	"github.com/block/proto-fleet/server/internal/domain/diagnostics"
	infrastructureDomain "github.com/block/proto-fleet/server/internal/domain/infrastructure"
	"github.com/block/proto-fleet/server/internal/domain/ipscanner"
	"github.com/block/proto-fleet/server/internal/domain/plugins"
	"github.com/block/proto-fleet/server/internal/domain/pools"
	"github.com/block/proto-fleet/server/internal/domain/session"
	"github.com/block/proto-fleet/server/internal/domain/telemetry"
	"github.com/block/proto-fleet/server/internal/domain/telemetry/scheduler"
	"github.com/block/proto-fleet/server/internal/domain/token"
	"github.com/block/proto-fleet/server/internal/domain/updates"
	"github.com/block/proto-fleet/server/internal/ha"
	"github.com/block/proto-fleet/server/internal/infrastructure/db"
	"github.com/block/proto-fleet/server/internal/infrastructure/encrypt"
	"github.com/block/proto-fleet/server/internal/infrastructure/files"
	fleet_telemetry "github.com/block/proto-fleet/server/internal/infrastructure/fleet-telemetry"
	"github.com/block/proto-fleet/server/internal/infrastructure/logging"
	"github.com/block/proto-fleet/server/internal/infrastructure/metrics"
	"github.com/block/proto-fleet/server/internal/infrastructure/queue"
	"github.com/block/proto-fleet/server/internal/infrastructure/sysmon"
	"github.com/block/proto-fleet/server/internal/infrastructure/timescaledb"
)

type HTTPConfig struct {
	Address           string        `help:"Address to listen on" default:"127.0.0.1:8080" env:"LISTEN_ADDRESS"`
	ReadHeaderTimeout time.Duration `help:"Read header timeout" default:"3s" env:"READ_HEADER_TIMEOUT"`
	WriteByteTimeout  time.Duration `help:"HTTP/2 timeout for stalled response writes." default:"30s" env:"WRITE_BYTE_TIMEOUT"`
	SuppressCors      bool          `help:"Suppress CORS" default:"false" env:"SUPPRESS_CORS"`
	PprofAddr         string        `help:"Address to listen for pprof debug server, e.g. 127.0.0.1:6060 (empty disables it; use a non-loopback address only if you intentionally want remote access)" default:"" env:"PPROF_ADDR"`
}
type Config struct {
	Mode string `help:"Execution mode" enum:"server,agent,combined" default:"combined" env:"MODE"`

	// Operator-facing base URL of this instance, used in outbound alert notification links.
	PublicURL string `help:"Base URL of the proto-fleet instance, used in alert notification links (e.g. https://fleet.example.com)" default:"" env:"FLEET_PUBLIC_URL"`

	DB             db.Config                    `embed:"" prefix:"db-" envprefix:"DB_"`
	Log            logging.Config               `embed:"" prefix:"logging-" envprefix:"LOG_"`
	HTTP           HTTPConfig                   `embed:"" prefix:"http-" envprefix:"HTTP_"`
	Auth           token.Config                 `embed:"" prefix:"auth-" envprefix:"AUTH_"`
	Session        session.Config               `embed:"" prefix:"session-" envprefix:"SESSION_"`
	Pools          pools.Config                 `embed:"" prefix:"pools-" envprefix:"POOLS_"`
	Encrypt        encrypt.Config               `embed:"" prefix:"encrypt-" envprefix:"ENCRYPT_"`
	Command        command.Config               `embed:"" prefix:"fleet-command-" envprefix:"FLEET_COMMAND_"`
	Curtailment    curtailmentReconciler.Config `embed:"" prefix:"curtailment-" envprefix:"CURTAILMENT_"`
	Queue          queue.Config                 `embed:"" prefix:"fleet-queue-" envprefix:"FLEET_QUEUE_"`
	TimescaleDB    timescaledb.Config           `embed:"" prefix:"timescaledb-" envprefix:"TIMESCALEDB_"`
	Telemetry      telemetry.Config             `embed:"" prefix:"telemetry-" envprefix:"TELEMETRY_"`
	Scheduler      scheduler.Config             `embed:"" prefix:"scheduler-" envprefix:"SCHEDULER_"`
	Plugins        plugins.Config               `embed:"" prefix:"plugins-" envprefix:"PLUGINS_"`
	IPScanner      ipscanner.Config             `embed:"" prefix:"ipscanner-" envprefix:"IPSCANNER_"`
	Diagnostics    diagnostics.Config           `embed:"" prefix:"diagnostics-" envprefix:"DIAGNOSTICS_"`
	Updates        updates.Config               `embed:"" prefix:"updates-" envprefix:"UPDATES_"`
	Infrastructure infrastructureDomain.Config  `embed:"" prefix:"infrastructure-" envprefix:"INFRASTRUCTURE_"`
	Files          files.Config                 `embed:"" prefix:"files-" envprefix:"FILES_"`
	FleetTelemetry fleet_telemetry.Config       `embed:"" prefix:"fleet-telemetry-" envprefix:"FLEET_TELEMETRY_"`
	Metrics        metrics.Config               `embed:"" prefix:"metrics-" envprefix:"FLEET_ALERTS_"`
	HA             ha.Config                    `embed:"" prefix:"ha-" envprefix:"FLEET_HA_"`

	SystemMonitoring sysmon.Config `embed:"" prefix:"system-monitoring-" envprefix:"FLEET_SYSTEM_MONITORING_"`
}