genaicode 0.0.37 → 0.1.1

package/src/cli/validate-cli-params.js

@@ -1,137 +0,0 @@
-import fs from 'fs';
-import path from 'path';
-import globRegex from 'glob-regex';
-import { rcConfig } from '../main/config.js';
-
-// List of allowed CLI parameters
-const allowedParameters = [
-  '--dry-run',
-  '--consider-all-files',
-  '--allow-file-create',
-  '--allow-file-delete',
-  '--allow-directory-create',
-  '--allow-file-move',
-  '--chat-gpt',
-  '--vertex-ai',
-  '--vertex-ai-claude',
-  '--anthropic',
-  '--explicit-prompt=',
-  '--task-file=',
-  '--dependency-tree',
-  '--verbose-prompt',
-  '--require-explanations',
-  '--disable-context-optimization',
-  '--gemini-block-none',
-  '--disable-initial-lint',
-  '--temperature=',
-  '--vision',
-  '--imagen=',
-  '--cheap',
-  '--ask-question',
-  '--help',
-  '--content-mask=',
-  '--disable-cache',
-  '--ignore-pattern=',
-];
-
-/**
- * Validate CLI parameters according to those mentioned in README.md
- * Fail the process if not valid, or if an unknown parameter is passed
- * @throws {Error} If an invalid parameter is provided
- */
-export function validateCliParams() {
-  const providedParameters = process.argv.slice(2);
-
-  // Check if --help is present
-  const helpRequested = providedParameters.includes('--help');
-
-  if (helpRequested) {
-    // If --help is present, no other parameters should be allowed
-    if (providedParameters.length > 1) {
-      console.error('The --help option cannot be used with other parameters.');
-      process.exit(1);
-    }
-    return; // Exit the function early as no further validation is needed
-  }
-
-  providedParameters.forEach((param) => {
-    if (!param.startsWith('--')) {
-      console.error(`Invalid parameter: ${param}, all parameters must start with --`);
-      process.exit(1);
-    }
-    if (!allowedParameters.some((p) => (p.endsWith('=') && param.startsWith(p)) || param === p)) {
-      console.error(`Invalid parameter: ${param}, allowed parameters are: ${allowedParameters.join(', ')}`);
-      process.exit(1);
-    }
-  });
-
-  // Validate temperature parameter, it must be a number between 0.0 and 2.0
-  const temperatureParam = providedParameters.find((param) => param.startsWith('--temperature='));
-  if (temperatureParam) {
-    const temperatureValue = parseFloat(temperatureParam.split('=')[1]);
-    if (isNaN(temperatureValue) || temperatureValue < 0.0 || temperatureValue > 2.0) {
-      console.error('Invalid temperature value. It must be a number between 0.0 and 2.0.');
-      process.exit(1);
-    }
-  }
-
-  // Validate --imagen parameter
-  const imagenParam = providedParameters.find((param) => param.startsWith('--imagen='));
-  if (imagenParam) {
-    const imagenValue = imagenParam.split('=')[1];
-    if (imagenValue !== 'vertex-ai' && imagenValue !== 'dall-e') {
-      throw new Error('Invalid --imagen value. It must be either "vertex-ai" or "dall-e".');
-    }
-  }
-
-  if (providedParameters.includes('--vision') && providedParameters.includes('--vertex-ai')) {
-    throw new Error('--vision and --vertex-ai are currently not supported together.');
-  }
-
-  // Validate content mask parameter
-  const contentMaskParam = providedParameters.find((param) => param.startsWith('--content-mask='));
-  if (contentMaskParam) {
-    const contentMaskValue = contentMaskParam.split('=')[1];
-    const fullPath = path.join(rcConfig.rootDir, contentMaskValue);
-    if (!fs.existsSync(fullPath) || !fs.statSync(fullPath).isDirectory()) {
-      throw new Error(
-        `Invalid --content-mask value. The path "${contentMaskValue}" does not exist or is not a directory within the project.`,
-      );
-    }
-  }
-
-  // Validate ignore pattern parameter
-  const ignorePatternParams = providedParameters.filter((param) => param.startsWith('--ignore-pattern='));
-  if (ignorePatternParams.length > 0) {
-    for (const ignorePatternParam of ignorePatternParams) {
-      const ignorePatternValue = ignorePatternParam.split('=')[1];
-      if (!ignorePatternValue) {
-        throw new Error('Invalid --ignore-pattern value. The pattern cannot be empty.');
-      }
-      try {
-        globRegex(ignorePatternValue);
-      } catch (e) {
-        throw new Error(`Invalid --ignore-pattern value. The pattern "${ignorePatternValue}" is not valid.`, e);
-      }
-    }
-  }
-}
-
-/**
- * Get the value of a CLI parameter
- * @param {string} paramName - The name of the parameter to get the value for
- * @returns {string|null} The value of the parameter, or null if not found
- */
-export function getCliParamValue(paramName) {
-  const param = process.argv.find((arg) => arg.startsWith(`${paramName}=`));
-  return param ? param.split('=')[1] : null;
-}
-
-/**
- * Check if a CLI parameter is present
- * @param {string} paramName - The name of the parameter to check
- * @returns {boolean} True if the parameter is present, false otherwise
- */
-export function hasCliParam(paramName) {
-  return process.argv.includes(paramName) || !!getCliParamValue(paramName);
-}

