cloudcompute

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

type BatchEvent struct {
	EventIdentifier   string          `json:"eventIdentifier"`
	ManifestOverrides ComputeManifest `json:"manifest"`
}

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

type CcJobStore interface {
	SaveJob(computeId uuid.UUID, payloadId uuid.UUID, event string, job *Job) error
}

type CcMessageQueue interface {
	SendMessage(channel string, message []byte) error
	Subscribe(channel string) (<-chan any, error) //amqp.Delivery
}

type CloudCompute struct {
	//Compute Identifier
	ID uuid.UUID `json:"id"`

	//User friendly Name for the compute
	Name string `json:"name"`

	//JobQueue to push the events to
	JobQueue string `json:"jobQueue"`

	//Event generator
	Events EventGenerator `json:"events"`

	//compute provider for the compute (typically AwsBatchProvider)
	ComputeProvider ComputeProvider `json:"computeProvider"`

	//Job store is for saving the set of jobs being sent to the compute environment to a
	//user defined storage location.  Typically something like a relational database
	JobStore CcJobStore

	//Function that will postprocess an event after it is generated by the event generator
	EventProcessor EventProcessor
}

type ComputeManifest struct {
	ManifestName         string                `json:"manifest_name,omitempty" jsonschema:"title=Manifest Name,description=The name for the compute manifest"`
	ManifestID           uuid.UUID             `json:"manifest_id,omitempty" jsonschema:"-"`
	Command              []string              `json:"command,omitempty" jsonschema:"title=Command Override,description=An optional command override for the plugin"`
	Dependencies         []uuid.UUID           `json:"dependencies,omitempty" jsonschema:"-"`
	Stores               []DataStore           `json:"stores,omitempty" jsonschema:"title=Stores"`
	Inputs               PluginInputs          `json:"inputs,omitempty" jsonschema:"title=Inputs"`
	Outputs              []DataSource          `json:"outputs,omitempty" jsonschema:"title=Outputs"`
	Actions              []Action              `json:"actions,omitempty" jsonschema:"title=Actions"`
	PluginDefinition     string                `json:"plugin_definition,omitempty" jsonschema:"title=Plugin Definition"` //plugin resource name. "name:version"
	Tags                 map[string]string     `json:"tags,omitempty" jsonschema:"title=Tags"`
	RetryAttemts         int32                 `json:"retry_attempts,omitempty" jsonschema:"title=Retry Attempts Override"`
	JobTimeout           int32                 `json:"job_timeout,omitempty" jsonschema:"title=Job Timeout Override"`
	ResourceRequirements []ResourceRequirement `json:"resource_requirements,omitempty" jsonschema:"title=Resource Requirement Overrides"`
	// contains filtered or unexported fields
}

type ComputeManifests []ComputeManifest

type ComputeProvider interface {
	SubmitJob(job *Job) error
	TerminateJobs(input TerminateJobInput) error
	Status(jobQueue string, query JobsSummaryQuery) error
	JobLog(submittedJobId string, token *string) (JobLogOutput, error)
	RegisterPlugin(plugin *Plugin) (PluginRegistrationOutput, error)
	UnregisterPlugin(nameAndRevision string) error
}

type ComputeProviderType int

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

type ContainerOverrides struct {
	Command              []string
	Environment          KeyValuePairs
	ResourceRequirements []ResourceRequirement
}

type Event struct {
	ID              uuid.UUID         `json:"id"`
	EventIdentifier string            `json:"event"` //RULES ONLY NUMBERS, STRINGS, DASH, AND UNDERSCORE
	Manifests       []ComputeManifest `json:"manifests"`

	AdditionalEventEnvVars KeyValuePairs
	// contains filtered or unexported fields
}

type EventGenerator interface {
	NextEvent() (Event, bool, error)
}

type EventGeneratorOld interface {
	HasNextEvent() bool
	NextEvent() (Event, error)
}

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

type EventProcessor interface {
	Process(event Event) (Event, error)
}

type Job struct {
	ID                 uuid.UUID
	EventID            uuid.UUID
	ManifestID         uuid.UUID
	JobName            string
	JobQueue           string
	JobDefinition      string
	ContainerOverrides ContainerOverrides
	DependsOn          []string //compute provider dependencies
	Parameters         map[string]string
	Tags               map[string]string
	RetryAttemts       int32
	JobTimeout         int32            //duration in seconds
	SubmittedJob       *SubmitJobResult //reference to the job information from the compute environment
}

type JobLogOutput struct {
	Logs  []string
	Token *string
}

type JobNameParts struct {
	Compute string
	Event   string
	Job     string
}

type JobSummary struct {
	//identifier for the compute environment being used.  e.g. AWS Batch Job ID
	JobId string

	//cloud compute job name
	JobName string

	//unix timestamp in milliseconds for when the job was created
	CreatedAt *int64

	//unix timestamp in milliseconds for when the job was started
	StartedAt *int64

	//status string value
	Status string

	//human readable string of the status
	StatusDetail *string

	//unix timestamp in milliseconds for when the job was stopped
	StoppedAt *int64

	//Compute Vendor resource name for the job.  e.g. the Job ARN for AWS
	ResourceName string
}

type JobSummaryFunction func(summaries []JobSummary)

type JobsLogQuery struct {
	QueryLevel       string
	QueryValue       JobNameParts
	MatchValue       string //regex to match and only print matching lines
	JsScriptFunction any
}

