securityhub

type ActionTarget struct {
	ActionTargetArn string `json:"ActionTargetArn"`
	Name            string `json:"Name"`
	Description     string `json:"Description"`
}

type AdminAccount struct {
	AccountId          string `json:"AccountId"`    //nolint:revive,staticcheck // existing issue.
	InvitationId       string `json:"InvitationId"` //nolint:revive,staticcheck // existing issue.
	InvitedAt          string `json:"InvitedAt"`
	RelationshipStatus string `json:"RelationshipStatus"`
}

type AggregatorV2 struct {
	AggregatorV2Arn   string   `json:"AggregatorV2Arn"`
	AggregationRegion string   `json:"AggregationRegion"`
	RegionLinkingMode string   `json:"RegionLinkingMode"`
	CreatedAt         string   `json:"CreatedAt"`
	UpdatedAt         string   `json:"UpdatedAt"`
	Regions           []string `json:"Regions"`
}

type AutomationRule struct {
	Criteria    map[string]any   `json:"Criteria"`
	RuleArn     string           `json:"RuleArn"`
	RuleStatus  string           `json:"RuleStatus"`
	RuleName    string           `json:"RuleName"`
	Description string           `json:"Description"`
	CreatedAt   string           `json:"CreatedAt"`
	UpdatedAt   string           `json:"UpdatedAt"`
	CreatedBy   string           `json:"CreatedBy"`
	Actions     []map[string]any `json:"Actions"`
	RuleOrder   int32            `json:"RuleOrder"`
	IsTerminal  bool             `json:"IsTerminal"`
}

type AutomationRuleMetadata struct {
	RuleArn     string
	RuleStatus  string
	RuleName    string
	Description string
	CreatedAt   string
	UpdatedAt   string
	CreatedBy   string
	RuleOrder   int32
	IsTerminal  bool
}

type AutomationRuleV2 struct {
	Identifier  string           `json:"Identifier"`
	RuleArn     string           `json:"RuleArn"`
	RuleName    string           `json:"RuleName"`
	RuleStatus  string           `json:"RuleStatus"`
	Description string           `json:"Description"`
	CreatedAt   string           `json:"CreatedAt"`
	UpdatedAt   string           `json:"UpdatedAt"`
	Criteria    map[string]any   `json:"Criteria"`
	Actions     []map[string]any `json:"Actions"`
	RuleOrder   float64          `json:"RuleOrder"`
	IsTerminal  bool             `json:"IsTerminal"`
}

type ConfigurationPolicy struct {
	ConfigurationPolicy map[string]any    `json:"ConfigurationPolicy"`
	Tags                map[string]string `json:"Tags"`
	Arn                 string            `json:"Arn"`
	Id                  string            `json:"Id"` //nolint:revive,staticcheck // existing issue.
	Name                string            `json:"Name"`
	Description         string            `json:"Description"`
	CreatedAt           string            `json:"CreatedAt"`
	UpdatedAt           string            `json:"UpdatedAt"`
}

type ConfigurationPolicyAssociation struct {
	ConfigurationPolicyId    string `json:"ConfigurationPolicyId"` //nolint:revive,staticcheck // existing issue.
	TargetId                 string `json:"TargetId"`              //nolint:revive,staticcheck // existing issue.
	TargetType               string `json:"TargetType"`
	AssociationType          string `json:"AssociationType"`
	UpdatedAt                string `json:"UpdatedAt"`
	AssociationStatus        string `json:"AssociationStatus"`
	AssociationStatusMessage string `json:"AssociationStatusMessage"`
}

type ConnectorV2 struct {
	Provider        map[string]any    `json:"Provider"`
	Tags            map[string]string `json:"Tags"`
	ConnectorId     string            `json:"ConnectorId"` //nolint:revive,staticcheck // existing issue.
	ConnectorArn    string            `json:"ConnectorArn"`
	Name            string            `json:"Name"`
	Description     string            `json:"Description"`
	CreatedAt       string            `json:"CreatedAt"`
	UpdatedAt       string            `json:"UpdatedAt"`
	ConnectorStatus string            `json:"ConnectorStatus"`
}

