bizgate-mcp-server 0.4.0 → 0.5.0

package/README.md

@@ -60,8 +60,8 @@ Mac の場合：
 
 ```smalltalk
 claude mcp add bizgate --scope user \
-  -e "BIZGATE_USERNAME=digi-man_bizg1" \
-  -e "BIZGATE_PASSWORD=digi-man_bizg1" \
+  -e "BIZGATE_USERNAME=digi-man_testbizg1" \
+  -e "BIZGATE_PASSWORD=digi-man_testbizg1" \
   -e "BIZGATE_AUTH_MODE=basic" \
   -e "BIZGATE_DAILY_LIMIT=200" \
   -e "BIZGATE_SKEY_COMPANY=/EhMJ9YMCJtJgo73.DjLuew8rnnTlb.F6/MuiESFXmZwlCKvG8bMm" \

package/dist/cloud/neon-client.d.ts

@@ -2,35 +2,28 @@ import { type PoolClient } from "pg";
 export declare function isConfigured(): boolean;
 export declare function ensureSchema(): Promise<void>;
 export declare function withTx<T>(fn: (c: PoolClient) => Promise<T>): Promise<T>;
-export declare function upsertUser(args: {
-    slack_user_id: string;
-    papatto_email: string;
-    display_name?: string;
-}): Promise<void>;
 export interface CampaignRow {
     campaign_id: string;
     owner_slack_user_id: string;
     natural_language_query: string | null;
-    papatto_conditions: unknown;
+    enrich_input: unknown;
     extracted_count: number | null;
     status: "extracting" | "done" | "failed" | string;
     created_at: number;
     updated_at: number;
     r2_extracted_url: string | null;
-    sheet_url: string | null;
 }
 export declare function insertCampaign(args: {
     campaign_id: string;
     owner_slack_user_id: string;
     natural_language_query?: string;
-    papatto_conditions?: unknown;
+    enrich_input?: unknown;
 }): Promise<void>;
 export declare function updateCampaign(args: {
     campaign_id: string;
     status?: "extracting" | "done" | "failed";
     extracted_count?: number;
     r2_extracted_url?: string;
-    sheet_url?: string;
 }): Promise<void>;
 export declare function getCampaign(campaign_id: string): Promise<CampaignRow | null>;
 export declare function listCampaignsByOwner(owner_slack_user_id: string, limit?: number): Promise<CampaignRow[]>;

package/dist/cloud/neon-client.js

@@ -14,7 +14,7 @@ function pool() {
     if (!cachedPool) {
         const url = process.env.NEON_DATABASE_URL;
         if (!url)
-            throw new Error("NEON_DATABASE_URL 환경변수가 설정되지 않았습니다");
+            throw new Error("NEON_DATABASE_URL 環境変数が設定されていません");
         cachedPool = new Pool({
             connectionString: url,
             max: 4,
@@ -51,31 +51,21 @@ export async function withTx(fn) {
         client.release();
     }
 }
-export async function upsertUser(args) {
-    await ensureSchema();
-    const now = Date.now();
-    await pool().query(`INSERT INTO papatto.users (slack_user_id, papatto_email, display_name, created_at, updated_at)
-     VALUES ($1, $2, $3, $4, $4)
-     ON CONFLICT (slack_user_id) DO UPDATE
-       SET papatto_email = EXCLUDED.papatto_email,
-           display_name  = COALESCE(EXCLUDED.display_name, papatto.users.display_name),
-           updated_at    = EXCLUDED.updated_at`, [args.slack_user_id, args.papatto_email, args.display_name ?? null, now]);
-}
 export async function insertCampaign(args) {
     await ensureSchema();
     const now = Date.now();
     await pool().query(`INSERT INTO papatto.campaigns
