types.go

// Copyright 2022 Namespace Labs Inc; All rights reserved.
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.

package api

import (
	"fmt"
	"time"

	"google.golang.org/protobuf/types/known/timestamppb"
)

type CreateKubernetesClusterRequest struct {
	Ephemeral         bool                   `json:"ephemeral,omitempty"`
	DocumentedPurpose string                 `json:"documented_purpose,omitempty"`
	AuthorizedSshKeys []string               `json:"authorized_ssh_keys,omitempty"`
	MachineType       string                 `json:"machine_type,omitempty"`
	Feature           []string               `json:"feature,omitempty"`
	Interactive       bool                   `json:"interactive,omitempty"`
	Label             []*LabelEntry          `json:"label,omitempty"`
	UniqueTag         string                 `json:"unique_tag,omitempty"`
	InternalExtra     string                 `json:"internal_extra,omitempty"`
	Deadline          *timestamppb.Timestamp `json:"deadline,omitempty"`
	Experimental      any                    `json:"experimental,omitempty"`
	AvailableSecrets  []*SecretRef           `json:"available_secrets,omitempty"`
}

type SecretRef struct {
	SecretID string `json:"secret_id"`
}

type GetKubernetesClusterRequest struct {
	ClusterId string `json:"cluster_id,omitempty"`
}

type EnsureKubernetesClusterRequest struct {
	ClusterId string `json:"cluster_id,omitempty"`
}

type WaitKubernetesClusterRequest struct {
	ClusterId string `json:"cluster_id,omitempty"`
}

type CreateKubernetesClusterResponse struct {
	Status       string             `json:"status,omitempty"`
	ClusterId    string             `json:"cluster_id,omitempty"`
	Cluster      *KubernetesCluster `json:"cluster,omitempty"`
	Registry     *ImageRegistry     `json:"registry,omitempty"`
	BuildCluster *BuildCluster      `json:"build_cluster,omitempty"`
	Deadline     string             `json:"deadline,omitempty"`
}

type GetKubernetesClusterResponse struct {
	Cluster      *KubernetesCluster `json:"cluster,omitempty"`
	Registry     *ImageRegistry     `json:"registry,omitempty"`
	BuildCluster *BuildCluster      `json:"build_cluster,omitempty"`
	Deadline     string             `json:"deadline,omitempty"`
}

type StartCreateKubernetesClusterResponse struct {
	ClusterId       string             `json:"cluster_id,omitempty"`
	ClusterFragment *KubernetesCluster `json:"cluster_fragment,omitempty"`
	Registry        *ImageRegistry     `json:"registry,omitempty"`
	Deadline        string             `json:"deadline,omitempty"`
}

type CreateContainersRequest struct {
	MachineType     string                 `json:"machine_type,omitempty"`
	Container       []*ContainerRequest    `json:"container,omitempty"`
	Compose         []*ComposeRequest      `json:"compose,omitempty"`
	Interactive     bool                   `json:"interactive,omitempty"`
	DevelopmentMode bool                   `json:"development_mode,omitempty"`
	Feature         []string               `json:"feature,omitempty"`
	Label           []*LabelEntry          `json:"label,omitempty"`
	InternalExtra   string                 `json:"internal_extra,omitempty"`
	Deadline        *timestamppb.Timestamp `json:"deadline,omitempty"`
}

type StartContainersRequest struct {
	Id        string              `json:"id,omitempty"`
	Container []*ContainerRequest `json:"container,omitempty"`
}

type ContainerRequest struct {
	Name           string            `json:"name,omitempty"`
	Image          string            `json:"image,omitempty"`
	Args           []string          `json:"args,omitempty"`
	Env            map[string]string `json:"env,omitempty"`
	Flag           []string          `json:"flag,omitempty"`
	ExportPort     []*ContainerPort  `json:"export_port,omitempty"`
	DockerSockPath string            `json:"docker_sock_path,omitempty"`
	NscStatePath   string            `json:"nsc_state_path,omitempty"`
	Network        string            `json:"network,omitempty"`
}

type ComposeRequest struct {
	Contents []byte `json:"contents,omitempty"`
}

type ContainerPort struct {
	Proto         string                         `json:"proto,omitempty"`
	Port          int32                          `json:"port,omitempty"`
	HttpMatchRule []*ContainerPort_HttpMatchRule `json:"http_match_rule,omitempty"`
}

type ContainerPort_HttpMatchRule struct {
	Match              *ContainerPort_HttpMatch `json:"match,omitempty"`
	DoesNotRequireAuth bool                     `json:"does_not_require_auth,omitempty"`
}