type CspmConnector struct {
	Provider         map[string]any    `json:"Provider"`
	Tags             map[string]string `json:"Tags"`
	ConnectorId      string            `json:"ConnectorId"` //nolint:revive,staticcheck // existing issue.
	ConnectorArn     string            `json:"ConnectorArn"`
	Name             string            `json:"Name"`
	Description      string            `json:"Description"`
	CreatedAt        string            `json:"CreatedAt"`
	LastUpdatedAt    string            `json:"LastUpdatedAt"`
	CreatedBy        string            `json:"CreatedBy"`
	EnablementStatus string            `json:"EnablementStatus"`
	// ConnectorStatus/HealthMessage/HealthCheckedAt make up the connector's
	// health check (types.CspmHealthCheck) -- kept as flat fields here rather
	// than a nested struct since nothing else needs to address them as a unit.
	ConnectorStatus string `json:"ConnectorStatus"`
	HealthMessage   string `json:"HealthMessage"`
	HealthCheckedAt string `json:"HealthCheckedAt"`
	ProviderName    string `json:"ProviderName"`
}

type FindingAggregator struct {
	FindingAggregatorArn     string   `json:"FindingAggregatorArn"`
	FindingAggregationRegion string   `json:"FindingAggregationRegion"`
	RegionLinkingMode        string   `json:"RegionLinkingMode"`
	Regions                  []string `json:"Regions"`
}

type Handler struct {
	Backend StorageBackend
}

type Hub struct {
	HubArn                  string `json:"HubArn"`
	SubscribedAt            string `json:"SubscribedAt"`
	AutoEnableStandards     string `json:"AutoEnableStandards"`
	ControlFindingGenerator string `json:"ControlFindingGenerator"`
	AutoEnableControls      bool   `json:"AutoEnableControls"`
}

type HubV2 struct {
	Features  map[string]*HubV2Feature `json:"Features"`
	HubV2Arn  string                   `json:"HubV2Arn"`
	CreatedAt string                   `json:"CreatedAt"`
	UpdatedAt string                   `json:"UpdatedAt"`
}

type HubV2Feature struct {
	FeatureStatus string `json:"FeatureStatus"`
	UpdatedAt     string `json:"UpdatedAt"`
}

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

type Insight struct {
	Filters          map[string]any `json:"Filters"`
	InsightArn       string         `json:"InsightArn"`
	Name             string         `json:"Name"`
	GroupByAttribute string         `json:"GroupByAttribute"`
}

type InsightResults struct {
	InsightArn       string
	GroupByAttribute string
	ResultValues     []map[string]any
}

type Invitation struct {
	AccountId    string `json:"AccountId"`    //nolint:revive,staticcheck // existing issue.
	InvitationId string `json:"InvitationId"` //nolint:revive,staticcheck // existing issue.
	InvitedAt    string `json:"InvitedAt"`
	MemberStatus string `json:"MemberStatus"`
}

type Member struct {
	AccountId       string `json:"AccountId"`       //nolint:revive,staticcheck // existing issue.
	AdministratorId string `json:"AdministratorId"` //nolint:revive,staticcheck // existing issue.
	MasterId        string `json:"MasterId"`        // deprecated alias //nolint:revive,staticcheck // existing issue.
	Email           string `json:"Email"`
	MemberStatus    string `json:"MemberStatus"`
	InvitedAt       string `json:"InvitedAt"`
	UpdatedAt       string `json:"UpdatedAt"`
}

type OrgAdminAccount struct {
	AccountId string `json:"AccountId"` //nolint:revive,staticcheck // existing issue.
	Status    string `json:"Status"`
}

type OrgConfig struct {
	AutoEnableStandards           string `json:"AutoEnableStandards"`
	OrganizationConfigurationType string `json:"OrganizationConfigurationType"`
	AutoEnable                    bool   `json:"AutoEnable"`
	MemberAccountLimitReached     bool   `json:"MemberAccountLimitReached"`
}

type Product struct {
	ProductArn                        string
	ProductName                       string
	CompanyName                       string
	Description                       string
	MarketplaceURL                    string
	ActivationURL                     string
	ProductSubscriptionResourcePolicy string
	Categories                        []string
	IntegrationTypes                  []string
}

type Provider struct{}

type RecommendedPolicyV2 struct {
	MetadataUid    string `json:"MetadataUid"` //nolint:revive,staticcheck // existing issue.
	Policy         string `json:"Policy"`
	GenerationTime string `json:"GenerationTime"`
}

type SecurityControl struct {
	SecurityControlID     string
	SecurityControlArn    string
	Title                 string
	Description           string
	RemediationURL        string
	SeverityRating        string
	SecurityControlStatus string
	UpdateStatus          string
	Parameters            map[string]any
	LastUpdateReason      string
}

