jarvice

type HpcReq struct {
	JobEnvConfig string            `json:"hpc_job_env_config"`
	JobScript    string            `json:"hpc_job_script"`
	JobShell     string            `json:"hpc_job_shell"`
	Queue        string            `json:"hpc_queue"`
	Umask        int               `json:"hpc_umask"`
	Envs         map[string]string `json:"hpc_envs,omitempty"`
	Resources    map[string]string `json:"hpc_resources"`
}

type JarviceApiSubmission struct {
	Machine JarviceMachine `json:"machine"`
	Queue   string         `json:"queue"`
}

type JarviceApplication struct {
	Command    string                 `json:"command"`
	Walltime   string                 `json:"walltime,omitempty"`
	Geometry   string                 `json:"geometry"`
	Parameters map[string]interface{} `json:"parameters,omitempty"`
}

type JarviceCluster struct {
	Endpoint string       `json:"jarvice_endpoint"`
	Insecure bool         `json:"jarvice_insecure"`
	Vault    string       `json:"jarvice_vault"`
	Creds    JarviceCreds `json:"jarvice_user"`
}

type JarviceConfig map[string]JarviceCluster

type JarviceCreds struct {
	Username string `json:"username"`
	Apikey   string `json:"apikey"`
}

type JarviceJob struct {
	Label         string               `json:"job_label"`
	User          string               `json:"job_owner_username"`
	Status        string               `json:"job_status"`
	SubmitTime    int                  `json:"job_submit_time"`
	StartTime     int                  `json:"job_start_time"`
	EndTime       int                  `json:"job_end_time"`
	ExitCode      int                  `json:"job_exitcode"`
	App           string               `json:"job_application"`
	ApiSubmission JarviceApiSubmission `json:"job_api_submission"`
}

type JarviceJobRequest struct {
	App         string             `json:"app"`
	Staging     bool               `json:"staging"`
	Checkedout  bool               `json:"checkedout"`
	Application JarviceApplication `json:"application"`
	Machine     JarviceMachine     `json:"machine"`
	Vault       JarviceVault       `json:"vault"`
	JobLabel    string             `json:"job_label,omitempty"`
	User        JarviceCreds       `json:"user"`
	Hpc         HpcReq             `json:"hpc"`
	Licenses    *string            `json:"licenses,omitempty"`
	JobProject  *string            `json:"job_project,omitempty"`
}

type JarviceJobResponse struct {
	Name   string `json:"name"`
	Number int    `json:"number"`
}

type JarviceJobs = map[int]JarviceJob

type JarviceMachine struct {
	Type  string `json:"type"`
	Nodes int    `json:"nodes"`
}

type JarviceQueue struct {
	Name           string `json:"name"`
	App            string `json:"app"`
	DefaultMachine string `json:"machine"`
	MachineScale   int    `json:"size"`
}

type JarviceQueues = map[string]JarviceQueue

type JarviceVault struct {
	Name     string `json:"name"`
	ReadOnly bool   `json:"readonly"`
	Force    bool   `json:"force"`
}

type JobScript struct {
	Shell string `json:"hpc_shell"`
	// Args parsed from SBATCH directive
	Args   []string `json:"hpc_args"`
	Script []byte   `json:"hpc_script"`
}

type JobSpec struct {
	// Job environment variables
	SubmitDirectory string            `json:"hpc_submit_directory"`
	SubmitHost      string            `json:"hpc_submit_host"`
	QueueName       string            `json:"hpc_queue_name"`
	JobName         string            `json:"hpc_job_name"`
	UserEnv         map[string]string `json:"hpc_user_env"`

	Queue             string `json:"hpc_queue"`
	NodeCount         int    `json:"hpc_node_count"`
	CpuCount          int    `json:"hpc_cpu_count"`
	WallClockLimit    string `json:"hpc_wall_clock_limit"`
	OutputFile        string `json:"hpc_output_file"`
	ErrorFile         string `json:"hpc_error_file"`
	CopyEnvironment   string `json:"hpc_opy_environment"`
	EventNotification string `json:"hpc_event_notification"`
	EmailAddress      string `json:"hpc_email_address"`
	JobRestart        bool   `json:"hpc_job_restart"`
	WorkingDirectory  string `json:"hpc_working_directory"`
	Exclusive         bool   `json:"hpc_exclusive"`
	Memory            string `json:"hpc_memory"`
	ChargeAccount     string `json:"hpc_charge_account"`
	TasksPerNode      int    `json:"hpc_tasks_per_node"`
	CpusPerTask       int    `json:"hpc_cpus_per_task"`
	JobDependency     string `json:"hpc_job_dependency"`
	JobProject        string `json:"hpc_job_project"`
	GenericResources  string `json:"hpc_generic_resources"`
	Licenses          string `json:"hpc_licenses"`
	BeginTime         string `json:"hpc_begin_time"`
}