[WORKFLOW] AI update based on proto changes from commit 28f2f05

Author: viambot

src/app/app-client.spec.ts

@@ -69,6 +69,8 @@ describe('AppClient tests', () => {
     secrets: [sharedSecret],
     secret: 'secret',
     fqdn: 'fqdn',
+    isMainPart: false,
+    dnsName: 'dns.name.example.com',
   });
 
   const logEntry = new LogEntry({

src/app/billing-client.spec.ts

@@ -7,12 +7,23 @@ import {
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 import { BillingService } from '../gen/app/v1/billing_connect';
 import {
+  CreateInvoiceAndChargeImmediatelyRequest,
+  CreateInvoiceAndChargeImmediatelyResponse,
+  GetAvailableBillingTiersRequest,
+  GetAvailableBillingTiersResponse,
+  GetCurrentMonthUsageRequest,
+  GetCurrentMonthUsageResponse,
+  GetInvoicePdfRequest,
   GetInvoicePdfResponse,
+  GetInvoicesSummaryRequest,
   GetInvoicesSummaryResponse,
+  GetOrgBillingInformationRequest,
   PaymentMethodType,
   ResourceUsageCosts,
   ResourceUsageCostsBySource,
+  SendPaymentRequiredEmailRequest,
   SourceType,
+  UpdateOrganizationBillingTierRequest,
   UsageCost,
   UsageCostType,
 } from '../gen/app/v1/billing_pb';
@@ -162,4 +173,48 @@ describe('BillingClient tests', () => {
     const array = new Uint8Array([1, 2, 3, 4]);
     await expect(promise).resolves.toStrictEqual(array);
   });
-});
+
+  describe('createInvoiceAndChargeImmediately tests', () => {
+    let capReq: CreateInvoiceAndChargeImmediatelyRequest;
+    beforeEach(() => {
+      mockTransport = createRouterTransport(({ service }) => {
+        service(BillingService, {
+          createInvoiceAndChargeImmediately: (req) => {
+            capReq = req;
+            return new CreateInvoiceAndChargeImmediatelyResponse();
+          },
+        });
+      });
+    });
+
+    it('creates and charges invoice immediately', async () => {
+      const expectedRequest = new CreateInvoiceAndChargeImmediatelyRequest({
+        orgIdToCharge: 'orgIdToCharge',
+        amount: 100.00,
+        description: 'test description',
+        orgIdForBranding: 'orgIdForBranding',
+      });
+
+      await subject().createInvoiceAndChargeImmediately(
+        'orgIdToCharge',
+        100.00,
+        'test description',
+        'orgIdForBranding'
+      );
+      expect(capReq).toStrictEqual(expectedRequest);
+    });
+
+    it('creates and charges invoice immediately with optional fields omitted', async () => {
+      const expectedRequest = new CreateInvoiceAndChargeImmediatelyRequest({
+        orgIdToCharge: 'orgIdToCharge',
+        amount: 100.00,
+      });
+
+      await subject().createInvoiceAndChargeImmediately(
+        'orgIdToCharge',
+        100.00
+      );
+      expect(capReq).toStrictEqual(expectedRequest);
+    });
+  });
+});

src/app/billing-client.ts

@@ -5,6 +5,7 @@ import {
 } from '@connectrpc/connect';
 import { BillingService } from '../gen/app/v1/billing_connect';
 import type { GetCurrentMonthUsageResponse as PBGetCurrentMonthUsageResponse } from '../gen/app/v1/billing_pb';
+import { CreateInvoiceAndChargeImmediatelyRequest, CreateInvoiceAndChargeImmediatelyResponse, GetAvailableBillingTiersRequest, GetAvailableBillingTiersResponse, GetInvoicePdfRequest, GetInvoicePdfResponse, GetInvoicesSummaryRequest, GetInvoicesSummaryResponse, GetOrgBillingInformationRequest, GetOrgBillingInformationResponse, SendPaymentRequiredEmailRequest, SendPaymentRequiredEmailResponse, UpdateOrganizationBillingTierRequest, UpdateOrganizationBillingTierResponse } from '../gen/app/v1/billing_pb';
 
 export type GetCurrentMonthUsageResponse =
   Partial<PBGetCurrentMonthUsageResponse> & {
@@ -119,6 +120,39 @@ export class BillingClient {
     }
     return concatArrayU8(chunks);
   }
+
+  /**
+   * Directly create a flat fee invoice for an organization and charge on the spot
+   *
+   * @example
+   *
+   * ```ts
+   * await billing.createInvoiceAndChargeImmediately(
+   *   '<organization-id-to-charge>',
+   *   100.00,
+   *   'One-time charge for services',
+   *   '<organization-id-for-branding>'
+   * );
+   * ```
+   *
+   * @param orgIdToCharge The ID of the organization to charge.
+   * @param amount The amount to charge.
+   * @param description Optional description for the invoice.
+   * @param orgIdForBranding Optional organization ID for branding the invoice.
+   */
+  async createInvoiceAndChargeImmediately(
+    orgIdToCharge: string,
+    amount: number,
+    description?: string,
+    orgIdForBranding?: string
+  ): Promise<void> {
+    await this.client.createInvoiceAndChargeImmediately({
+      orgIdToCharge,
+      amount,
+      description,
+      orgIdForBranding,
+    });
+  }
 }
 
 const concatArrayU8 = (arrays: Uint8Array[]) => {

src/app/data-client.spec.ts

@@ -64,6 +64,8 @@ import {
   ListDatasetsByIDsResponse,
   ListDatasetsByOrganizationIDRequest,
   ListDatasetsByOrganizationIDResponse,
+  MergeDatasetsRequest,
+  MergeDatasetsResponse,
   RenameDatasetRequest,
   RenameDatasetResponse,
 } from '../gen/app/dataset/v1/dataset_pb';
@@ -408,7 +410,11 @@ describe('DataClient tests', () => {
             capReq = req;
             if (!once) {
               once = true;
-              return binDataResponse;
+              const response = new BinaryDataByFilterResponse();
+              response.deletedCount = req.includeInternalData
+                ? BigInt(20)
+                : BigInt(10);
+              return response;
             }
             return new BinaryDataByFilterResponse();
           },
@@ -1335,6 +1341,38 @@ describe('DatasetClient tests', () => {
       expect(set2?.created).toEqual(dataset2.timeCreated?.toDate());
     });
   });
