@cmdoss/memwal-sdk 0.6.0

package/dist/services/CrossContextPermissionService.js

@@ -0,0 +1,337 @@
+/**
+ * Cross-Context Permission Service
+ *
+ * Manages cross-context access permissions for the Personal Data Wallet.
+ * Enables apps to request and manage access to data from other app contexts.
+ */
+import { Transaction } from '@mysten/sui/transactions';
+import { normalizeSuiAddress } from '@mysten/sui/utils';
+/**
+ * Service for managing cross-context permissions
+ */
+export class CrossContextPermissionService {
+    constructor(config, client) {
+        this.packageId = config.packageId;
+        this.accessRegistryId = config.accessRegistryId;
+        this.client = client;
+    }
+    /**
+     * Register a new context wallet for an app
+     *
+     * @param options - Context registration options
+     * @param signer - Transaction signer
+     * @returns Transaction digest
+     */
+    async registerContextWallet(options, signer) {
+        const tx = this.buildRegisterContextWalletTransaction(options);
+        const result = await this.client.signAndExecuteTransaction({
+            transaction: tx,
+            signer,
+            options: {
+                showEffects: true,
+                showEvents: true,
+            },
+        });
+        // Wait for transaction to be finalized to prevent gas coin version conflicts
+        if (result.digest) {
+            await this.client.waitForTransaction({ digest: result.digest });
+        }
+        if (result.effects?.status?.status !== 'success') {
+            throw new Error(`Failed to register context: ${result.effects?.status?.error}`);
+        }
+        return result.digest;
+    }
+    /**
+     * Build transaction to register a context wallet
+     *
+     * @param options - Context registration options
+     * @returns Transaction object
+     */
+    buildRegisterContextWalletTransaction(options) {
+        const tx = new Transaction();
+        tx.moveCall({
+            target: `${this.packageId}::seal_access_control::register_context_wallet`,
+            arguments: [
+                tx.object(this.accessRegistryId),
+                tx.pure.address(normalizeSuiAddress(options.contextWallet)),
+                tx.pure.u64(options.derivationIndex),
+                tx.pure.string(options.appHint ?? ''),
+                tx.object('0x6'), // Clock object
+            ],
+        });
+        return tx;
+    }
+    /**
+     * Grant cross-context access permission
+     *
+     * @param options - Permission grant options
+     * @param signer - Transaction signer
+     * @returns Transaction digest
+     */
+    async grantWalletAllowlistAccess(options, signer) {
+        const tx = this.buildGrantWalletAllowlistTransaction(options);
+        const result = await this.client.signAndExecuteTransaction({
+            transaction: tx,
+            signer,
+            options: {
+                showEffects: true,
+                showEvents: true,
+            },
+        });
+        // Wait for transaction to be finalized to prevent gas coin version conflicts
+        if (result.digest) {
+            await this.client.waitForTransaction({ digest: result.digest });
+        }
+        if (result.effects?.status?.status !== 'success') {
+            throw new Error(`Failed to grant access: ${result.effects?.status?.error}`);
+        }
+        return result.digest;
+    }
+    /**
+     * Build transaction to grant cross-context access
+     *
+     * @param options - Permission grant options
+     * @returns Transaction object
+     */
+    buildGrantWalletAllowlistTransaction(options) {
+        const tx = new Transaction();
+        tx.moveCall({
+            target: `${this.packageId}::seal_access_control::grant_wallet_allowlist_access`,
+            arguments: [
+                tx.object(this.accessRegistryId),
+                tx.pure.address(normalizeSuiAddress(options.requestingWallet)),
+                tx.pure.address(normalizeSuiAddress(options.targetWallet)),
+                tx.pure.string(options.scope ?? 'read'),
+                tx.pure.string(options.accessLevel),
+                tx.pure.u64(options.expiresAt),
+                tx.object('0x6'), // Clock object
+            ],
+        });
+        return tx;
+    }
+    /**
+     * Revoke cross-context access permission
+     *
+     * @param options - Permission revocation options
+     * @param signer - Transaction signer
+     * @returns Transaction digest
+     */
+    async revokeWalletAllowlistAccess(options, signer) {
+        const tx = this.buildRevokeWalletAllowlistTransaction(options);
+        const result = await this.client.signAndExecuteTransaction({
+            transaction: tx,
+            signer,
+            options: {
+                showEffects: true,
+                showEvents: true,
+            },
+        });
+        // Wait for transaction to be finalized to prevent gas coin version conflicts
+        if (result.digest) {
+            await this.client.waitForTransaction({ digest: result.digest });
+        }
+        if (result.effects?.status?.status !== 'success') {
+            throw new Error(`Failed to revoke access: ${result.effects?.status?.error}`);
+        }
+        return result.digest;
+    }
+    /**
+     * Build transaction to revoke cross-context access
+     *
+     * @param options - Permission revocation options
+     * @returns Transaction object
+     */
+    buildRevokeWalletAllowlistTransaction(options) {
+        const tx = new Transaction();
+        tx.moveCall({
+            target: `${this.packageId}::seal_access_control::revoke_wallet_allowlist_access`,
+            arguments: [
+                tx.object(this.accessRegistryId),
+                tx.pure.address(normalizeSuiAddress(options.requestingWallet)),
+                tx.pure.address(normalizeSuiAddress(options.targetWallet)),
+                tx.pure.string(options.scope ?? 'read'),
+            ],
+        });
+        return tx;
+    }
+    /**
+     * Build seal_approve transaction using capability module
+     *
+     * Uses pdw::capability::seal_approve which requires:
+     * - id: vector<u8> - SEAL key identifier (MUST be first parameter!)
+     * - cap: &MemoryCap - Reference to the capability object
+     *
+     * IMPORTANT: SEAL key server extracts 'id' from the FIRST PTB argument
+     *
+     * @param keyId - SEAL key ID bytes (computed from owner + nonce)
+     * @param memoryCapId - MemoryCap object ID on Sui
+     * @returns Transaction object
+     */
+    buildSealApproveTransaction(keyId, memoryCapId) {
+        const tx = new Transaction();
+        tx.moveCall({
+            target: `${this.packageId}::capability::seal_approve`,
+            arguments: [
+                tx.pure.vector('u8', Array.from(keyId)), // id MUST be first (SEAL requirement)
+                tx.object(memoryCapId), // MemoryCap object reference
+            ],
+        });
+        return tx;
+    }
+    /**
+     * Build seal_approve transaction (legacy - for backward compatibility)
+     * @deprecated Use buildSealApproveTransaction with memoryCapId instead
+     */
+    buildSealApproveTransactionLegacy(contentId, requestingWallet) {
+        const tx = new Transaction();
+        tx.moveCall({
+            target: `${this.packageId}::seal_access_control::seal_approve`,
+            arguments: [
+                tx.pure.vector('u8', Array.from(contentId)),
+                tx.pure.address(normalizeSuiAddress(requestingWallet)),
+                tx.object(this.accessRegistryId),
+                tx.object('0x6'), // Clock object
+            ],
+        });
+        return tx;
+    }
+    /**
+     * Query wallet allowlist permissions filtered by requester, target, or scope
+     */
+    async queryWalletPermissions(options) {
+        const events = await this.fetchWalletAllowlistEvents();
+        const state = this.reduceWalletAllowlistEvents(events);
+        const normalizedRequester = options.requestingWallet ? normalizeSuiAddress(options.requestingWallet) : undefined;
+        const normalizedTarget = options.targetWallet ? normalizeSuiAddress(options.targetWallet) : undefined;
+        const scopeFilter = options.scope ?? undefined;
+        return Array.from(state.values())
+            .filter((permission) => {
+            if (normalizedRequester && permission.requestingWallet !== normalizedRequester) {
+                return false;
+            }
+            if (normalizedTarget && permission.targetWallet !== normalizedTarget) {
+                return false;
+            }
+            if (scopeFilter && permission.scope !== scopeFilter) {
+                return false;
+            }
+            return true;
+        });
+    }
+    async listGrantsByTarget(targetWallet, scope) {
+        return this.queryWalletPermissions({ targetWallet, scope });
+    }
+    async listGrantsByRequester(requestingWallet, scope) {
+        return this.queryWalletPermissions({ requestingWallet, scope });
+    }
+    /**
+     * Determine whether a wallet currently has allowlist permission
+     */
+    async hasWalletPermission(options) {
+        const permissions = await this.queryWalletPermissions(options);
+        const now = Date.now();
+        return permissions.some(permission => {
+            const expiry = permission.expiresAt;
+            return expiry === 0 || expiry > now;
+        });
+    }
+    /**
+     * List target wallets this requester can access for an optional scope
+     */
+    async getAccessibleWallets(requestingWallet, scope = 'read') {
+        const permissions = await this.queryWalletPermissions({ requestingWallet, scope });
+        const now = Date.now();
+        return permissions
+            .filter(permission => permission.expiresAt === 0 || permission.expiresAt > now)
+            .map(permission => permission.targetWallet);
+    }
+    async getWalletAllowlistHistory(filter) {
+        const events = await this.fetchWalletAllowlistEvents();
+        const normalizedRequester = filter?.requestingWallet
+            ? normalizeSuiAddress(filter.requestingWallet)
+            : undefined;
+        const normalizedTarget = filter?.targetWallet
+            ? normalizeSuiAddress(filter.targetWallet)
+            : undefined;
+        return events
+            .filter((event) => {
+            if (normalizedRequester && event.requestingWallet !== normalizedRequester) {
+                return false;
+            }
+            if (normalizedTarget && event.targetWallet !== normalizedTarget) {
+                return false;
+            }
+            return true;
+        })
+            .map((event) => ({
+            timestamp: event.grantedAt,
+            action: event.granted ? 'grant' : 'revoke',
+            requestingWallet: event.requestingWallet,
+            targetWallet: event.targetWallet,
+            scope: event.scope,
+            accessLevel: event.accessLevel,
+            expiresAt: event.expiresAt,
+            grantedBy: event.grantedBy,
+        }))
+            .sort((a, b) => a.timestamp - b.timestamp);
+    }
+    async fetchWalletAllowlistEvents() {
+        const response = await this.client.queryEvents({
+            query: {
+                MoveEventType: `${this.packageId}::seal_access_control::WalletAllowlistChanged`,
+            },
+            limit: 1000,
+            order: 'ascending',
+        });
+        const events = [];
+        for (const event of response.data) {
+            const parsed = event.parsedJson;
+            if (!parsed) {
+                continue;
+            }
+            const requestingWallet = normalizeSuiAddress(String(parsed.requester_wallet));
+            const targetWallet = normalizeSuiAddress(String(parsed.target_wallet));
+            const scope = String(parsed.scope ?? 'read');
+            const accessLevel = String(parsed.access_level ?? 'read');
+            const granted = Boolean(parsed.granted);
+            const expiresAt = Number(parsed.expires_at ?? 0);
+            const grantedBy = normalizeSuiAddress(String(parsed.granted_by ?? requestingWallet));
+            const grantedAt = Number(event.timestampMs ?? Date.now());
+            const key = `${requestingWallet}-${targetWallet}-${scope}`;
+            events.push({
+                key,
+                requestingWallet,
+                targetWallet,
+                scope,
+                accessLevel,
+                granted,
+                expiresAt,
+                grantedAt,
+                grantedBy,
+            });
+        }
+        return events;
+    }
+    reduceWalletAllowlistEvents(events) {
+        const state = new Map();
+        const sorted = [...events].sort((a, b) => a.grantedAt - b.grantedAt);
+        for (const event of sorted) {
+            if (event.granted) {
+                state.set(event.key, {
+                    requestingWallet: event.requestingWallet,
+                    targetWallet: event.targetWallet,
+                    scope: event.scope,
+                    accessLevel: event.accessLevel,
+                    grantedAt: event.grantedAt,
+                    expiresAt: event.expiresAt,
+                    grantedBy: event.grantedBy,
+                });
+            }
+            else {
+                state.delete(event.key);
+            }
+        }
+        return state;
+    }
+}
+//# sourceMappingURL=CrossContextPermissionService.js.map

