mirror of
https://github.com/j3ssie/osmedeus.git
synced 2026-08-30 11:49:44 +02:00
j3ssie
8d413aecb9
- Refactor table display using tablewriter with box-drawing characters and proper Unicode width handling - Add --as-cron flag to register cron schedules instead of executing immediately - Implement new 'osmedeus assets' command for asset querying with filtering and statistics - Enhance workflow parameter display with categorization and per-module summaries - Improve column width handling and parameter documentation
329 lines
13 KiB
Go
329 lines
13 KiB
Go
package client
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
// ErrorResponse represents an error response from the API
|
|
type ErrorResponse struct {
|
|
Error bool `json:"error"`
|
|
Message string `json:"message"`
|
|
}
|
|
|
|
// APIError represents an API error with status code and message
|
|
type APIError struct {
|
|
StatusCode int
|
|
Message string
|
|
}
|
|
|
|
// Error implements the error interface
|
|
func (e *APIError) Error() string {
|
|
return fmt.Sprintf("API error (status %d): %s", e.StatusCode, e.Message)
|
|
}
|
|
|
|
// Pagination represents pagination information in API responses
|
|
type Pagination struct {
|
|
Total int `json:"total"`
|
|
Offset int `json:"offset"`
|
|
Limit int `json:"limit"`
|
|
}
|
|
|
|
// PaginatedResponse represents a paginated API response
|
|
type PaginatedResponse struct {
|
|
Data interface{} `json:"data"`
|
|
Pagination Pagination `json:"pagination"`
|
|
}
|
|
|
|
// RunsResponse represents the response from listing runs
|
|
type RunsResponse struct {
|
|
Data []Run `json:"data"`
|
|
Pagination Pagination `json:"pagination"`
|
|
}
|
|
|
|
// Run represents a workflow run from the API
|
|
type Run struct {
|
|
ID int64 `json:"id"`
|
|
RunUUID string `json:"run_uuid"`
|
|
WorkflowName string `json:"workflow_name"`
|
|
WorkflowKind string `json:"workflow_kind"`
|
|
Target string `json:"target"`
|
|
Params map[string]interface{} `json:"params,omitempty"`
|
|
Status string `json:"status"`
|
|
TriggerType string `json:"trigger_type,omitempty"`
|
|
RunGroupID string `json:"run_group_id,omitempty"`
|
|
TotalSteps int `json:"total_steps"`
|
|
CompletedSteps int `json:"completed_steps"`
|
|
Workspace string `json:"workspace"`
|
|
RunPriority string `json:"run_priority,omitempty"`
|
|
RunMode string `json:"run_mode,omitempty"`
|
|
StartedAt *time.Time `json:"started_at,omitempty"`
|
|
CompletedAt *time.Time `json:"completed_at,omitempty"`
|
|
ErrorMessage string `json:"error_message,omitempty"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|
|
|
|
// CreateRunRequest represents a request to create a new run
|
|
type CreateRunRequest struct {
|
|
Flow string `json:"flow,omitempty"`
|
|
Module string `json:"module,omitempty"`
|
|
Target string `json:"target,omitempty"`
|
|
Targets []string `json:"targets,omitempty"`
|
|
Params map[string]string `json:"params,omitempty"`
|
|
Concurrency int `json:"concurrency,omitempty"`
|
|
Priority string `json:"priority,omitempty"`
|
|
RunMode string `json:"run_mode,omitempty"`
|
|
Timeout int `json:"timeout,omitempty"`
|
|
RunnerType string `json:"runner_type,omitempty"`
|
|
ThreadsHold int `json:"threads_hold,omitempty"`
|
|
EmptyTarget bool `json:"empty_target,omitempty"`
|
|
Repeat bool `json:"repeat,omitempty"`
|
|
RepeatWaitTime string `json:"repeat_wait_time,omitempty"`
|
|
HeuristicsCheck string `json:"heuristics_check,omitempty"`
|
|
}
|
|
|
|
// CreateRunResponse represents the response from creating a run
|
|
type CreateRunResponse struct {
|
|
Message string `json:"message"`
|
|
Workflow string `json:"workflow"`
|
|
Kind string `json:"kind"`
|
|
TargetCount int `json:"target_count"`
|
|
Priority string `json:"priority"`
|
|
JobID string `json:"job_id"`
|
|
Status string `json:"status"`
|
|
PollURL string `json:"poll_url"`
|
|
Target string `json:"target,omitempty"`
|
|
RunUUID string `json:"run_uuid,omitempty"`
|
|
Targets []string `json:"targets,omitempty"`
|
|
Concurrency int `json:"concurrency,omitempty"`
|
|
}
|
|
|
|
// CancelRunResponse represents the response from cancelling a run
|
|
type CancelRunResponse struct {
|
|
Message string `json:"message"`
|
|
ID int64 `json:"id"`
|
|
RunUUID string `json:"run_uuid"`
|
|
}
|
|
|
|
// AssetsResponse represents the response from listing assets
|
|
type AssetsResponse struct {
|
|
Data []Asset `json:"data"`
|
|
Pagination Pagination `json:"pagination"`
|
|
}
|
|
|
|
// Asset represents an asset from the API
|
|
type Asset struct {
|
|
ID int64 `json:"id"`
|
|
Workspace string `json:"workspace"`
|
|
AssetType string `json:"asset_type"`
|
|
AssetValue string `json:"asset_value"`
|
|
URL string `json:"url,omitempty"`
|
|
Title string `json:"title,omitempty"`
|
|
StatusCode int `json:"status_code,omitempty"`
|
|
ContentLength int `json:"content_length,omitempty"`
|
|
HostIP string `json:"host_ip,omitempty"`
|
|
TechStack []string `json:"tech_stack,omitempty"`
|
|
ContentType string `json:"content_type,omitempty"`
|
|
IsCDN bool `json:"is_cdn,omitempty"`
|
|
IsCloud bool `json:"is_cloud,omitempty"`
|
|
IsWAF bool `json:"is_waf,omitempty"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|
|
|
|
// WorkspacesResponse represents the response from listing workspaces
|
|
type WorkspacesResponse struct {
|
|
Data []Workspace `json:"data"`
|
|
Pagination Pagination `json:"pagination"`
|
|
}
|
|
|
|
// Workspace represents a workspace from the API
|
|
type Workspace struct {
|
|
ID int64 `json:"id,omitempty"`
|
|
Name string `json:"name"`
|
|
LocalPath string `json:"local_path,omitempty"`
|
|
DataSource string `json:"data_source,omitempty"`
|
|
TotalAssets int `json:"total_assets,omitempty"`
|
|
Tags []string `json:"tags,omitempty"`
|
|
CreatedAt time.Time `json:"created_at,omitempty"`
|
|
UpdatedAt time.Time `json:"updated_at,omitempty"`
|
|
}
|
|
|
|
// VulnerabilitiesResponse represents the response from listing vulnerabilities
|
|
type VulnerabilitiesResponse struct {
|
|
Data []Vulnerability `json:"data"`
|
|
Pagination Pagination `json:"pagination"`
|
|
}
|
|
|
|
// Vulnerability represents a vulnerability from the API
|
|
type Vulnerability struct {
|
|
ID int64 `json:"id"`
|
|
Workspace string `json:"workspace"`
|
|
VulnInfo string `json:"vuln_info,omitempty"`
|
|
VulnTitle string `json:"vuln_title,omitempty"`
|
|
VulnDesc string `json:"vuln_desc,omitempty"`
|
|
VulnPOC string `json:"vuln_poc,omitempty"`
|
|
Severity string `json:"severity,omitempty"`
|
|
Confidence string `json:"confidence,omitempty"`
|
|
AssetType string `json:"asset_type,omitempty"`
|
|
AssetValue string `json:"asset_value,omitempty"`
|
|
Tags []string `json:"tags,omitempty"`
|
|
DetailHTTPRequest string `json:"detail_http_request,omitempty"`
|
|
DetailHTTPResponse string `json:"detail_http_response,omitempty"`
|
|
RawVulnJSON string `json:"raw_vuln_json,omitempty"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|
|
|
|
// FunctionEvalRequest represents a request to evaluate a function
|
|
type FunctionEvalRequest struct {
|
|
Script string `json:"script"`
|
|
Target string `json:"target,omitempty"`
|
|
Params map[string]string `json:"params,omitempty"`
|
|
}
|
|
|
|
// FunctionEvalResponse represents the response from evaluating a function
|
|
type FunctionEvalResponse struct {
|
|
Result interface{} `json:"result"`
|
|
RenderedScript string `json:"rendered_script,omitempty"`
|
|
Error bool `json:"error,omitempty"`
|
|
Message string `json:"message,omitempty"`
|
|
}
|
|
|
|
// StepResultsResponse represents the response from listing step results
|
|
type StepResultsResponse struct {
|
|
Data []StepResult `json:"data"`
|
|
Pagination Pagination `json:"pagination"`
|
|
}
|
|
|
|
// StepResult represents a step execution result from the API
|
|
type StepResult struct {
|
|
ID string `json:"id"`
|
|
RunID int64 `json:"run_id"`
|
|
StepName string `json:"step_name"`
|
|
StepType string `json:"step_type"`
|
|
Status string `json:"status"`
|
|
Command string `json:"command,omitempty"`
|
|
Output string `json:"output,omitempty"`
|
|
ErrorMessage string `json:"error_message,omitempty"`
|
|
Exports map[string]interface{} `json:"exports,omitempty"`
|
|
DurationMs int64 `json:"duration_ms"`
|
|
LogFile string `json:"log_file,omitempty"`
|
|
StartedAt *time.Time `json:"started_at,omitempty"`
|
|
CompletedAt *time.Time `json:"completed_at,omitempty"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
}
|
|
|
|
// ArtifactsResponse represents the response from listing artifacts
|
|
type ArtifactsResponse struct {
|
|
Data []Artifact `json:"data"`
|
|
Pagination Pagination `json:"pagination"`
|
|
}
|
|
|
|
// Artifact represents an artifact from the API
|
|
type Artifact struct {
|
|
ID string `json:"id"`
|
|
RunID int64 `json:"run_id"`
|
|
Workspace string `json:"workspace"`
|
|
Name string `json:"name"`
|
|
ArtifactPath string `json:"artifact_path"`
|
|
ArtifactType string `json:"artifact_type,omitempty"`
|
|
ContentType string `json:"content_type,omitempty"`
|
|
SizeBytes int64 `json:"size_bytes"`
|
|
LineCount int `json:"line_count"`
|
|
Description string `json:"description,omitempty"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
}
|
|
|
|
// EventLogsResponse represents the response from listing event logs
|
|
type EventLogsResponse struct {
|
|
Data []EventLog `json:"data"`
|
|
Pagination Pagination `json:"pagination"`
|
|
}
|
|
|
|
// EventLog represents an event log from the API
|
|
type EventLog struct {
|
|
ID int64 `json:"id"`
|
|
Topic string `json:"topic"`
|
|
EventID string `json:"event_id,omitempty"`
|
|
Name string `json:"name,omitempty"`
|
|
SourceType string `json:"source_type,omitempty"` // "run", "eval", "api" - origin of the event
|
|
Source string `json:"source,omitempty"`
|
|
DataType string `json:"data_type,omitempty"`
|
|
Data string `json:"data,omitempty"`
|
|
Workspace string `json:"workspace,omitempty"`
|
|
RunID string `json:"run_id,omitempty"`
|
|
WorkflowName string `json:"workflow_name,omitempty"`
|
|
Processed bool `json:"processed"`
|
|
ProcessedAt *time.Time `json:"processed_at,omitempty"`
|
|
Error string `json:"error,omitempty"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
}
|
|
|
|
// SchedulesResponse represents the response from listing schedules
|
|
type SchedulesResponse struct {
|
|
Data []Schedule `json:"data"`
|
|
Pagination Pagination `json:"pagination"`
|
|
}
|
|
|
|
// Schedule represents a schedule from the API
|
|
type Schedule struct {
|
|
ID string `json:"id"`
|
|
Name string `json:"name"`
|
|
WorkflowName string `json:"workflow_name"`
|
|
WorkflowKind string `json:"workflow_kind,omitempty"`
|
|
Target string `json:"target,omitempty"`
|
|
Workspace string `json:"workspace,omitempty"`
|
|
Params map[string]interface{} `json:"params,omitempty"`
|
|
TriggerName string `json:"trigger_name"`
|
|
TriggerType string `json:"trigger_type"`
|
|
Schedule string `json:"schedule,omitempty"`
|
|
EventTopic string `json:"event_topic,omitempty"`
|
|
WatchPath string `json:"watch_path,omitempty"`
|
|
IsEnabled bool `json:"is_enabled"`
|
|
LastRun *time.Time `json:"last_run,omitempty"`
|
|
NextRun *time.Time `json:"next_run,omitempty"`
|
|
RunCount int `json:"run_count"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|
|
|
|
// AssetDiffsResponse represents the response from listing asset diffs
|
|
type AssetDiffsResponse struct {
|
|
Data []AssetDiff `json:"data"`
|
|
Pagination Pagination `json:"pagination"`
|
|
}
|
|
|
|
// AssetDiff represents an asset diff snapshot from the API
|
|
type AssetDiff struct {
|
|
ID int64 `json:"id"`
|
|
WorkspaceName string `json:"workspace_name"`
|
|
FromTime time.Time `json:"from_time"`
|
|
ToTime time.Time `json:"to_time"`
|
|
TotalAdded int `json:"total_added"`
|
|
TotalRemoved int `json:"total_removed"`
|
|
TotalChanged int `json:"total_changed"`
|
|
DiffData string `json:"diff_data,omitempty"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
}
|
|
|
|
// VulnDiffsResponse represents the response from listing vulnerability diffs
|
|
type VulnDiffsResponse struct {
|
|
Data []VulnDiff `json:"data"`
|
|
Pagination Pagination `json:"pagination"`
|
|
}
|
|
|
|
// VulnDiff represents a vulnerability diff snapshot from the API
|
|
type VulnDiff struct {
|
|
ID int64 `json:"id"`
|
|
WorkspaceName string `json:"workspace_name"`
|
|
FromTime time.Time `json:"from_time"`
|
|
ToTime time.Time `json:"to_time"`
|
|
TotalAdded int `json:"total_added"`
|
|
TotalRemoved int `json:"total_removed"`
|
|
TotalChanged int `json:"total_changed"`
|
|
DiffData string `json:"diff_data,omitempty"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
}
|