@donkeylabs/mcp 0.4.4 → 0.4.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@donkeylabs/mcp",
-  "version": "0.4.4",
+  "version": "0.4.5",
   "description": "MCP server for AI-assisted development with @donkeylabs/server",
   "type": "module",
   "main": "./src/server.ts",

package/src/server.ts

@@ -624,7 +624,7 @@ const tools = [
   },
   {
     name: "add_migration",
-    description: "Create a numbered SQL migration file for a plugin's database schema.",
+    description: "Create a numbered Kysely migration file for a plugin's database schema. Use Kysely schema builder - NOT raw SQL.",
     inputSchema: {
       type: "object" as const,
       properties: {
@@ -636,16 +636,16 @@ const tools = [
           type: "string",
           description: "Descriptive name (e.g., 'create_users', 'add_email_column')",
         },
-        upSql: {
+        upCode: {
           type: "string",
-          description: "SQL for the up migration (CREATE TABLE, ALTER TABLE, etc.)",
+          description: "Kysely schema builder code for up migration. Example: 'await db.schema.createTable(\"users\").addColumn(\"id\", \"text\", (col) => col.primaryKey()).addColumn(\"email\", \"text\", (col) => col.notNull().unique()).execute();'",
         },
-        downSql: {
+        downCode: {
           type: "string",
-          description: "SQL for the down migration (DROP TABLE, etc.)",
+          description: "Kysely schema builder code for down migration. Example: 'await db.schema.dropTable(\"users\").execute();'",
         },
       },
-      required: ["pluginName", "migrationName", "upSql"],
+      required: ["pluginName", "migrationName", "upCode"],
     },
   },
   {
@@ -972,7 +972,7 @@ async function getArchitectureGuidance(args: { task: string }): Promise<string>
     guidance += `\`\`\`\nTool: create_plugin\n  name: "auth"\n  hasSchema: true\n\`\`\`\n\n`;
 
     guidance += `### Step 2: Add Database Migration\n`;
-    guidance += `\`\`\`\nTool: add_migration\n  pluginName: "auth"\n  migrationName: "create_users"\n  upSql: "CREATE TABLE users (id INTEGER PRIMARY KEY, email TEXT UNIQUE, password_hash TEXT, created_at TEXT DEFAULT CURRENT_TIMESTAMP)"\n\`\`\`\n\n`;
+    guidance += `\`\`\`\nTool: add_migration\n  pluginName: "auth"\n  migrationName: "create_users"\n  upCode: 'await db.schema.createTable("users").addColumn("id", "integer", (col) => col.primaryKey().autoIncrement()).addColumn("email", "text", (col) => col.notNull().unique()).addColumn("password_hash", "text", (col) => col.notNull()).addColumn("created_at", "text", (col) => col.defaultTo("CURRENT_TIMESTAMP")).execute();'\n  downCode: 'await db.schema.dropTable("users").execute();'\n\`\`\`\n\n`;
 
     guidance += `### Step 3: Add Service Methods\n`;
     guidance += `Add these methods using \`add_service_method\`:\n`;
@@ -1192,9 +1192,7 @@ async function createPlugin(args: {
   // Generate imports
   let imports = `import { createPlugin } from "@donkeylabs/server";\n`;
   imports += `import { z } from "zod";\n`;
-  if (dependencies.length > 0) {
-    imports += dependencies.map(d => `import { ${d}Plugin } from "../${d}";`).join("\n") + "\n";
-  }
+  // Note: dependencies are just string names, no imports needed
   if (hasSchema) {
     // Note: schema.ts is auto-generated after running `donkeylabs generate`
     // Once generated, uncomment this import:
@@ -1209,7 +1207,7 @@ async function createPlugin(args: {
   }
 
   const depsArray = dependencies.length > 0
-    ? `\n  dependencies: [${dependencies.map(d => `${d}Plugin`).join(", ")}] as const,`
+    ? `\n  dependencies: [${dependencies.map(d => `"${d}"`).join(", ")}] as const,`
     : "";
 
   // Build the service context comment based on what's available
@@ -1258,34 +1256,9 @@ ${ctxComment}
 
   // Create migrations folder if plugin has schema
   // Note: schema.ts is auto-generated by `donkeylabs generate` from migrations
+  // No initial migration is created - use add_migration tool to add migrations
   if (hasSchema) {
     mkdirSync(join(pluginDir, "migrations"), { recursive: true });
-
-    const migrationContent = `import { Kysely } from "kysely";
-
-/**
- * Migration: 001_initial
- * Created: ${new Date().toISOString()}
- * Plugin: ${name}
- *
- * Use Kysely schema builder methods - NOT raw SQL.
- * See docs/database.md for examples.
- */
-
-export async function up(db: Kysely<any>): Promise<void> {
-  await db.schema
-    .createTable("${name}")
-    .addColumn("id", "text", (col) => col.primaryKey())
-    .addColumn("name", "text", (col) => col.notNull())
-    .addColumn("created_at", "text", (col) => col.notNull())
-    .execute();
-}
-
-export async function down(db: Kysely<any>): Promise<void> {
-  await db.schema.dropTable("${name}").execute();
-}
-`;
-    await Bun.write(join(pluginDir, "migrations", "001_initial.ts"), migrationContent);
   }
 
   // Build registration example based on plugin type
@@ -1308,7 +1281,7 @@ server.registerPlugin(${name}Plugin);
 
 **Location:** ${projectConfig.pluginsDir}/${name}/
 **Files created:**
-- index.ts (plugin definition)${hasSchema ? "\n- migrations/001_initial.ts (Kysely migration)\n- schema.ts (auto-generated by donkeylabs generate)" : ""}
+- index.ts (plugin definition)${hasSchema ? "\n- migrations/ (empty - use add_migration to add)" : ""}
 
 **Plugin features:**
 ${hasSchema ? "- ✅ Database schema (withSchema<>)\n" : ""}${hasConfig ? "- ✅ Configurable (withConfig<>)\n" : ""}- Service with ctx.core (logger, cache, events, cron, jobs, sse, rateLimiter)
@@ -1320,9 +1293,9 @@ ${registrationExample}
 
 ### Next Steps
 
-1. ${hasSchema ? "Edit the migration file with your schema" : "Add service methods using `add_service_method`"}
-2. ${hasSchema ? "Run \`donkeylabs generate\` to create types" : "Register the plugin in your server"}
-3. Use \`extend_plugin\` to add errors, events, or middleware
+1. ${hasSchema ? "Use \\`add_migration\\` to create your database schema" : "Add service methods using `add_service_method`"}
+2. ${hasSchema ? "**IMPORTANT:** Run \\`donkeylabs generate\\` after migrations to generate schema types" : "Register the plugin in your server"}
+3. ${hasSchema ? "Update plugin to use generated DB type: \\`.withSchema<DB>()\\` and import from ./schema" : "Use \\`extend_plugin\\` to add errors, events, or middleware"}
 
 ### Example Usage
 
@@ -1410,10 +1383,10 @@ const result = await ctx.plugins.${pluginName}.${methodName}(${params ? "..." :
 async function addMigration(args: {
   pluginName: string;
   migrationName: string;
-  upSql: string;
-  downSql?: string;
+  upCode: string;
+  downCode?: string;
 }): Promise<string> {
-  const { pluginName, migrationName, upSql, downSql = "" } = args;
+  const { pluginName, migrationName, upCode, downCode = "" } = args;
 
   const pluginValid = validatePluginExists(pluginName);
   if (!pluginValid.valid) {
@@ -1426,9 +1399,9 @@ async function addMigration(args: {
     mkdirSync(migrationsDir, { recursive: true });
   }
 
-  // Find next migration number - check both .ts and .sql files for backwards compatibility
+  // Find next migration number
   const existing = readdirSync(migrationsDir)
-    .filter((f) => f.endsWith(".ts") || f.endsWith(".sql"))
+    .filter((f) => f.endsWith(".ts"))
     .map((f) => parseInt(f.split("_")[0], 10))
     .filter((n) => !isNaN(n));
 
@@ -1436,11 +1409,7 @@ async function addMigration(args: {
   const numStr = String(nextNum).padStart(3, "0");
   const filename = `${numStr}_${migrationName}.ts`;
 
-  // Escape backticks in SQL for template literal
-  const escapedUpSql = upSql.replace(/`/g, "\\`");
-  const escapedDownSql = (downSql || "").replace(/`/g, "\\`");
-
-  const content = `import { Kysely, sql } from "kysely";
+  const content = `import { Kysely } from "kysely";
 
 /**
  * Migration: ${numStr}_${migrationName}
@@ -1449,11 +1418,11 @@ async function addMigration(args: {
  */
 
 export async function up(db: Kysely<any>): Promise<void> {
-  await sql\`${escapedUpSql}\`.execute(db);
+  ${upCode}
 }
 
 export async function down(db: Kysely<any>): Promise<void> {
-  ${escapedDownSql ? `await sql\`${escapedDownSql}\`.execute(db);` : "// Add rollback logic here"}
+  ${downCode || "// Add rollback logic here"}
 }
 `;
 
@@ -1464,14 +1433,27 @@ export async function down(db: Kysely<any>): Promise<void> {
 **Plugin:** ${pluginName}
 **Path:** ${projectConfig.pluginsDir}/${pluginName}/migrations/${filename}
 
-### Next Steps
+### ⚠️ REQUIRED: Run Codegen
+
+After creating/editing migrations, you MUST run:
+\`\`\`bash
+donkeylabs generate
+\`\`\`
+
+This generates \`${pluginName}/schema.ts\` with typed DB interfaces.
+
+### After Codegen
 
-1. Review the migration
-2. Run \`donkeylabs generate\` to update schema types
-3. The migration will run automatically on server start
+Update the plugin to use the generated types:
+\`\`\`typescript
+import type { DB } from "./schema";
+
+export const ${pluginName}Plugin = createPlugin
+  .withSchema<DB>()
+  .define({ ... });
+\`\`\`
 
-### Note
-Make sure to update the plugin's schema.ts file with the new table types.
+The migration will run automatically on server start.
 `;
 }