-       (campaign_id, owner_slack_user_id, natural_language_query, papatto_conditions,
+       (campaign_id, owner_slack_user_id, natural_language_query, enrich_input,
         status, created_at, updated_at)
      VALUES ($1, $2, $3, $4, 'extracting', $5, $5)
      ON CONFLICT (campaign_id) DO UPDATE
        SET natural_language_query = COALESCE(EXCLUDED.natural_language_query, papatto.campaigns.natural_language_query),
-           papatto_conditions     = COALESCE(EXCLUDED.papatto_conditions, papatto.campaigns.papatto_conditions),
+           enrich_input           = COALESCE(EXCLUDED.enrich_input, papatto.campaigns.enrich_input),
            updated_at             = EXCLUDED.updated_at`, [
         args.campaign_id,
         args.owner_slack_user_id,
         args.natural_language_query ?? null,
-        args.papatto_conditions ? JSON.stringify(args.papatto_conditions) : null,
+        args.enrich_input ? JSON.stringify(args.enrich_input) : null,
         now,
     ]);
 }
@@ -96,10 +86,6 @@ export async function updateCampaign(args) {
         sets.push(`r2_extracted_url = $${i++}`);
         params.push(args.r2_extracted_url);
     }
-    if (args.sheet_url !== undefined) {
-        sets.push(`sheet_url = $${i++}`);
-        params.push(args.sheet_url);
-    }
     if (sets.length === 0)
         return;
     sets.push(`updated_at = $${i++}`);
@@ -110,8 +96,8 @@ export async function updateCampaign(args) {
 export async function getCampaign(campaign_id) {
     await ensureSchema();
     const res = await pool().query(`SELECT campaign_id, owner_slack_user_id, natural_language_query,
-            papatto_conditions, extracted_count, status,
-            created_at, updated_at, r2_extracted_url, sheet_url
+            enrich_input, extracted_count, status,
+            created_at, updated_at, r2_extracted_url
        FROM papatto.campaigns
        WHERE campaign_id = $1`, [campaign_id]);
     return res.rows[0] ?? null;
@@ -119,8 +105,8 @@ export async function getCampaign(campaign_id) {
 export async function listCampaignsByOwner(owner_slack_user_id, limit = 20) {
     await ensureSchema();
     const res = await pool().query(`SELECT campaign_id, owner_slack_user_id, natural_language_query,
-            papatto_conditions, extracted_count, status,
-            created_at, updated_at, r2_extracted_url, sheet_url
+            enrich_input, extracted_count, status,
+            created_at, updated_at, r2_extracted_url
        FROM papatto.campaigns
        WHERE owner_slack_user_id = $1
        ORDER BY created_at DESC

package/dist/enrich/core.d.ts

@@ -0,0 +1,38 @@
+export interface CompanyInput {
+    name: string;
+    corporate_number?: string;
+}
+export interface EnrichedCompany {
+    name: string;
+    bizgate_matched: boolean;
+    match_pattern?: string;
+    corporate_number?: string;
+    ceo?: string;
+    revenue?: string;
+    industry?: string;
+    hp_url?: string;
+    email?: string;
+    capital?: string;
+    emp?: string;
+    address?: string;
+    phone?: string;
+    departments?: Array<{
+        name: string;
+        phone?: string;
+    }>;
+    keymen?: Array<{
+        role: string;
+    }>;
+    priority_score: number;
+    priority_reason?: string;
+}
+export declare function heuristicScore(c: EnrichedCompany): {
+    score: number;
+    reason: string;
+};
+export declare function enrichWithCompanySearch(rows: CompanyInput[]): Promise<{
+    items: EnrichedCompany[];
+    api_calls: number;
+}>;
+export declare function enrichDepartmentsAndKeymen(targets: EnrichedCompany[]): Promise<number>;
+export declare function rowsToCsv(items: EnrichedCompany[]): string;

package/dist/enrich/core.js

@@ -0,0 +1,171 @@
+import { client as bizgateClient } from "../index.js";
+import { first } from "../bizgate-client.js";
+import { log } from "../shared/logger.js";
+export function heuristicScore(c) {
+    let s = 50;
+    const reasons = [];
+    const cap = c.capital ?? "";
+    if (/1兆|5000億|1000億/.test(cap)) {
+        s += 25;
+        reasons.push("超大企業規模");
+    }
+    else if (/500億|300億/.test(cap)) {
+        s += 18;
+        reasons.push("大企業");
+    }
+    else if (/100億|50億/.test(cap)) {
+        s += 10;
+        reasons.push("中堅規模");
+    }
+    const emp = c.emp ?? "";
+    if (/5000人|1000人/.test(emp)) {
+        s += 12;
+        reasons.push("従業員 1000+");
+    }
+    else if (/500人|100人/.test(emp)) {
+        s += 6;
+        reasons.push("従業員 100~");
+    }
+    if (c.bizgate_matched) {
+        s += 5;
+        reasons.push("BizGate データ確保");
+    }
+    if (c.departments && c.departments.length >= 3) {
+        s += 5;
+        reasons.push(`部署 ${c.departments.length} 件`);
+    }
+    if (c.keymen && c.keymen.length > 0) {
+        s += 10;
+        reasons.push("キーマン情報あり");
+    }
+    if (c.hp_url) {
+        s += 2;
+    }
+    if (c.email) {
+        s += 3;
+        reasons.push("メール公開");
+    }
+    return {
+        score: Math.min(100, s),
+        reason: reasons.slice(0, 3).join(" / ") || "基本情報のみ",
+    };
+}
+export async function enrichWithCompanySearch(rows) {
+    const items = [];
+    let calls = 0;
+    for (const r of rows) {
+        const ec = {
+            name: r.name,
+            corporate_number: r.corporate_number,
+            bizgate_matched: false,
+            priority_score: 0,
+        };
+        try {
+            const params = { shogo: r.name };
+            if (r.corporate_number)
+                params.compno = r.corporate_number;
+            const { matchPattern, docs } = await bizgateClient.searchCompany(params);
+            calls++;
+            if (docs.length > 0) {
+                const d = docs[0];
+                ec.bizgate_matched = true;
+                ec.match_pattern = matchPattern;
+                ec.corporate_number = d.compno ?? ec.corporate_number;
+                ec.ceo = first(d.ceo);
+                ec.revenue = d.revenue ? String(d.revenue) : undefined;
+                ec.industry = first(d.gyoshu_facet);
+                ec.hp_url = first(d.hpurl);
+                ec.email = first(d.mail);
+                ec.capital = d.shihon ? String(d.shihon) : undefined;
+                ec.emp = d.emp ? String(d.emp) : undefined;
+                ec.address = first(d.add);
+                ec.phone = first(d.tel);
+            }
+        }
+        catch (e) {
+            log("warn", "bizgate company_search failed", { name: r.name, error: String(e) });
+        }
+        const sc = heuristicScore(ec);
+        ec.priority_score = sc.score;
+        ec.priority_reason = sc.reason;
+        items.push(ec);
+    }
+    return { items, api_calls: calls };
+}
+export async function enrichDepartmentsAndKeymen(targets) {
+    let calls = 0;
+    for (const c of targets) {
+        if (!c.bizgate_matched)
+            continue;
+        try {
+            const dept = await bizgateClient.searchDepartments({
+                compno: c.corporate_number,
+                shogo: c.name,
+            });
+            calls++;
+            c.departments = dept.docs.slice(0, 10).map((d) => ({
+                name: first(d.bumon) || "(不明)",
+                phone: first(d.tel),
+            }));
+        }
+        catch (e) {
+            log("warn", "department_search failed", { name: c.name, error: String(e) });
+        }
+        try {
+            const km = await bizgateClient.searchKeyman({
+                compno: c.corporate_number,
+                shogo: c.name,
+            });
+            calls++;
+            c.keymen = km.docs.slice(0, 5).map((d) => ({ role: first(d.bumon) || "(不明)" }));
+        }
+        catch (e) {
+            log("warn", "keyman_search failed", { name: c.name, error: String(e) });
+        }
+        const sc = heuristicScore(c);
+        c.priority_score = sc.score;
+        c.priority_reason = sc.reason;
+    }
+    return calls;
+}
+function csvEscape(v) {
+    if (/[",\r\n]/.test(v))
+        return `"${v.replace(/"/g, '""')}"`;
+    return v;
+}
+export function rowsToCsv(items) {
+    const headers = [
+        "priority_score",
+        "name",
+        "corporate_number",
+        "ceo",
+        "industry",
+        "revenue",
+        "capital",
+        "emp",
+        "address",
+        "phone",
+        "hp_url",
+        "email",
+        "departments",
+        "keymen",
+        "priority_reason",
+    ];
+    const lines = [headers.join(",")];
+    for (const c of items) {
+        const row = headers.map((h) => {
+            switch (h) {
+                case "departments":
+                    return csvEscape(c.departments?.map((d) => `${d.name}${d.phone ? `(${d.phone})` : ""}`).join("; ") ?? "");
+                case "keymen":
+                    return csvEscape(c.keymen?.map((k) => k.role).join("; ") ?? "");
+                default: {
+                    const v = c[h];
+                    return csvEscape(v != null ? String(v) : "");
+                }
+            }
+        });
+        lines.push(row.join(","));
+    }
+    return lines.join("\n");
+}

package/dist/{papatto → enrich}/tools.d.ts

@@ -1,2 +1,2 @@
 import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-export declare function registerPapattoTools(server: McpServer): void;
+export declare function registerEnrichTools(server: McpServer): void;

package/dist/enrich/tools.js

@@ -0,0 +1,206 @@
+import { z } from "zod";
+import { existsSync, mkdirSync, writeFileSync } from "node:fs";
+import { homedir } from "node:os";
+import { join } from "node:path";
+import { enrichWithCompanySearch, enrichDepartmentsAndKeymen, rowsToCsv, } from "./core.js";
+import * as r2 from "../cloud/r2-client.js";
+import * as neon from "../cloud/neon-client.js";
+import { log } from "../shared/logger.js";
+function ok(text) {
+    return { content: [{ type: "text", text }] };
+}
+function fail(text) {
+    return { content: [{ type: "text", text: `エラー: ${text}` }] };
+}
+function genCampaignId() {
+    const d = new Date();
+    const pad = (n) => String(n).padStart(2, "0");
+    const dt = `${d.getFullYear()}${pad(d.getMonth() + 1)}${pad(d.getDate())}_${pad(d.getHours())}${pad(d.getMinutes())}`;
+    const rand = Math.random().toString(36).slice(2, 6);
+    return `enrich_${dt}_${rand}`;
+}
+export function registerEnrichTools(server) {
+    server.tool("bizgate__enrich_companies", "会社名 / 法人番号 のリストを受け取り BizGate API で詳細情報を補強する。" +
+        "depth=basic は会社基本情報のみ、smart (デフォルト) は優先度上位 N 社のみ部署 + キーマンまで深掘り、" +
+        "full は全社の部署 + キーマンを取得。" +
+        "結果は priority_score (0–100) 降順で返し、CSV をローカル + R2 に保存し、Neon campaigns に記録する。", {
+        companies: z
+            .array(z.object({
+            name: z.string().describe("会社名"),
+            corporate_number: z.string().optional().describe("法人番号 (13桁。マッチング精度向上)"),
+        }))
+            .min(1)
+            .describe("補強対象の会社リスト (最低1社)"),
+        depth: z
+            .enum(["basic", "smart", "full"])
+            .optional()
+            .describe("補強深度: basic=会社基本情報のみ / smart=上位 N 社のみ部署+キーマン (デフォルト) / full=全社の部署+キーマン"),
+        smart_top_n: z.number().optional().describe("smart 時の深堀対象数 (デフォルト 20)"),
+        campaign_id: z.string().optional().describe("キャンペーン ID (未指定時は enrich_YYYYMMDD_HHMM_xxxx 自動生成)"),
+        owner_slack_user_id: z.string().optional().describe("オーナー Slack ユーザー ID (Neon 記録用、任意)"),
+        natural_language: z.string().optional().describe("元の自然言語クエリ (Neon 記録用、任意)"),
+    }, async ({ companies, depth, smart_top_n, campaign_id, owner_slack_user_id, natural_language, }) => {
+        const t0 = Date.now();
+        const cid = campaign_id ?? genCampaignId();
+        const d = depth ?? "smart";
+        const topN = Math.max(1, Math.min(smart_top_n ?? 20, companies.length));
+        const dbActive = neon.isConfigured();
+        if (dbActive) {
+            try {
+                await neon.insertCampaign({
+                    campaign_id: cid,
+                    owner_slack_user_id: owner_slack_user_id ?? "(anonymous)",
+                    natural_language_query: natural_language,
+                    enrich_input: { companies },
+                });
+            }
+            catch (e) {
+                log("warn", "neon insertCampaign (enrich) failed", { campaign_id: cid, error: String(e) });
+            }
+        }
+        try {
+            const { items: enriched, api_calls: searchCalls } = await enrichWithCompanySearch(companies);
+            let deepCalls = 0;
+            let deeplyEnriched = 0;
+            if (d === "smart") {
+                const sorted = [...enriched].sort((a, b) => b.priority_score - a.priority_score);
+                const top = sorted.slice(0, topN).filter((c) => c.bizgate_matched);
+                deepCalls = await enrichDepartmentsAndKeymen(top);
+                deeplyEnriched = top.length;
+            }
+            else if (d === "full") {
+                const targets = enriched.filter((c) => c.bizgate_matched);
+                deepCalls = await enrichDepartmentsAndKeymen(targets);
+                deeplyEnriched = targets.length;
+            }
+            enriched.sort((a, b) => b.priority_score - a.priority_score);
+            const dir = join(homedir(), ".bizgate-mcp", "enrich", cid);
+            if (!existsSync(dir))
+                mkdirSync(dir, { recursive: true });
+            const csvPath = join(dir, `enrich__${Date.now()}.csv`);
+            writeFileSync(csvPath, rowsToCsv(enriched), "utf-8");
+            let r2Url;
+            if (r2.isConfigured()) {
+                try {
+                    const key = r2.csvKey(cid, "enrich.csv");
+                    const up = await r2.uploadFile(key, csvPath, "text/csv; charset=utf-8");
+                    r2Url = up.url;
+                }
+                catch (e) {
+                    log("warn", "r2 upload (enrich) failed", { campaign_id: cid, error: String(e) });
+                }
+            }
+            if (dbActive) {
+                try {
+                    await neon.updateCampaign({
+                        campaign_id: cid,
+                        status: "done",
+                        extracted_count: enriched.length,
+                        r2_extracted_url: r2Url,
+                    });
+                }
+                catch (e) {
+                    log("warn", "neon updateCampaign (enrich) failed", { campaign_id: cid, error: String(e) });
+                }
+            }
+            const matched = enriched.filter((c) => c.bizgate_matched).length;
+            const top5 = enriched.slice(0, 5);
+            const lines = [
+                `## エンリッチメント完了 (campaign_id=${cid})`,
+                `- 入力: ${companies.length} 社`,
+                `- BizGate マッチ: ${matched} 社`,
+                `- 部署 / キーマン深掘り: ${deeplyEnriched} 社 (depth=${d})`,
+                `- BizGate API 利用: ${searchCalls + deepCalls} 回`,
+                `- 所要時間: ${((Date.now() - t0) / 1000).toFixed(1)} 秒`,
+                `- CSV (ローカル): ${csvPath}`,
+            ];
+            if (r2Url)
+                lines.push(`- CSV (R2): ${r2Url}`);
+            if (top5.length > 0) {
+                lines.push("", `## 上位 ${top5.length} 社 (priority_score 降順)`);
+                for (let i = 0; i < top5.length; i++) {
+                    const c = top5[i];
+                    const meta = [];
+                    if (c.industry)
+                        meta.push(`業: ${c.industry}`);
+                    if (c.revenue)
+                        meta.push(`売: ${c.revenue}`);
+                    if (c.emp)
+                        meta.push(`従: ${c.emp}`);
+                    if (c.address)
+                        meta.push(`所: ${c.address}`);
+                    const deptText = c.departments && c.departments.length > 0
+                        ? ` / 部署 ${c.departments.length} 件`
+                        : "";
+                    const keymanText = c.keymen && c.keymen.length > 0
+                        ? ` / キーマン ${c.keymen.length} 件`
+                        : "";
+                    lines.push(`${i + 1}. **${c.name}** (score=${c.priority_score}) — ${c.priority_reason ?? "-"}`);
+                    if (meta.length > 0 || deptText || keymanText) {
+                        lines.push(`   ${meta.join(" / ")}${deptText}${keymanText}`);
+                    }
+                }
+            }
+            return ok(lines.join("\n"));
+        }
+        catch (e) {
+            if (dbActive) {
+                await neon.updateCampaign({ campaign_id: cid, status: "failed" }).catch(() => undefined);
+            }
+            return fail(e instanceof Error ? e.message : String(e));
+        }
+    });
+    server.tool("bizgate__campaign_status", "Neon DB に記録された enrich キャンペーンの状態を照会する。campaign_id 単体または owner_slack_user_id で最近 N 件のリスト。R2 URL も返す。NEON_DATABASE_URL 未設定時は利用不可。", {
+        campaign_id: z.string().optional().describe("特定キャンペーン照会"),
+        owner_slack_user_id: z.string().optional().describe("オーナー別最近リスト (campaign_id 未指定時)"),
+        limit: z.number().optional().describe("リスト取得件数 (デフォルト 20)"),
+    }, async ({ campaign_id, owner_slack_user_id, limit }) => {
+        if (!neon.isConfigured()) {
+            return fail("NEON_DATABASE_URL 未設定。データベース連携が必要です。");
+        }
+        try {
+            if (campaign_id) {
+                const c = await neon.getCampaign(campaign_id);
+                if (!c)
+                    return ok(`campaign_id=${campaign_id} は見つかりませんでした。`);
+                const lines = [
+                    `## キャンペーン: ${c.campaign_id}`,
+                    `- オーナー: ${c.owner_slack_user_id}`,
+                    `- ステータス: ${c.status}`,
+                    `- 抽出件数: ${c.extracted_count ?? "-"}`,
+                    `- 作成: ${new Date(Number(c.created_at)).toISOString()}`,
+                    `- 更新: ${new Date(Number(c.updated_at)).toISOString()}`,
+                    `- R2: ${c.r2_extracted_url ?? "-"}`,
+                ];
+                if (c.enrich_input) {
+                    lines.push("", "### 入力", "```json", JSON.stringify(c.enrich_input, null, 2), "```");
+                }
+                if (c.natural_language_query) {
+                    lines.push("", "### 自然言語クエリ", `> ${c.natural_language_query}`);
+                }
+                return ok(lines.join("\n"));
+            }
+            if (!owner_slack_user_id) {
+                return fail("campaign_id または owner_slack_user_id のいずれかが必要です");
+            }
+            const rows = await neon.listCampaignsByOwner(owner_slack_user_id, limit ?? 20);
+            if (rows.length === 0)
+                return ok(`${owner_slack_user_id} の campaign は見つかりませんでした。`);
+            const header = `| campaign_id | status | 抽出件数 | 作成日時 | R2 |\n|-------------|--------|----------|----------|----|`;
+            const lines = rows.map((c) => {
+                const dt = new Date(Number(c.created_at)).toISOString().replace("T", " ").slice(0, 16);
+                const r2Short = c.r2_extracted_url ? "✓" : "-";
+                return `| ${c.campaign_id} | ${c.status} | ${c.extracted_count ?? "-"} | ${dt} | ${r2Short} |`;
+            });
+            return ok([
+                `## ${owner_slack_user_id} のキャンペーン (最近 ${rows.length} 件)`,
+                "",
+                header,
+                ...lines,
+            ].join("\n"));
+        }
+        catch (e) {
+            return fail(e instanceof Error ? e.message : String(e));
+        }
+    });
+}

package/dist/http-server.d.ts

@@ -0,0 +1,9 @@
+import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+export interface HttpServerOptions {
+    /** Factory function: 한 client connection 마다 새 McpServer 인스턴스 생성. */
+    createMcpServer: () => McpServer;
+    port?: number;
+    host?: string;
+    authToken?: string;
+}
+export declare function startHttpServer(opts: HttpServerOptions): Promise<void>;