rapid7

type APIError struct {
	Message       string `json:"message"`
	CorrelationID string `json:"correlation_id"`
}

type AssetCountQuery struct {
	Organization struct {
		Assets struct {
			TotalCount graphql.Int `graphql:"totalCount"`
		} `graphql:"assets(first: $first)"`
	} `graphql:"organization(id: $id)"`
}

type Assignee struct {
	Email string `json:"email"`
	Name  string `json:"name"`
}

type Client struct {
	IDR *IDR
	VM  *VM
}

type CommentAttachment struct {
	CreatedTime time.Time `json:"created_time"`
	Creator     Creator   `json:"creator"`
	FileName    string    `json:"file_name"`
	MimeType    string    `json:"mime_type"`
	RRN         string    `json:"rrn"`
	ScanStatus  string    `json:"scan_status"`
	Size        int64     `json:"size"`
}

type Creator struct {
	Name string `json:"name"`
	Type string `json:"type"`
}

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

type IDR struct {
	BaseURL *url.URL
	// contains filtered or unexported fields
}

type IDRAsset struct {
	RRN  string `json:"rrn"`
	Name string `json:"name"`
}

type IDRAssetQueryParams struct {
	Index *int32 `json:"index"`
	// Size field is not used here, but it is included to reflect the data structure.
	// Set ASSET_SEARCH_PAGE_SIZE instead.
	Size *int32 `json:"size"`
}

type IDRAssetRequest struct {
	Search []IDRAssetSearchQuery `json:"search"`
	Sort   []IDRAssetSortQuery   `json:"sort,omitempty"`
}

type IDRAssetSearchPageSize int32

type IDRAssetSearchQuery struct {
	Field    string         `json:"field"`
	Operator SearchOperator `json:"operator"`
	Value    string         `json:"value"`
}

type IDRAssetSortQuery struct {
	Field string        `json:"field"`
	Order SortDirection `json:"order"`
}

type Investigation struct {
	Assignee        *Assignee                `json:"assignee"`
	CreatedTime     time.Time                `json:"created_time"`
	Disposition     InvestigationDisposition `json:"disposition"`
	FirstAlertTime  *time.Time               `json:"first_alert_time"`
	LastAccessed    time.Time                `json:"last_accessed"`
	LatestAlertTime *time.Time               `json:"latest_alert_time"`
	OrganizationID  string                   `json:"organization_id"`
	Priority        InvestigationPriority    `json:"priority"`
	Responsibility  string                   `json:"responsibility"`
	RRN             string                   `json:"rrn"`
	Source          InvestigationSource      `json:"source"`
	Status          InvestigationStatus      `json:"status"`
	Tags            []string                 `json:"tags"`
	Title           string                   `json:"title"`
}

type InvestigationAssignee struct {
	Email string `json:"email"`
}

type InvestigationCommentData struct {
	Body        string              `json:"body"`
	CreatedTime time.Time           `json:"created_time"`
	Creator     Creator             `json:"creator"`
	RRN         string              `json:"rrn"`
	Target      string              `json:"target"`
	Visibility  string              `json:"visibility"`
	Attachments []CommentAttachment `json:"attachments"`
}

type InvestigationComments struct {
	Data     []InvestigationCommentData `json:"data"`
	Metadata Metadata                   `json:"metadata"`
}

type InvestigationDisposition string

type InvestigationPriority string

type InvestigationSource string

type InvestigationStatus string

type InvestigationUpdateRequest struct {
	Assignee    *InvestigationAssignee   `json:"assignee,omitempty"`
	Disposition InvestigationDisposition `json:"disposition,omitempty"`
	Priority    InvestigationPriority    `json:"priority,omitempty"`
	Status      InvestigationStatus      `json:"status,omitempty"`
	Title       string                   `json:"title,omitempty"`
}