type SecurityControlDefinition struct {
	ParameterDefinitions      map[string]any
	SecurityControlID         string
	Title                     string
	Description               string
	RemediationURL            string
	SeverityRating            string
	CurrentRegionAvailability string
	CustomizableProperties    []string
}

type Standard struct {
	StandardsManagedBy map[string]any
	StandardsArn       string
	Name               string
	Description        string
	EnabledByDefault   bool
}

type StandardsControl struct {
	StandardsControlArn    string   `json:"StandardsControlArn"`
	ControlStatus          string   `json:"ControlStatus"`
	DisabledReason         string   `json:"DisabledReason"`
	ControlStatusUpdatedAt string   `json:"ControlStatusUpdatedAt"`
	ControlID              string   `json:"ControlId"`
	Title                  string   `json:"Title"`
	Description            string   `json:"Description"`
	RemediationURL         string   `json:"RemediationUrl"`
	SeverityRating         string   `json:"SeverityRating"`
	RelatedRequirements    []string `json:"RelatedRequirements"`
}

type StandardsControlAssociation struct {
	SecurityControlID           string
	StandardsArn                string
	AssociationStatus           string
	RelatedRequirements         []string
	UpdatedAt                   string
	UpdatedReason               string
	StandardsControlTitle       string
	StandardsControlDescription string
	StandardsControlArns        []string
}

type StandardsSubscription struct {
	StandardsInput           map[string]string `json:"StandardsInput"`
	StatusReason             map[string]any    `json:"StatusReason"`
	StandardsSubscriptionArn string            `json:"StandardsSubscriptionArn"`
	StandardsArn             string            `json:"StandardsArn"`
	StandardsStatus          string            `json:"StandardsStatus"`
}