type ContainerPort_HttpMatch struct {
	Method []string `json:"method,omitempty"` // If empty, matches any method.
	Path   string   `json:"path,omitempty"`   // If empty, matches any path.
}

type CreateContainersResponse struct {
	ClusterId  string       `json:"cluster_id,omitempty"`
	ClusterUrl string       `json:"cluster_url,omitempty"`
	Container  []*Container `json:"container,omitempty"`
}

type StartContainersResponse struct {
	Container []*Container `json:"container,omitempty"`
}

type Container struct {
	Id           string                             `json:"id,omitempty"`
	Name         string                             `json:"name,omitempty"`
	ExportedPort []*Container_ExportedContainerPort `json:"exported_port,omitempty"`
}

type Container_ExportedContainerPort struct {
	Proto         string                         `json:"proto,omitempty"`
	ContainerPort int32                          `json:"container_port,omitempty"`
	ExportedPort  int32                          `json:"exported_port,omitempty"` // Port exposed within the guest.
	IngressFqdn   string                         `json:"ingress_fqdn,omitempty"`
	HttpMatchRule []*ContainerPort_HttpMatchRule `json:"http_match_rule,omitempty"`
}

type ListKubernetesClustersRequest struct {
	IncludePreviousRuns bool                `json:"include_previous_runs,omitempty"`
	PaginationCursor    []byte              `json:"pagination_cursor,omitempty"`
	MaxEntries          int64               `json:"max_entries,omitempty"`
	NotOlderThan        *time.Time          `json:"not_older_than,omitempty"`
	KindFilter          string              `json:"kind_filter,omitempty"`
	LabelFilter         []*LabelFilterEntry `json:"label_filter,omitempty"`
}

type LabelFilterEntry struct {
	Name  string `json:"name,omitempty"`
	Value string `json:"value,omitempty"`
	Op    string `json:"op,omitempty"`
}

type ListKubernetesClustersResponse struct {
	Clusters []KubernetesClusterMetadata `json:"cluster"`
}

type KubernetesClusterMetadata struct {
	ClusterId         string        `json:"cluster_id,omitempty"`
	Created           string        `json:"created,omitempty"`
	DestroyedAt       string        `json:"destroyed_at,omitempty"`
	Deadline          string        `json:"deadline,omitempty"`
	DocumentedPurpose string        `json:"documented_purpose,omitempty"`
	Shape             *ClusterShape `json:"shape,omitempty"`

	KubernetesDistribution string   `json:"kubernetes_distribution,omitempty"`
	Platform               []string `json:"platform,omitempty"`

	IngressDomain string `json:"ingress_domain,omitempty"`

	Label     []*LabelEntry     `json:"label,omitempty"`
	UserLabel map[string]string `json:"user_label,omitempty" protobuf_key:"bytes,1,opt,name=key,proto3" protobuf_val:"bytes,2,opt,name=value,proto3"`

	CreatorId      string              `json:"creator_id,omitempty"`
	GithubWorkflow *GithubWorkflowInfo `json:"github_workflow,omitempty"`
}

type KubernetesCluster struct {
	AppURL            string        `json:"app_url,omitempty"`
	ClusterId         string        `json:"cluster_id,omitempty"`
	Created           string        `json:"created,omitempty"`
	DestroyedAt       string        `json:"destroyed_at,omitempty"`
	Deadline          string        `json:"deadline,omitempty"`
	SSHProxyEndpoint  string        `json:"ssh_proxy_endpoint,omitempty"`
	SshPrivateKey     []byte        `json:"ssh_private_key,omitempty"`
	DocumentedPurpose string        `json:"documented_purpose,omitempty"`
	Shape             *ClusterShape `json:"shape,omitempty"`

	EndpointAddress          string `json:"endpoint_address,omitempty"`
	CertificateAuthorityData []byte `json:"certificate_authority_data,omitempty"`
	ClientCertificateData    []byte `json:"client_certificate_data,omitempty"`
	ClientKeyData            []byte `json:"client_key_data,omitempty"`

	KubernetesDistribution string   `json:"kubernetes_distribution,omitempty"`
	Platform               []string `json:"platform,omitempty"`

	IngressDomain string `json:"ingress_domain,omitempty"`

	Label []*LabelEntry `json:"label,omitempty"`

	CreatorId      string              `json:"creator_id,omitempty"`
	GithubWorkflow *GithubWorkflowInfo `json:"github_workflow,omitempty"`

	ServiceState []*Cluster_ServiceState `json:"service_state,omitempty"`

	TlsBackedPort []*KubernetesCluster_TlsBackedPort `json:"tls_backed_port,omitempty"`
}

