registration

type InviteRecord struct {
	ID        string
	FleetID   string
	TokenHash string
	ExpiresAt time.Time
	CreatedAt time.Time
}

type InviteToken struct {
	ID        string
	FleetID   string
	Token     string
	ExpiresAt time.Time
}

type NodeRegistration struct {
	InviteTokenHash         string
	NodeID                  string
	KeyID                   string
	PublicKey               string
	HardwareFingerprintHash string
	BootstrapTokenID        string
	BootstrapTokenHash      string
	BootstrapTokenExpiresAt time.Time
	Now                     time.Time
}

type NodeSummary struct {
	ID              string     `json:"id"`
	State           string     `json:"state"`
	CurrentModelID  string     `json:"current_model_id,omitempty"`
	LastSeenAt      *time.Time `json:"last_seen_at,omitempty"`
	SessionStatus   string     `json:"session_status"`
	LastHeartbeatAt *time.Time `json:"last_heartbeat_at,omitempty"`
	ScheduleState   string     `json:"schedule_state,omitempty"`
	ThermalState    string     `json:"thermal_state,omitempty"`
	Paused          bool       `json:"paused"`
	Draining        bool       `json:"draining"`
}

type PGStore struct {
	Pool *pgxpool.Pool
}

type RegisterRequest struct {
	InviteToken             string `json:"invite_token"`
	PublicKey               string `json:"public_key"`
	HardwareFingerprintHash string `json:"hardware_fingerprint_hash"`
}

type RegisterResult struct {
	NodeID                  string            `json:"node_id"`
	BootstrapToken          string            `json:"bootstrap_token"`
	BootstrapTokenExpiresAt time.Time         `json:"bootstrap_token_expires_at"`
	FleetSchedule           *ScheduleDefaults `json:"fleet_schedule,omitempty"`
}

type RegistrationCreated struct {
	NodeID                  string
	BootstrapTokenExpiresAt time.Time
	FleetSchedule           *ScheduleDefaults
}

type Repository interface {
	CreateInvite(ctx context.Context, invite InviteRecord) error
	RegisterNode(ctx context.Context, registration NodeRegistration) (RegistrationCreated, error)
	ConsumeBootstrap(ctx context.Context, nodeID, bootstrapHash string, now time.Time) error
}

type ScheduleDefaults struct {
	From     string `json:"from"`
	Until    string `json:"until"`
	Timezone string `json:"timezone"`
}

type Service struct {
	Repository   Repository
	Now          func() time.Time
	InviteTTL    time.Duration
	BootstrapTTL time.Duration
	TokenBytes   int
}