+
+  describe('mergeDatasets tests', () => {
+    let capReq: MergeDatasetsRequest;
+    beforeEach(() => {
+      mockTransport = createRouterTransport(({ service }) => {
+        service(DatasetService, {
+          mergeDatasets: (req) => {
+            capReq = req;
+            return new MergeDatasetsResponse({
+              datasetId: 'newDatasetId',
+            });
+          },
+        });
+      });
+    });
+
+    it('merges datasets', async () => {
+      const expectedRequest = new MergeDatasetsRequest({
+        datasetIds: ['id1', 'id2'],
+        name: 'newName',
+        organizationId: 'orgId',
+      });
+
+      const promise = await subject().mergeDatasets(
+        ['id1', 'id2'],
+        'newName',
+        'orgId'
+      );
+      expect(capReq).toStrictEqual(expectedRequest);
+      expect(promise).toEqual('newDatasetId');
+    });
+  });
 });
 
 describe('DataSyncClient tests', () => {

src/app/data-client.ts

@@ -17,7 +17,7 @@ import {
   TabularDataSourceType,
 } from '../gen/app/data/v1/data_pb';
 import { DatasetService } from '../gen/app/dataset/v1/dataset_connect';
-import type { Dataset as PBDataset } from '../gen/app/dataset/v1/dataset_pb';
+import type { Dataset as PBDataset, MergeDatasetsRequest, MergeDatasetsResponse } from '../gen/app/dataset/v1/dataset_pb';
 import { DataSyncService } from '../gen/app/datasync/v1/data_sync_connect';
 import { DataPipelinesService } from '../gen/app/datapipelines/v1/data_pipelines_connect';
 import {
@@ -1086,6 +1086,37 @@ export class DataClient {
     });
   }
 
+  /**
+   * MergeDatasets merges multiple datasets into a new dataset.
+   *
+   * @example
+   *
+   * ```ts
+   * const newDatasetId = await dataClient.mergeDatasets(
+   *   ['dataset-id-1', 'dataset-id-2'],
+   *   'merged-dataset-name',
+   *   'organization-id'
+   * );
+   * ```
+   *
+   * @param datasetIds The IDs of the datasets to merge.
+   * @param name The name of the new merged dataset.
+   * @param organizationId The ID of the organization that owns the datasets.
+   * @returns The dataset ID of the newly created merged dataset.
+   */
+  async mergeDatasets(
+    datasetIds: string[],
+    name: string,
+    organizationId: string
+  ): Promise<string> {
+    const resp = await this.datasetClient.mergeDatasets({
+      datasetIds,
+      name,
+      organizationId,
+    });
+    return resp.datasetId;
+  }
+
   /**
    * Uploads the content and metadata for tabular data.
    *
@@ -1568,4 +1599,4 @@ export class ListDataPipelineRunsPage {
 }
 
 export { type BinaryID, type Order } from '../gen/app/data/v1/data_pb';
-export { type UploadMetadata } from '../gen/app/datasync/v1/data_sync_pb';
+export { type UploadMetadata } from '../gen/app/datasync/v1/data_sync_pb';

src/components/camera/camera.ts

@@ -62,7 +62,7 @@ export interface Camera extends Resource {
    * @param mimeType - A specific MIME type to request. This is not necessarily
    *   the same type that will be returned.
    */
-  getImage: (mimeType?: MimeType, extra?: Struct) => Promise<Uint8Array>;
+  getImage: (mimeType?: MimeType, filterSourceNames?: string[], extra?: Struct) => Promise<Uint8Array>;
 
   /**
    * Render a frame from a camera to an HTTP response.
@@ -112,4 +112,4 @@ export interface Camera extends Resource {
    * API](https://docs.viam.com/dev/reference/apis/components/camera/#getproperties).
    */
   getProperties: () => Promise<Properties>;
-}
+}

src/components/camera/client.ts

@@ -44,12 +44,14 @@ export class CameraClient implements Camera {
 
   async getImage(
     mimeType: MimeType = '',
+    filterSourceNames: string[] = [],
     extra = {},
     callOptions = this.callOptions
   ) {
     const request = new GetImageRequest({
       name: this.name,
       mimeType,
+      filterSourceNames,
       extra: Struct.fromJson(extra),
     });
 
@@ -111,4 +113,4 @@ export class CameraClient implements Camera {
       callOptions
     );
   }
-}
+}