type InvestigationsQuery struct {
	// A user's email address. Only investigations assigned to that user will be included.
	AssigneeEmail string `url:"assignee.email,omitempty"`
	// The time an investigation is closed. Only investigations whose created_time is before this
	// date will be returned by the API. Must be an ISO-formatted timestamp.
	EndTime time.Time `url:"end_time,omitempty"`
	// The 0-based index of the first page to retrieve. Must be an integer greater than 0.
	//
	// Default: 0
	Index int32 `url:"index,omitempty"`
	// Indicates whether the requester has multi-customer access. If set to true, a user API key
	// must be provided. Investigations will be returned from all organizations the calling user
	// has access to.
	//
	// Default: false
	MultiCustomer bool `url:"multi-customer,omitempty"`
	// A comma-separated list of investigation priorities to include in the result.
	Priorities []InvestigationPriority `url:"priorities,omitempty,comma"`
	// The maximum number of investigations to retrieve. Must be an integer greater than 0, or less
	// than or equal to 100.
	//
	// Default: 20
	Size int32 `url:"size,omitempty"`
	// Sort investigations by field and direction,  separated by a comma. Sortable fields are
	// `created_time`, `priority`, `rrn`, `alerts_most_recent_created_time`, and
	// `alerts_most_recent_detection_created_time`.
	//
	// Default: "priority,DESC"
	Sort string `url:"sort,omitempty"`
	// A comma-separated list of investigation sources to include in the result.
	Sources []string `url:"sources,omitempty,comma"`
	// The time an investigation is opened. Only investigations whose created_time is after this
	// date will be returned by the API. Must be an ISO-formatted timestamp.
	//
	// Default: 28 days prior to current time.
	StartTime time.Time `url:"start_time,omitempty,comma"`
	// A comma-separated list of investigation statuses to include in the result.
	Statuses []InvestigationStatus `url:"statuses,omitempty,comma"`
	// A comma-separated list of tags to include in the result. Only investigations who have all
	// specified tags will be included.
	Tags []string `url:"tags,omitempty,comma"`
}

type InvestigationsResponse = Rapid7PagedResponse[Investigation]

type Metadata struct {
	// The current page, starting from 0. This value will always be provided.
	Index int32 `json:"index"`
	// The number of data items in the current page. This value will always be provided.
	Size int32 `json:"size"`
	// The attributes used to sort the complete response. This will be provided if the response is sorted.
	Sort string `json:"sort,omitempty"`
	// The total number of data items that make up the complete response. This will be provided if possible.
	TotalData int64 `json:"total_data,omitempty"`
	// The total number of pages that make up the complete response. This will be provided if possible.
	TotalPages int32 `json:"total_pages,omitempty"`
}

type RRN struct {
	OrganizationID string   `json:"organizationId"`
	Partition      string   `json:"partition"`
	RegionCode     string   `json:"regionCode"`
	Resource       string   `json:"resource"`
	ResourceTypes  []string `json:"resourceTypes"`
	Service        string   `json:"service"`
}

type Rapid7PagedResponse[T any] struct {
	Data     []*T     `json:"data"`
	Metadata Metadata `json:"metadata"`
}

type Rapid7VMPagedResponse[T any] struct {
	Data     []T         `json:"data"`
	Links    []VMLink    `json:"links"`
	Metadata *VMMetadata `json:"metadata"`
}

type SearchOperator string

type SortDirection string

type SortField string

type VM struct {
	BaseURL *url.URL
	// contains filtered or unexported fields
}