package/src/cli/validate-cli-params.test.js

@@ -1,134 +0,0 @@
-import { describe, it, expect, beforeEach, afterEach } from 'vitest';
-import { validateCliParams, getCliParamValue, hasCliParam } from './validate-cli-params.js';
-
-describe('validateCliParams', () => {
-  const originalArgv = process.argv;
-
-  beforeEach(() => {
-    process.argv = ['node', 'script.js'];
-  });
-
-  afterEach(() => {
-    process.argv = originalArgv;
-  });
-
-  it('should not throw for valid parameters', () => {
-    process.argv.push('--dry-run', '--consider-all-files');
-    expect(() => validateCliParams()).not.toThrow();
-  });
-
-  it('should throw for invalid parameters', () => {
-    process.argv.push('--invalid-param');
-    expect(() => validateCliParams()).toThrow();
-  });
-
-  it('should throw for parameters without --', () => {
-    process.argv.push('invalid-param');
-    expect(() => validateCliParams()).toThrow();
-  });
-
-  it('should not throw for valid --temperature parameter', () => {
-    process.argv.push('--temperature=0.5');
-    expect(() => validateCliParams()).not.toThrow();
-  });
-
-  it('should throw for invalid --temperature parameter', () => {
-    process.argv.push('--temperature=invalid');
-    expect(() => validateCliParams()).toThrow();
-  });
-
-  it('should throw an error when both vision and vertexAi flags are true', () => {
-    process.argv.push('--vision');
-    process.argv.push('--vertex-ai');
-
-    expect(() => validateCliParams()).toThrow('--vision and --vertex-ai are currently not supported together.');
-  });
-
-  // New tests for --imagen parameter
-  it('should not throw for valid --imagen parameter with vertex-ai', () => {
-    process.argv.push('--imagen=vertex-ai');
-    expect(() => validateCliParams()).not.toThrow();
-  });
-
-  it('should not throw for valid --imagen parameter with dall-e', () => {
-    process.argv.push('--imagen=dall-e');
-    expect(() => validateCliParams()).not.toThrow();
-  });
-
-  it('should throw for invalid --imagen parameter value', () => {
-    process.argv.push('--imagen=invalid-service');
-    expect(() => validateCliParams()).toThrow('Invalid --imagen value. It must be either "vertex-ai" or "dall-e".');
-  });
-
-  it('should throw for --imagen parameter without value', () => {
-    process.argv.push('--imagen=');
-    expect(() => validateCliParams()).toThrow('Invalid --imagen value. It must be either "vertex-ai" or "dall-e".');
-  });
-});
-
-describe('getCliParamValue', () => {
-  const originalArgv = process.argv;
-
-  beforeEach(() => {
-    process.argv = ['node', 'script.js'];
-  });
-
-  afterEach(() => {
-    process.argv = originalArgv;
-  });
-
-  it('should return the value for a parameter with a value', () => {
-    process.argv.push('--explicit-prompt=Test prompt');
-    expect(getCliParamValue('--explicit-prompt')).toBe('Test prompt');
-  });
-
-  it('should return null for a parameter without a value', () => {
-    process.argv.push('--dry-run');
-    expect(getCliParamValue('--dry-run')).toBeNull();
-  });
-
-  it('should return null for a non-existent parameter', () => {
-    expect(getCliParamValue('--non-existent')).toBeNull();
-  });
-
-  it('should return the value for --temperature parameter', () => {
-    process.argv.push('--temperature=0.5');
-    expect(getCliParamValue('--temperature')).toBe('0.5');
-  });
-
-  it('should return the value for --imagen parameter', () => {
-    process.argv.push('--imagen=vertex-ai');
-    expect(getCliParamValue('--imagen')).toBe('vertex-ai');
-  });
-});
-
-describe('hasCliParam', () => {
-  const originalArgv = process.argv;
-
-  beforeEach(() => {
-    process.argv = ['node', 'script.js'];
-  });
-
-  afterEach(() => {
-    process.argv = originalArgv;
-  });
-
-  it('should return true for an existing parameter', () => {
-    process.argv.push('--dry-run');
-    expect(hasCliParam('--dry-run')).toBe(true);
-  });
-
-  it('should return false for a non-existent parameter', () => {
-    expect(hasCliParam('--non-existent')).toBe(false);
-  });
-
-  it('should return true for --temperature parameter', () => {
-    process.argv.push('--temperature=0.5');
-    expect(hasCliParam('--temperature')).toBe(true);
-  });
-
-  it('should return true for --imagen parameter', () => {
-    process.argv.push('--imagen=dall-e');
-    expect(hasCliParam('--imagen')).toBe(true);
-  });
-});