type StorageBackend interface {
	// Hub management
	EnableHub(enableDefaultStandards bool, tags map[string]string) error
	DisableHub() error
	DescribeHub() (*Hub, error)
	UpdateHubConfiguration(autoEnableControls *bool, autoEnableStandards *string, controlFindingGenerator *string) error

	// Findings
	ImportFindings(findings []map[string]any) (int, int, []map[string]any)
	GetFindings(
		filters map[string]any,
		sortCriteria []map[string]any,
		nextToken string,
		maxResults int,
	) ([]map[string]any, string)
	UpdateFindings(filters map[string]any, note map[string]any, recordState string) error
	BatchUpdateFindings(
		findingIdentifiers []map[string]any,
		updates map[string]any,
	) ([]map[string]any, []map[string]any)
	GetFindingHistory(
		findingIdentifier map[string]any,
		startTime, endTime string,
		nextToken string,
		maxResults int,
	) ([]map[string]any, string)

	// Insights
	CreateInsight(name, groupByAttribute string, filters map[string]any) (string, error)
	GetInsights(insightArns []string, nextToken string, maxResults int) ([]*Insight, string, error)
	UpdateInsight(insightArn, name, groupByAttribute string, filters map[string]any) error
	DeleteInsight(insightArn string) (string, error)
	GetInsightResults(insightArn string) (*InsightResults, error)

	// Standards
	BatchEnableStandards(requests []map[string]any) ([]*StandardsSubscription, []map[string]any)
	BatchDisableStandards(subscriptionArns []string) ([]*StandardsSubscription, []map[string]any)
	GetEnabledStandards(subscriptionArns []string, nextToken string, maxResults int) ([]*StandardsSubscription, string)
	DescribeStandards(nextToken string, maxResults int) ([]*Standard, string)
	DescribeStandardsControls(subscriptionArn, nextToken string, maxResults int) ([]*StandardsControl, string)
	UpdateStandardsControl(controlArn, controlStatus, disabledReason string) error
	ListStandardsControlAssociations(
		securityControlID, nextToken string,
		maxResults int,
	) ([]*StandardsControlAssociation, string)
	BatchGetStandardsControlAssociations(requests []map[string]any) ([]*StandardsControlAssociation, []map[string]any)
	BatchUpdateStandardsControlAssociations(updates []map[string]any) ([]map[string]any, error)

	// Action Targets
	CreateActionTarget(name, description, id string) (string, error)
	DescribeActionTargets(actionTargetArns []string, nextToken string, maxResults int) ([]*ActionTarget, string)
	UpdateActionTarget(actionTargetArn, name, description string) error
	DeleteActionTarget(actionTargetArn string) (string, error)

	// Products
	DescribeProducts(productArn, nextToken string, maxResults int) ([]*Product, string)
	EnableImportFindingsForProduct(productArn string) (string, error)
	DisableImportFindingsForProduct(productSubscriptionArn string) error
	ListEnabledProductsForImport(nextToken string, maxResults int) ([]string, string)

	// Security Controls
	GetSecurityControlDefinition(securityControlID string) (*SecurityControlDefinition, error)
	ListSecurityControlDefinitions(
		standardsArn, nextToken string,
		maxResults int,
	) ([]*SecurityControlDefinition, string)
	BatchGetSecurityControls(securityControlIDs []string) ([]*SecurityControl, []map[string]any)
	UpdateSecurityControl(securityControlID string, parameters map[string]any, lastUpdateReason string) error

	// Automation Rules
	CreateAutomationRule(rule map[string]any) (string, string)
	ListAutomationRules(nextToken string, maxResults int) ([]*AutomationRuleMetadata, string)
	BatchGetAutomationRules(automationRulesArns []string) ([]*AutomationRule, []map[string]any)
	BatchDeleteAutomationRules(automationRulesArns []string) ([]string, []map[string]any)
	BatchUpdateAutomationRules(updates []map[string]any) ([]string, []map[string]any)

	// Tags
	TagResource(resourceArn string, tags map[string]string) error
	UntagResource(resourceArn string, tagKeys []string) error
	ListTagsForResource(resourceArn string) (map[string]string, error)

	// Members
	CreateMembers(accounts []map[string]any) ([]*Member, []map[string]any)
	DeleteMembers(accountIDs []string) ([]string, []map[string]any)
	GetMembers(accountIDs []string) ([]*Member, []map[string]any)
	InviteMembers(accountIDs []string) []map[string]any
	ListMembers(onlyAssociated bool, nextToken string, maxResults int) ([]*Member, string)
	DisassociateMembers(accountIDs []string) error

	// Invitations / Admin
	AcceptAdministratorInvitation(administratorID, invitationID string) error
	AcceptInvitation(masterID, invitationID string) error
	DeclineInvitations(accountIDs []string) ([]map[string]any, []map[string]any)
	DeleteInvitations(accountIDs []string) ([]map[string]any, []map[string]any)
	GetInvitationsCount() int
	ListInvitations(nextToken string, maxResults int) ([]*Invitation, string)
	GetAdministratorAccount() (*AdminAccount, error)
	GetMasterAccount() (*AdminAccount, error)
	DisassociateFromAdministratorAccount() error
	DisassociateFromMasterAccount() error

	// Organization
	DescribeOrganizationConfiguration() *OrgConfig
	UpdateOrganizationConfiguration(autoEnable bool, autoEnableStandards string, orgConfigType string) error
	EnableOrganizationAdminAccount(accountID string) error
	DisableOrganizationAdminAccount(accountID string) error
	ListOrganizationAdminAccounts(nextToken string, maxResults int) ([]*OrgAdminAccount, string)

	// Finding Aggregator
	CreateFindingAggregator(regionLinkingMode string, regions []string) (*FindingAggregator, error)
	GetFindingAggregator(arn string) (*FindingAggregator, error)
	ListFindingAggregators(nextToken string, maxResults int) ([]*FindingAggregator, string)
	UpdateFindingAggregator(arn, regionLinkingMode string, regions []string) (*FindingAggregator, error)
	DeleteFindingAggregator(arn string) error

	// Configuration Policy
	CreateConfigurationPolicy(
		name, description string,
		policy map[string]any,
		tags map[string]string,
	) (*ConfigurationPolicy, error)
	GetConfigurationPolicy(identifier string) (*ConfigurationPolicy, error)
	UpdateConfigurationPolicy(identifier, name, description string, policy map[string]any) (*ConfigurationPolicy, error)
	DeleteConfigurationPolicy(identifier string) error
	ListConfigurationPolicies(nextToken string, maxResults int) ([]*ConfigurationPolicy, string)
	StartConfigurationPolicyAssociation(
		configPolicyIdentifier, targetID, targetType string,
	) (*ConfigurationPolicyAssociation, error)
	StartConfigurationPolicyDisassociation(configPolicyIdentifier, targetID, targetType string) error
	GetConfigurationPolicyAssociation(targetID, targetType string) (*ConfigurationPolicyAssociation, error)
	ListConfigurationPolicyAssociations(
		filterPolicyID, filterType, nextToken string,
		maxResults int,
	) ([]*ConfigurationPolicyAssociation, string)
	BatchGetConfigurationPolicyAssociations(
		requests []map[string]any,
	) ([]*ConfigurationPolicyAssociation, []map[string]any)

	// Hub V2
	EnableSecurityHubV2(tags map[string]string) error
	DisableSecurityHubV2() error
	DescribeSecurityHubV2() (*HubV2, error)
	EnableSecurityHubFeatureV2(featureName string) error
	DisableSecurityHubFeatureV2(featureName string) error

	// CSPM Connectors (third-party cloud provider connectors, distinct from
	// Connectors V2's ticketing-system connectors)
	CreateConnector(name, description string, provider map[string]any, tags map[string]string) (*CspmConnector, error)
	GetConnector(connectorID string) (*CspmConnector, error)
	UpdateConnector(connectorID, description string, provider map[string]any) (*CspmConnector, error)
	DeleteConnector(connectorID string) (string, error)
	ListConnectors(
		connectorStatus, enablementStatus, providerName, nextToken string,
		maxResults int,
	) ([]*CspmConnector, string)

	// Aggregator V2
	CreateAggregatorV2(regionLinkingMode string, regions []string) (*AggregatorV2, error)
	GetAggregatorV2(arn string) (*AggregatorV2, error)
	ListAggregatorsV2(nextToken string, maxResults int) ([]*AggregatorV2, string)
	UpdateAggregatorV2(arn, regionLinkingMode string, regions []string) (*AggregatorV2, error)
	DeleteAggregatorV2(arn string) error

	// Automation Rules V2
	CreateAutomationRuleV2(
		ruleName, ruleStatus, description string,
		criteria map[string]any,
		actions []map[string]any,
		ruleOrder float64,
		isTerminal bool,
		tags map[string]string,
	) (*AutomationRuleV2, error)
	GetAutomationRuleV2(identifier string) (*AutomationRuleV2, error)
	ListAutomationRulesV2(nextToken string, maxResults int) ([]*AutomationRuleV2, string)
	UpdateAutomationRuleV2(identifier string, updates map[string]any) (*AutomationRuleV2, error)
	DeleteAutomationRuleV2(identifier string) error

	// Connectors V2
	CreateConnectorV2(name, description string, provider map[string]any, tags map[string]string) (*ConnectorV2, error)
	GetConnectorV2(connectorID string) (*ConnectorV2, error)
	ListConnectorsV2(nextToken string, maxResults int) ([]*ConnectorV2, string)
	UpdateConnectorV2(connectorID, name, description string, provider map[string]any) (*ConnectorV2, error)
	DeleteConnectorV2(connectorID string) error
	RegisterConnectorV2(connectorID string, provider map[string]any) (*ConnectorV2, error)

	// Tickets V2
	CreateTicketV2(ticketConfig map[string]any, tags map[string]string) (*TicketV2, error)

	// Findings V2
	GetFindingsV2(
		filters map[string]any,
		sortCriteria []map[string]any,
		nextToken string,
		maxResults int,
	) ([]map[string]any, string)
	BatchUpdateFindingsV2(
		findingIdentifiers []map[string]any,
		metadataUids []string,
		updates map[string]any,
	) ([]map[string]any, []map[string]any)
	GetFindingStatisticsV2(groupByAttributes []string) []map[string]any
	GetFindingsTrendsV2(groupByAttribute string, startTime, endTime string) []map[string]any

	// Resources V2
	GetResourcesV2(filters map[string]any, nextToken string, maxResults int) ([]map[string]any, string)
	GetResourcesStatisticsV2(groupByAttributes []string) []map[string]any
	GetResourcesTrendsV2(groupByAttribute string, startTime, endTime string) []map[string]any

	// Products V2
	DescribeProductsV2(nextToken string, maxResults int) ([]*Product, string)

	// Recommended Policy V2
	GenerateRecommendedPolicyV2(metadataUID string) (*RecommendedPolicyV2, error)
	GetRecommendedPolicyV2(metadataUID string) (*RecommendedPolicyV2, error)

	// Metadata
	AccountID() string
	Region() string
	Reset()
	Snapshot(ctx context.Context) []byte
	Restore(ctx context.Context, data []byte) error
}

type TicketV2 struct {
	TicketConfigurationArn string `json:"TicketConfigurationArn"`
	CreatedAt              string `json:"CreatedAt"`
}