store

type APIKey struct {
	gorm.Model

	APIKeyID string `gorm:"uniqueIndex"`

	Name string `gorm:"uniqueIndex:idx_api_key_name_user_id"`

	TenantID string

	OrganizationID string
	ProjectID      string
	UserID         string `gorm:"uniqueIndex:idx_api_key_name_user_id"`

	IsServiceAccount bool
	// ExcludedFromRateLimiting indicates whether this API key is excluded from rate limiting
	ExcludedFromRateLimiting bool

	// Secret is set when kms encryption is disabled.
	Secret string
	// EncryptedSecret is encrypted by data key, and it is set when kms encryption is enabled.
	EncryptedSecret []byte
}

type APIKeySpec struct {
	APIKeyID       string
	TenantID       string
	OrganizationID string
	ProjectID      string
	UserID         string

	IsServiceAccount bool
	// ExcludedFromRateLimiting indicates whether this API key is excluded from rate limiting
	ExcludedFromRateLimiting bool

	Name   string
	Secret string
	// EncryptedSecret is encrypted by data key.
	EncryptedSecret []byte
}

type CreateProjectParams struct {
	ProjectID      string
	OrganizationID string
	TenantID       string
	Title          string
	Assignments    []*v1.ProjectAssignment
	IsDefault      bool
}

type CreateProjectUserParams struct {
	ProjectID      string
	OrganizationID string
	UserID         string
	Role           v1.ProjectRole
}

type DataKey struct {
	gorm.Model

	EncryptedDataKey []byte
}

type DataKeyManagementClient interface {
	CreateDataKey(ctx context.Context) ([]byte, []byte, error)
	DecryptDataKey(ctx context.Context, encryptedKey []byte) ([]byte, error)
}

type GetProjectParams struct {
	TenantID       string
	OrganizationID string
	ProjectID      string
}

type Organization struct {
	gorm.Model

	OrganizationID string `gorm:"uniqueIndex"`

	TenantID string `gorm:"uniqueIndex:idx_orgs_tenant_id_title"`
	Title    string `gorm:"uniqueIndex:idx_orgs_tenant_id_title"`

	IsDefault bool
}

type OrganizationUser struct {
	gorm.Model

	OrganizationID string `gorm:"uniqueIndex:user_id_org_id"`
	UserID         string `gorm:"uniqueIndex:user_id_org_id"`

	Role string

	// Hidden is set to true if the user is not visible from the list/get API call.
	Hidden bool
}

type Project struct {
	gorm.Model

	ProjectID      string `gorm:"uniqueIndex"`
	OrganizationID string

	TenantID string `gorm:"uniqueIndex:idx_projects_tenant_id_title"`
	Title    string `gorm:"uniqueIndex:idx_projects_tenant_id_title"`

	IsDefault bool

	// KubernetesNamespace is the namespace where the fine-tuning jobs for the organization run.
	// TODO(kenji): Currently we don't set the unique constraint so that multiple orgs can use the same namespace,
	// but revisit the design.
	KubernetesNamespace string

	// Assignments is the assignments of the project. It is a marshaled v1.ProjectAssignments.
	Assignments []byte
}

type ProjectUser struct {
	gorm.Model

	ProjectID      string `gorm:"uniqueIndex:user_id_project_id"`
	OrganizationID string
	UserID         string `gorm:"uniqueIndex:user_id_project_id"`

	Role string

	// Hidden is set to true if the user is not visible from the list/get API call.
	Hidden bool
}

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

type User struct {
	gorm.Model

	UserID         string `gorm:"uniqueIndex"`
	InternalUserID string `gorm:"uniqueIndex"`

	Hidden bool
}