consumerapi

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

type ArtifactMutationCommit func(context.Context) (artifact.Artifact, error)

type BuiltinStore interface {
	EnsureBuiltIn(
		ctx context.Context,
		request EnsureBuiltInRequest,
	) (Bundle, error)

	EnsureBuiltInCurrent(
		ctx context.Context,
		collectionRef collection.CollectionRef,
	) error

	Get(
		ctx context.Context,
		collectionRef collection.CollectionRef,
	) (Bundle, error)

	ListServers(
		ctx context.Context,
		collectionRef collection.CollectionRef,
	) ([]artifact.Artifact, error)

	ListPolicies(
		ctx context.Context,
		collectionRef collection.CollectionRef,
	) ([]artifact.Artifact, error)
}

type Bundle struct {
	Collection      collection.Collection          `json:"collection"`
	Data            mcpDomainBundle.CollectionData `json:"data"`
	Attachment      collection.Attachment          `json:"attachment"`
	Source          source.Summary                 `json:"source"`
	PackageAddress  source.ManagedPackageAddress   `json:"packageAddress"`
	DocumentLocator basespec.Locator               `json:"documentLocator"`
}

type BundleInstallationView struct {
	Bundle             collection.CollectionRef `json:"bundle"`
	BuiltIn            bool                     `json:"builtIn"`
	CollectionRevision uint64                   `json:"collectionRevision"`
	OverlayRevision    uint64                   `json:"overlayRevision"`
	RuntimeEnabled     bool                     `json:"runtimeEnabled"`
}

type BundleMutationCommit func(context.Context) (Bundle, error)

type BundleMutator interface {
	PrepareReplaceDocument(
		ctx context.Context,
		request ReplaceDocumentRequest,
	) (BundleMutationCommit, error)

	PrepareRefresh(
		ctx context.Context,
		collectionRef collection.CollectionRef,
		allowProtected bool,
	) (BundleMutationCommit, error)

	PrepareUpdateBundleEnabled(
		ctx context.Context,
		collectionRef collection.CollectionRef,
		bundleID uint64,
		enabled bool,
	) (BundleMutationCommit, error)

	PrepareRetire(
		ctx context.Context,
		collectionRef collection.CollectionRef,
		bundleID uint64,
	) (CollectionMutationCommit, error)

	PreparePurge(
		ctx context.Context,
		collectionRef collection.CollectionRef,
		bundleID uint64,
	) (MutationCommit, error)

	PrepareUpdateProtectedBundleInstallation(
		ctx context.Context,
		collectionRef collection.CollectionRef,
		bundleID uint64,
		protected bool,
	) (MutationCommit, error)

	PrepareUpdateServerInstallation(
		ctx context.Context,
		artifactRef artifact.ArtifactRef,
		installationID uint64,
		serverData mcpDomainServer.ServerData,
	) (ArtifactMutationCommit, error)

	PrepareUpdateProtectedServerInstallation(
		ctx context.Context,
		artifactRef artifact.ArtifactRef,
		installationID uint64,
		protected bool,
		serverData mcpDomainServer.ServerData,
	) (MutationCommit, error)
}

type BundleServerStore interface {
	ListServers(
		ctx context.Context,
		collectionRef collection.CollectionRef,
	) ([]artifact.Artifact, error)

	GetServerInstallation(
		ctx context.Context,
		artifactRef artifact.ArtifactRef,
	) (ServerInstallationView, error)
}

type CollectionMutationCommit func(
	context.Context,
) (collection.Collection, error)

type CreateMCPBundleBody struct {
	RootID           root.RootID
	CollectionID     collection.CollectionID
	SourceID         source.SourceID
	SourceStorageKey basespec.StorageKey
	Document         json.RawMessage
	Registrations    []Registration
}

type CreateMCPBundleRequest struct {
	Body *CreateMCPBundleBody `json:"body"`
}

type CreateMCPBundleResponse struct {
	Body *Bundle `json:"body"`
}

type EnsureBuiltInRequest struct {
	RootID         root.RootID
	CollectionID   collection.CollectionID
	SourceID       source.SourceID
	PackageAddress source.ManagedPackageAddress

	PackageFiles  []source.ManagedPackageFile
	Document      schema.ParsedDocument
	Registrations []Registration
}

type GetMCPBundleDocumentRequest struct {
	Bundle collection.CollectionRef `json:"bundle"`
}

type GetMCPBundleDocumentResponse struct {
	Body *mcpDomainBundle.BundleDocument `json:"body"`
}

