added assets support in export-query

.github/workflows/unit-test.yml

@@ -25,7 +25,7 @@ jobs:
       - name: Prune pnpm store
         run: pnpm store prune
       - name: Install Dependencies
-        run: pnpm install --frozen-lockfile
+        run: pnpm install --no-frozen-lockfile
 
       - name: Build all plugins
         run: |

.talismanrc

@@ -1,4 +1,12 @@
 fileignoreconfig:
-- filename: pnpm-lock.yaml
-  checksum: 2b0f2461ea1bb240a9210b9cf99dc403a756199712b7270f9792a590480451bd
+  - filename: pnpm-lock.yaml
+    checksum: 2b0f2461ea1bb240a9210b9cf99dc403a756199712b7270f9792a590480451bd
+  - filename: packages/contentstack-import/test/unit/import/modules/base-class.test.ts
+    checksum: fe372852d5f2f3f57ef62c603406c30ccecdb444c17133ac0b21dda399b962c0
+  - filename: packages/contentstack-bulk-operations/test/unit/commands/bulk-am-assets.test.ts
+    checksum: f8d21db7db0ca2eebe7cc40af0a59f02e74e1689efb6d50a1072dc5ca3e03e9b
+  - filename: packages/contentstack-export/src/export/modules/taxonomies.ts
+    checksum: b6d077118280bc88385405f504f921468a9fd490ac37a4a21f741be729fd1ca3
+  - filename: packages/contentstack-export/test/unit/export/modules/taxonomies.test.ts
+    checksum: cab2ad4d897d23f04f988c1f018a9583ab7f0ee1815994d7bc9fce23dea70073
 version: '1.0'

packages/contentstack-apps-cli/package.json

@@ -110,4 +110,4 @@
       "app:deploy": "APDP"
     }
   }
-}
+}

packages/contentstack-asset-management/src/index.ts

@@ -3,4 +3,5 @@ export * from './types';
 export * from './utils';
 export * from './export';
 export * from './import';
+export * from './query-export';
 export * from './import-setup';

packages/contentstack-asset-management/src/query-export/cs-assets-query-exporter.ts

