@stigmer/sdk 3.11.0 → 3.12.0

package/src/gen/datastore.ts

@@ -1,381 +0,0 @@
-// Code generated by stigmer-codegen. DO NOT EDIT.
-
-import { wrapError } from "./errors.js";
-import { stripUndefined } from "./proto-utils.js";
-import { type DeleteResourceInput, type ListParams, type ListResult, type ResourceRef } from "./types.js";
-import { create, fromJson, type JsonValue } from "@bufbuild/protobuf";
-import { ValueSchema } from "@bufbuild/protobuf/wkt";
-import { createClient, type Client, type Transport } from "@connectrpc/connect";
-import { DatastoreSchema, type Datastore } from "@stigmer/protos/ai/stigmer/agentic/datastore/v1/api_pb";
-import { DatastoreCommandController } from "@stigmer/protos/ai/stigmer/agentic/datastore/v1/command_pb";
-import { DatastoreQueryController } from "@stigmer/protos/ai/stigmer/agentic/datastore/v1/query_pb";
-import { DatastoreRecordCommandController } from "@stigmer/protos/ai/stigmer/agentic/datastore/v1/record_command_pb";
-import { InsertRecordRequestSchema, RecordEnvelopeSchema, UpdateRecordRequestSchema, DeleteRecordRequestSchema, FindRecordsRequestSchema, RecordListSchema, DescribeDatastoreRequestSchema, DatastoreDescriptionSchema, type InsertRecordRequest, type RecordEnvelope, type UpdateRecordRequest, type DeleteRecordRequest, type FindRecordsRequest, type RecordList, type DescribeDatastoreRequest, type DatastoreDescription } from "@stigmer/protos/ai/stigmer/agentic/datastore/v1/record_io_pb";
-import { DatastoreRecordQueryController } from "@stigmer/protos/ai/stigmer/agentic/datastore/v1/record_query_pb";
-import { DatastoreSpecSchema, FieldType, DatastoreVerb, DatastoreGrantScope, DatastoreRoleSchema, ChannelSenderSubjectSchema, DatastoreSubjectSchema, DatastoreRoleBindingSchema, DatastoreAuthorizationSchema, FieldDeclarationSchema, UniqueWhereSchema, UniqueConstraintSchema, CheckConstraintSchema, ExistsConstraintSchema, DatastoreGrantSchema, CollectionDeclarationSchema } from "@stigmer/protos/ai/stigmer/agentic/datastore/v1/spec_pb";
-import { ApiResourceKind } from "@stigmer/protos/ai/stigmer/commons/apiresource/apiresourcekind/api_resource_kind_pb";
-import { ApiResourceVisibility } from "@stigmer/protos/ai/stigmer/commons/apiresource/enum_pb";
-import { ApiResourceIdSchema, ApiResourceReferenceSchema, ApiResourceDeleteInputSchema, type UpdateVisibilityInput } from "@stigmer/protos/ai/stigmer/commons/apiresource/io_pb";
-import { ApiResourceMetadataSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/metadata_pb";
-import { PageInfoSchema } from "@stigmer/protos/ai/stigmer/commons/rpc/pagination_pb";
-import { ApiResourceRefSchema } from "@stigmer/protos/ai/stigmer/iam/iampolicy/v1/spec_pb";
-import { SearchRequestSchema } from "@stigmer/protos/ai/stigmer/search/v1/io_pb";
-import { SearchService } from "@stigmer/protos/ai/stigmer/search/v1/query_pb";
-
-/** Provides operations on datastore resources. */
-export class DatastoreClient {
-  private readonly command: Client<typeof DatastoreCommandController>;
-  private readonly query: Client<typeof DatastoreQueryController>;
-  private readonly datastoreRecordCommand: Client<typeof DatastoreRecordCommandController>;
-  private readonly datastoreRecordQuery: Client<typeof DatastoreRecordQueryController>;
-  private readonly search: Client<typeof SearchService>;
-
-  constructor(transport: Transport) {
-    this.command = createClient(DatastoreCommandController, transport);
-    this.query = createClient(DatastoreQueryController, transport);
-    this.datastoreRecordCommand = createClient(DatastoreRecordCommandController, transport);
-    this.datastoreRecordQuery = createClient(DatastoreRecordQueryController, transport);
-    this.search = createClient(SearchService, transport);
-  }
-
-  async apply(input: DatastoreInput): Promise<Datastore> {
-    try {
-      return await this.command.apply(buildDatastoreProto(input));
-    } catch (e) { throw wrapError(e); }
-  }
-
-  async create(input: DatastoreInput): Promise<Datastore> {
-    try {
-      return await this.command.create(buildDatastoreProto(input));
-    } catch (e) { throw wrapError(e); }
-  }
-
-  async update(input: DatastoreInput): Promise<Datastore> {
-    try {
-      return await this.command.update(buildDatastoreProto(input));
-    } catch (e) { throw wrapError(e); }
-  }
-
-  async updateVisibility(input: UpdateVisibilityInput): Promise<Datastore> {
-    try {
-      return await this.command.updateVisibility(input);
-    } catch (e) { throw wrapError(e); }
-  }
-
-  async delete(input: DeleteResourceInput): Promise<Datastore> {
-    try {
-      return await this.command.delete(create(ApiResourceDeleteInputSchema, {
-        resourceId: input.resourceId,
-        versionMessage: input.versionMessage,
-        force: input.force,
-      }));
-    } catch (e) { throw wrapError(e); }
-  }
-
-  async get(id: string): Promise<Datastore> {
-    try {
-      return await this.query.get(create(ApiResourceIdSchema, { value: id }));
-    } catch (e) { throw wrapError(e); }
-  }
-
-  async getByReference(ref: ResourceRef): Promise<Datastore> {
-    try {
-      return await this.query.getByReference(create(ApiResourceReferenceSchema, { ...ref, kind: ApiResourceKind.datastore }));
-    } catch (e) { throw wrapError(e); }
-  }
-
-  async insertRecord(input: InsertRecordRequest): Promise<RecordEnvelope> {
-    try {
-      return await this.datastoreRecordCommand.insertRecord(input);
-    } catch (e) { throw wrapError(e); }
-  }
-
-  async updateRecord(input: UpdateRecordRequest): Promise<RecordEnvelope> {
-    try {
-      return await this.datastoreRecordCommand.updateRecord(input);
-    } catch (e) { throw wrapError(e); }
-  }
-
-  async deleteRecord(input: DeleteRecordRequest): Promise<RecordEnvelope> {
-    try {
-      return await this.datastoreRecordCommand.deleteRecord(input);
-    } catch (e) { throw wrapError(e); }
-  }
-
-  async findRecords(input: FindRecordsRequest): Promise<RecordList> {
-    try {
-      return await this.datastoreRecordQuery.findRecords(input);
-    } catch (e) { throw wrapError(e); }
-  }
-
-  async describeDatastore(input: DescribeDatastoreRequest): Promise<DatastoreDescription> {
-    try {
-      return await this.datastoreRecordQuery.describeDatastore(input);
-    } catch (e) { throw wrapError(e); }
-  }
-
-  async list(params: ListParams): Promise<ListResult> {
-    try {
-      const resp = await this.search.search(create(SearchRequestSchema, {
-        kinds: [ApiResourceKind.datastore],
-        query: params.query,
-        org: params.org,
-        excludePublic: params.excludePublic ?? false,
-        crossOrgPublic: params.crossOrgPublic ?? false,
-        page: params.page ? create(PageInfoSchema, params.page) : undefined,
-      }));
-      return {
-        entries: resp.entries,
-        totalCount: resp.totalCount,
-        totalPages: resp.totalPages,
-      };
-    } catch (e) { throw wrapError(e); }
-  }
-}
-
-/** Input for creating/updating a Datastore. */
-export interface DatastoreInput {
-  name: string;
-  slug?: string;
-  org: string;
-  labels?: Record<string, string>;
-  visibility?: ApiResourceVisibility;
-  description?: string;
-  timezone?: string;
-  authorization?: DatastoreAuthorizationInput;
-  collections?: CollectionDeclarationInput[];
-}
-
-/** SDK input type for DatastoreAuthorization. */
-export interface DatastoreAuthorizationInput {
-  roles?: DatastoreRoleInput[];
-  bindings?: DatastoreRoleBindingInput[];
-  defaultRole?: string;
-}
-
-/** SDK input type for DatastoreRole. */
-export interface DatastoreRoleInput {
-  name: string;
-}
-
-/** SDK input type for DatastoreRoleBinding. */
-export interface DatastoreRoleBindingInput {
-  subject: DatastoreSubjectInput;
-  role: string;
-}
-
-/** SDK input type for DatastoreSubject. */
-export interface DatastoreSubjectInput {
-  channelSender?: ChannelSenderSubjectInput;
-  principal?: ApiResourceRefInput;
-}
-
-/** SDK input type for ChannelSenderSubject. */
-export interface ChannelSenderSubjectInput {
-  senderKind: string;
-  value: string;
-}
-
-/** SDK input type for ApiResourceRef. */
-export interface ApiResourceRefInput {
-  kind: string;
-  id: string;
-  relation?: string;
-}
-
-/** SDK input type for CollectionDeclaration. */
-export interface CollectionDeclarationInput {
-  name: string;
-  description?: string;
-  fields?: FieldDeclarationInput[];
-  uniques?: UniqueConstraintInput[];
-  checks?: CheckConstraintInput[];
-  exists?: ExistsConstraintInput[];
-  notExists?: ExistsConstraintInput[];
-  grants?: DatastoreGrantInput[];
-}
-
-/** SDK input type for FieldDeclaration. */
-export interface FieldDeclarationInput {
-  name: string;
-  type: FieldType;
-  required?: boolean;
-  default?: JsonValue;
-  enumValues?: string[];
-  description?: string;
-}
-
-/** SDK input type for UniqueConstraint. */
-export interface UniqueConstraintInput {
-  name: string;
-  fields?: string[];
-  where?: UniqueWhereInput;
-  message: string;
-}
-
-/** SDK input type for UniqueWhere. */
-export interface UniqueWhereInput {
-  field: string;
-  equals: JsonValue;
-}
-
-/** SDK input type for CheckConstraint. */
-export interface CheckConstraintInput {
-  name: string;
-  expression: string;
-  when?: string;
-  message: string;
-}
-
-/** SDK input type for ExistsConstraint. */
-export interface ExistsConstraintInput {
-  name: string;
-  collection: string;
-  where: string;
-  when?: string;
-  message: string;
-}
-
-/** SDK input type for DatastoreGrant. */
-export interface DatastoreGrantInput {
-  role: string;
-  verbs?: DatastoreVerb[];
-  scope?: DatastoreGrantScope;
-  readFields?: string[];
-}
-
-function buildDatastoreRoleProto(input: DatastoreRoleInput) {
-  return Object.assign(create(DatastoreRoleSchema), stripUndefined({
-    name: input.name,
-  }));
-}
-
-function buildChannelSenderSubjectProto(input: ChannelSenderSubjectInput) {
-  return Object.assign(create(ChannelSenderSubjectSchema), stripUndefined({
-    senderKind: input.senderKind,
-    value: input.value,
-  }));
-}
-
-function buildApiResourceRefProto(input: ApiResourceRefInput) {
-  return Object.assign(create(ApiResourceRefSchema), stripUndefined({
-    kind: input.kind,
-    id: input.id,
-    relation: input.relation,
-  }));
-}
-
-function buildDatastoreSubjectProto(input: DatastoreSubjectInput) {
-  const msg = create(DatastoreSubjectSchema);
-  if (input.channelSender) {
-    msg.kind = { case: "channelSender", value: buildChannelSenderSubjectProto(input.channelSender) };
-  } else if (input.principal) {
-    msg.kind = { case: "principal", value: buildApiResourceRefProto(input.principal) };
-  }
-  return msg;
-}
-
-function buildDatastoreRoleBindingProto(input: DatastoreRoleBindingInput) {
-  const msg = create(DatastoreRoleBindingSchema);
-  if (input.subject) msg.subject = buildDatastoreSubjectProto(input.subject);
-  if (input.role !== undefined) msg.role = input.role;
-  return msg;
-}
-
-function buildDatastoreAuthorizationProto(input: DatastoreAuthorizationInput) {
-  const msg = create(DatastoreAuthorizationSchema);
-  if (input.roles) msg.roles = input.roles.map(buildDatastoreRoleProto);
-  if (input.bindings) msg.bindings = input.bindings.map(buildDatastoreRoleBindingProto);
-  if (input.defaultRole !== undefined) msg.defaultRole = input.defaultRole;
-  return msg;
-}
-
-function buildFieldDeclarationProto(input: FieldDeclarationInput) {
-  const msg = create(FieldDeclarationSchema);
-  if (input.name !== undefined) msg.name = input.name;
-  if (input.type !== undefined) msg.type = input.type;
-  if (input.required !== undefined) msg.required = input.required;
-  if (input.default !== undefined) msg.default = fromJson(ValueSchema, input.default);
-  if (input.enumValues) msg.enumValues = input.enumValues;
-  if (input.description !== undefined) msg.description = input.description;
-  return msg;
-}
-
-function buildUniqueWhereProto(input: UniqueWhereInput) {
-  const msg = create(UniqueWhereSchema);
-  if (input.field !== undefined) msg.field = input.field;
-  if (input.equals !== undefined) msg.equals = fromJson(ValueSchema, input.equals);
-  return msg;
-}
-
-function buildUniqueConstraintProto(input: UniqueConstraintInput) {
-  const msg = create(UniqueConstraintSchema);
-  if (input.name !== undefined) msg.name = input.name;
-  if (input.fields) msg.fields = input.fields;
-  if (input.where) msg.where = buildUniqueWhereProto(input.where);
-  if (input.message !== undefined) msg.message = input.message;
-  return msg;
-}
-
-function buildCheckConstraintProto(input: CheckConstraintInput) {
-  return Object.assign(create(CheckConstraintSchema), stripUndefined({
-    name: input.name,
-    expression: input.expression,
-    when: input.when,
-    message: input.message,
-  }));
-}
-
-function buildExistsConstraintProto(input: ExistsConstraintInput) {
-  return Object.assign(create(ExistsConstraintSchema), stripUndefined({
-    name: input.name,
-    collection: input.collection,
-    where: input.where,
-    when: input.when,
-    message: input.message,
-  }));
-}
-
-function buildDatastoreGrantProto(input: DatastoreGrantInput) {
-  return Object.assign(create(DatastoreGrantSchema), stripUndefined({
-    role: input.role,
-    verbs: input.verbs,
-    scope: input.scope,
-    readFields: input.readFields,
-  }));
-}
-
-function buildCollectionDeclarationProto(input: CollectionDeclarationInput) {
-  const msg = create(CollectionDeclarationSchema);
-  if (input.name !== undefined) msg.name = input.name;
-  if (input.description !== undefined) msg.description = input.description;
-  if (input.fields) msg.fields = input.fields.map(buildFieldDeclarationProto);
-  if (input.uniques) msg.uniques = input.uniques.map(buildUniqueConstraintProto);
-  if (input.checks) msg.checks = input.checks.map(buildCheckConstraintProto);
-  if (input.exists) msg.exists = input.exists.map(buildExistsConstraintProto);
-  if (input.notExists) msg.notExists = input.notExists.map(buildExistsConstraintProto);
-  if (input.grants) msg.grants = input.grants.map(buildDatastoreGrantProto);
-  return msg;
-}
-
-export function buildDatastoreProto(input: DatastoreInput): Datastore {
-  const authorization = input.authorization ? buildDatastoreAuthorizationProto(input.authorization) : undefined;
-  const collections = input.collections?.map(buildCollectionDeclarationProto);
-  return Object.assign(create(DatastoreSchema), {
-    apiVersion: "agentic.stigmer.ai/v1",
-    kind: "Datastore",
-    metadata: Object.assign(create(ApiResourceMetadataSchema), {
-      name: input.name,
-      org: input.org,
-      ...(input.slug && { slug: input.slug }),
-      ...(input.labels && { labels: input.labels }),
-      ...(input.visibility && { visibility: input.visibility }),
-    }),
-    spec: Object.assign(create(DatastoreSpecSchema), stripUndefined({
-      description: input.description,
-      timezone: input.timezone,
-      authorization,
-      collections,
-    })),
-  }) as Datastore;
-}