type GetMCPBundleInstallationRequest struct {
	Bundle collection.CollectionRef `json:"bundle"`
}

type GetMCPBundleInstallationResponse struct {
	Body *BundleInstallationView `json:"body"`
}

type GetMCPBundleRequest struct {
	Bundle collection.CollectionRef `json:"bundle"`
}

type GetMCPBundleResponse struct {
	Body *Bundle `json:"body"`
}

type GetMCPServerInstallationRequest struct {
	Server artifact.ArtifactRef `json:"server"`
}

type GetMCPServerInstallationResponse struct {
	Body *ServerInstallationView `json:"body"`
}

type InspectMCPPolicyRequest struct {
	Policy artifact.ArtifactRef `json:"policy"`
}

type InspectMCPPolicyResponse struct {
	Body *PolicyView `json:"body"`
}

type InspectMCPServerRequest struct {
	Server artifact.ArtifactRef `json:"server"`
}

type InspectMCPServerResponse struct {
	Body *mcpDomainServer.Resolved `json:"body"`
}

type ListMCPBundlePoliciesRequest struct {
	Bundle collection.CollectionRef `json:"bundle"`
}

type ListMCPBundlePoliciesResponse struct {
	Body *ListMCPBundlePoliciesResponseBody `json:"body"`
}

type ListMCPBundlePoliciesResponseBody struct {
	Policies []artifact.Artifact `json:"policies"`
}

type ListMCPBundleServersRequest struct {
	Bundle collection.CollectionRef `json:"bundle"`
}

type ListMCPBundleServersResponse struct {
	Body *ListMCPBundleServersResponseBody `json:"body"`
}

type ListMCPBundleServersResponseBody struct {
	Servers []artifact.Artifact `json:"servers"`
}

type ListMCPBundlesRequest struct {
	RootID root.RootID `json:"rootID"`
}

type ListMCPBundlesResponse struct {
	Body *ListMCPBundlesResponseBody `json:"body"`
}

type ListMCPBundlesResponseBody struct {
	Bundles []Bundle `json:"bundles"`
}

type MCPDocumentSchemaIdentity struct {
	Kind          artifact.ArtifactKind `json:"kind"`
	SchemaID      schema.SchemaID       `json:"schemaID"`
	SchemaVersion string                `json:"schemaVersion"`
}

type MCPServerSchemaIdentity struct {
	Server MCPDocumentSchemaIdentity `json:"server"`
	Policy MCPDocumentSchemaIdentity `json:"policy"`
}

type MutationCommit func(context.Context) error

type PolicyView struct {
	Artifact         artifact.Artifact        `json:"artifact"`
	Collection       collection.CollectionRef `json:"collection"`
	CatalogRevision  uint64                   `json:"catalogRevision"`
	Definition       definition.Definition    `json:"definition"`
	Body             mcpPolicy.MCPPolicy      `json:"body"`
	EffectiveEnabled bool                     `json:"effectiveEnabled"`
	BuiltIn          bool                     `json:"builtIn"`
}

type Registration struct {
	ArtifactID  artifact.ArtifactID
	Subresource basespec.SubresourceLocator
	Kind        artifact.ArtifactKind
	Enabled     bool
	Data        json.RawMessage
}

type ReplaceDocumentRequest struct {
	Bundle                     collection.CollectionRef
	ExpectedCollectionRevision uint64
	Document                   json.RawMessage
	Registrations              []Registration
	AllowProtected             bool
}

type ServerInstallationView struct {
	Artifact             artifact.Artifact              `json:"artifact"`
	Collection           collection.CollectionRef       `json:"collection"`
	CatalogRevision      uint64                         `json:"catalogRevision"`
	Document             mcpDomainServer.ServerDocument `json:"document"`
	Installation         mcpDomainServer.ServerData     `json:"installation"`
	InstallationRevision uint64                         `json:"installationRevision"`
	InstallationEnabled  bool                           `json:"installationEnabled"`
	RuntimeEnabled       bool                           `json:"runtimeEnabled"`
	BuiltIn              bool                           `json:"builtIn"`
}

type ServerStore interface {
	ResolveMCPServer(
		ctx context.Context,
		ref artifact.ArtifactRef,
	) (mcpDomainServer.Resolved, error)
	InspectMCPServerForRuntime(
		ctx context.Context,
		ref artifact.ArtifactRef,
	) (mcpDomainServer.Resolved, error)
}