@@ -0,0 +1,246 @@
+import { resolve as pResolve } from 'node:path';
+import { mkdir, writeFile } from 'node:fs/promises';
+import { Readable } from 'node:stream';
+import { log, handleAndLogError, configHandler } from '@contentstack/cli-utilities';
+
+import type { CsAssetsQueryExportOptions, CSAssetsAPIConfig, LinkedWorkspace } from '../types/cs-assets-api';
+import type { ExportContext } from '../types/export-types';
+import ExportAssetTypes from '../export/asset-types';
+import ExportFields from '../export/fields';
+import { CSAssetsExportAdapter } from '../export/base';
+import { getAssetItems, writeStreamToFile } from '../utils/export-helpers';
+import { runInBatches } from '../utils/concurrent-batch';
+
+const DEFAULT_ASSET_BATCH_SIZE = 100;
+const SEARCH_PAGE_LIMIT = 100;
+
+/**
+ * Query-based Contentstack Assets exporter.
+ * Exports only referenced asset UIDs from entries into the `spaces/` directory layout.
+ */
+export class CsAssetsQueryExporter {
+  private readonly options: CsAssetsQueryExportOptions;
+
+  constructor(options: CsAssetsQueryExportOptions) {
+    this.options = options;
+  }
+
+  async export(assetUIDs: string[]): Promise<void> {
+    const { linkedWorkspaces, exportDir, context } = this.options;
+
+    if (!assetUIDs.length) {
+      log.info('No asset UIDs to export for Contentstack Assets query export', context);
+      return;
+    }
+
+    if (!linkedWorkspaces.length) {
+      log.warn('No linked workspaces configured for Contentstack Assets query export', context);
+      return;
+    }
+
+    log.info(
+      `Starting Contentstack Assets query export (${assetUIDs.length} UID(s), ${linkedWorkspaces.length} space(s))`,
+      context,
+    );
+
+    const spacesRootPath = pResolve(exportDir, 'spaces');
+    await mkdir(spacesRootPath, { recursive: true });
+
+    const apiConfig: CSAssetsAPIConfig = {
+      baseURL: this.options.csAssetsUrl,
+      headers: { organization_uid: this.options.org_uid },
+      context,
+    };
+
+    const exportContext: ExportContext = {
+      spacesRootPath,
+      context,
+      securedAssets: this.options.securedAssets,
+      chunkFileSizeMb: this.options.chunkFileSizeMb,
+      apiConcurrency: this.options.apiConcurrency,
+      downloadAssetsConcurrency: this.options.downloadAssetsConcurrency,
+    };
+
+    const batchSize = this.options.assetBatchSize ?? DEFAULT_ASSET_BATCH_SIZE;
+
+    try {
+      await this.bootstrapSharedModules(apiConfig, exportContext, linkedWorkspaces[0].space_uid);
+
+      for (const workspace of linkedWorkspaces) {
+        try {
+          await this.exportWorkspaceAssets(apiConfig, exportContext, workspace, assetUIDs, batchSize);
+        } catch (err) {
+          handleAndLogError(
+            err,
+            { ...(context as Record<string, unknown>), spaceUid: workspace.space_uid },
+            `Failed Contentstack Assets query export for space ${workspace.space_uid}`,
+          );
+        }
+      }
+
+      log.success('Contentstack Assets query export completed', context);
+    } catch (err) {
+      handleAndLogError(err, context as Record<string, unknown>, 'Contentstack Assets query export failed');
+      throw err;
+    }
+  }
+
+  private async bootstrapSharedModules(
+    apiConfig: CSAssetsAPIConfig,
+    exportContext: ExportContext,
+    firstSpaceUid: string,
+  ): Promise<void> {
+    const sharedFieldsDir = pResolve(exportContext.spacesRootPath, 'fields');
+    const sharedAssetTypesDir = pResolve(exportContext.spacesRootPath, 'asset_types');
+    await mkdir(sharedFieldsDir, { recursive: true });
+    await mkdir(sharedAssetTypesDir, { recursive: true });
+
+    const exportAssetTypes = new ExportAssetTypes(apiConfig, exportContext);
+    const exportFields = new ExportFields(apiConfig, exportContext);
+    await Promise.all([exportAssetTypes.start(firstSpaceUid), exportFields.start(firstSpaceUid)]);
+  }
+
+  private async exportWorkspaceAssets(
+    apiConfig: CSAssetsAPIConfig,
+    exportContext: ExportContext,
+    workspace: LinkedWorkspace,
+    assetUIDs: string[],
+    batchSize: number,
+  ): Promise<void> {
+    const { branchName, context } = this.options;
+    const workspaceExporter = new QueryExportWorkspaceAdapter(apiConfig, exportContext);
+    await workspaceExporter.start(workspace, assetUIDs, branchName || 'main', batchSize);
+    log.debug(`Contentstack Assets query export finished for space ${workspace.space_uid}`, context);
+  }
+}
+
+/**
+ * Per-space export: search by UID, write metadata/files, download binaries.
+ */
+class QueryExportWorkspaceAdapter extends CSAssetsExportAdapter {
+  async start(
+    workspace: LinkedWorkspace,
+    assetUIDs: string[],
+    branchName: string,
+    uidBatchSize: number,
+  ): Promise<void> {
+    await this.init();
+
+    const spaceDir = pResolve(this.exportContext.spacesRootPath, workspace.space_uid);
+    await mkdir(spaceDir, { recursive: true });
+
+    const spaceResponse = await this.getSpace(workspace.space_uid);
+    const space = spaceResponse.space;
+    const metadata = {
+      ...space,
+      workspace_uid: workspace.uid,
+      is_default: workspace.is_default,
+      branch: branchName,
+    };
+    await writeFile(pResolve(spaceDir, 'metadata.json'), JSON.stringify(metadata, null, 2));
+
+    const assetsDir = pResolve(spaceDir, 'assets');
+    await mkdir(assetsDir, { recursive: true });
+
+    const spaceRef = { space_uid: workspace.space_uid, workspace: workspace.uid };
+    const assetItems = await this.searchAllAssets(assetUIDs, spaceRef, uidBatchSize);
+
+    const folders = assetItems.filter((item) => (item as { is_dir?: boolean }).is_dir === true);
+    const files = assetItems.filter((item) => (item as { is_dir?: boolean }).is_dir !== true);
+
+    await writeFile(pResolve(assetsDir, 'folders.json'), JSON.stringify(folders, null, 2));
+
+    await this.writeItemsToChunkedJson(
+      assetsDir,
+      'assets.json',
+      'assets',
+      ['uid', 'url', 'filename', 'file_name', 'parent_uid'],
+      files,
+    );
+
+    await this.downloadAssets(files, assetsDir, workspace.space_uid);
+  }
+
+  private async searchAllAssets(
+    assetUIDs: string[],
+    spaceRef: { space_uid: string; workspace: string },
+    uidBatchSize: number,
+  ): Promise<Array<Record<string, unknown>>> {
+    const seen = new Set<string>();
+    const results: Array<Record<string, unknown>> = [];
+
+    for (let i = 0; i < assetUIDs.length; i += uidBatchSize) {
+      const uidBatch = assetUIDs.slice(i, i + uidBatchSize);
+      let skip = 0;
+      let pageItems: unknown[];
+
+      do {
+        const response = await this.searchAssets({
+          assetUIDs: uidBatch,
+          spaces: [spaceRef],
+          skip,
+          limit: SEARCH_PAGE_LIMIT,
+        });
+        pageItems = getAssetItems(response);
+
+        if (pageItems.length === 0 && skip === 0) {
+          log.warn(
+            `Search returned 0 assets in space ${spaceRef.space_uid} for UID(s): [${uidBatch.join(', ')}]`,
+            this.exportContext.context,
+          );
+        }
+
+        for (const item of pageItems) {
+          const record = item as Record<string, unknown>;
+          const key = String(record.uid ?? record.asset_id ?? record._uid ?? '');
+          if (key && !seen.has(key)) {
+            seen.add(key);
+            results.push(record);
+          }
+        }
+
+        skip += pageItems.length;
+      } while (pageItems.length === SEARCH_PAGE_LIMIT);
+    }
+
+    return results;
+  }
+
+  private async downloadAssets(
+    items: Array<Record<string, unknown>>,
+    assetsDir: string,
+    spaceUid: string,
+  ): Promise<void> {
+    const downloadable = items.filter((asset) => Boolean(asset.url && (asset.uid ?? asset._uid)));
+    if (downloadable.length === 0) {
+      log.debug(`No downloadable assets for space ${spaceUid}`, this.exportContext.context);
+      return;
+    }
+
+    const filesDir = pResolve(assetsDir, 'files');
+    await mkdir(filesDir, { recursive: true });
+
+    const securedAssets = this.exportContext.securedAssets ?? false;
+    const authtoken = securedAssets ? configHandler.get('authtoken') : null;
+
+    await runInBatches(downloadable, this.downloadAssetsBatchConcurrency, async (asset) => {
+      const uid = String(asset.uid ?? asset._uid);
+      const url = String(asset.url);
+      const filename = String(asset.filename ?? asset.file_name ?? 'asset');
+      try {
+        const separator = url.includes('?') ? '&' : '?';
+        const downloadUrl = securedAssets && authtoken ? `${url}${separator}authtoken=${authtoken}` : url;
+        const response = await fetch(downloadUrl);
+        if (!response.ok) throw new Error(`HTTP ${response.status}`);
+        const body = response.body;
+        if (!body) throw new Error('No response body');
+        const nodeStream = Readable.fromWeb(body as Parameters<typeof Readable.fromWeb>[0]);
+        const assetFolderPath = pResolve(filesDir, uid);
+        await mkdir(assetFolderPath, { recursive: true });
+        await writeStreamToFile(nodeStream, pResolve(assetFolderPath, filename));
+      } catch (e) {
+        log.debug(`Failed to download asset ${uid} in space ${spaceUid}: ${e}`, this.exportContext.context);
+      }
+    });
+  }
+}

