postgres

type Branch struct {
	// A timestamp indicating when the branch was created.
	CreateTime *time.Time `json:"create_time,omitempty"`
	// The resource name of the branch. This field is output-only and
	// constructed by the system. Format:
	// `projects/{project_id}/branches/{branch_id}`
	Name string `json:"name,omitempty"`
	// The project containing this branch (API resource hierarchy). Format:
	// projects/{project_id}
	//
	// Note: This field indicates where the branch exists in the resource
	// hierarchy. For point-in-time branching from another branch, see
	// `spec.source_branch`.
	Parent string `json:"parent,omitempty"`
	// The spec contains the branch configuration.
	Spec *BranchSpec `json:"spec,omitempty"`
	// The current status of a Branch.
	Status *BranchStatus `json:"status,omitempty"`
	// System-generated unique ID for the branch.
	Uid string `json:"uid,omitempty"`
	// A timestamp indicating when the branch was last updated.
	UpdateTime *time.Time `json:"update_time,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type BranchOperationMetadata struct {
}

type BranchSpec struct {
	// Absolute expiration timestamp. When set, the branch will expire at this
	// time.
	ExpireTime *time.Time `json:"expire_time,omitempty"`
	// When set to true, protects the branch from deletion and reset. Associated
	// compute endpoints and the project cannot be deleted while the branch is
	// protected.
	IsProtected bool `json:"is_protected,omitempty"`
	// Explicitly disable expiration. When set to true, the branch will not
	// expire. If set to false, the request is invalid; provide either ttl or
	// expire_time instead.
	NoExpiry bool `json:"no_expiry,omitempty"`
	// The name of the source branch from which this branch was created (data
	// lineage for point-in-time recovery). If not specified, defaults to the
	// project's default branch. Format:
	// projects/{project_id}/branches/{branch_id}
	SourceBranch string `json:"source_branch,omitempty"`
	// The Log Sequence Number (LSN) on the source branch from which this branch
	// was created.
	SourceBranchLsn string `json:"source_branch_lsn,omitempty"`
	// The point in time on the source branch from which this branch was
	// created.
	SourceBranchTime *time.Time `json:"source_branch_time,omitempty"`
	// Relative time-to-live duration. When set, the branch will expire at
	// creation_time + ttl.
	Ttl *duration.Duration `json:"ttl,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type BranchStatus struct {
	// The branch's state, indicating if it is initializing, ready for use, or
	// archived.
	CurrentState BranchStatusState `json:"current_state,omitempty"`
	// Whether the branch is the project's default branch.
	Default bool `json:"default,omitempty"`
	// Absolute expiration time for the branch. Empty if expiration is disabled.
	ExpireTime *time.Time `json:"expire_time,omitempty"`
	// Whether the branch is protected.
	IsProtected bool `json:"is_protected,omitempty"`
	// The logical size of the branch.
	LogicalSizeBytes int64 `json:"logical_size_bytes,omitempty"`
	// The pending state of the branch, if a state transition is in progress.
	PendingState BranchStatusState `json:"pending_state,omitempty"`
	// The name of the source branch from which this branch was created. Format:
	// projects/{project_id}/branches/{branch_id}
	SourceBranch string `json:"source_branch,omitempty"`
	// The Log Sequence Number (LSN) on the source branch from which this branch
	// was created.
	SourceBranchLsn string `json:"source_branch_lsn,omitempty"`
	// The point in time on the source branch from which this branch was
	// created.
	SourceBranchTime *time.Time `json:"source_branch_time,omitempty"`
	// A timestamp indicating when the `current_state` began.
	StateChangeTime *time.Time `json:"state_change_time,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type BranchStatusState string

type CreateBranchOperationInterface interface {

	// Wait blocks until the long-running operation is completed. If no timeout is
	// specified, this will poll indefinitely. If a timeout is provided and the operation
	// didn't finish within the timeout, this function will return an error, otherwise
	// returns successful response and any errors encountered.
	Wait(ctx context.Context, opts ...api.Option) (*Branch, error)

	// Name returns the name of the long-running operation. The name is assigned
	// by the server and is unique within the service from which the operation is created.
	Name() string

	// Metadata returns metadata associated with the long-running operation.
	// If the metadata is not available, the returned metadata and error are both nil.
	Metadata() (*BranchOperationMetadata, error)

	// Done reports whether the long-running operation has completed.
	Done() (bool, error)
}

type CreateBranchRequest struct {
	// The Branch to create.
	Branch Branch `json:"branch"`
	// The ID to use for the Branch. This becomes the final component of the
	// branch's resource name. The ID must be 1-63 characters long, start with a
	// lowercase letter, and contain only lowercase letters, numbers, and
	// hyphens (RFC 1123). Examples: - With custom ID: `staging` → name
	// becomes `projects/{project_id}/branches/staging` - Without custom ID:
	// system generates slug → name becomes
	// `projects/{project_id}/branches/br-example-name-x1y2z3a4`
	BranchId string `json:"-" url:"branch_id"`
	// The Project where this Branch will be created. Format:
	// projects/{project_id}
	Parent string `json:"-" url:"-"`
}

type CreateEndpointOperationInterface interface {

	// Wait blocks until the long-running operation is completed. If no timeout is
	// specified, this will poll indefinitely. If a timeout is provided and the operation
	// didn't finish within the timeout, this function will return an error, otherwise
	// returns successful response and any errors encountered.
	Wait(ctx context.Context, opts ...api.Option) (*Endpoint, error)

	// Name returns the name of the long-running operation. The name is assigned
	// by the server and is unique within the service from which the operation is created.
	Name() string

	// Metadata returns metadata associated with the long-running operation.
	// If the metadata is not available, the returned metadata and error are both nil.
	Metadata() (*EndpointOperationMetadata, error)

	// Done reports whether the long-running operation has completed.
	Done() (bool, error)
}

type CreateEndpointRequest struct {
	// The Endpoint to create.
	Endpoint Endpoint `json:"endpoint"`
	// The ID to use for the Endpoint. This becomes the final component of the
	// endpoint's resource name. The ID must be 1-63 characters long, start with
	// a lowercase letter, and contain only lowercase letters, numbers, and
	// hyphens (RFC 1123). Examples: - With custom ID: `primary` → name
	// becomes `projects/{project_id}/branches/{branch_id}/endpoints/primary` -
	// Without custom ID: system generates slug → name becomes
	// `projects/{project_id}/branches/{branch_id}/endpoints/ep-example-name-x1y2z3a4`
	EndpointId string `json:"-" url:"endpoint_id"`
	// The Branch where this Endpoint will be created. Format:
	// projects/{project_id}/branches/{branch_id}
	Parent string `json:"-" url:"-"`
}

type CreateProjectOperationInterface interface {

	// Wait blocks until the long-running operation is completed. If no timeout is
	// specified, this will poll indefinitely. If a timeout is provided and the operation
	// didn't finish within the timeout, this function will return an error, otherwise
	// returns successful response and any errors encountered.
	Wait(ctx context.Context, opts ...api.Option) (*Project, error)

	// Name returns the name of the long-running operation. The name is assigned
	// by the server and is unique within the service from which the operation is created.
	Name() string

	// Metadata returns metadata associated with the long-running operation.
	// If the metadata is not available, the returned metadata and error are both nil.
	Metadata() (*ProjectOperationMetadata, error)

	// Done reports whether the long-running operation has completed.
	Done() (bool, error)
}

type CreateProjectRequest struct {
	// The Project to create.
	Project Project `json:"project"`
	// The ID to use for the Project. This becomes the final component of the
	// project's resource name. The ID must be 1-63 characters long, start with
	// a lowercase letter, and contain only lowercase letters, numbers, and
	// hyphens (RFC 1123). Examples: - With custom ID: `production` → name
	// becomes `projects/production` - Without custom ID: system generates UUID
	// → name becomes `projects/a7f89b2c-3d4e-5f6g-7h8i-9j0k1l2m3n4o`
	ProjectId string `json:"-" url:"project_id"`
}

type CreateRoleOperationInterface interface {

	// Wait blocks until the long-running operation is completed. If no timeout is
	// specified, this will poll indefinitely. If a timeout is provided and the operation
	// didn't finish within the timeout, this function will return an error, otherwise
	// returns successful response and any errors encountered.
	Wait(ctx context.Context, opts ...api.Option) (*Role, error)

	// Name returns the name of the long-running operation. The name is assigned
	// by the server and is unique within the service from which the operation is created.
	Name() string

	// Metadata returns metadata associated with the long-running operation.
	// If the metadata is not available, the returned metadata and error are both nil.
	Metadata() (*RoleOperationMetadata, error)

	// Done reports whether the long-running operation has completed.
	Done() (bool, error)
}

type CreateRoleRequest struct {
	// The Branch where this Role is created. Format:
	// projects/{project_id}/branches/{branch_id}
	Parent string `json:"-" url:"-"`
	// The desired specification of a Role.
	Role Role `json:"role"`
	// The ID to use for the Role, which will become the final component of the
	// role's resource name. This ID becomes the role in Postgres.
	//
	// This value should be 4-63 characters, and valid characters are lowercase
	// letters, numbers, and hyphens, as defined by RFC 1123.
	RoleId string `json:"-" url:"role_id"`
}

type DatabaseCredential struct {
	// Timestamp in UTC of when this credential expires.
	ExpireTime *time.Time `json:"expire_time,omitempty"`
	// The OAuth token that can be used as a password when connecting to a
	// database.
	Token string `json:"token,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type DatabricksServiceExceptionWithDetailsProto struct {
	Details []json.RawMessage `json:"details,omitempty"`

	ErrorCode ErrorCode `json:"error_code,omitempty"`

	Message string `json:"message,omitempty"`

	StackTrace string `json:"stack_trace,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type DeleteBranchOperationInterface interface {

	// Wait blocks until the long-running operation is completed. If no timeout is
	// specified, this will poll indefinitely. If a timeout is provided and the operation
	// didn't finish within the timeout, this function will return an error, otherwise
	// returns successful response and any errors encountered.
	Wait(ctx context.Context, opts ...api.Option) error

	// Name returns the name of the long-running operation. The name is assigned
	// by the server and is unique within the service from which the operation is created.
	Name() string

	// Metadata returns metadata associated with the long-running operation.
	// If the metadata is not available, the returned metadata and error are both nil.
	Metadata() (*BranchOperationMetadata, error)

	// Done reports whether the long-running operation has completed.
	Done() (bool, error)
}

type DeleteBranchRequest struct {
	// The name of the Branch to delete. Format:
	// projects/{project_id}/branches/{branch_id}
	Name string `json:"-" url:"-"`
}

type DeleteEndpointOperationInterface interface {

	// Wait blocks until the long-running operation is completed. If no timeout is
	// specified, this will poll indefinitely. If a timeout is provided and the operation
	// didn't finish within the timeout, this function will return an error, otherwise
	// returns successful response and any errors encountered.
	Wait(ctx context.Context, opts ...api.Option) error

	// Name returns the name of the long-running operation. The name is assigned
	// by the server and is unique within the service from which the operation is created.
	Name() string

	// Metadata returns metadata associated with the long-running operation.
	// If the metadata is not available, the returned metadata and error are both nil.
	Metadata() (*EndpointOperationMetadata, error)

	// Done reports whether the long-running operation has completed.
	Done() (bool, error)
}

type DeleteEndpointRequest struct {
	// The name of the Endpoint to delete. Format:
	// projects/{project_id}/branches/{branch_id}/endpoints/{endpoint_id}
	Name string `json:"-" url:"-"`
}

type DeleteProjectOperationInterface interface {

	// Wait blocks until the long-running operation is completed. If no timeout is
	// specified, this will poll indefinitely. If a timeout is provided and the operation
	// didn't finish within the timeout, this function will return an error, otherwise
	// returns successful response and any errors encountered.
	Wait(ctx context.Context, opts ...api.Option) error

	// Name returns the name of the long-running operation. The name is assigned
	// by the server and is unique within the service from which the operation is created.
	Name() string

	// Metadata returns metadata associated with the long-running operation.
	// If the metadata is not available, the returned metadata and error are both nil.
	Metadata() (*ProjectOperationMetadata, error)

	// Done reports whether the long-running operation has completed.
	Done() (bool, error)
}

type DeleteProjectRequest struct {
	// The name of the Project to delete. Format: projects/{project_id}
	Name string `json:"-" url:"-"`
}

type DeleteRoleOperationInterface interface {

	// Wait blocks until the long-running operation is completed. If no timeout is
	// specified, this will poll indefinitely. If a timeout is provided and the operation
	// didn't finish within the timeout, this function will return an error, otherwise
	// returns successful response and any errors encountered.
	Wait(ctx context.Context, opts ...api.Option) error

	// Name returns the name of the long-running operation. The name is assigned
	// by the server and is unique within the service from which the operation is created.
	Name() string

	// Metadata returns metadata associated with the long-running operation.
	// If the metadata is not available, the returned metadata and error are both nil.
	Metadata() (*RoleOperationMetadata, error)

	// Done reports whether the long-running operation has completed.
	Done() (bool, error)
}

type DeleteRoleRequest struct {
	// The resource name of the postgres role. Format:
	// projects/{project_id}/branches/{branch_id}/roles/{role_id}
	Name string `json:"-" url:"-"`
	// Reassign objects. If this is set, all objects owned by the role are
	// reassigned to the role specified in this parameter.
	//
	// NOTE: setting this requires spinning up a compute to succeed, since it
	// involves running SQL queries.
	//
	// TODO: #LKB-7187 implement reassign_owned_to on LBM side. This might
	// end-up being a synchronous query when this parameter is used.
	ReassignOwnedTo string `json:"-" url:"reassign_owned_to,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type Endpoint struct {
	// A timestamp indicating when the compute endpoint was created.
	CreateTime *time.Time `json:"create_time,omitempty"`
	// The resource name of the endpoint. This field is output-only and
	// constructed by the system. Format:
	// `projects/{project_id}/branches/{branch_id}/endpoints/{endpoint_id}`
	Name string `json:"name,omitempty"`
	// The branch containing this endpoint (API resource hierarchy). Format:
	// projects/{project_id}/branches/{branch_id}
	Parent string `json:"parent,omitempty"`
	// The spec contains the compute endpoint configuration, including
	// autoscaling limits, suspend timeout, and disabled state.
	Spec *EndpointSpec `json:"spec,omitempty"`
	// Current operational status of the compute endpoint.
	Status *EndpointStatus `json:"status,omitempty"`
	// System-generated unique ID for the endpoint.
	Uid string `json:"uid,omitempty"`
	// A timestamp indicating when the compute endpoint was last updated.
	UpdateTime *time.Time `json:"update_time,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type EndpointHosts struct {
	// The hostname to connect to this endpoint. For read-write endpoints, this
	// is a read-write hostname which connects to the primary compute. For
	// read-only endpoints, this is a read-only hostname which allows read-only
	// operations.
	Host string `json:"host,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type EndpointOperationMetadata struct {
}