type Cluster_ServiceState struct {
	Name     string `json:"name,omitempty"`
	Status   string `json:"status,omitempty"`
	Endpoint string `json:"endpoint,omitempty"` // Service-specific endpoint.
	Public   bool   `json:"public,omitempty"`
}

type KubernetesCluster_TlsBackedPort struct {
	Name       string `json:"name,omitempty"`
	Port       int32  `json:"port,omitempty"`
	ServerName string `json:"server_name,omitempty"`
}

type GithubWorkflowInfo struct {
	Repository string `json:"repository,omitempty"`
	RunId      string `json:"run_id,omitempty"`
	RunAttempt string `json:"run_attempt,omitempty"`
	Sha        string `json:"sha,omitempty"`
	Ref        string `json:"ref,omitempty"`
	Actor      string `json:"actor,omitempty"`
	Workflow   string `json:"workflow,omitempty"`
}

type GetImageRegistryResponse struct {
	Registry *ImageRegistry `json:"registry,omitempty"`
	NSCR     *ImageRegistry `json:"nscr,omitempty"`
}

type TailLogsRequest struct {
	ClusterID      string          `json:"cluster_id,omitempty"`
	UseBlockLabels bool            `json:"use_block_labels,omitempty"`
	Include        []*LogsSelector `json:"include,omitempty"`
	Exclude        []*LogsSelector `json:"exclude,omitempty"`
}

type GetLogsRequest struct {
	ClusterID      string          `json:"cluster_id,omitempty"`
	UseBlockLabels bool            `json:"use_block_labels,omitempty"`
	StartTs        *time.Time      `json:"start_ts,omitempty"`
	EndTs          *time.Time      `json:"end_ts,omitempty"`
	Include        []*LogsSelector `json:"include,omitempty"`
	Exclude        []*LogsSelector `json:"exclude,omitempty"`
}

type GetLogsResponse struct {
	LogBlock []LogBlock `json:"log_block,omitempty"`
}

type LogsSelector struct {
	Source        string `json:"source,omitempty"`
	Namespace     string `json:"namespace,omitempty"`
	PodName       string `json:"pod_name,omitempty"`
	ContainerName string `json:"container_name,omitempty"`
	ContainerID   string `json:"container_id,omitempty"`
}

type LogBlock struct {
	Line   []LogLine         `json:"line,omitempty"`
	Labels map[string]string `json:"labels,omitempty"`
}

type LogLine struct {
	Timestamp time.Time `json:"timestamp,omitempty"`
	Content   string    `json:"content,omitempty"`
	Stream    string    `json:"stream,omitempty"`
}

func (l LogLine) String() string {
	return fmt.Sprintf("%s stream=%s msg=%s", l.Timestamp.Format(time.RFC3339), l.Stream, l.Content)
}

type ImageRegistry struct {
	EndpointAddress string `json:"endpoint_address,omitempty"`
	Repository      string `json:"repository,omitempty"`
}

type ClusterShape struct {
	VirtualCpu      int32  `json:"virtual_cpu,omitempty"`
	MemoryMegabytes int32  `json:"memory_megabytes,omitempty"`
	MachineArch     string `json:"machine_arch,omitempty"`
}

type DestroyKubernetesClusterRequest struct {
	ClusterId string `json:"cluster_id,omitempty"`
}

type SuspendKubernetesClusterRequest struct {
	ClusterId string `json:"cluster_id,omitempty"`
}

type ReleaseKubernetesClusterRequest struct {
	ClusterId string `json:"cluster_id,omitempty"`
}

type WakeKubernetesClusterRequest struct {
	ClusterId string `json:"cluster_id,omitempty"`
}

type LabelEntry struct {
	Name  string `json:"name,omitempty"`
	Value string `json:"value,omitempty"`
}

type BuildCluster struct {
	Colocated             *BuildCluster_ColocatedPort `json:"colocated,omitempty"`
	Resumable             bool                        `json:"resumable,omitempty"`
	DoesNotRequireRefresh bool                        `json:"does_not_require_refresh,omitempty"`
}

type BuildCluster_ColocatedPort struct {
	TargetPort int32  `json:"target_port,omitempty"`
	ClusterId  string `json:"cluster_id,omitempty"`
}

type Attachment struct {
	TypeURL string `json:"type_url,omitempty"`
	Content []byte `json:"content,omitempty"`
}