packages/contentstack-asset-management/src/query-export/index.ts

@@ -0,0 +1 @@
+export { CsAssetsQueryExporter } from './cs-assets-query-exporter';

packages/contentstack-asset-management/src/types/cs-assets-api.ts

@@ -141,6 +141,30 @@ export type BulkMoveAssetsResponse = {
  * Adapter interface for Contentstack Assets API calls.
  * Used by export and (future) import.
  */
+/** Space + workspace pair for Contentstack Assets search API. */
+export type SearchSpaceRef = {
+  space_uid: string;
+  workspace: string;
+};
+
+/** Parameters for POST /api/search (asset query export). */
+export type SearchAssetsParams = {
+  assetUIDs: string[];
+  spaces: SearchSpaceRef[];
+  skip?: number;
+  limit?: number;
+};
+
+/** Response shape from POST /api/search for assets. */
+export type SearchAssetsResponse = {
+  count?: number;
+  relation?: string;
+  assets?: unknown[];
+  items?: unknown[];
+  results?: unknown[];
+  folders?: unknown[];
+};
+
 export interface ICSAssetsAdapter {
   init(): Promise<void>;
   listSpaces(): Promise<SpacesListResponse>;
@@ -149,6 +173,7 @@ export interface ICSAssetsAdapter {
   getWorkspaceAssets(spaceUid: string, workspaceUid?: string): Promise<unknown>;
   getWorkspaceFolders(spaceUid: string, workspaceUid?: string): Promise<unknown>;
   getWorkspaceAssetTypes(spaceUid: string): Promise<AssetTypesResponse>;
+  searchAssets(params: SearchAssetsParams): Promise<SearchAssetsResponse>;
   bulkDeleteAssets(
     spaceUid: string,
     workspaceUid: string | undefined,
@@ -161,6 +186,23 @@ export interface ICSAssetsAdapter {
   ): Promise<BulkMoveAssetsResponse>;
 }
 
+/** Options for query-based Contentstack Assets export (referenced assets from entries). */
+export type CsAssetsQueryExportOptions = {
+  linkedWorkspaces: LinkedWorkspace[];
+  exportDir: string;
+  branchName: string;
+  csAssetsUrl: string;
+  org_uid: string;
+  apiKey?: string;
+  context?: Record<string, unknown>;
+  securedAssets?: boolean;
+  chunkFileSizeMb?: number;
+  apiConcurrency?: number;
+  downloadAssetsConcurrency?: number;
+  /** Max UIDs per search request ($in batch). */
+  assetBatchSize?: number;
+};
+
 /**
  * Options for exporting space structure (used by export app after fetching linked workspaces).
  */