refactor: replace component creation instruction with generic task step functionality

Author: YusukeHirao

src/get-component-creation-instruction.ts

@@ -8,7 +8,7 @@ import { z } from 'zod';
 import { getCodingGuidelines } from './coding-guideline.js';
 import { getFigmaData } from './figma/get-data.js';
 import { getFigmaImage } from './figma/get-image.js';
-import { getComponentCreationInstruction } from './get-component-creation-instruction.js';
+import { getTaskStep } from './utils/get-task-step.js';
 
 const packageJsonPath = path.resolve(import.meta.dirname, '../package.json');
 const packageJson = JSON.parse(fs.readFileSync(packageJsonPath, 'utf8'));
@@ -43,13 +43,18 @@ server.tool(
 );
 
 server.tool(
-	'get_component_creation_instruction',
-	'Get component creation instruction',
+	'get_task_step',
+	'Get task step',
 	{
+		cwd: z.string().describe('Current working directory (Absolute path)'),
+		filePath: z.string().describe('Task file path (Relative to cwd)'),
 		step: z.optional(z.number().min(1).int()).describe('Step number (default: 1)'),
 	},
-	({ step }) => {
-		const content = getComponentCreationInstruction(step);
+	async ({ cwd, filePath, step }) => {
+		const content = await getTaskStep(
+			path.isAbsolute(filePath) ? filePath : path.join(cwd, filePath),
+			step,
+		);
 		return {
 			content: [
 				{

src/index.ts

@@ -0,0 +1,36 @@
+import { promises as fs } from 'node:fs';
+
+import { splitTaskSteps } from './split-task-steps.js';
+
+const cache = new Map<string, string[]>();
+
+/**
+ *
+ * @param filePath
+ * @param step
+ */
+export async function getTaskStep(filePath: string, step = 1) {
+	if (cache.has(filePath)) {
+		const steps = cache.get(filePath);
+		if (!steps) {
+			throw new Error('No steps found');
+		}
+		return getStep(steps, step);
+	}
+	const task = await fs.readFile(filePath, 'utf8');
+	const steps = splitTaskSteps(task);
+	cache.set(filePath, steps);
+	return getStep(steps, step);
+}
+
+/**
+ *
+ * @param steps
+ * @param step
+ */
+function getStep(steps: string[], step: number) {
+	if (steps.length < step) {
+		return '全てのステップが完了しました。';
+	}
+	return steps.at(step - 1) ?? 'ステップが見つかりません。次のステップに進んでください。';
+}

src/utils/get-task-step.ts

@@ -0,0 +1,25 @@
+import { describe, it, expect } from 'vitest';
+
+import { splitTaskSteps } from './split-task-steps.js';
+
+describe('splitTaskSteps', () => {
+	it('should split the task into steps', () => {
+		const task = `
+# 1
+Step 1
+
+---
+
+# 2
+Step 2
+
+---
+
+# 3
+Step 3
+
+`;
+		const steps = splitTaskSteps(task);
+		expect(steps).toEqual(['Step 1\n\n---', 'Step 2\n\n---', 'Step 3']);
+	});
+});

src/utils/split-task-steps.spec.ts

@@ -0,0 +1,15 @@
+/**
+ *
+ * @param task
+ */
+export function splitTaskSteps(task: string) {
+	const steps = task.trim().split(/(?:^|\n\r?)(?=#\s)/);
+	return steps.map((step) => {
+		const lines = step
+			.trim()
+			.split('\n')
+			.map((line) => line.trim())
+			.filter((line) => !/^#\s+\d+/.test(line));
+		return lines.join('\n');
+	});
+}