@codeyam/codeyam-cli 0.1.0-staging.1 → 0.1.0-staging.28f73cf

package/analyzer-template/packages/analyze/src/lib/files/analyze/gatherEntityMap.ts

@@ -73,14 +73,16 @@ export default async function gatherEntityMap({
     );
 
     if (!importedFile) {
+      // Skip dependencies when the file is not found in project.files.
+      // This commonly happens when a universal mock intercepts the import:
+      // 1. TypeScript resolves @formbricks/types/errors to the mock file
+      // 2. The original packages/types/errors.ts is never imported
+      // 3. So it's not in getSourceFiles() and thus not in project.files
+      // 4. Since the mock handles the import, we don't need to analyze the original
       console.log(
-        `CodeYam Error: File not found while gathering entity map: ${importedExportFilePath}`,
-        entity.filePath,
-        entity.name,
-      );
-      throw new Error(
-        `File not found while gathering entity map: ${importedExportFilePath}`,
+        `CodeYam Warning: Skipping dependency - file not in project.files (likely mocked): ${importedExportFilePath}`,
       );
+      continue;
     }
 
     let importedEntity = context.getEntity(

package/analyzer-template/packages/analyze/src/lib/files/analyze/trackEntityCircularDependencies.ts

@@ -99,19 +99,11 @@ function traverseEntity(
     );
 
     if (!dependentEntity) {
-      console.log(
-        'CodeYam Error: trackEntityCircularDependencies: Dependent entity not found',
-        {
-          entityFilePath: entity.filePath,
-          entityName: entity.name,
-          dependencyFilePath,
-          dependencyName,
-          fileEntities: context.getEntityNamesForFile(dependencyFilePath),
-        },
-      );
-      throw new Error(
-        'trackEntityCircularDependencies: Dependent entity not found',
-      );
+      // Skip dependencies when the entity is not found in context.
+      // This commonly happens when a universal mock intercepts the import:
+      // - gatherEntityMap skips adding the entity because the file isn't in project.files
+      // - The mock handles the import at runtime, so we don't need to track it
+      continue;
     }
 
     // Recursively traverse (will skip if already visited)

package/analyzer-template/packages/analyze/src/lib/files/analyzeChange.ts

@@ -16,7 +16,12 @@ import { AI } from '~codeyam/ai';
 import extractDiffLines from './extractDiffLines';
 import analyzeCodeChange from './analyzeCodeChange';
 import relevantDiffPart from './relevantDiffPart';
-import { awsLog, awsLogDebugLevel, pushAnalysisError } from '~codeyam/utils';
+import {
+  awsLog,
+  awsLogDebugLevel,
+  pushAnalysisError,
+  Semaphore,
+} from '~codeyam/utils';
 
 export interface AnalyzeChangeArgs {
   previousAnalysis: Analysis;
@@ -29,6 +34,8 @@ export interface AnalyzeChangeArgs {
   steps?: string[];
   updateProgress?: (detail: string) => void;
   model: AI.Model;
+  /** Semaphore to serialize CPU-intensive Structure phase */
+  structureSemaphore?: Semaphore;
 }
 
 export default async function analyzeChange({
@@ -42,6 +49,7 @@ export default async function analyzeChange({
   steps,
   updateProgress,
   model,
+  structureSemaphore,
 }: AnalyzeChangeArgs) {
   analysis.entity = entity;
   awsLog('CodeYam: Analyzing change entity', {
@@ -79,24 +87,35 @@ export default async function analyzeChange({
       });
     }
 
+    // Structure phase: CPU-intensive, serialize via semaphore if provided
     if (!steps || steps.includes(Step.Structure)) {
-      awsLogDebugLevel(
-        1,
-        `Merging dependent data structures for ${entity.name}...`,
-      );
+      if (structureSemaphore) {
+        await structureSemaphore.acquire();
+      }
+      try {
+        awsLogDebugLevel(
+          1,
+          `Merging dependent data structures for ${entity.name}...`,
+        );
 
-      const { step: dataStructureStep } = await recordStep(
-        'generateDataStructure',
-        () =>
-          generateDataStructure({
-            entity,
-            dependentAnalyses,
-            analysis,
-          }),
-      );
-      analysis.status.steps.push(dataStructureStep);
+        const { step: dataStructureStep } = await recordStep(
+          'generateDataStructure',
+          () =>
+            generateDataStructure({
+              entity,
+              dependentAnalyses,
+              analysis,
+            }),
+        );
+        analysis.status.steps.push(dataStructureStep);
+      } finally {
+        if (structureSemaphore) {
+          structureSemaphore.release();
+        }
+      }
     }
 
+    // KeyAttributes phase: I/O-bound LLM call, can run in parallel with other entities' Structure
     if (!steps || steps.includes(Step.KeyAttributes)) {
       awsLogDebugLevel(1, `Determining key attributes for ${entity.name}...`);
 

package/analyzer-template/packages/analyze/src/lib/files/analyzeEntity.ts

@@ -31,7 +31,12 @@ import relevantDiffPart from './relevantDiffPart';
 import { Step } from './enums/steps';
 import { ImmutableAnalysisContext } from '../analysisContext';
 import generateAnalyzedTreeSha from './analyze/generateAnalyzedTreeSha';
-import { awsLog, awsLogDebugLevel, pushAnalysisError } from '~codeyam/utils';
+import {
+  awsLog,
+  awsLogDebugLevel,
+  pushAnalysisError,
+  Semaphore,
+} from '~codeyam/utils';
 
 export interface AnalyzeEntityArgs {
   project: Project;
@@ -50,6 +55,8 @@ export interface AnalyzeEntityArgs {
   forceAll?: boolean;
   updateProgress?: (detail: string) => void;
   modelString?: string;
+  /** Semaphore to serialize CPU-intensive Structure phase */
+  structureSemaphore?: Semaphore;
 }
 
 /**
@@ -77,6 +84,7 @@ export default async function analyzeEntity({
   forceAll,
   updateProgress,
   modelString,
+  structureSemaphore,
 }: AnalyzeEntityArgs): Promise<{
   analysis: Analysis;
   dependentAnalyses?: ReadonlyAnalysisMap;
@@ -476,6 +484,7 @@ export default async function analyzeEntity({
             steps,
             updateProgress,
             model,
+            structureSemaphore,
           })
         : analyzeInitial({
             analysis,
@@ -485,6 +494,7 @@ export default async function analyzeEntity({
             steps,
             updateProgress,
             model,
+            structureSemaphore,
           }),
     ]);
 
@@ -498,8 +508,12 @@ export default async function analyzeEntity({
     analysis = analysisResult.analysis;
 
     analysis.completedAt = new Date().toISOString();
-    // Don't set analyzedTreeSha when ONLY_DATA_STRUCTURE=true, so full analysis can re-analyze later
-    if (process.env.ONLY_DATA_STRUCTURE !== 'true') {
+    // Don't set analyzedTreeSha when:
+    // 1. ONLY_DATA_STRUCTURE=true - so full analysis can re-analyze later
+    // 2. analyzeAsDependency=true - dependency analyses only generate Structure+KeyAttributes,
+    //    not scenarios. Leaving analyzedTreeSha unset allows partial analysis detection to work
+    //    when the entity is later analyzed directly.
+    if (process.env.ONLY_DATA_STRUCTURE !== 'true' && !analyzeAsDependency) {
       analysis.analyzedTreeSha = generateAnalyzedTreeSha(analysis);
     }
   } catch (error) {

package/analyzer-template/packages/analyze/src/lib/files/analyzeInitial.ts

@@ -12,7 +12,12 @@ import {
   type ReadonlyAnalysisMap,
 } from '~codeyam/types';
 import { upsertAnalysesWithScenarios } from '~codeyam/supabase';
-import { awsLog, awsLogDebugLevel, pushAnalysisError } from '~codeyam/utils';
+import {
+  awsLog,
+  awsLogDebugLevel,
+  pushAnalysisError,
+  Semaphore,
+} from '~codeyam/utils';
 import generateKeyAttributes from './scenarios/generateKeyAttributes';
 
 export interface AnalyzeInitialArgs {
@@ -23,6 +28,8 @@ export interface AnalyzeInitialArgs {
   steps?: string[];
   updateProgress?: (detail: string) => void;
   model: AI.Model;
+  /** Semaphore to serialize CPU-intensive Structure phase */
+  structureSemaphore?: Semaphore;
 }
 
 export default async function analyzeInitial({
@@ -33,6 +40,7 @@ export default async function analyzeInitial({
   steps,
   updateProgress,
   model,
+  structureSemaphore,
 }: AnalyzeInitialArgs) {
   analysis.entity = entity;
   awsLog('CodeYam: Analyzing initial entity', {
@@ -47,22 +55,32 @@ export default async function analyzeInitial({
   });
 
   try {
+    // Structure phase: CPU-intensive, serialize via semaphore if provided
     if (!steps || steps.includes(Step.Structure)) {
-      awsLogDebugLevel(
-        1,
-        `Merging dependent data structures for ${entity.name}...`,
-      );
-
-      const { step: dataStructureStep } = await recordStep(
-        'generateDataStructure',
-        () =>
-          generateDataStructure({
-            entity,
-            dependentAnalyses,
-            analysis,
-          }),
-      );
-      analysis.status.steps.push(dataStructureStep);
+      if (structureSemaphore) {
+        await structureSemaphore.acquire();
+      }
+      try {
+        awsLogDebugLevel(
+          1,
+          `Merging dependent data structures for ${entity.name}...`,
+        );
+
+        const { step: dataStructureStep } = await recordStep(
+          'generateDataStructure',
+          () =>
+            generateDataStructure({
+              entity,
+              dependentAnalyses,
+              analysis,
+            }),
+        );
+        analysis.status.steps.push(dataStructureStep);
+      } finally {
+        if (structureSemaphore) {
+          structureSemaphore.release();
+        }
+      }
     }
 
     // Short-circuit for test mocking: only generate data structure, skip key attributes and scenarios
@@ -73,6 +91,7 @@ export default async function analyzeInitial({
       return { analysis };
     }
 
+    // KeyAttributes phase: I/O-bound LLM call, can run in parallel with other entities' Structure
     if (!steps || steps.includes(Step.KeyAttributes)) {
       awsLogDebugLevel(1, `Determining key attributes for ${entity.name}...`);
 

package/analyzer-template/packages/analyze/src/lib/files/scenarios/generateDataStructure.ts

@@ -54,8 +54,14 @@ export default function generateDataStructure({
     const nonMockedDependentAnalyses: ReadonlyAnalysisMap =
       allImportedExports.reduce((acc, importedExport) => {
         if (importedExport.isMocked) return acc;
+        // Use resolvedFilePath when available (for re-exports from index files)
+        // to find the actual analysis of the component
+        const lookupPath =
+          ('resolvedFilePath' in importedExport &&
+            importedExport.resolvedFilePath) ||
+          importedExport.filePath;
         const relevantDependencyAnalysis =
-          dependentAnalyses[importedExport.filePath]?.[importedExport.name];
+          dependentAnalyses[lookupPath]?.[importedExport.name];
 
         if (!relevantDependencyAnalysis) {
           // console.info(

package/analyzer-template/packages/analyze/src/lib/files/scenarios/generateKeyAttributes.ts

@@ -8,6 +8,7 @@ import measureAndReportExecutionTime from '../../utils/measureAndReportExecution
 import { AI, generateEntityKeyAttributes } from '~codeyam/ai';
 import { awsLog } from '~codeyam/utils';
 import { gatherRelevantDependentKeyAttributes } from '~codeyam/ai';
+import { gatherDataForMocks } from './generateScenarioData';
 
 export interface GenerateEntityKeyAttributesArgs {
   entity: Entity;
@@ -28,7 +29,14 @@ export default async function generateKeyAttributes({
 }: GenerateEntityKeyAttributesArgs): Promise<void> {
   if (!['visual', 'library'].includes(entity.entityType)) return;
 
-  const structure = { ...analysis.metadata.scenariosDataStructure };
+  // Merge direct dataForMocks with recursively gathered data from dependencies
+  const structure = {
+    arguments: analysis.metadata.scenariosDataStructure?.arguments ?? [],
+    dataForMocks: {
+      ...(analysis.metadata.scenariosDataStructure?.dataForMocks ?? {}),
+      ...gatherDataForMocks(analysis, dependentAnalyses),
+    },
+  };
 
   awsLog(`🔮 Generating ${error ? 'error ' : ''}scenarios for`, {
     filePath: entity.filePath,

package/analyzer-template/packages/analyze/src/lib/files/scenarios/generateScenarioData.ts

@@ -104,8 +104,13 @@ export function gatherDataForMocks(
       // regardless of whether we found a relevantKey above.
       // This ensures transitive dependencies are included.
       // For example: Activity -> CurrentActivityTab -> useLastLogLine
+      // Use resolvedFilePath when available (for re-exports from index files)
+      const lookupPath =
+        ('resolvedFilePath' in importedExport &&
+          importedExport.resolvedFilePath) ||
+        importedExport.filePath;
       const dependentAnalysis =
-        dependentAnalyses?.[importedExport.filePath]?.[importedExport.name];
+        dependentAnalyses?.[lookupPath]?.[importedExport.name];
 
       if (dependentAnalysis && !analysesGathered.has(dependentAnalysis.id)) {
         analysesGathered.add(dependentAnalysis.id);

package/analyzer-template/packages/analyze/src/lib/files/scenarios/generateScenarios.ts

@@ -7,6 +7,7 @@ import {
 import measureAndReportExecutionTime from '../../utils/measureAndReportExecutionTime';
 import { AI, generateEntityScenarios } from '~codeyam/ai';
 import { awsLog } from '~codeyam/utils';
+import { gatherDataForMocks } from './generateScenarioData';
 
 export interface GenerateScenariosArgs {
   entity: Entity;
@@ -27,7 +28,14 @@ export default async function generateScenarios({
 }: GenerateScenariosArgs): Promise<void> {
   if (!['visual', 'library'].includes(entity.entityType)) return;
 
-  const structure = { ...analysis.metadata.scenariosDataStructure };
+  // Merge direct dataForMocks with recursively gathered data from dependencies
+  const structure = {
+    arguments: analysis.metadata.scenariosDataStructure?.arguments ?? [],
+    dataForMocks: {
+      ...(analysis.metadata.scenariosDataStructure?.dataForMocks ?? {}),
+      ...gatherDataForMocks(analysis, dependentAnalyses),
+    },
+  };
 
   awsLog(`🔮 Generating ${error ? 'error ' : ''}scenarios for`, {
     filePath: entity.filePath,

package/analyzer-template/packages/analyze/src/lib/files/scenarios/mergeInDependentDataStructure.ts

@@ -696,13 +696,23 @@ export default function mergeInDependentDataStructure({
     });
 
     // First, normalize any returnValue. paths that were written by equivalency processing
-    // to the standard functionName().functionCallReturnValue. format
+    // to the standard functionName().functionCallReturnValue. format.
+    // Also delete returnValue[] artifact paths that don't represent actual array structure.
     const pathsToNormalize: [string, string][] = [];
+    const pathsToDelete: string[] = [];
     for (const path in depSchema.returnValueSchema) {
       if (path === 'returnValue' || path.startsWith('returnValue.')) {
         pathsToNormalize.push([path, depSchema.returnValueSchema[path]]);
+      } else if (path.match(/^returnValue\[\d*\]/)) {
+        // Delete returnValue[] and returnValue[n] paths - these are artifacts from
+        // equivalency processing and don't represent the actual structure of mocked dependencies.
+        // The correct structure comes from srcSchema (dependencySchemas).
+        pathsToDelete.push(path);
       }
     }
+    for (const path of pathsToDelete) {
+      delete depSchema.returnValueSchema[path];
+    }
     for (const [path, value] of pathsToNormalize) {
       delete depSchema.returnValueSchema[path];
       const normalizedPath =
@@ -734,9 +744,8 @@ export default function mergeInDependentDataStructure({
                 `${dependency.name}().functionCallReturnValue.`,
               );
 
-        if (!(normalizedPath in depSchema.returnValueSchema)) {
-          depSchema.returnValueSchema[normalizedPath] = value;
-        }
+        // Always write srcSchema values - they take precedence over equivalency-derived values
+        depSchema.returnValueSchema[normalizedPath] = value;
         continue;
       }
 
@@ -744,9 +753,8 @@ export default function mergeInDependentDataStructure({
       // These are needed for constructMockCode to build the proper mock data hierarchy
       // Example: supabase.auth.getSession().functionCallReturnValue.data.session
       if (path.includes('.functionCallReturnValue')) {
-        if (!(path in depSchema.returnValueSchema)) {
-          depSchema.returnValueSchema[path] = value;
-        }
+        // Always write srcSchema values - they take precedence over equivalency-derived values
+        depSchema.returnValueSchema[path] = value;
         continue;
       }
 

package/analyzer-template/packages/aws/dist/src/lib/s3/getPresignedUrl.d.ts

@@ -0,0 +1,23 @@
+interface GetPresignedUploadUrlArgs {
+    bucketName: string;
+    key: string;
+    contentType?: string;
+    expiresInSeconds?: number;
+}
+interface GetPresignedDownloadUrlArgs {
+    bucketName: string;
+    key: string;
+    expiresInSeconds?: number;
+}
+/**
+ * Generate a presigned URL for uploading a file to S3.
+ * The client can PUT directly to this URL without AWS credentials.
+ */
+export declare function getPresignedUploadUrl({ bucketName, key, contentType, expiresInSeconds, }: GetPresignedUploadUrlArgs): Promise<string>;
+/**
+ * Generate a presigned URL for downloading a file from S3.
+ * Anyone with this URL can GET the file without AWS credentials.
+ */
+export declare function getPresignedDownloadUrl({ bucketName, key, expiresInSeconds, }: GetPresignedDownloadUrlArgs): Promise<string>;
+export {};
+//# sourceMappingURL=getPresignedUrl.d.ts.map

package/analyzer-template/packages/aws/dist/src/lib/s3/getPresignedUrl.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"getPresignedUrl.d.ts","sourceRoot":"","sources":["../../../../src/lib/s3/getPresignedUrl.ts"],"names":[],"mappings":"AAOA,UAAU,yBAAyB;IACjC,UAAU,EAAE,MAAM,CAAC;IACnB,GAAG,EAAE,MAAM,CAAC;IACZ,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B;AAED,UAAU,2BAA2B;IACnC,UAAU,EAAE,MAAM,CAAC;IACnB,GAAG,EAAE,MAAM,CAAC;IACZ,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B;AAKD;;;GAGG;AACH,wBAAsB,qBAAqB,CAAC,EAC1C,UAAU,EACV,GAAG,EACH,WAAgC,EAChC,gBAAwC,GACzC,EAAE,yBAAyB,GAAG,OAAO,CAAC,MAAM,CAAC,CAU7C;AAED;;;GAGG;AACH,wBAAsB,uBAAuB,CAAC,EAC5C,UAAU,EACV,GAAG,EACH,gBAA0C,GAC3C,EAAE,2BAA2B,GAAG,OAAO,CAAC,MAAM,CAAC,CAS/C"}

package/analyzer-template/packages/aws/dist/src/lib/s3/getPresignedUrl.js

@@ -0,0 +1,30 @@
+import { S3Client, PutObjectCommand, GetObjectCommand, } from '@aws-sdk/client-s3';
+import { getSignedUrl } from '@aws-sdk/s3-request-presigner';
+const DEFAULT_UPLOAD_EXPIRY = 2 * 60 * 60; // 2 hours
+const DEFAULT_DOWNLOAD_EXPIRY = 24 * 60 * 60; // 24 hours
+/**
+ * Generate a presigned URL for uploading a file to S3.
+ * The client can PUT directly to this URL without AWS credentials.
+ */
+export async function getPresignedUploadUrl({ bucketName, key, contentType = 'application/gzip', expiresInSeconds = DEFAULT_UPLOAD_EXPIRY, }) {
+    const s3Client = new S3Client({});
+    const command = new PutObjectCommand({
+        Bucket: bucketName,
+        Key: key,
+        ContentType: contentType,
+    });
+    return getSignedUrl(s3Client, command, { expiresIn: expiresInSeconds });
+}
+/**
+ * Generate a presigned URL for downloading a file from S3.
+ * Anyone with this URL can GET the file without AWS credentials.
+ */
+export async function getPresignedDownloadUrl({ bucketName, key, expiresInSeconds = DEFAULT_DOWNLOAD_EXPIRY, }) {
+    const s3Client = new S3Client({});
+    const command = new GetObjectCommand({
+        Bucket: bucketName,
+        Key: key,
+    });
+    return getSignedUrl(s3Client, command, { expiresIn: expiresInSeconds });
+}
+//# sourceMappingURL=getPresignedUrl.js.map

package/analyzer-template/packages/aws/dist/src/lib/s3/getPresignedUrl.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"getPresignedUrl.js","sourceRoot":"","sources":["../../../../src/lib/s3/getPresignedUrl.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,QAAQ,EACR,gBAAgB,EAChB,gBAAgB,GACjB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAe7D,MAAM,qBAAqB,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,UAAU;AACrD,MAAM,uBAAuB,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,WAAW;AAEzD;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,EAC1C,UAAU,EACV,GAAG,EACH,WAAW,GAAG,kBAAkB,EAChC,gBAAgB,GAAG,qBAAqB,GACd;IAC1B,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC;IAElC,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC;QACnC,MAAM,EAAE,UAAU;QAClB,GAAG,EAAE,GAAG;QACR,WAAW,EAAE,WAAW;KACzB,CAAC,CAAC;IAEH,OAAO,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,SAAS,EAAE,gBAAgB,EAAE,CAAC,CAAC;AAC1E,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAAC,EAC5C,UAAU,EACV,GAAG,EACH,gBAAgB,GAAG,uBAAuB,GACd;IAC5B,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC;IAElC,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC;QACnC,MAAM,EAAE,UAAU;QAClB,GAAG,EAAE,GAAG;KACT,CAAC,CAAC;IAEH,OAAO,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,SAAS,EAAE,gBAAgB,EAAE,CAAC,CAAC;AAC1E,CAAC"}

package/analyzer-template/packages/aws/package.json

@@ -11,14 +11,15 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
-    "@aws-sdk/client-codebuild": "^3.940.0",
+    "@aws-sdk/client-codebuild": "^3.948.0",
     "@aws-sdk/client-cloudwatch-logs": "^3.940.0",
     "@aws-sdk/client-dynamodb": "^3.899.0",
-    "@aws-sdk/client-ecr": "^3.946.0",
+    "@aws-sdk/client-ecr": "^3.948.0",
     "@aws-sdk/client-ecs": "^3.937.0",
-    "@aws-sdk/client-s3": "^3.899.0",
+    "@aws-sdk/client-s3": "^3.940.0",
     "@aws-sdk/client-sqs": "^3.940.0",
-    "@aws-sdk/lib-storage": "^3.899.0",
+    "@aws-sdk/lib-storage": "^3.940.0",
+    "@aws-sdk/s3-request-presigner": "^3.940.0",
     "@aws-sdk/util-dynamodb": "^3.902.0"
   }
 }

package/analyzer-template/packages/aws/s3/index.ts

@@ -1,2 +1,6 @@
 // Direct export for s3 functions only
 export { default as s3UploadFile } from '../src/lib/s3/uploadFileToS3';
+export {
+  getPresignedUploadUrl,
+  getPresignedDownloadUrl,
+} from '../src/lib/s3/getPresignedUrl';

package/analyzer-template/packages/aws/src/lib/s3/getPresignedUrl.ts

@@ -0,0 +1,62 @@
+import {
+  S3Client,
+  PutObjectCommand,
+  GetObjectCommand,
+} from '@aws-sdk/client-s3';
+import { getSignedUrl } from '@aws-sdk/s3-request-presigner';
+
+interface GetPresignedUploadUrlArgs {
+  bucketName: string;
+  key: string;
+  contentType?: string;
+  expiresInSeconds?: number;
+}
+
+interface GetPresignedDownloadUrlArgs {
+  bucketName: string;
+  key: string;
+  expiresInSeconds?: number;
+}
+
+const DEFAULT_UPLOAD_EXPIRY = 2 * 60 * 60; // 2 hours
+const DEFAULT_DOWNLOAD_EXPIRY = 24 * 60 * 60; // 24 hours
+
+/**
+ * Generate a presigned URL for uploading a file to S3.
+ * The client can PUT directly to this URL without AWS credentials.
+ */
+export async function getPresignedUploadUrl({
+  bucketName,
+  key,
+  contentType = 'application/gzip',
+  expiresInSeconds = DEFAULT_UPLOAD_EXPIRY,
+}: GetPresignedUploadUrlArgs): Promise<string> {
+  const s3Client = new S3Client({});
+
+  const command = new PutObjectCommand({
+    Bucket: bucketName,
+    Key: key,
+    ContentType: contentType,
+  });
+
+  return getSignedUrl(s3Client, command, { expiresIn: expiresInSeconds });
+}
+
+/**
+ * Generate a presigned URL for downloading a file from S3.
+ * Anyone with this URL can GET the file without AWS credentials.
+ */
+export async function getPresignedDownloadUrl({
+  bucketName,
+  key,
+  expiresInSeconds = DEFAULT_DOWNLOAD_EXPIRY,
+}: GetPresignedDownloadUrlArgs): Promise<string> {
+  const s3Client = new S3Client({});
+
+  const command = new GetObjectCommand({
+    Bucket: bucketName,
+    Key: key,
+  });
+
+  return getSignedUrl(s3Client, command, { expiresIn: expiresInSeconds });
+}

package/analyzer-template/packages/generate/src/lib/componentScenarioPage/componentScenarioPageNext.ts

@@ -39,7 +39,6 @@ export default function componentScenarioPageNext({
   type ReactNode,
   type ChangeEvent
 } from 'react';
-import Head from 'next/head';
 ${analysis.scenarios?.length > 0 && `import { scenarios } from "${relativeMocksDir}/MockData_${safeFileName(scenario.name)}";`}
 ${getComponentImportStatement({
   file,
@@ -58,17 +57,7 @@ ${getComponentImportStatement({
 // Inject image interception code at module level to run before any rendering
 ${getImageReplacementCode()}
 
-
-const NextHead = () => {
-  return (
-    <Head>
-      <title>${fileName}: ${escapeQuotes(scenario.name)} CodeYam Demo v${VERSION}</title>
-      <meta name="description" content="This is a demo of the ${fileName} component" />
-    </Head>
-  )
-};
-
-function Scenario({ name, width, children }: { name: string, width: string, children: ReactNode }) {
+function Scenario({ name, width, height, children }: { name: string, width: string, height: string, children: ReactNode }) {
   
   useEffect(() => {
     if (typeof window === 'undefined') return;
@@ -89,14 +78,14 @@ function Scenario({ name, width, children }: { name: string, width: string, chil
       return Math.max(100, bodyHeight, htmlHeight, scrollHeight);
     }
 
-    const height = calculateHeight();
-    parent.postMessage({ type: 'codeyam-resize', name, height }, '*');
+    const calculatedHeight = calculateHeight();
+    parent.postMessage({ type: 'codeyam-resize', name, height: calculatedHeight }, '*');
 
     const eventListener = (event) => {
       if (event.data.name === name) {
         if (event.data.type === 'codeyam-respond') {
-          const height = calculateHeight();
-          parent.postMessage({ type: 'codeyam-resize', name, height }, '*');
+          const calculatedHeight = calculateHeight();
+          parent.postMessage({ type: 'codeyam-resize', name, height: calculatedHeight }, '*');
         }
       }
     };
@@ -108,9 +97,9 @@ function Scenario({ name, width, children }: { name: string, width: string, chil
   return (
     <div className='w-full flex items-center justify-center'>
       <div id='component' style={{
-        width: width,
-        minWidth: width,
-        maxWidth: width,
+        minWidth: width !== 'auto' ? \`min(\${width}, 100vw)\` : undefined,
+        height: height !== 'auto' ? height : undefined,
+        minHeight: height !== 'auto' ? height : undefined,
         display: 'block',
         padding: '6px'
       }}>
@@ -121,13 +110,31 @@ function Scenario({ name, width, children }: { name: string, width: string, chil
 }
 
 export default function Index() {
+  const [searchParams] = useState(() =>
+    typeof window !== 'undefined'
+      ? new URLSearchParams(window.location.search)
+      : new URLSearchParams()
+  );
+  const widthOverride = searchParams.get('width');
+  const heightOverride = searchParams.get('height') || 'auto';
+
+  useEffect(() => {
+    document.title = "${fileName}: ${escapeQuotes(scenario.name)} CodeYam Demo v${VERSION}";
+    let metaDesc = document.querySelector('meta[name="description"]');
+    if (!metaDesc) {
+      metaDesc = document.createElement('meta');
+      metaDesc.setAttribute('name', 'description');
+      document.head.appendChild(metaDesc);
+    }
+    metaDesc.setAttribute('content', "This is a demo of the ${fileName} component");
+  }, []);
+
   ${getIFrameMessageListenerCode({ scenarioName: scenario.name })}
   ${getNextJsErrorClosingCode()}
 
   return (
     <div className='flex flex-col justify-center items-center h-screen' data-version={version}>
-      <NextHead />
-      ${scenarioComponent(entity, analysis, scenario)}
+      ${scenarioComponent(entity, analysis, scenario, { useOverrides: true })}
     </div>
   );
 }`;