revisium 2.0.1 → 2.2.0

package/src/services/sync/row-sync.service.ts

@@ -29,7 +29,6 @@ export interface OrderBy {
 
 export interface ApiClient {
   rows(
-    revisionId: string,
     tableId: string,
     options: { first: number; after?: string; orderBy?: OrderBy[] },
   ): Promise<{
@@ -41,13 +40,11 @@ export interface ApiClient {
   }>;
 
   createRows(
-    revisionId: string,
     tableId: string,
     data: { rows: { rowId: string; data: object }[]; isRestore?: boolean },
   ): Promise<{ data?: unknown; error?: unknown }>;
 
   updateRows(
-    revisionId: string,
     tableId: string,
     data: { rows: { rowId: string; data: object }[]; isRestore?: boolean },
   ): Promise<{ data?: unknown; error?: unknown }>;
@@ -65,7 +62,6 @@ const DEFAULT_BATCH_SIZE = 100;
 export class RowSyncService {
   async syncTableRows(
     api: ApiClient,
-    targetRevisionId: string,
     tableId: string,
     sourceRows: RowData[],
     batchSize: number = DEFAULT_BATCH_SIZE,
@@ -80,12 +76,7 @@ export class RowSyncService {
       updateErrors: 0,
     };
 
-    const existingRows = await this.getExistingRows(
-      api,
-      targetRevisionId,
-      tableId,
-      onProgress,
-    );
+    const existingRows = await this.getExistingRows(api, tableId, onProgress);
 
     const { rowsToCreate, rowsToUpdate, skippedCount } = this.categorizeRows(
       sourceRows,
@@ -96,7 +87,6 @@ export class RowSyncService {
     if (rowsToCreate.length > 0) {
       const createResult = await this.processBatchCreate(
         api,
-        targetRevisionId,
         tableId,
         rowsToCreate,
         batchSize,
@@ -109,7 +99,6 @@ export class RowSyncService {
     if (rowsToUpdate.length > 0) {
       const updateResult = await this.processBatchUpdate(
         api,
-        targetRevisionId,
         tableId,
         rowsToUpdate,
         batchSize,
@@ -124,7 +113,6 @@ export class RowSyncService {
 
   async getExistingRows(
     api: ApiClient,
-    revisionId: string,
     tableId: string,
     onProgress?: ProgressCallback,
   ): Promise<Map<string, JsonValue>> {
@@ -133,7 +121,7 @@ export class RowSyncService {
     let hasMore = true;
 
     while (hasMore) {
-      const result = await api.rows(revisionId, tableId, {
+      const result = await api.rows(tableId, {
         first: 100,
         after,
         orderBy: [{ field: 'id', direction: 'asc' }],
@@ -194,7 +182,6 @@ export class RowSyncService {
 
   private async processBatchCreate(
     api: ApiClient,
-    revisionId: string,
     tableId: string,
     rows: RowData[],
     batchSize: number,
@@ -207,7 +194,7 @@ export class RowSyncService {
       operation: 'create',
       onProgress,
       executeBatch: (batch) =>
-        api.createRows(revisionId, tableId, {
+        api.createRows(tableId, {
           rows: batch.map((r) => ({ rowId: r.id, data: r.data as object })),
           isRestore: true,
         }),
@@ -217,7 +204,6 @@ export class RowSyncService {
 
   private async processBatchUpdate(
     api: ApiClient,
-    revisionId: string,
     tableId: string,
     rows: RowData[],
     batchSize: number,
@@ -230,7 +216,7 @@ export class RowSyncService {
       operation: 'update',
       onProgress,
       executeBatch: (batch) =>
-        api.updateRows(revisionId, tableId, {
+        api.updateRows(tableId, {
           rows: batch.map((r) => ({ rowId: r.id, data: r.data as object })),
           isRestore: true,
         }),

package/src/services/sync/sync-data.service.ts

@@ -1,12 +1,9 @@
 import { Injectable } from '@nestjs/common';
+import { RevisionScope } from '@revisium/client';
 import { SyncApiService, ConnectionInfo } from './sync-api.service';
 import { TableDependencyService } from './table-dependency.service';
-import {
-  RowSyncService,
-  RowData,
-  ApiClient,
-  RowSyncError,
-} from './row-sync.service';
+import { RowSyncService, RowData, RowSyncError } from './row-sync.service';
+import { createApiClientAdapter } from '../connection';
 import { LoggerService } from '../common';
 import {
   DataSyncResult,
@@ -60,17 +57,12 @@ export class SyncDataService {
     tables: string[],
   ): Promise<Record<string, JsonSchema>> {
     const schemas: Record<string, JsonSchema> = {};
+    const revisionScope = this.getRevisionScope(connection);
 
     for (const tableId of tables) {
       try {
-        const result = await connection.client.api.tableSchema(
-          connection.revisionId,
-          tableId,
-        );
-
-        if (result.data) {
-          schemas[tableId] = result.data as JsonSchema;
-        }
+        const schema = await revisionScope.getTableSchema(tableId);
+        schemas[tableId] = schema as JsonSchema;
       } catch (error) {
         this.logger.indentWarn(
           `Could not fetch schema for table ${tableId}: ${error instanceof Error ? error.message : String(error)}`,
@@ -105,30 +97,24 @@ export class SyncDataService {
     tableFilter?: string[],
   ): Promise<string[]> {
     const tables: string[] = [];
+    const revisionScope = this.getRevisionScope(connection);
     let cursor: string | undefined;
 
     do {
-      const result = await connection.client.api.tables({
-        revisionId: connection.revisionId,
+      const result = await revisionScope.getTables({
         first: 100,
         after: cursor,
       });
 
-      if (result.error) {
-        throw new Error(
-          `Failed to get tables: ${JSON.stringify(result.error)}`,
-        );
-      }
-
-      for (const edge of result.data.edges) {
+      for (const edge of result.edges) {
         const tableId = edge.node.id;
         if (!tableFilter || tableFilter.includes(tableId)) {
           tables.push(tableId);
         }
       }
 
-      cursor = result.data.pageInfo.hasNextPage
-        ? result.data.pageInfo.endCursor
+      cursor = result.pageInfo.hasNextPage
+        ? (result.pageInfo.endCursor ?? undefined)
         : undefined;
     } while (cursor);
 
@@ -140,24 +126,17 @@ export class SyncDataService {
     tableId: string,
   ): Promise<RowData[]> {
     const rows: RowData[] = [];
+    const revisionScope = this.getRevisionScope(connection);
     let cursor: string | undefined;
 
     do {
-      const result = await connection.client.api.rows(
-        connection.revisionId,
-        tableId,
-        {
-          first: 100,
-          after: cursor,
-          orderBy: [{ field: 'id', direction: 'asc' }],
-        },
-      );
-
-      if (result.error) {
-        throw new Error(`Failed to get rows: ${JSON.stringify(result.error)}`);
-      }
+      const result = await revisionScope.getRows(tableId, {
+        first: 100,
+        after: cursor,
+        orderBy: [{ field: 'id', direction: 'asc' }],
+      });
 
-      for (const edge of result.data.edges) {
+      for (const edge of result.edges) {
         rows.push({
           id: edge.node.id,
           data: edge.node.data as Record<string, unknown>,
@@ -169,8 +148,8 @@ export class SyncDataService {
         { labels: { fetch: 'Loading from source' }, indent: '    ' },
       );
 
-      cursor = result.data.pageInfo.hasNextPage
-        ? result.data.pageInfo.endCursor
+      cursor = result.pageInfo.hasNextPage
+        ? (result.pageInfo.endCursor ?? undefined)
         : undefined;
     } while (cursor);
 
@@ -195,9 +174,11 @@ export class SyncDataService {
 
       let existingRows: Map<string, JsonValue>;
       try {
+        const targetApiClient = createApiClientAdapter(
+          this.getRevisionScope(target),
+        );
         existingRows = await this.rowSync.getExistingRows(
-          target.client.api as unknown as ApiClient,
-          target.draftRevisionId,
+          targetApiClient,
           tableId,
         );
       } catch {
@@ -310,9 +291,11 @@ export class SyncDataService {
     const sourceRows = await this.getSourceRows(source, tableId);
     this.logger.syncFound(sourceRows.length, 'rows in source');
 
+    const targetApiClient = createApiClientAdapter(
+      this.getRevisionScope(target),
+    );
     const stats = await this.rowSync.syncTableRows(
-      target.client.api as unknown as ApiClient,
-      target.draftRevisionId,
+      targetApiClient,
       tableId,
       sourceRows,
       batchSize,
@@ -334,6 +317,10 @@ export class SyncDataService {
     };
   }
 
+  private getRevisionScope(connection: ConnectionInfo): RevisionScope {
+    return connection.revisionScope;
+  }
+
   private createEmptyResult(): DataSyncResult {
     return {
       tables: [],

package/src/services/sync/sync-schema.service.ts

@@ -1,4 +1,5 @@
 import { Injectable } from '@nestjs/common';
+import { RevisionScope } from '@revisium/client';
 import { SyncApiService, ConnectionInfo } from './sync-api.service';
 import { LoggerService } from '../common';
 import { SchemaSyncResult } from 'src/types/sync.types';
@@ -41,17 +42,9 @@ export class SyncSchemaService {
   private async getMigrations(
     connection: ConnectionInfo,
   ): Promise<Migration[]> {
-    const result = await connection.client.api.migrations(
-      connection.revisionId,
-    );
-
-    if (result.error) {
-      throw new Error(
-        `Failed to get migrations: ${JSON.stringify(result.error)}`,
-      );
-    }
-
-    return result.data as Migration[];
+    const revisionScope = this.getRevisionScope(connection);
+    const migrations = await revisionScope.getMigrations();
+    return migrations as Migration[];
   }
 
   private analyzeMigrations(migrations: Migration[]): SchemaSyncResult {
@@ -99,19 +92,14 @@ export class SyncSchemaService {
     const tablesRemoved: string[] = [];
     let migrationsApplied = 0;
 
-    for (const migration of migrations) {
-      const result = await connection.client.api.applyMigrations(
-        connection.draftRevisionId,
-        [migration],
-      );
+    const revisionScope = this.getRevisionScope(connection);
 
-      if (result.error) {
-        throw new Error(
-          `Failed to apply migration: ${JSON.stringify(result.error)}`,
-        );
-      }
+    for (const migration of migrations) {
+      const results = await revisionScope.applyMigrationsWithStatus([
+        migration,
+      ]);
 
-      const response = result.data[0];
+      const response = results[0];
 
       if (response.status === 'failed') {
         throw new Error(
@@ -152,4 +140,8 @@ export class SyncSchemaService {
       tablesRemoved,
     };
   }
+
+  private getRevisionScope(connection: ConnectionInfo): RevisionScope {
+    return connection.revisionScope;
+  }
 }

package/src/services/url/auth-prompt.service.ts

@@ -68,6 +68,6 @@ export class AuthPromptService {
     const normalizedUrl = baseUrl.endsWith('/')
       ? baseUrl.slice(0, -1)
       : baseUrl;
-    return `${normalizedUrl}/get-mcp-token`;
+    return `${normalizedUrl}/get-token`;
   }
 }

package/src/types/migration.types.ts

@@ -1,9 +1,9 @@
-import {
+import type {
   InitMigrationDto,
   RemoveMigrationDto,
   RenameMigrationDto,
   UpdateMigrationDto,
-} from 'src/__generated__/api';
+} from '@revisium/client';
 
 export type Migration =
   | InitMigrationDto