package/src/files/file-utils.js

@@ -1,7 +0,0 @@
-import path from 'path';
-
-/** Check if directory is ancestor of given directory */
-export function isAncestorDirectory(parent, dir) {
-  const relative = path.relative(parent, dir);
-  return parent === dir || (relative && !relative.startsWith('..') && !path.isAbsolute(relative));
-}

package/src/files/file-utils.test.js

@@ -1,21 +0,0 @@
-import { vi, describe, beforeEach, it, expect } from 'vitest';
-import { isAncestorDirectory } from './file-utils.js';
-
-// Test for isAncestorDirectory
-describe('isAncestorDirectory', () => {
-  beforeEach(() => {
-    vi.resetAllMocks();
-  });
-
-  it('should return true if directories are the same', () => {
-    expect(isAncestorDirectory('/project', '/project')).toBe(true);
-  });
-
-  it('should return true if parent is ancestor of dir', () => {
-    expect(isAncestorDirectory('/project', '/project/src')).toBe(true);
-  });
-
-  it('should return false if parent is not ancestor of dir', () => {
-    expect(isAncestorDirectory('/project', '/other')).toBe(false);
-  });
-});

package/src/files/find-files.js

@@ -1,86 +0,0 @@
-import fs from 'fs';
-import path from 'path';
-import { rcConfig, sourceExtensions, IMAGE_ASSET_EXTENSIONS, ignorePaths } from '../main/config.js';
-
-// List of possible extensions for dependency resolution
-const POSSIBLE_DEPENDENCY_EXTENSIONS = ['.ts', '.js', '.tsx', '.jsx'];
-
-function findFiles(dir, recursive, extensions) {
-  const files = [];
-  const items = fs.readdirSync(dir);
-  for (const item of items) {
-    const fullPath = path.join(dir, item);
-
-    if (ignorePaths.some((ignorePath) => fullPath.endsWith(ignorePath))) {
-      continue;
-    }
-
-    if (fs.statSync(fullPath).isDirectory()) {
-      if (recursive) {
-        files.push(...findFiles(fullPath, true, extensions));
-      }
-    } else if (extensions.includes(path.extname(fullPath))) {
-      files.push(fullPath);
-    }
-  }
-  return files;
-}
-
-function getDependencies(filePath) {
-  const content = fs.readFileSync(filePath, 'utf-8');
-  const dependencyRegex = /import\s+.+?\s+from\s+['"](.+?\/?[^'"]+)['"]/g;
-  const dependencies = [];
-  let match;
-  while ((match = dependencyRegex.exec(content)) !== null) {
-    const dependencyPath = match[1];
-    // Resolve relative paths from the file's directory
-    let resolvedPath = path.resolve(path.dirname(filePath), dependencyPath);
-
-    // Only add the dependency if it's a local file and not a module
-    if (fs.existsSync(resolvedPath)) {
-      dependencies.push(resolvedPath);
-    } else {
-      const possibleExtensions = POSSIBLE_DEPENDENCY_EXTENSIONS;
-      for (const ext of possibleExtensions) {
-        const extendedPath = resolvedPath + ext;
-        if (fs.existsSync(extendedPath)) {
-          dependencies.push(extendedPath);
-        }
-      }
-    }
-  }
-  return dependencies;
-}
-
-/** Generates a dependency list for given file */
-export function getDependencyList(entryFile) {
-  const visitedFiles = new Set();
-  const result = new Set();
-
-  function traverse(file) {
-    if (visitedFiles.has(file)) return;
-    visitedFiles.add(file);
-    const dependencies = getDependencies(file);
-    dependencies.forEach((dependency) => result.add(dependency));
-    dependencies.forEach(traverse);
-  }
-
-  result.add(path.resolve(entryFile));
-  traverse(entryFile);
-
-  return Array.from(result);
-}
-
-const sourceFiles = findFiles(rcConfig.rootDir, true, sourceExtensions);
-
-/** Get source files of the application */
-export function getSourceFiles() {
-  return [...sourceFiles];
-}
-
-const imageAssetFiles = findFiles(rcConfig.rootDir, true, IMAGE_ASSET_EXTENSIONS);
-
-/** Get source files of the application */
-export function getImageAssetFiles() {
-  return [...imageAssetFiles];
-}