type VMAsset struct {
	AssessedForPolicies            bool                     `json:"assessed_for_policies"`
	AssessedForVulnerabilities     bool                     `json:"assessed_for_vulnerabilities"`
	CredentialAssessments          []VMCredentialAssessment `json:"credential_assessments"`
	CriticalVulnerabilities        int32                    `json:"critical_vulnerabilities"`
	Exploits                       int32                    `json:"exploits"`
	HostName                       string                   `json:"host_name"`
	ID                             string                   `json:"id"`
	IP                             string                   `json:"ip"`
	LastAssessedForVulnerabilities time.Time                `json:"last_assessed_for_vulnerabilities"`
	LastScanEnd                    time.Time                `json:"last_scan_end"`
	LastScanStart                  time.Time                `json:"last_scan_start"`
	MAC                            string                   `json:"mac"`
	MalwareKits                    int32                    `json:"malware_kits"`
	ModerateVulnerabilities        int32                    `json:"moderate_vulnerabilities"`
	New                            []VMVulnerability        `json:"new"`
	OSArchitecture                 string                   `json:"os_architecture"`
	OSDescription                  string                   `json:"os_description"`
	OSFamily                       string                   `json:"os_family"`
	OSName                         string                   `json:"os_name"`
	OSSystemName                   string                   `json:"os_system_name"`
	OSType                         string                   `json:"os_type"`
	OSVendor                       string                   `json:"os_vendor"`
	OSVersion                      string                   `json:"os_version"`
	Remediated                     []VMVulnerability        `json:"remediated"`
	RiskScore                      float32                  `json:"risk_score"`
	Same                           []VMVulnerability        `json:"same"`
	SevereVulnerabilities          int32                    `json:"severe_vulnerabilities"`
	Tags                           []VMAssetTag             `json:"tags"`
	TotalVulnerabilities           int32                    `json:"total_vulnerabilities"`
	Type                           VMType                   `json:"type"`
	UniqueIdentifiers              []VMUniqueIdentifier     `json:"unique_identifiers"`
}

type VMAssetSearchPageSize int

type VMAssetSearchQuery struct {
	Cursor                   string        `json:"cursor,omitempty"`
	CurrentTime              time.Time     `json:"currentTime,omitempty"`
	ComparisonTime           time.Time     `json:"comparisonTime,omitempty"`
	IncludeSame              bool          `json:"includeSame,omitempty"`
	IncludeUniqueIdentifiers bool          `json:"includeUniqueIdentifiers,omitempty"`
	Page                     int           `json:"page,omitempty"`
	Size                     int           `json:"size,omitempty"`
	Sort                     SortDirection `json:"sort,omitempty"`
}

type VMAssetSearchRequest struct {
	Asset         string `json:"asset,omitempty"`
	Vulnerability string `json:"vulnerability,omitempty"`
}

type VMAssetTag struct {
	Name string `json:"name"`
	Type string `json:"type"`
}

type VMCredentialAssessment struct {
	Port     int64  `json:"port"`
	Protocol string `json:"protocol"`
	Status   string `json:"status"`
}

type VMLink struct {
	HREF string `json:"href"`
	Rel  string `json:"rel"`
}

type VMMetadata struct {
	// The index (zero-based) of the current page returned.
	Number int64 `json:"number"`
	// The maximum size of the page returned.
	Size int64 `json:"size"`
	// The stateless cursor associated with the series of page requests being made.
	Cursor string `json:"cursor"`
	// The total number of resources available across all pages.
	TotalResources int64 `json:"totalResources"`
	// The total number of pages available.
	TotalPages int64 `json:"totalPages"`
}

type VMType string

type VMUniqueIdentifier struct {
	ID     string `json:"id"`
	Source string `json:"source"`
}

type VMVulnerability struct {
	CheckID         string                `json:"check_id"`
	FirstFound      time.Time             `json:"first_found"`
	Key             string                `json:"key"`
	LastFound       time.Time             `json:"last_found"`
	NIC             string                `json:"nic"`
	Port            int32                 `json:"port"`
	Proof           string                `json:"proof"`
	Protocol        string                `json:"protocol"`
	SolutionFix     string                `json:"solution_fix"`
	SolutionID      string                `json:"solution_id"`
	SolutionSummary string                `json:"solution_summary"`
	SolutionType    string                `json:"solution_type"`
	Status          VMVulnerabilityStatus `json:"status"`
	VulnerabilityID string                `json:"vulnerability_id"`
}

type VMVulnerabilityStatus string