package/dist/services/CrossContextPermissionService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"CrossContextPermissionService.js","sourceRoot":"","sources":["../../src/services/CrossContextPermissionService.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AACvD,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAyExD;;GAEG;AACH,MAAM,OAAO,6BAA6B;IAKxC,YAAY,MAAoC,EAAE,MAAiB;QACjE,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QAClC,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,CAAC;QAChD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,qBAAqB,CACzB,OAAqC,EACrC,MAAc;QAEd,MAAM,EAAE,GAAG,IAAI,CAAC,qCAAqC,CAAC,OAAO,CAAC,CAAC;QAE/D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,yBAAyB,CAAC;YACzD,WAAW,EAAE,EAAE;YACf,MAAM;YACN,OAAO,EAAE;gBACP,WAAW,EAAE,IAAI;gBACjB,UAAU,EAAE,IAAI;aACjB;SACF,CAAC,CAAC;QAEH,6EAA6E;QAC7E,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,MAAM,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QAClE,CAAC;QAED,IAAI,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,KAAK,SAAS,EAAE,CAAC;YACjD,MAAM,IAAI,KAAK,CAAC,+BAA+B,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QAClF,CAAC;QAED,OAAO,MAAM,CAAC,MAAM,CAAC;IACvB,CAAC;IAED;;;;;OAKG;IACH,qCAAqC,CAAC,OAAqC;QACzE,MAAM,EAAE,GAAG,IAAI,WAAW,EAAE,CAAC;QAE7B,EAAE,CAAC,QAAQ,CAAC;YACV,MAAM,EAAE,GAAG,IAAI,CAAC,SAAS,gDAAgD;YACzE,SAAS,EAAE;gBACT,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC;gBAChC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;gBAC3D,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,eAAe,CAAC;gBACpC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;gBACrC,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,eAAe;aAClC;SACF,CAAC,CAAC;QAEH,OAAO,EAAE,CAAC;IACZ,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,0BAA0B,CAC9B,OAAoC,EACpC,MAAc;QAEd,MAAM,EAAE,GAAG,IAAI,CAAC,oCAAoC,CAAC,OAAO,CAAC,CAAC;QAE9D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,yBAAyB,CAAC;YACzD,WAAW,EAAE,EAAE;YACf,MAAM;YACN,OAAO,EAAE;gBACP,WAAW,EAAE,IAAI;gBACjB,UAAU,EAAE,IAAI;aACjB;SACF,CAAC,CAAC;QAEH,6EAA6E;QAC7E,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,MAAM,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QAClE,CAAC;QAED,IAAI,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,KAAK,SAAS,EAAE,CAAC;YACjD,MAAM,IAAI,KAAK,CAAC,2BAA2B,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QAC9E,CAAC;QAED,OAAO,MAAM,CAAC,MAAM,CAAC;IACvB,CAAC;IAED;;;;;OAKG;IACH,oCAAoC,CAClC,OAAoC;QAEpC,MAAM,EAAE,GAAG,IAAI,WAAW,EAAE,CAAC;QAE7B,EAAE,CAAC,QAAQ,CAAC;YACV,MAAM,EAAE,GAAG,IAAI,CAAC,SAAS,sDAAsD;YAC/E,SAAS,EAAE;gBACT,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC;gBAChC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;gBAC9D,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;gBAC1D,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,IAAI,MAAM,CAAC;gBACvC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;gBACnC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC;gBAC9B,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,eAAe;aAClC;SACF,CAAC,CAAC;QAEH,OAAO,EAAE,CAAC;IACZ,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,2BAA2B,CAC/B,OAAqC,EACrC,MAAc;QAEd,MAAM,EAAE,GAAG,IAAI,CAAC,qCAAqC,CAAC,OAAO,CAAC,CAAC;QAE/D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,yBAAyB,CAAC;YACzD,WAAW,EAAE,EAAE;YACf,MAAM;YACN,OAAO,EAAE;gBACP,WAAW,EAAE,IAAI;gBACjB,UAAU,EAAE,IAAI;aACjB;SACF,CAAC,CAAC;QAEH,6EAA6E;QAC7E,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,MAAM,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QAClE,CAAC;QAED,IAAI,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,KAAK,SAAS,EAAE,CAAC;YACjD,MAAM,IAAI,KAAK,CAAC,4BAA4B,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/E,CAAC;QAED,OAAO,MAAM,CAAC,MAAM,CAAC;IACvB,CAAC;IAED;;;;;OAKG;IACH,qCAAqC,CACnC,OAAqC;QAErC,MAAM,EAAE,GAAG,IAAI,WAAW,EAAE,CAAC;QAE7B,EAAE,CAAC,QAAQ,CAAC;YACV,MAAM,EAAE,GAAG,IAAI,CAAC,SAAS,uDAAuD;YAChF,SAAS,EAAE;gBACT,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC;gBAChC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;gBAC9D,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;gBAC1D,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,IAAI,MAAM,CAAC;aACxC;SACF,CAAC,CAAC;QAEH,OAAO,EAAE,CAAC;IACZ,CAAC;IAED;;;;;;;;;;;;OAYG;IACH,2BAA2B,CACzB,KAAiB,EACjB,WAAmB;QAEnB,MAAM,EAAE,GAAG,IAAI,WAAW,EAAE,CAAC;QAE7B,EAAE,CAAC,QAAQ,CAAC;YACV,MAAM,EAAE,GAAG,IAAI,CAAC,SAAS,4BAA4B;YACrD,SAAS,EAAE;gBACT,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,sCAAsC;gBAC/E,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,6BAA6B;aACtD;SACF,CAAC,CAAC;QAEH,OAAO,EAAE,CAAC;IACZ,CAAC;IAED;;;OAGG;IACH,iCAAiC,CAC/B,SAAqB,EACrB,gBAAwB;QAExB,MAAM,EAAE,GAAG,IAAI,WAAW,EAAE,CAAC;QAE7B,EAAE,CAAC,QAAQ,CAAC;YACV,MAAM,EAAE,GAAG,IAAI,CAAC,SAAS,qCAAqC;YAC9D,SAAS,EAAE;gBACT,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC3C,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,CAAC;gBACtD,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC;gBAChC,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,eAAe;aAClC;SACF,CAAC,CAAC;QAEH,OAAO,EAAE,CAAC;IACZ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,sBAAsB,CAAC,OAA0C;QACrE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACvD,MAAM,KAAK,GAAG,IAAI,CAAC,2BAA2B,CAAC,MAAM,CAAC,CAAC;QAEvD,MAAM,mBAAmB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC,CAAC,mBAAmB,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACjH,MAAM,gBAAgB,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,mBAAmB,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACtG,MAAM,WAAW,GAAG,OAAO,CAAC,KAAK,IAAI,SAAS,CAAC;QAE/C,OAAO,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;aAC9B,MAAM,CAAC,CAAC,UAAU,EAAE,EAAE;YACrB,IAAI,mBAAmB,IAAI,UAAU,CAAC,gBAAgB,KAAK,mBAAmB,EAAE,CAAC;gBAC/E,OAAO,KAAK,CAAC;YACf,CAAC;YACD,IAAI,gBAAgB,IAAI,UAAU,CAAC,YAAY,KAAK,gBAAgB,EAAE,CAAC;gBACrE,OAAO,KAAK,CAAC;YACf,CAAC;YACD,IAAI,WAAW,IAAI,UAAU,CAAC,KAAK,KAAK,WAAW,EAAE,CAAC;gBACpD,OAAO,KAAK,CAAC;YACf,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,YAAoB,EAAE,KAAc;QAC3D,OAAO,IAAI,CAAC,sBAAsB,CAAC,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,gBAAwB,EAAE,KAAc;QAClE,OAAO,IAAI,CAAC,sBAAsB,CAAC,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC,CAAC;IAClE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,mBAAmB,CAAC,OAAiC;QACzD,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC;QAC/D,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,OAAO,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE;YACnC,MAAM,MAAM,GAAG,UAAU,CAAC,SAAS,CAAC;YACpC,OAAO,MAAM,KAAK,CAAC,IAAI,MAAM,GAAG,GAAG,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,oBAAoB,CAAC,gBAAwB,EAAE,QAAgB,MAAM;QACzE,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC,CAAC;QACnF,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,OAAO,WAAW;aACf,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,KAAK,CAAC,IAAI,UAAU,CAAC,SAAS,GAAG,GAAG,CAAC;aAC9E,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,yBAAyB,CAC7B,MAAqC;QAErC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACvD,MAAM,mBAAmB,GAAG,MAAM,EAAE,gBAAgB;YAClD,CAAC,CAAC,mBAAmB,CAAC,MAAM,CAAC,gBAAgB,CAAC;YAC9C,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,gBAAgB,GAAG,MAAM,EAAE,YAAY;YAC3C,CAAC,CAAC,mBAAmB,CAAC,MAAM,CAAC,YAAY,CAAC;YAC1C,CAAC,CAAC,SAAS,CAAC;QAEd,OAAO,MAAM;aACV,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YAChB,IAAI,mBAAmB,IAAI,KAAK,CAAC,gBAAgB,KAAK,mBAAmB,EAAE,CAAC;gBAC1E,OAAO,KAAK,CAAC;YACf,CAAC;YACD,IAAI,gBAAgB,IAAI,KAAK,CAAC,YAAY,KAAK,gBAAgB,EAAE,CAAC;gBAChE,OAAO,KAAK,CAAC;YACf,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC;aACD,GAAG,CAA8B,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YAC5C,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,MAAM,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ;YAC1C,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;YACxC,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,WAAW,EAAE,KAAK,CAAC,WAAW;YAC9B,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,SAAS,EAAE,KAAK,CAAC,SAAS;SAC3B,CAAC,CAAC;aACF,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;IAC/C,CAAC;IAEO,KAAK,CAAC,0BAA0B;QACtC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;YAC7C,KAAK,EAAE;gBACL,aAAa,EAAE,GAAG,IAAI,CAAC,SAAS,+CAA+C;aAChF;YACD,KAAK,EAAE,IAAI;YACX,KAAK,EAAE,WAAW;SACnB,CAAC,CAAC;QAEH,MAAM,MAAM,GAA2B,EAAE,CAAC;QAE1C,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;YAClC,MAAM,MAAM,GAAG,KAAK,CAAC,UAAiB,CAAC;YACvC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,SAAS;YACX,CAAC;YAED,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,CAAC;YAC9E,MAAM,YAAY,GAAG,mBAAmB,CAAC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC;YACvE,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,CAAC;YAC7C,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,YAAY,IAAI,MAAM,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACxC,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,UAAU,IAAI,CAAC,CAAC,CAAC;YACjD,MAAM,SAAS,GAAG,mBAAmB,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,IAAI,gBAAgB,CAAC,CAAC,CAAC;YACrF,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,WAAW,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;YAC1D,MAAM,GAAG,GAAG,GAAG,gBAAgB,IAAI,YAAY,IAAI,KAAK,EAAE,CAAC;YAE3D,MAAM,CAAC,IAAI,CAAC;gBACV,GAAG;gBACH,gBAAgB;gBAChB,YAAY;gBACZ,KAAK;gBACL,WAAW;gBACX,OAAO;gBACP,SAAS;gBACT,SAAS;gBACT,SAAS;aACV,CAAC,CAAC;QACL,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,2BAA2B,CAAC,MAA8B;QAChE,MAAM,KAAK,GAAG,IAAI,GAAG,EAAqC,CAAC;QAE3D,MAAM,MAAM,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;QAErE,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;gBAClB,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE;oBACnB,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;oBACxC,YAAY,EAAE,KAAK,CAAC,YAAY;oBAChC,KAAK,EAAE,KAAK,CAAC,KAAK;oBAClB,WAAW,EAAE,KAAK,CAAC,WAAW;oBAC9B,SAAS,EAAE,KAAK,CAAC,SAAS;oBAC1B,SAAS,EAAE,KAAK,CAAC,SAAS;oBAC1B,SAAS,EAAE,KAAK,CAAC,SAAS;iBAC3B,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;CACF"}