type RefreshKubernetesClusterRequest struct {
	ClusterId         string `json:"cluster_id,omitempty"`
	ExtendBySecs      int32  `json:"extend_by_secs,omitempty"`
	EnsureMinimumSecs int32  `json:"ensure_minimum_secs,omitempty"`
}

type RefreshKubernetesClusterResponse struct {
	NewDeadline string `json:"new_deadline,omitempty"`
}

type GetKubernetesClusterSummaryRequest struct {
	ClusterId string   `json:"cluster_id,omitempty"`
	Resource  []string `json:"resource,omitempty"`
}

type GetKubernetesClusterSummaryResponse struct {
	Summary []ResourceSummary `json:"summary,omitempty"`
}

type ResourceSummary struct {
	Resource    string              `json:"resource,omitempty"`
	PerResource map[string]Resource `json:"per_resource,omitempty"`
	GenVersion  int32               `json:"gen_version,omitempty"`
}

type Resource struct {
	Namespace    string                          `json:"namespace,omitempty"`
	Name         string                          `json:"name,omitempty"`
	Uid          string                          `json:"uid,omitempty"` // Union value - depends on resource type.
	Parent       []*ResourceSummary_Resource_Ref `json:"parent,omitempty"`
	CreationTime *time.Time                      `json:"creation_time,omitempty"`
	UpdatedTime  *time.Time                      `json:"updated_time,omitempty"`
	Tombstone    *time.Time                      `json:"tombstone,omitempty"` // If set, the resource has been removed.
	// For pods.
	Container []*ResourceSummary_Resource_Container `json:"container,omitempty"`
	Phase     string                                `json:"phase,omitempty"`
	// For apps.
	ReplicaCount       string `json:"replica_count,omitempty"`
	ReplicaUpdated     string `json:"replica_updated,omitempty"`
	ReplicaReady       string `json:"replica_ready,omitempty"`
	ReplicaAvailable   string `json:"replica_available,omitempty"`
	ObservedGeneration string `json:"observed_generation,omitempty"`
	// For ingress.
	HttpMatchRule      []*ContainerPort_HttpMatchRule `json:"http_match_rule,omitempty"`
	TargetExportedPort int32                          `json:"target_exported_port,omitempty"`
}

type ResourceSummary_Resource_Container struct {
	Id               string                             `json:"id,omitempty"`
	Name             string                             `json:"name,omitempty"`
	StartedAt        *time.Time                         `json:"started_at,omitempty"`
	TerminatedAt     *time.Time                         `json:"terminated_at,omitempty"`
	Ready            bool                               `json:"ready,omitempty"`
	RestartCount     int64                              `json:"restart_count,omitempty"`
	Status           string                             `json:"status,omitempty"`
	NotRunningReason string                             `json:"not_running_reason,omitempty"`
	ExportedPort     []*Container_ExportedContainerPort `json:"exported_port,omitempty"`
}

type ResourceSummary_Resource_Ref struct {
	Resource string `json:"resource,omitempty"`
	Name     string `json:"name,omitempty"`
	Uid      string `json:"uid,omitempty"`
}

type GetKubernetesConfigRequest struct {
	ClusterId string `json:"cluster_id,omitempty"`
}

type GetKubernetesConfigResponse struct {
	Kubeconfig string `json:"kubeconfig,omitempty"`
}

type GetProfileResponse struct {
	ClusterPlatform []string `json:"cluster_platform,omitempty"`
}

type RegisterIngressRequest struct {
	ClusterId       string                         `json:"cluster_id,omitempty"`
	Name            string                         `json:"name,omitempty"`
	BackendEndpoint *IngressBackendEndpoint        `json:"backend_endpoint,omitempty"`
	HttpMatchRule   []*ContainerPort_HttpMatchRule `json:"http_match_rule,omitempty"`
}

type IngressBackendEndpoint struct {
	IpAddr string `json:"ip_addr,omitempty"`
	Port   int32  `json:"port,omitempty"`
}

type RegisterIngressResponse struct {
	Fqdn string `json:"fqdn,omitempty"`
}

type ListIngressesRequest struct {
	ClusterId string `json:"cluster_id,omitempty"`
}

type ListIngressesResponse struct {
	ExportedInstancePort []*ExportedInstancePort `json:"exported_instance_port,omitempty"`
}

type ExportedInstancePort struct {
	Proto         string                         `json:"proto,omitempty"`
	Port          int32                          `json:"port,omitempty"`
	IngressFqdn   string                         `json:"ingress_fqdn,omitempty"`
	HttpMatchRule []*ContainerPort_HttpMatchRule `json:"http_match_rule,omitempty"`
	Description   string                         `json:"description,omitempty"`
}