genaicode 0.0.37 → 0.2.0

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;
-}

package/src/files/update-files.js

@@ -1,205 +0,0 @@
-import fs from 'fs';
-import path from 'path';
-import assert from 'node:assert';
-import * as diff from 'diff';
-
-import { getSourceFiles } from './find-files.js';
-import { isAncestorDirectory } from './file-utils.js';
-import { rcConfig } from '../main/config.js';
-import {
-  allowDirectoryCreate,
-  allowFileCreate,
-  allowFileDelete,
-  allowFileMove,
-  anthropic,
-  chatGpt,
-  vertexAiClaude,
-} from '../cli/cli-params.js';
-import { getTempBuffer } from './temp-buffer.js';
-import { imglyRemoveBackground } from '../images/imgly-remove-background.js';
-import { splitImage } from '../images/split-image.js';
-import { resizeImageFile } from '../images/resize-image.js';
-
-/**
- * @param functionCalls Result of the code generation, a map of file paths to new content
- */
-export async function updateFiles(functionCalls) {
-  for (const { name, args } of functionCalls) {
-    let {
-      filePath,
-      newContent,
-      source,
-      destination,
-      patch,
-      inputFilePath,
-      outputFilePath,
-      backgroundColor,
-      parts,
-      size,
-    } = args;
-
-    // Check if filePath is absolute, if not use rootDir as baseline
-    if (name !== 'moveFile' && name !== 'imglyRemoveBackground' && name !== 'splitImage') {
-      filePath = path.isAbsolute(filePath) ? filePath : path.join(rcConfig.rootDir, filePath);
-    } else if (name === 'moveFile') {
-      source = path.isAbsolute(source) ? source : path.join(rcConfig.rootDir, source);
-      destination = path.isAbsolute(destination) ? destination : path.join(rcConfig.rootDir, destination);
-    } else if (name === 'imglyRemoveBackground' || name === 'splitImage') {
-      inputFilePath = path.isAbsolute(inputFilePath) ? inputFilePath : path.join(rcConfig.rootDir, inputFilePath);
-      if (name === 'imglyRemoveBackground') {
-        outputFilePath = path.isAbsolute(outputFilePath) ? outputFilePath : path.join(rcConfig.rootDir, outputFilePath);
-      } else if (name === 'splitImage') {
-        parts.forEach(
-          (part) =>
-            (part.outputFilePath = path.isAbsolute(part.outputFilePath)
-              ? part.outputFilePath
-              : path.join(rcConfig.rootDir, part.outputFilePath)),
-        );
-      }
-    }
-
-    // ignore files which are not located inside project directory (sourceFiles)
-    if (
-      (name !== 'moveFile' && name !== 'imglyRemoveBackground' && name !== 'splitImage' && !isProjectPath(filePath)) ||
-      (name === 'moveFile' && (!isProjectPath(source) || !isProjectPath(destination))) ||
-      (name === 'imglyRemoveBackground' && (!isProjectPath(inputFilePath) || !isProjectPath(outputFilePath))) ||
-      (name === 'splitImage' &&
-        (!isProjectPath(inputFilePath) || parts.some((part) => !isProjectPath(part.outputFilePath))))
-    ) {
-      console.log(`Skipping file: ${filePath || source || inputFilePath}`);
-      throw new Error(
-        `File ${filePath || source || inputFilePath} is not located inside project directory, something is wrong?`,
-      );
-    }
-
-    if (name === 'deleteFile') {
-      assert(allowFileDelete, 'File delete option was not enabled');
-      console.log(`Removing file: ${filePath}`);
-      fs.unlinkSync(filePath);
-    } else if (name === 'createDirectory') {
-      assert(allowDirectoryCreate, 'Directory create option was not enabled');
-      console.log(`Creating directory: ${filePath}`);
-      fs.mkdirSync(filePath, { recursive: true });
-    } else if (name === 'updateFile' || name === 'createFile' || name === 'patchFile') {
-      if (name === 'patchFile') {
-        console.log(`Applying a patch: ${filePath} content`);
-        newContent = diff.applyPatch(fs.readFileSync(filePath, 'utf-8'), patch);
-        assert(!!newContent, 'Patch was not successful');
-      }
-
-      assert(!!newContent, 'newContent must not be empty');
-      if (name === 'createFile') {
-        console.log(`Creating file: ${filePath}`);
-        assert(allowFileCreate, 'File create option was not enabled');
-        assert(!fs.existsSync(filePath), 'File already exists');
-        if (allowDirectoryCreate) {
-          fs.mkdirSync(path.dirname(filePath), { recursive: true });
-        }
-      } else {
-        console.log(`Updating file: ${filePath}`);
-        assert(fs.existsSync(filePath), 'File does not exist');
-      }
-      fs.writeFileSync(
-        filePath,
-        chatGpt || anthropic || vertexAiClaude
-          ? newContent
-          : // Fixing a problem caused by vertex function calling. Possibly not a good fix
-            newContent.replace(/\\n/g, '\n').replace(/\\'/g, "'"),
-        'utf-8',
-      );
-    } else if (name === 'moveFile') {
-      console.log(`Moving file from ${source} to ${destination}`);
-      assert(fs.existsSync(source), 'Source file does not exist');
-      assert(!fs.existsSync(destination), 'Destination file already exists');
-      assert(allowFileMove, 'File move option was not enabled');
-      if (allowDirectoryCreate) {
-        fs.mkdirSync(path.dirname(destination), { recursive: true });
-      }
-      fs.renameSync(source, destination);
-    } else if (name === 'downloadFile') {
-      console.log(`Downloading image: ${filePath}`);
-      assert(fs.existsSync(filePath) || allowFileCreate, 'File create option was not enabled and file does not exist');
-      if (allowDirectoryCreate) {
-        fs.mkdirSync(path.dirname(filePath), { recursive: true });
-      }
-      try {
-        assert(args.downloadUrl, 'image url is not empty');
-        if (args.downloadUrl.startsWith('temp://')) {
-          assert(getTempBuffer(args.downloadUrl), 'Temp buffer not present but expected');
-          fs.writeFileSync(filePath, getTempBuffer(args.downloadUrl));
-        } else {
-          const imageResponse = await fetch(args.downloadUrl);
-          const arrayBuffer = await imageResponse.arrayBuffer();
-          const buffer = Buffer.from(arrayBuffer);
-          fs.writeFileSync(filePath, buffer);
-        }
-        console.log(`Image download and saved to: ${filePath}`);
-      } catch (error) {
-        console.error(`Failed to download image: ${error.message}`);
-        throw error;
-      }
-    } else if (name === 'imglyRemoveBackground') {
-      console.log(`Removing background from image: ${inputFilePath}`);
-      assert(fs.existsSync(inputFilePath), 'Input file does not exist');
-      assert(
-        allowFileCreate || fs.existsSync(outputFilePath),
-        'File create option was not enabled and output file does not exist',
-      );
-      if (allowDirectoryCreate) {
-        fs.mkdirSync(path.dirname(outputFilePath), { recursive: true });
-      }
-      try {
-        await imglyRemoveBackground(inputFilePath, outputFilePath, backgroundColor);
-        console.log(`Background removed and image saved to: ${outputFilePath}`);
-      } catch (error) {
-        console.error(`Failed to remove background: ${error.message}`);
-        throw error;
-      }
-    } else if (name === 'splitImage') {
-      console.log(`Splitting image: ${inputFilePath}`, parts);
-      assert(fs.existsSync(inputFilePath), 'Input file does not exist');
-      assert(Array.isArray(parts) && parts.length > 0, 'Parts array must not be empty');
-      for (const part of parts) {
-        assert(
-          allowFileCreate || fs.existsSync(part.outputFilePath),
-          'File create option was not enabled and output file does not exist',
-        );
-        if (allowDirectoryCreate) {
-          fs.mkdirSync(path.dirname(part.outputFilePath), { recursive: true });
-        }
-      }
-      try {
-        await splitImage(inputFilePath, parts);
-        console.log(`Image split successfully`);
-      } catch (error) {
-        console.error(`Failed to split image: ${error.message}`);
-        throw error;
-      }
-    } else if (name === 'resizeImage') {
-      console.log(`Resizing image: ${filePath}`, size);
-      assert(fs.existsSync(filePath), 'Input file does not exist');
-
-      try {
-        await resizeImageFile(filePath, size);
-        console.log(`Image resized successfully`);
-      } catch (error) {
-        console.error(`Failed to resize image: ${error.message}`);
-        throw error;
-      }
-    }
-  }
-}
-
-function isProjectPath(filePath) {
-  const sourceFiles = getSourceFiles();
-
-  return (
-    isAncestorDirectory(rcConfig.rootDir, filePath) ||
-    sourceFiles.includes(filePath) ||
-    !sourceFiles.some(
-      (sourceFile) =>
-        path.dirname(filePath) === path.dirname(sourceFile) ||
-        isAncestorDirectory(path.dirname(sourceFile), path.dirname(filePath)),
-    )
-  );
-}

package/src/images/ensure-alpha.js

@@ -1,5 +0,0 @@
-import sharp from 'sharp';
-
-export async function ensureAlpha(image) {
-  return sharp(image).ensureAlpha().toBuffer();
-}

package/src/images/imgly-remove-background.js

@@ -1,23 +0,0 @@
-import fs from 'fs';
-import { createRequire } from 'node:module';
-
-import { removeBackground } from '@imgly/background-removal-node';
-
-/** Converts white color to transparency on a image and saves to destination path */
-export async function imglyRemoveBackground(inputFilePath, outputFilePath) {
-  try {
-    console.log(`Removing background for image: ${inputFilePath}`);
-
-    const publicPath = 'file://' + createRequire(import.meta.url).resolve('@imgly/background-removal-node');
-    const blob = await removeBackground(inputFilePath, {
-      publicPath,
-    });
-    fs.writeFileSync(outputFilePath, Buffer.from(await blob.arrayBuffer()));
-
-    console.log(`Background removed successfully. Saved to: ${outputFilePath}`);
-    return outputFilePath;
-  } catch (error) {
-    console.error('Error removing background:', error);
-    throw error;
-  }
-}

package/src/images/resize-image.js

@@ -1,23 +0,0 @@
-import sharp from 'sharp';
-import fs from 'fs';
-
-/**
- *
- * @param {Buffer} input
- * @param {{width: number, height: number}} size
- * @returns {Buffer}
- */
-export async function resizeImageBuffer(input, size) {
-  return await sharp(input).resize(size.width, size.height).toBuffer();
-}
-
-/**
- *
- * @param {string} filePath
- * @param {{width: number, height: number}} size
- * @returns {Buffer}
- */
-export async function resizeImageFile(filePath, size) {
-  const buffer = await resizeImageBuffer(filePath, size);
-  fs.writeFileSync(filePath, buffer);
-}

package/src/images/split-image.js

@@ -1,26 +0,0 @@
-import sharp from 'sharp';
-
-/**
- * This function opens the input file, and splits its into parts, extracting parts of the input image, and saving to outputFilePaths
- *
- * @param {string} inputFilePath
- * @param {Array<{rect: {x: number, y: number, width: number, height: number}, outputFilePath: string}>} parts
- */
-export async function splitImage(inputFilePath, parts) {
-  try {
-    // Process each part
-    for (const part of parts) {
-      const { rect, outputFilePath } = part;
-      const { x, y, width, height } = rect;
-
-      // Extract the specified rectangle from the input image
-      console.log('Extracting', rect, outputFilePath);
-      await sharp(inputFilePath).extract({ left: x, top: y, width, height }).toFile(outputFilePath);
-    }
-
-    console.log('Image splitting completed successfully.');
-  } catch (error) {
-    console.error('Error splitting image:', error);
-    throw error;
-  }
-}