package/dist/services/EmbeddingService.d.ts

@@ -0,0 +1,206 @@
+/**
+ * EmbeddingService - AI SDK Integration
+ *
+ * Refactored to use Vercel AI SDK as the underlying embedding provider.
+ * Supports any AI SDK compatible provider (OpenAI, Google, Cohere, etc.)
+ * while maintaining backward compatibility with existing PDW code.
+ *
+ * OpenRouter now uses the official @openrouter/sdk instead of raw fetch calls.
+ *
+ * Key features:
+ * - Provider-agnostic: Accept any ai-sdk EmbeddingModel
+ * - Backward compatible: Existing code continues to work
+ * - Flexible configuration: Direct model OR provider config
+ */
+import type { EmbeddingModelV2 } from '@ai-sdk/provider';
+type EmbeddingModel<VALUE> = EmbeddingModelV2<VALUE>;
+export interface EmbeddingConfig {
+    /**
+     * Option 1: Direct ai-sdk model (most flexible)
+     * User provides their own EmbeddingModel from any provider
+     *
+     * For backward compatibility, also accepts string (treated as modelName)
+     *
+     * @example
+     * ```typescript
+     * import { openai } from '@ai-sdk/openai';
+     * const service = new EmbeddingService({
+     *   model: openai.embedding('text-embedding-3-large')
+     * });
+     *
+     * // Backward compatible:
+     * const service = new EmbeddingService({
+     *   model: 'text-embedding-004', // Treated as modelName
+     *   apiKey: 'your-key'
+     * });
+     * ```
+     */
+    model?: EmbeddingModel<string> | string;
+    /**
+     * Option 2: Provider-based configuration
+     * PDW creates the model from provider settings
+     *
+     * - google: Direct Google AI API
+     * - openai: Direct OpenAI API
+     * - openrouter: OpenRouter API gateway (supports multiple models)
+     * - cohere: Direct Cohere API
+     */
+    provider?: 'google' | 'openai' | 'openrouter' | 'cohere';
+    /**
+     * API key for the provider
+     * Falls back to environment variables:
+     * - GEMINI_API_KEY or GOOGLE_AI_API_KEY (for google)
+     * - OPENAI_API_KEY (for openai)
+     * - OPENROUTER_API_KEY (for openrouter)
+     * - COHERE_API_KEY (for cohere)
+     */
+    apiKey?: string;
+    /**
+     * Model name to use
+     * - Google: 'text-embedding-004', 'gemini-embedding-001'
+     * - OpenAI: 'text-embedding-3-small', 'text-embedding-3-large'
+     * - OpenRouter: 'google/gemini-embedding-001', 'openai/text-embedding-3-small', etc.
+     * - Cohere: 'embed-english-v3.0', 'embed-multilingual-v3.0'
+     */
+    modelName?: string;
+    /**
+     * Embedding dimensions (optional, provider-dependent)
+     * - Google: Up to 768
+     * - OpenAI: 256, 512, 1024, 1536, 3072 (depending on model)
+     * - OpenRouter: Depends on the underlying model
+     * - Cohere: Model-specific
+     */
+    dimensions?: number;
+    /**
+     * Rate limiting
+     */
+    requestsPerMinute?: number;
+}
+export interface EmbeddingOptions {
+    text: string;
+    type?: 'content' | 'metadata' | 'query';
+    taskType?: 'RETRIEVAL_QUERY' | 'RETRIEVAL_DOCUMENT' | 'SEMANTIC_SIMILARITY';
+}
+export interface EmbeddingResult {
+    vector: number[];
+    dimension: number;
+    model: string;
+    processingTime: number;
+    tokenCount?: number;
+}
+export interface BatchEmbeddingResult {
+    vectors: number[][];
+    dimension: number;
+    model: string;
+    totalProcessingTime: number;
+    averageProcessingTime: number;
+    successCount: number;
+    failedCount: number;
+}
+/**
+ * Embedding service using Vercel AI SDK
+ * Supports all AI SDK compatible providers
+ * OpenRouter uses the official @openrouter/sdk for better type safety
+ */
+export declare class EmbeddingService {
+    private embeddingModel;
+    private modelName;
+    private dimensions;
+    private requestCount;
+    private lastReset;
+    private readonly maxRequestsPerMinute;
+    private provider;
+    private apiKey;
+    private openRouterClient;
+    constructor(config?: EmbeddingConfig);
+    /**
+     * Resolve API key from config or environment
+     */
+    private resolveApiKey;
+    /**
+     * Get default model name for provider
+     */
+    private getDefaultModelName;
+    /**
+     * Get default dimensions for provider
+     */
+    private getDefaultDimensions;
+    /**
+     * Create embedding model from provider
+     */
+    private createModel;
+    /**
+     * Generate embedding for a single text
+     */
+    embedText(options: EmbeddingOptions): Promise<EmbeddingResult>;
+    /**
+     * Generate embedding using OpenRouter SDK
+     * Uses official @openrouter/sdk for embeddings
+     */
+    private embedTextOpenRouter;
+    /**
+     * Generate embeddings for multiple texts (batched)
+     */
+    embedBatch(texts: string[], options?: Omit<EmbeddingOptions, 'text'>): Promise<BatchEmbeddingResult>;
+    /**
+     * Generate batch embeddings using OpenRouter SDK
+     */
+    private embedBatchOpenRouter;
+    /**
+     * Get provider-specific options
+     */
+    private getProviderOptions;
+    /**
+     * Map PDW type to Google task type
+     */
+    private getGoogleTaskType;
+    /**
+     * Map PDW type to Cohere input type
+     */
+    private getCohereInputType;
+    /**
+     * Calculate cosine similarity between two vectors
+     */
+    calculateCosineSimilarity(vectorA: number[], vectorB: number[]): number;
+    /**
+     * Calculate Euclidean distance between two vectors
+     */
+    calculateEuclideanDistance(vectorA: number[], vectorB: number[]): number;
+    /**
+     * Normalize a vector to unit length
+     */
+    normalizeVector(vector: number[]): number[];
+    /**
+     * Find the most similar vectors to a query vector
+     */
+    findMostSimilar(queryVector: number[], candidateVectors: number[][], k?: number): Array<{
+        index: number;
+        similarity: number;
+        distance: number;
+    }>;
+    /**
+     * Get embedding statistics
+     */
+    getStats(): {
+        totalRequests: number;
+        requestsThisMinute: number;
+        model: string;
+        dimensions: number;
+        rateLimit: number;
+        provider: string;
+    };
+    /**
+     * Reset rate limiting counters
+     */
+    private resetRateLimit;
+    /**
+     * Check rate limiting and wait if necessary
+     */
+    private checkRateLimit;
+    /**
+     * Utility delay function
+     */
+    private delay;
+}
+export default EmbeddingService;
+//# sourceMappingURL=EmbeddingService.d.ts.map

