mirror of
https://github.com/catlog22/Claude-Code-Workflow.git
synced 2026-02-12 02:37:45 +08:00
feat(ccw): migrate backend to TypeScript
- Convert 40 JS files to TypeScript (CLI, tools, core, MCP server) - Add Zod for runtime parameter validation - Add type definitions in src/types/ - Keep src/templates/ as JavaScript (dashboard frontend) - Update bin entries to use dist/ - Add tsconfig.json with strict mode - Add backward-compatible exports for tests - All 39 tests passing Breaking changes: None (backward compatible) 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
245
ccw/src/tools/classify-folders.ts
Normal file
245
ccw/src/tools/classify-folders.ts
Normal file
@@ -0,0 +1,245 @@
|
||||
/**
|
||||
* Classify Folders Tool
|
||||
* Categorize folders by type for documentation generation
|
||||
* Types: code (API.md + README.md), navigation (README.md only), skip (empty)
|
||||
*/
|
||||
|
||||
import { z } from 'zod';
|
||||
import type { ToolSchema, ToolResult } from '../types/tool.js';
|
||||
import { readdirSync, statSync, existsSync } from 'fs';
|
||||
import { join, resolve, extname } from 'path';
|
||||
|
||||
// Code file extensions
|
||||
const CODE_EXTENSIONS = [
|
||||
'.ts', '.tsx', '.js', '.jsx',
|
||||
'.py', '.go', '.java', '.rs',
|
||||
'.c', '.cpp', '.cs', '.rb',
|
||||
'.php', '.swift', '.kt'
|
||||
];
|
||||
|
||||
// Define Zod schema for validation
|
||||
const ParamsSchema = z.object({
|
||||
input: z.string().optional(),
|
||||
path: z.string().optional(),
|
||||
}).refine(data => data.input || data.path, {
|
||||
message: 'Either "input" or "path" parameter is required'
|
||||
});
|
||||
|
||||
type Params = z.infer<typeof ParamsSchema>;
|
||||
|
||||
interface FolderClassification {
|
||||
type: 'code' | 'navigation' | 'skip';
|
||||
codeFiles: number;
|
||||
subdirs: number;
|
||||
}
|
||||
|
||||
interface ClassificationResult {
|
||||
path: string;
|
||||
type: 'code' | 'navigation' | 'skip';
|
||||
code_files: number;
|
||||
subdirs: number;
|
||||
}
|
||||
|
||||
interface ToolOutput {
|
||||
total: number;
|
||||
by_type: {
|
||||
code: number;
|
||||
navigation: number;
|
||||
skip: number;
|
||||
};
|
||||
results: ClassificationResult[];
|
||||
output: string;
|
||||
}
|
||||
|
||||
/**
|
||||
* Count code files in a directory (non-recursive)
|
||||
*/
|
||||
function countCodeFiles(dirPath: string): number {
|
||||
try {
|
||||
const entries = readdirSync(dirPath, { withFileTypes: true });
|
||||
return entries.filter(e => {
|
||||
if (!e.isFile()) return false;
|
||||
const ext = extname(e.name).toLowerCase();
|
||||
return CODE_EXTENSIONS.includes(ext);
|
||||
}).length;
|
||||
} catch (e) {
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Count subdirectories in a directory
|
||||
*/
|
||||
function countSubdirs(dirPath: string): number {
|
||||
try {
|
||||
const entries = readdirSync(dirPath, { withFileTypes: true });
|
||||
return entries.filter(e => e.isDirectory() && !e.name.startsWith('.')).length;
|
||||
} catch (e) {
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Determine folder type
|
||||
*/
|
||||
function classifyFolder(dirPath: string): FolderClassification {
|
||||
const codeFiles = countCodeFiles(dirPath);
|
||||
const subdirs = countSubdirs(dirPath);
|
||||
|
||||
if (codeFiles > 0) {
|
||||
return { type: 'code', codeFiles, subdirs }; // Generates API.md + README.md
|
||||
} else if (subdirs > 0) {
|
||||
return { type: 'navigation', codeFiles, subdirs }; // README.md only
|
||||
} else {
|
||||
return { type: 'skip', codeFiles, subdirs }; // Empty or no relevant content
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Parse input from get_modules_by_depth format
|
||||
* Format: depth:N|path:./path|files:N|types:[ext,ext]|has_claude:yes/no
|
||||
*/
|
||||
function parseModuleInput(line: string): Record<string, string> {
|
||||
const parts: Record<string, string> = {};
|
||||
line.split('|').forEach(part => {
|
||||
const [key, value] = part.split(':');
|
||||
if (key && value !== undefined) {
|
||||
parts[key] = value;
|
||||
}
|
||||
});
|
||||
return parts;
|
||||
}
|
||||
|
||||
// Tool schema for MCP
|
||||
export const schema: ToolSchema = {
|
||||
name: 'classify_folders',
|
||||
description: `Classify folders by type for documentation generation.
|
||||
Types:
|
||||
- code: Contains code files (generates API.md + README.md)
|
||||
- navigation: Contains subdirectories only (generates README.md only)
|
||||
- skip: Empty or no relevant content
|
||||
|
||||
Input: Either piped output from get_modules_by_depth or a single directory path.`,
|
||||
inputSchema: {
|
||||
type: 'object',
|
||||
properties: {
|
||||
input: {
|
||||
type: 'string',
|
||||
description: 'Piped input from get_modules_by_depth (one module per line)'
|
||||
},
|
||||
path: {
|
||||
type: 'string',
|
||||
description: 'Single directory path to classify'
|
||||
}
|
||||
},
|
||||
required: []
|
||||
}
|
||||
};
|
||||
|
||||
// Handler function
|
||||
export async function handler(params: Record<string, unknown>): Promise<ToolResult<ToolOutput>> {
|
||||
const parsed = ParamsSchema.safeParse(params);
|
||||
if (!parsed.success) {
|
||||
return { success: false, error: `Invalid params: ${parsed.error.message}` };
|
||||
}
|
||||
|
||||
const { input, path: targetPath } = parsed.data;
|
||||
|
||||
const results: ClassificationResult[] = [];
|
||||
|
||||
try {
|
||||
// Mode 1: Process piped input from get_modules_by_depth
|
||||
if (input) {
|
||||
let lines: string[];
|
||||
|
||||
// Check if input is JSON (from ccw tool exec output)
|
||||
if (input.trim().startsWith('{')) {
|
||||
try {
|
||||
const jsonInput = JSON.parse(input);
|
||||
// Handle output from get_modules_by_depth tool (wrapped in result)
|
||||
const output = jsonInput.result?.output || jsonInput.output;
|
||||
if (output) {
|
||||
lines = output.split('\n');
|
||||
} else {
|
||||
lines = [input];
|
||||
}
|
||||
} catch {
|
||||
// Not JSON, treat as line-delimited text
|
||||
lines = input.split('\n');
|
||||
}
|
||||
} else {
|
||||
lines = input.split('\n');
|
||||
}
|
||||
|
||||
for (const line of lines) {
|
||||
if (!line.trim()) continue;
|
||||
|
||||
const parsed = parseModuleInput(line);
|
||||
const folderPath = parsed.path;
|
||||
|
||||
if (!folderPath) continue;
|
||||
|
||||
const basePath = targetPath ? resolve(process.cwd(), targetPath) : process.cwd();
|
||||
const fullPath = resolve(basePath, folderPath);
|
||||
|
||||
if (!existsSync(fullPath) || !statSync(fullPath).isDirectory()) {
|
||||
continue;
|
||||
}
|
||||
|
||||
const classification = classifyFolder(fullPath);
|
||||
|
||||
results.push({
|
||||
path: folderPath,
|
||||
type: classification.type,
|
||||
code_files: classification.codeFiles,
|
||||
subdirs: classification.subdirs
|
||||
});
|
||||
}
|
||||
}
|
||||
// Mode 2: Classify a single directory
|
||||
else if (targetPath) {
|
||||
const fullPath = resolve(process.cwd(), targetPath);
|
||||
|
||||
if (!existsSync(fullPath)) {
|
||||
return { success: false, error: `Directory not found: ${fullPath}` };
|
||||
}
|
||||
|
||||
if (!statSync(fullPath).isDirectory()) {
|
||||
return { success: false, error: `Not a directory: ${fullPath}` };
|
||||
}
|
||||
|
||||
const classification = classifyFolder(fullPath);
|
||||
|
||||
results.push({
|
||||
path: targetPath,
|
||||
type: classification.type,
|
||||
code_files: classification.codeFiles,
|
||||
subdirs: classification.subdirs
|
||||
});
|
||||
}
|
||||
|
||||
// Format output
|
||||
const output = results.map(r =>
|
||||
`${r.path}|${r.type}|code:${r.code_files}|dirs:${r.subdirs}`
|
||||
).join('\n');
|
||||
|
||||
return {
|
||||
success: true,
|
||||
result: {
|
||||
total: results.length,
|
||||
by_type: {
|
||||
code: results.filter(r => r.type === 'code').length,
|
||||
navigation: results.filter(r => r.type === 'navigation').length,
|
||||
skip: results.filter(r => r.type === 'skip').length
|
||||
},
|
||||
results,
|
||||
output
|
||||
}
|
||||
};
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: `Failed to classify folders: ${(error as Error).message}`
|
||||
};
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user