grpc

type AppBudgetData struct {
	// Usage is the current period usage in dollars
	Usage float64 `json:"usage"`

	// PeriodStart is the start of this app's budget period
	PeriodStart time.Time `json:"period_start"`

	// PeriodEnd is the end of this app's budget period
	PeriodEnd time.Time `json:"period_end"`
}

type BudgetSyncPayload struct {
	// AppUsages maps app_id to current period usage in dollars
	// Deprecated: Use AppBudgets instead for per-app budget periods
	AppUsages map[uint32]float64 `json:"app_usages"`

	// AppBudgets maps app_id to budget data with per-app period info
	// This supports custom budget_start_date per app
	AppBudgets map[uint32]AppBudgetData `json:"app_budgets,omitempty"`

	// PeriodStart is the start of the budget period (1st of month)
	// Deprecated: Use per-app periods in AppBudgets instead
	PeriodStart time.Time `json:"period_start"`

	// PeriodEnd is the end of the budget period (last moment of month)
	// Deprecated: Use per-app periods in AppBudgets instead
	PeriodEnd time.Time `json:"period_end"`

	// ControlTimestamp is when the control server generated this payload
	// Used by edges to detect clock skew
	ControlTimestamp time.Time `json:"control_timestamp"`

	// SequenceNumber for ordering and deduplication
	SequenceNumber uint64 `json:"sequence_number"`

	// Blocks maps app_id to the reason edges must refuse the App on budget
	// grounds (Enterprise): its budget is 0, or its hard-blocking team has
	// spent its budget. Edges read a synced App budget of 0 as "no limit",
	// so this is how a zero budget reaches them. When BlocksIncluded is set
	// it is the complete set: edges replace theirs, so a released App is
	// served again.
	Blocks         map[uint32]string `json:"blocks,omitempty"`
	BlocksIncluded bool              `json:"blocks_included,omitempty"`
}

type BudgetSyncService struct {
	// contains filtered or unexported fields
}

type Config struct {
	GRPCPort    int
	GRPCHost    string
	TLSEnabled  bool
	TLSCertPath string
	TLSKeyPath  string
	// TLSConfig, when set, is used for TLS instead of TLSCertPath and
	// TLSKeyPath (and whatever TLSEnabled says): an embedding host supplies
	// its own certificates, cipher suites and minimum version.
	TLSConfig            *tls.Config
	AuthToken            string
	NextAuthToken        string
	MaxConcurrentStreams int // Maximum number of concurrent gRPC streams (default 1000)
	// EncryptionKey is the 32-character key edges use to decrypt the
	// credentials sent to them. Empty means MICROGATEWAY_ENCRYPTION_KEY.
	EncryptionKey string
	// NodeID identifies this Studio replica (pkg/cluster). It is recorded as
	// the owner of every edge stream this server holds. Empty means
	// "control", for a single replica.
	NodeID string
	// BudgetSyncInterval is how often budget usage is synced to edges.
	// Zero means BUDGET_SYNC_INTERVAL, or 30s.
	BudgetSyncInterval time.Duration

	// MaxMessageSize bounds a message in either direction, in bytes (zero:
	// 16 MB, the edge's own GRPC_MAX_MESSAGE_SIZE default). Configuration
	// snapshots and analytics pulses outgrow gRPC's 4 MB default.
	MaxMessageSize int
	// KeepaliveMinTime is the shortest interval between client pings the
	// server accepts, with or without an open stream (zero: 10s; edges ping
	// every 30s). A client pinging more often is disconnected.
	KeepaliveMinTime time.Duration
	// MaxConnectionAge, when set, closes each connection after about this
	// long, so edges spread again over replicas behind a load balancer;
	// streams get MaxConnectionAgeGrace to finish. Zero keeps connections
	// open: an edge's stream is long-lived, and pushes in flight on a closed
	// stream go back to pending.
	MaxConnectionAge      time.Duration
	MaxConnectionAgeGrace time.Duration
}

type ControlServer struct {
	pb.UnimplementedConfigurationSyncServiceServer
	// contains filtered or unexported fields
}

type EdgeBudgetSource interface {
	EdgeBlocks() (map[uint]string, error)
	AnalyzeApps(appIDs []uint)
}

type EdgeInstanceConnection struct {
	EdgeID        string
	Namespace     string
	Status        string
	Version       string
	SessionID     string
	Stream        pb.ConfigurationSyncService_SubscribeToChangesServer
	LastHeartbeat time.Time
	// contains filtered or unexported fields
}

type EdgePayloadForwarder interface {
	ForwardEdgePayloads(ctx context.Context, payloads []*pb.PluginControlPayload) error
}

type EdgePayloadRouter interface {
	RouteEdgePayload(ctx context.Context, payload *pb.PluginControlPayload) error
}

type PushDelivery interface {
	StreamOpened(edgeID string)
	StreamClosed(edgeID, session string)
	HandleReloadResponse(*pb.ConfigurationReloadResponse)
}