package/dist/services/EmbeddingService.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"EmbeddingService.d.ts","sourceRoot":"","sources":["../../src/services/EmbeddingService.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAEH,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAOzD,KAAK,cAAc,CAAC,KAAK,IAAI,gBAAgB,CAAC,KAAK,CAAC,CAAC;AAQrD,MAAM,WAAW,eAAe;IAC9B;;;;;;;;;;;;;;;;;;;OAmBG;IACH,KAAK,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC,GAAG,MAAM,CAAC;IAExC;;;;;;;;OAQG;IACH,QAAQ,CAAC,EAAE,QAAQ,GAAG,QAAQ,GAAG,YAAY,GAAG,QAAQ,CAAC;IAEzD;;;;;;;OAOG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB;;;;;;OAMG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IAEnB;;;;;;OAMG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB;;OAEG;IACH,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AAED,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,SAAS,GAAG,UAAU,GAAG,OAAO,CAAC;IACxC,QAAQ,CAAC,EAAE,iBAAiB,GAAG,oBAAoB,GAAG,qBAAqB,CAAC;CAC7E;AAED,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,cAAc,EAAE,MAAM,CAAC;IACvB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,oBAAoB;IACnC,OAAO,EAAE,MAAM,EAAE,EAAE,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,mBAAmB,EAAE,MAAM,CAAC;IAC5B,qBAAqB,EAAE,MAAM,CAAC;IAC9B,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED;;;;GAIG;AACH,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,cAAc,CAAuC;IAC7D,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,YAAY,CAAK;IACzB,OAAO,CAAC,SAAS,CAAc;IAC/B,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAS;IAC9C,OAAO,CAAC,QAAQ,CAA2D;IAC3E,OAAO,CAAC,MAAM,CAAc;IAC5B,OAAO,CAAC,gBAAgB,CAA2B;gBAEvC,MAAM,GAAE,eAAoB;IAsExC;;OAEG;IACH,OAAO,CAAC,aAAa;IAiBrB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAe3B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAe5B;;OAEG;IACH,OAAO,CAAC,WAAW;IAgDnB;;OAEG;IACG,SAAS,CAAC,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,eAAe,CAAC;IA2CpE;;;OAGG;YACW,mBAAmB;IAuCjC;;OAEG;IACG,UAAU,CACd,KAAK,EAAE,MAAM,EAAE,EACf,OAAO,GAAE,IAAI,CAAC,gBAAgB,EAAE,MAAM,CAAM,GAC3C,OAAO,CAAC,oBAAoB,CAAC;IA2ChC;;OAEG;YACW,oBAAoB;IA2ClC;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAmC1B;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAazB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAW1B;;OAEG;IACH,yBAAyB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,MAAM;IAwBvE;;OAEG;IACH,0BAA0B,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,MAAM;IAcxE;;OAEG;IACH,eAAe,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,EAAE;IAU3C;;OAEG;IACH,eAAe,CACb,WAAW,EAAE,MAAM,EAAE,EACrB,gBAAgB,EAAE,MAAM,EAAE,EAAE,EAC5B,CAAC,GAAE,MAAU,GACZ,KAAK,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC;IAajE;;OAEG;IACH,QAAQ,IAAI;QACV,aAAa,EAAE,MAAM,CAAC;QACtB,kBAAkB,EAAE,MAAM,CAAC;QAC3B,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,MAAM,CAAC;QACnB,SAAS,EAAE,MAAM,CAAC;QAClB,QAAQ,EAAE,MAAM,CAAC;KAClB;IAcD;;OAEG;IACH,OAAO,CAAC,cAAc;IAQtB;;OAEG;YACW,cAAc;IAe5B;;OAEG;IACH,OAAO,CAAC,KAAK;CAGd;AAED,eAAe,gBAAgB,CAAC"}