ai-code-writer
Version:
An AI code writer application using OpenAI APIs for audio transcription and chat completion.
121 lines (120 loc) • 5.03 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
__setModuleDefault(result, mod);
return result;
};
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
const fs = __importStar(require("fs"));
const path = __importStar(require("path"));
class FileCollector {
constructor(sourceDir, patterns, excludeDirs, excludeFiles) {
this.sourceDir = sourceDir;
this.patterns = patterns;
this.excludeDirs = excludeDirs;
this.excludeFiles = excludeFiles;
}
collectFiles() {
return __awaiter(this, void 0, void 0, function* () {
const collectedFiles = [];
const files = yield this.findFiles(this.sourceDir);
for (const filePath of files) {
const relativeFilePath = path.relative(process.cwd(), filePath);
if (this.isExcluded(relativeFilePath))
continue;
const excludedByPart = relativeFilePath.split(path.sep)
.find((p) => this.isExcluded(p)) !== undefined;
if (excludedByPart)
continue;
const fileContent = yield this.readFileStream(filePath);
collectedFiles.push({ filePath: relativeFilePath, content: fileContent });
}
return collectedFiles;
});
}
isExcluded(filePath) {
for (const excludeDir of this.excludeDirs) {
if (filePath.includes(path.normalize(excludeDir) + path.sep))
return true;
}
for (const excludeFile of this.excludeFiles) {
if (this.matchPattern(filePath, excludeFile))
return true;
}
return false;
}
findFiles(dir) {
return __awaiter(this, void 0, void 0, function* () {
const entries = yield fs.promises.readdir(dir, { withFileTypes: true });
const files = [];
for (const entry of entries) {
const fullPath = path.resolve(dir, entry.name);
if (entry.isDirectory()) {
files.push(...yield this.findFiles(fullPath));
}
else {
for (const pattern of this.patterns) {
if (this.matchPattern(entry.name, pattern)) {
files.push(fullPath);
break;
}
}
}
}
return files;
});
}
matchPattern(filePath, pattern) {
const regexPattern = pattern
.replace(/\./g, '\\.')
.replace(/\*/g, '.*')
.replace(/\?/g, '.');
const regex = new RegExp(`^${regexPattern}$`);
return regex.test(filePath);
}
readFileStream(filePath) {
return __awaiter(this, void 0, void 0, function* () {
return new Promise((resolve, reject) => {
let content = '';
const stream = fs.createReadStream(filePath, { encoding: 'utf8' });
stream.on('data', (chunk) => {
content += chunk;
});
stream.on('end', () => {
resolve(content);
});
stream.on('error', (error) => {
reject(error);
});
});
});
}
}
exports.default = FileCollector;