type EndpointSettings struct {
	// A raw representation of Postgres settings.
	PgSettings map[string]string `json:"pg_settings,omitempty"`
}

type EndpointSpec struct {
	// The maximum number of Compute Units. Minimum value is 0.5.
	AutoscalingLimitMaxCu float64 `json:"autoscaling_limit_max_cu,omitempty"`
	// The minimum number of Compute Units. Minimum value is 0.5.
	AutoscalingLimitMinCu float64 `json:"autoscaling_limit_min_cu,omitempty"`
	// Whether to restrict connections to the compute endpoint. Enabling this
	// option schedules a suspend compute operation. A disabled compute endpoint
	// cannot be enabled by a connection or console action.
	Disabled bool `json:"disabled,omitempty"`
	// The endpoint type. A branch can only have one READ_WRITE endpoint.
	EndpointType EndpointType `json:"endpoint_type"`

	Settings *EndpointSettings `json:"settings,omitempty"`
	// Duration of inactivity after which the compute endpoint is automatically
	// suspended. Supported values: -1s (never suspend), 0s (use default), or
	// value should be between 60s and 604800s (1 minute to 1 week).
	SuspendTimeoutDuration *duration.Duration `json:"suspend_timeout_duration,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type EndpointStatus struct {
	// The maximum number of Compute Units.
	AutoscalingLimitMaxCu float64 `json:"autoscaling_limit_max_cu,omitempty"`
	// The minimum number of Compute Units.
	AutoscalingLimitMinCu float64 `json:"autoscaling_limit_min_cu,omitempty"`

	CurrentState EndpointStatusState `json:"current_state,omitempty"`
	// Whether to restrict connections to the compute endpoint. Enabling this
	// option schedules a suspend compute operation. A disabled compute endpoint
	// cannot be enabled by a connection or console action.
	Disabled bool `json:"disabled,omitempty"`
	// The endpoint type. A branch can only have one READ_WRITE endpoint.
	EndpointType EndpointType `json:"endpoint_type,omitempty"`
	// Contains host information for connecting to the endpoint.
	Hosts *EndpointHosts `json:"hosts,omitempty"`

	PendingState EndpointStatusState `json:"pending_state,omitempty"`

	Settings *EndpointSettings `json:"settings,omitempty"`
	// Duration of inactivity after which the compute endpoint is automatically
	// suspended.
	SuspendTimeoutDuration *duration.Duration `json:"suspend_timeout_duration,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type EndpointStatusState string

type EndpointType string

type ErrorCode string

type GenerateDatabaseCredentialRequest struct {
	// The returned token will be scoped to UC tables with the specified
	// permissions.
	Claims []RequestedClaims `json:"claims,omitempty"`
	// This field is not yet supported. The endpoint for which this credential
	// will be generated. Format:
	// projects/{project_id}/branches/{branch_id}/endpoints/{endpoint_id}
	Endpoint string `json:"endpoint"`
}

type GetBranchRequest struct {
	// The resource name of the branch to retrieve. Format:
	// `projects/{project_id}/branches/{branch_id}`
	Name string `json:"-" url:"-"`
}

type GetEndpointRequest struct {
	// The resource name of the endpoint to retrieve. Format:
	// `projects/{project_id}/branches/{branch_id}/endpoints/{endpoint_id}`
	Name string `json:"-" url:"-"`
}

type GetOperationRequest struct {
	// The name of the operation resource.
	Name string `json:"-" url:"-"`
}

type GetProjectRequest struct {
	// The resource name of the project to retrieve. Format:
	// `projects/{project_id}`
	Name string `json:"-" url:"-"`
}

type GetRoleRequest struct {
	// The name of the Role to retrieve. Format:
	// projects/{project_id}/branches/{branch_id}/roles/{role_id}
	Name string `json:"-" url:"-"`
}

type ListBranchesRequest struct {
	// Upper bound for items returned. Cannot be negative.
	PageSize int `json:"-" url:"page_size,omitempty"`
	// Page token from a previous response. If not provided, returns the first
	// page.
	PageToken string `json:"-" url:"page_token,omitempty"`
	// The Project that owns this collection of branches. Format:
	// projects/{project_id}
	Parent string `json:"-" url:"-"`

	ForceSendFields []string `json:"-" url:"-"`
}

type ListBranchesResponse struct {
	// List of database branches in the project.
	Branches []Branch `json:"branches,omitempty"`
	// Token to request the next page of database branches.
	NextPageToken string `json:"next_page_token,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type ListEndpointsRequest struct {
	// Upper bound for items returned. Cannot be negative.
	PageSize int `json:"-" url:"page_size,omitempty"`
	// Page token from a previous response. If not provided, returns the first
	// page.
	PageToken string `json:"-" url:"page_token,omitempty"`
	// The Branch that owns this collection of endpoints. Format:
	// projects/{project_id}/branches/{branch_id}
	Parent string `json:"-" url:"-"`

	ForceSendFields []string `json:"-" url:"-"`
}

type ListEndpointsResponse struct {
	// List of compute endpoints in the branch.
	Endpoints []Endpoint `json:"endpoints,omitempty"`
	// Token to request the next page of compute endpoints.
	NextPageToken string `json:"next_page_token,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type ListProjectsRequest struct {
	// Upper bound for items returned. Cannot be negative.
	PageSize int `json:"-" url:"page_size,omitempty"`
	// Page token from a previous response. If not provided, returns the first
	// page.
	PageToken string `json:"-" url:"page_token,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type ListProjectsResponse struct {
	// Token to request the next page of database projects.
	NextPageToken string `json:"next_page_token,omitempty"`
	// List of all database projects in the workspace that the user has
	// permission to access.
	Projects []Project `json:"projects,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type ListRolesRequest struct {
	// Upper bound for items returned. Cannot be negative.
	PageSize int `json:"-" url:"page_size,omitempty"`
	// Page token from a previous response. If not provided, returns the first
	// page.
	PageToken string `json:"-" url:"page_token,omitempty"`
	// The Branch that owns this collection of roles. Format:
	// projects/{project_id}/branches/{branch_id}
	Parent string `json:"-" url:"-"`

	ForceSendFields []string `json:"-" url:"-"`
}

type ListRolesResponse struct {
	// Token to request the next page of Postgres roles.
	NextPageToken string `json:"next_page_token,omitempty"`
	// List of Postgres roles in the branch.
	Roles []Role `json:"roles,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type Operation struct {
	// If the value is `false`, it means the operation is still in progress. If
	// `true`, the operation is completed, and either `error` or `response` is
	// available.
	Done bool `json:"done,omitempty"`
	// The error result of the operation in case of failure or cancellation.
	Error *DatabricksServiceExceptionWithDetailsProto `json:"error,omitempty"`
	// Service-specific metadata associated with the operation. It typically
	// contains progress information and common metadata such as create time.
	// Some services might not provide such metadata.
	Metadata json.RawMessage `json:"metadata,omitempty"`
	// The server-assigned name, which is only unique within the same service
	// that originally returns it. If you use the default HTTP mapping, the
	// `name` should be a resource name ending with `operations/{unique_id}`.
	Name string `json:"name,omitempty"`
	// The normal, successful response of the operation.
	Response json.RawMessage `json:"response,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

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

type PostgresInterface interface {
	CreateBranch(ctx context.Context, request CreateBranchRequest) (CreateBranchOperationInterface, error)

	CreateEndpoint(ctx context.Context, request CreateEndpointRequest) (CreateEndpointOperationInterface, error)

	CreateProject(ctx context.Context, request CreateProjectRequest) (CreateProjectOperationInterface, error)

	CreateRole(ctx context.Context, request CreateRoleRequest) (CreateRoleOperationInterface, error)

	DeleteBranch(ctx context.Context, request DeleteBranchRequest) (DeleteBranchOperationInterface, error)

	DeleteEndpoint(ctx context.Context, request DeleteEndpointRequest) (DeleteEndpointOperationInterface, error)

	DeleteProject(ctx context.Context, request DeleteProjectRequest) (DeleteProjectOperationInterface, error)

	DeleteRole(ctx context.Context, request DeleteRoleRequest) (DeleteRoleOperationInterface, error)

	// Generate OAuth credentials for a Postgres database.
	GenerateDatabaseCredential(ctx context.Context, request GenerateDatabaseCredentialRequest) (*DatabaseCredential, error)

	// Retrieves information about the specified database branch.
	GetBranch(ctx context.Context, request GetBranchRequest) (*Branch, error)

	// Retrieves information about the specified compute endpoint, including its
	// connection details and operational state.
	GetEndpoint(ctx context.Context, request GetEndpointRequest) (*Endpoint, error)

	// Retrieves the status of a long-running operation.
	GetOperation(ctx context.Context, request GetOperationRequest) (*Operation, error)

	// Retrieves information about the specified database project.
	GetProject(ctx context.Context, request GetProjectRequest) (*Project, error)

	// Retrieves information about the specified Postgres role, including its
	// authentication method and permissions.
	GetRole(ctx context.Context, request GetRoleRequest) (*Role, error)

	// Returns a paginated list of database branches in the project.
	//
	// This method is generated by Databricks SDK Code Generator.
	ListBranches(ctx context.Context, request ListBranchesRequest) listing.Iterator[Branch]

	// Returns a paginated list of database branches in the project.
	//
	// This method is generated by Databricks SDK Code Generator.
	ListBranchesAll(ctx context.Context, request ListBranchesRequest) ([]Branch, error)

	// Returns a paginated list of compute endpoints in the branch.
	//
	// This method is generated by Databricks SDK Code Generator.
	ListEndpoints(ctx context.Context, request ListEndpointsRequest) listing.Iterator[Endpoint]

	// Returns a paginated list of compute endpoints in the branch.
	//
	// This method is generated by Databricks SDK Code Generator.
	ListEndpointsAll(ctx context.Context, request ListEndpointsRequest) ([]Endpoint, error)

	// Returns a paginated list of database projects in the workspace that the user
	// has permission to access.
	//
	// This method is generated by Databricks SDK Code Generator.
	ListProjects(ctx context.Context, request ListProjectsRequest) listing.Iterator[Project]

	// Returns a paginated list of database projects in the workspace that the user
	// has permission to access.
	//
	// This method is generated by Databricks SDK Code Generator.
	ListProjectsAll(ctx context.Context, request ListProjectsRequest) ([]Project, error)

	// Returns a paginated list of Postgres roles in the branch.
	//
	// This method is generated by Databricks SDK Code Generator.
	ListRoles(ctx context.Context, request ListRolesRequest) listing.Iterator[Role]

	// Returns a paginated list of Postgres roles in the branch.
	//
	// This method is generated by Databricks SDK Code Generator.
	ListRolesAll(ctx context.Context, request ListRolesRequest) ([]Role, error)

	UpdateBranch(ctx context.Context, request UpdateBranchRequest) (UpdateBranchOperationInterface, error)

	UpdateEndpoint(ctx context.Context, request UpdateEndpointRequest) (UpdateEndpointOperationInterface, error)

	UpdateProject(ctx context.Context, request UpdateProjectRequest) (UpdateProjectOperationInterface, error)
}

type Project struct {
	// A timestamp indicating when the project was created.
	CreateTime *time.Time `json:"create_time,omitempty"`
	// The resource name of the project. This field is output-only and
	// constructed by the system. Format: `projects/{project_id}`
	Name string `json:"name,omitempty"`
	// The spec contains the project configuration, including display_name,
	// pg_version (Postgres version), history_retention_duration, and
	// default_endpoint_settings.
	Spec *ProjectSpec `json:"spec,omitempty"`
	// The current status of a Project.
	Status *ProjectStatus `json:"status,omitempty"`
	// System-generated unique ID for the project.
	Uid string `json:"uid,omitempty"`
	// A timestamp indicating when the project was last updated.
	UpdateTime *time.Time `json:"update_time,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type ProjectDefaultEndpointSettings struct {
	// The maximum number of Compute Units. Minimum value is 0.5.
	AutoscalingLimitMaxCu float64 `json:"autoscaling_limit_max_cu,omitempty"`
	// The minimum number of Compute Units. Minimum value is 0.5.
	AutoscalingLimitMinCu float64 `json:"autoscaling_limit_min_cu,omitempty"`
	// A raw representation of Postgres settings.
	PgSettings map[string]string `json:"pg_settings,omitempty"`
	// Duration of inactivity after which the compute endpoint is automatically
	// suspended. Supported values: -1s (never suspend), 0s (use default), or
	// value should be between 60s and 604800s (1 minute to 1 week).
	SuspendTimeoutDuration *duration.Duration `json:"suspend_timeout_duration,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type ProjectOperationMetadata struct {
}

type ProjectSpec struct {
	DefaultEndpointSettings *ProjectDefaultEndpointSettings `json:"default_endpoint_settings,omitempty"`
	// Human-readable project name. Length should be between 1 and 256
	// characters.
	DisplayName string `json:"display_name,omitempty"`
	// The number of seconds to retain the shared history for point in time
	// recovery for all branches in this project. Value should be between 0s and
	// 2592000s (up to 30 days).
	HistoryRetentionDuration *duration.Duration `json:"history_retention_duration,omitempty"`
	// The major Postgres version number. Supported versions are 16 and 17.
	PgVersion int `json:"pg_version,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type ProjectStatus struct {
	// The logical size limit for a branch.
	BranchLogicalSizeLimitBytes int64 `json:"branch_logical_size_limit_bytes,omitempty"`
	// The effective default endpoint settings.
	DefaultEndpointSettings *ProjectDefaultEndpointSettings `json:"default_endpoint_settings,omitempty"`
	// The effective human-readable project name.
	DisplayName string `json:"display_name,omitempty"`
	// The effective number of seconds to retain the shared history for point in
	// time recovery.
	HistoryRetentionDuration *duration.Duration `json:"history_retention_duration,omitempty"`
	// The email of the project owner.
	Owner string `json:"owner,omitempty"`
	// The effective major Postgres version number.
	PgVersion int `json:"pg_version,omitempty"`
	// The current space occupied by the project in storage.
	SyntheticStorageSizeBytes int64 `json:"synthetic_storage_size_bytes,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type RequestedClaims struct {
	PermissionSet RequestedClaimsPermissionSet `json:"permission_set,omitempty"`

	Resources []RequestedResource `json:"resources,omitempty"`
}

type RequestedClaimsPermissionSet string

type RequestedResource struct {
	TableName string `json:"table_name,omitempty"`

	UnspecifiedResourceName string `json:"unspecified_resource_name,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type Role struct {
	CreateTime *time.Time `json:"create_time,omitempty"`
	// The resource name of the role. Format:
	// projects/{project_id}/branches/{branch_id}/roles/{role_id}
	Name string `json:"name,omitempty"`
	// The Branch where this Role exists. Format:
	// projects/{project_id}/branches/{branch_id}
	Parent string `json:"parent,omitempty"`
	// The spec contains the role configuration, including identity type,
	// authentication method, and role attributes.
	Spec *RoleRoleSpec `json:"spec,omitempty"`
	// Current status of the role, including its identity type, authentication
	// method, and role attributes.
	Status *RoleRoleStatus `json:"status,omitempty"`

	UpdateTime *time.Time `json:"update_time,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type RoleAuthMethod string

type RoleIdentityType string

type RoleOperationMetadata struct {
}

type RoleRoleSpec struct {
	// If auth_method is left unspecified, a meaningful authentication method is
	// derived from the identity_type: * For the managed identities, OAUTH is
	// used. * For the regular postgres roles, authentication based on postgres
	// passwords is used.
	//
	// NOTE: this is ignored for the Databricks identity type GROUP, and
	// NO_LOGIN is implicitly assumed instead for the GROUP identity type.
	AuthMethod RoleAuthMethod `json:"auth_method,omitempty"`
	// The type of role. When specifying a managed-identity, the chosen role_id
	// must be a valid:
	//
	// * application ID for SERVICE_PRINCIPAL * user email for USER * group name
	// for GROUP
	IdentityType RoleIdentityType `json:"identity_type,omitempty"`
}

type RoleRoleStatus struct {
	AuthMethod RoleAuthMethod `json:"auth_method,omitempty"`
	// The type of the role.
	IdentityType RoleIdentityType `json:"identity_type,omitempty"`
}

type UpdateBranchOperationInterface interface {

	// Wait blocks until the long-running operation is completed. If no timeout is
	// specified, this will poll indefinitely. If a timeout is provided and the operation
	// didn't finish within the timeout, this function will return an error, otherwise
	// returns successful response and any errors encountered.
	Wait(ctx context.Context, opts ...api.Option) (*Branch, error)

	// Name returns the name of the long-running operation. The name is assigned
	// by the server and is unique within the service from which the operation is created.
	Name() string

	// Metadata returns metadata associated with the long-running operation.
	// If the metadata is not available, the returned metadata and error are both nil.
	Metadata() (*BranchOperationMetadata, error)

	// Done reports whether the long-running operation has completed.
	Done() (bool, error)
}

type UpdateBranchRequest struct {
	// The Branch to update.
	//
	// The branch's `name` field is used to identify the branch to update.
	// Format: projects/{project_id}/branches/{branch_id}
	Branch Branch `json:"branch"`
	// The resource name of the branch. This field is output-only and
	// constructed by the system. Format:
	// `projects/{project_id}/branches/{branch_id}`
	Name string `json:"-" url:"-"`
	// The list of fields to update. If unspecified, all fields will be updated
	// when possible.
	UpdateMask fieldmask.FieldMask `json:"-" url:"update_mask"`
}

type UpdateEndpointOperationInterface interface {

	// Wait blocks until the long-running operation is completed. If no timeout is
	// specified, this will poll indefinitely. If a timeout is provided and the operation
	// didn't finish within the timeout, this function will return an error, otherwise
	// returns successful response and any errors encountered.
	Wait(ctx context.Context, opts ...api.Option) (*Endpoint, error)

	// Name returns the name of the long-running operation. The name is assigned
	// by the server and is unique within the service from which the operation is created.
	Name() string

	// Metadata returns metadata associated with the long-running operation.
	// If the metadata is not available, the returned metadata and error are both nil.
	Metadata() (*EndpointOperationMetadata, error)

	// Done reports whether the long-running operation has completed.
	Done() (bool, error)
}

type UpdateEndpointRequest struct {
	// The Endpoint to update.
	//
	// The endpoint's `name` field is used to identify the endpoint to update.
	// Format:
	// projects/{project_id}/branches/{branch_id}/endpoints/{endpoint_id}
	Endpoint Endpoint `json:"endpoint"`
	// The resource name of the endpoint. This field is output-only and
	// constructed by the system. Format:
	// `projects/{project_id}/branches/{branch_id}/endpoints/{endpoint_id}`
	Name string `json:"-" url:"-"`
	// The list of fields to update. If unspecified, all fields will be updated
	// when possible.
	UpdateMask fieldmask.FieldMask `json:"-" url:"update_mask"`
}

type UpdateProjectOperationInterface interface {

	// Wait blocks until the long-running operation is completed. If no timeout is
	// specified, this will poll indefinitely. If a timeout is provided and the operation
	// didn't finish within the timeout, this function will return an error, otherwise
	// returns successful response and any errors encountered.
	Wait(ctx context.Context, opts ...api.Option) (*Project, error)

	// Name returns the name of the long-running operation. The name is assigned
	// by the server and is unique within the service from which the operation is created.
	Name() string

	// Metadata returns metadata associated with the long-running operation.
	// If the metadata is not available, the returned metadata and error are both nil.
	Metadata() (*ProjectOperationMetadata, error)

	// Done reports whether the long-running operation has completed.
	Done() (bool, error)
}

type UpdateProjectRequest struct {
	// The resource name of the project. This field is output-only and
	// constructed by the system. Format: `projects/{project_id}`
	Name string `json:"-" url:"-"`
	// The Project to update.
	//
	// The project's `name` field is used to identify the project to update.
	// Format: projects/{project_id}
	Project Project `json:"project"`
	// The list of fields to update. If unspecified, all fields will be updated
	// when possible.
	UpdateMask fieldmask.FieldMask `json:"-" url:"update_mask"`
}