package/src/files/read-files.js

@@ -1,65 +0,0 @@
-import fs from 'fs';
-import mime from 'mime-types';
-import sizeOf from 'image-size';
-import path from 'path';
-import globRegex from 'glob-regex';
-
-import { getSourceFiles, getImageAssetFiles } from './find-files.js';
-import { rcConfig } from '../main/config.js';
-import { verifySourceCodeLimit } from '../prompt/limits.js';
-import { taskFile, contentMask, ignorePatterns } from '../cli/cli-params.js';
-
-/**
- * Read contents of source files and create a map with file path as key and file content as value
- */
-function readSourceFiles(filterPaths) {
-  const sourceCode = {};
-  for (const file of getSourceFiles()) {
-    if (!filterPaths || filterPaths.includes(file)) {
-      // Apply content mask filter if it's set
-      if (!filterPaths && contentMask) {
-        const relativePath = path.relative(rcConfig.rootDir, file);
-        if (!relativePath.startsWith(contentMask)) {
-          sourceCode[file] = { content: null }; // Include the file path but set content to null
-          continue;
-        }
-      }
-      if (ignorePatterns.some((pattern) => globRegex(pattern).test(file))) {
-        sourceCode[file] = { content: null };
-      } else {
-        const content = fs.readFileSync(file, 'utf-8');
-        sourceCode[file] = { content };
-      }
-    }
-  }
-  return sourceCode;
-}
-
-/** Print source code of all source files */
-export function getSourceCode(filterPaths) {
-  const sourceCode = readSourceFiles(filterPaths);
-
-  if (taskFile && !sourceCode[taskFile]) {
-    sourceCode[taskFile] = {
-      content: fs.readFileSync(taskFile, 'utf-8'),
-    };
-  }
-
-  verifySourceCodeLimit(JSON.stringify(sourceCode));
-  return sourceCode;
-}
-
-/** Get image asset files summary */
-export function getImageAssets() {
-  const imageAssets = {};
-  for (const file of getImageAssetFiles()) {
-    const dimensions = sizeOf(file);
-    imageAssets[file] = {
-      mimeType: mime.lookup(file),
-      width: dimensions.width,
-      height: dimensions.height,
-    };
-  }
-  verifySourceCodeLimit(JSON.stringify(imageAssets));
-  return imageAssets;
-}

package/src/files/read-files.test.js