type JobsSummaryQuery struct {
	//The Level to request.  Must be one of three values
	//COMPUTE/EVENT/MANIFEST as represented by the SUMMARY_{level} constants
	QueryLevel string

	//the GUIDs representing the referenced levels
	//The value must include preceding levels, so COMPUTE level must have at least the compute GUID
	//EVENT level must have the compuyte and event levels and MANIFEST level has all three guids
	QueryValue JobNameParts

	//a required function to process each job returned in the query
	JobSummaryFunction JobSummaryFunction
}

type KeyValuePair struct {
	Name  string `json:"name"`
	Value string `json:"value"`
}

type KeyValuePairs []KeyValuePair

type LinuxDevice struct {
	HostPath      *string `json:"host_path" jsonschema:"title=Host Path"`
	ContainerPath *string `json:"container_path" jsonschema:"title=Container Path"`
}

type LinuxParameters struct {
	Devices []LinuxDevice `json:"devices" jsonschema:"title=Devices"`
}

type MountPoint struct {
	ContainerPath string
	ReadOnly      bool
	SourceVolume  string
}

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

type Plugin struct {
	Name string `json:"name" jsonschema:"title=Name"`
	//Revision           string                   `json:"revision" yaml:"revision"`
	ImageAndTag        string                   `json:"image_and_tag" jsonschema:"title=Image and Tag,description=The docker image and tag"`
	Description        string                   `json:"description" jsonschema:"title=Description"`
	Command            []string                 `json:"command" jsonschema:"title=Command,description=The docker command and arguments to run"`
	ComputeEnvironment PluginComputeEnvironment `json:"compute_environment" jsonschema:"title=Compute Environment,description=CPU and Memory runtime requirements"`
	DefaultEnvironment KeyValuePairs            `json:"environment" jsonschema:"title=Default Environment Variables,description=The list of default environment variables"` //default values for the container environment
	Volumes            []PluginComputeVolumes   ``                                                                                                                          //@NOTE: currently not used
	/* 128-byte string literal not displayed */
	Credentials      KeyValuePairs     ``                                                /* 213-byte string literal not displayed */
	Parameters       map[string]string `json:"parameters" jsonschema:"title=Parameters"` //@NOTE: currently not being used
	RetryAttemts     int32             `json:"retry_attempts" jsonschema:"title=Retry Attempts"`
	ExecutionTimeout *int32            `json:"execution_timeout" jsonschema:"title=Execution Timeout (sec)"`
	Privileged       bool              `json:"privileged" jsonschema:"title=Requires Privileged Execution"` //assign container privileged execution.  for example to mount linux devices
	LinuxParameters  LinuxParameters   `json:"linux_parameters" jsonschema:"title=Linux Parameters"`
	MountPoints      []MountPoint      `json:"mountpoints" jsonschema:"title=MountPoints"` //@NOTE: currently not being used
}

type PluginComputeEnvironment struct {
	VCPU       string   `json:"vcpu" jsonschema:"title=Virtual CPUs"`
	Memory     string   `json:"memory" jsonschema:"title=Memory in MB"`
	ExtraHosts []string `json:"extraHosts" jsonschema:"title=Extra Hosts for the Docker API"`
}

type PluginComputeVolumes struct {
	Name         string `json:"name" jsonschema:"title=Name"`
	ResourceName string `json:"resource_name" jsonschema:"title=ResourceName,description=The vendor resource name, e.g. ARN for AWS"`
	ReadOnly     bool   `json:"read_only" jsonschema:"title=Read Only,description=Check to make this resource read-only"`
	MountPoint   string `json:"mount_point" jsonschema:"title=Mount Point,description=Path in the running container to mount the volume"` //default is "/data"
}

type PluginInputs struct {
	Environment       KeyValuePairs     `json:"environment,omitempty"`
	Parameters        map[string]string `json:"parameters,omitempty"`
	DataSources       []DataSource      `json:"data_sources,omitempty"`
	PayloadAttributes PayloadAttributes `json:"payload_attributes,omitempty"`
}

type PluginRegistrationOutput struct {
	Name         string `json:"name"`
	ResourceName string `json:"resourceName"`
	Revision     int32  `json:"revision"`
}

type ResourceRequirement struct {
	Type  ResourceType `json:"resource_type"`
	Value string       `json:"value"`
}

type ResourceType string

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

type SubmitJobResult struct {

	//Vendor ID
	JobId *string

	//Vendor Resource Name
	ResourceName *string //ARN in AWS
}

type TerminateJobFunction func(output TerminateJobOutput)

type TerminateJobInput struct {
	//Users reason for terminating the job
	Reason string

	//The Vendor Job Queue the job was submitted to
	JobQueue string

	//Optional. A jobs summary query that will generate a list of jobs to terminate
	//if this value is provided the JobSummaryQuery JobSummaryFunction is ignored
	//and should be left empty
	Query JobsSummaryQuery

	//Optional. A list of VendorJobs to terminate
	VendorJobs VendorJobs

	//Optional.  A function to process the results of each terminated job
	TerminateJobFunction TerminateJobFunction
}

type TerminateJobOutput struct {

	//CloudCompute Job Name
	JobName string

	//Error if returned from the terminate operation
	Err error

	//Vendor Job ID
	JobId string
}

type TopoSortable[T comparable] interface {
	Node() T
	Deps() []T
}

type VendorJob interface {
	ID() string
	Name() string
}

type VendorJobs []VendorJob