@@ -1,103 +0,0 @@
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-import fs from 'fs';
-import mime from 'mime-types';
-import sizeOf from 'image-size';
-import { getSourceCode, getImageAssets } from './read-files.js';
-import { getSourceFiles, getImageAssetFiles } from './find-files.js';
-import { verifySourceCodeLimit } from '../prompt/limits.js';
-import * as cliParams from '../cli/cli-params.js';
-import { rcConfig } from '../main/config.js';
-
-vi.mock('fs');
-vi.mock('mime-types');
-vi.mock('image-size');
-vi.mock('./find-files.js', () => ({
-  getImageAssetFiles: vi.fn(),
-  getSourceFiles: vi.fn(),
-}));
-vi.mock('../prompt/limits.js');
-vi.mock('../cli/cli-params.js', () => ({
-  taskFile: null,
-  contentMask: null,
-  ignorePatterns: [],
-}));
-vi.mock('../main/config.js', () => ({
-  rcConfig: { rootDir: '/home/project' },
-}));
-
-describe('read-files', () => {
-  beforeEach(() => {
-    vi.resetAllMocks();
-    vi.mocked(cliParams).contentMask = undefined;
-  });
-
-  afterEach(() => {
-    vi.clearAllMocks();
-  });
-
-  describe('getSourceCode', () => {
-    it('should return source code for all files', () => {
-      const mockFiles = ['/home/project/file1.js', '/home/project/file2.js'];
-      getSourceFiles.mockReturnValue(mockFiles);
-      fs.readFileSync.mockImplementation((file) => `Content of ${file}`);
-      rcConfig.rootDir = '/home/project';
-
-      const result = getSourceCode();
-
-      expect(result).toEqual({
-        '/home/project/file1.js': { content: 'Content of /home/project/file1.js' },
-        '/home/project/file2.js': { content: 'Content of /home/project/file2.js' },
-      });
-      expect(verifySourceCodeLimit).toHaveBeenCalled();
-    });
-
-    it('should apply content mask when specified', () => {
-      const mockFiles = ['/home/project/file1.js', '/home/project/subfolder/file2.js'];
-      getSourceFiles.mockReturnValue(mockFiles);
-      fs.readFileSync.mockImplementation((file) => `Content of ${file}`);
-      rcConfig.rootDir = '/home/project';
-      vi.mocked(cliParams).contentMask = 'subfolder';
-
-      const result = getSourceCode();
-
-      expect(result).toEqual({
-        '/home/project/file1.js': { content: null },
-        '/home/project/subfolder/file2.js': {
-          content: 'Content of /home/project/subfolder/file2.js',
-        },
-      });
-    });
-
-    it('should include task file when specified', () => {
-      const mockFiles = ['/home/project/file1.js'];
-      getSourceFiles.mockReturnValue(mockFiles);
-      fs.readFileSync.mockImplementation((file) => `Content of ${file}`);
-      rcConfig.rootDir = '/home/project';
-      vi.mocked(cliParams).taskFile = '/home/project/task.md';
-
-      const result = getSourceCode();
-
-      expect(result).toEqual({
-        '/home/project/file1.js': { content: 'Content of /home/project/file1.js' },
-        '/home/project/task.md': { content: 'Content of /home/project/task.md' },
-      });
-    });
-  });
-
-  describe('getImageAssets', () => {
-    it('should return image assets information', () => {
-      const mockImageFiles = ['/home/project/image1.png', '/home/project/image2.jpg'];
-      getImageAssetFiles.mockReturnValue(mockImageFiles);
-      mime.lookup.mockImplementation((file) => (file.endsWith('.png') ? 'image/png' : 'image/jpeg'));
-      sizeOf.mockImplementation(() => ({ width: 100, height: 200 }));
-
-      const result = getImageAssets();
-
-      expect(result).toEqual({
-        '/home/project/image1.png': { mimeType: 'image/png', width: 100, height: 200 },
-        '/home/project/image2.jpg': { mimeType: 'image/jpeg', width: 100, height: 200 },
-      });
-      expect(verifySourceCodeLimit).toHaveBeenCalled();
-    });
-  });
-});

package/src/files/temp-buffer.js

@@ -1,13 +0,0 @@
-/** Temporary storage */
-let tempId = 1;
-const temp = {};
-
-export function getTempBuffer(url) {
-  return temp[url];
-}
-
-export function setTempBuffer(buffer) {
-  const imageUrl = 'temp://' + tempId++;
-  temp[imageUrl] = buffer;
-  return imageUrl;
-}