gm-mcp 1.0.10 → 1.1.1

package/src/services/lark-base.service.ts

@@ -0,0 +1,18 @@
+import axios from "axios";
+import { LarkRecord } from "./types/lark.type";
+import { getLarkTenantAccessToken } from "./lark.service";
+
+export async function fetchReportRecords(p: { view_id: string; field_names: string[] }) {
+  const access_token = await getLarkTenantAccessToken();
+  const appToken = "V5akbIXqfaBvXqs8twfl6rALgLd";
+  const tableId = "tblQGKjGd8MnK5qs";
+  const url = `https://open.larksuite.com/open-apis/bitable/v1/apps/${appToken}/tables/${tableId}/records/search`;
+  return axios
+    .post<{ data: { items: LarkRecord[] } }>(url, p, {
+      params: { page_size: 500 },
+      headers: {
+        Authorization: "Bearer " + access_token,
+      },
+    })
+    .then((res) => res.data.data.items);
+}

package/src/services/lark.service.ts

@@ -7,13 +7,15 @@ import { getEnv } from "../env";
 // const LARK_X3_GROUP_ID = 'oc_bb4e5b8dd8ada38127c790fa368c02b9';
 // Group dan choi front end
 
+const DEBT_COLLECTION_CARD_ID = "ctp_AAXr3Fr7xcjr";
+
 const larkClient = new lark.Client({
   appId: getEnv().LARK_APP_ID,
   appSecret: getEnv().LARK_APP_SECRET,
   disableTokenCache: true,
 });
 
-async function getAccessToken() {
+export async function getLarkTenantAccessToken() {
   const url = "https://open.larksuite.com/open-apis/auth/v3/tenant_access_token/internal";
   return axios
     .post<{
@@ -68,7 +70,7 @@ async function getAccessToken() {
 
 // Split later
 export async function sendStatisticX3Message(data: { date: string; content: string }) {
-  const access_token = await getAccessToken();
+  const access_token = await getLarkTenantAccessToken();
   const { date, content } = data;
   const messageContent = {
     type: "template",
@@ -96,9 +98,38 @@ export async function sendStatisticX3Message(data: { date: string; content: stri
   );
 }
 
+export async function sendReportOpsGroup(data: { title: string; message_content: string }) {
+  const access_token = await getLarkTenantAccessToken();
+  const { title, message_content } = data;
+  const messageContent = {
+    type: "template",
+    data: {
+      template_id: getEnv().LARK_X3_TEMPLATE_ID,
+      // template_id: 'ctp_AAdI5Sp48YUT',
+      template_variable: {
+        title,
+        message_content,
+      },
+    },
+  };
+  return larkClient.im.message.create(
+    {
+      params: {
+        receive_id_type: "chat_id",
+      },
+      data: {
+        receive_id: getEnv().LARK_X3_GROUP_ID,
+        msg_type: "interactive",
+        content: JSON.stringify(messageContent),
+      },
+    },
+    lark.withTenantToken(access_token)
+  );
+}
+
 export async function sendLarkTextMessage(p: { message_content: { text: string }; receive_id: string }) {
   const { message_content, receive_id } = p;
-  const access_token = await getAccessToken();
+  const access_token = await getLarkTenantAccessToken();
   return larkClient.im.message.create(
     {
       params: {

package/src/services/types/lark.type.ts

@@ -0,0 +1,4 @@
+export interface LarkRecord {
+  fields: Record<string, any>;
+  record_id: string;
+}

package/src/test.ts

@@ -1,4 +1,7 @@
 import { ensureEnvVariables, getEnv } from "./env";
+import { searchLike } from "./helpers/search-like";
+import { REPORT_COLUMS } from "./services/constants/report-columns";
+import { fetchReportRecords } from "./services/lark-base.service";
 import { sendLarkTextMessage, sendMessageToX3Group } from "./services/lark.service";
 import {
   statisticX3RangeTool,
@@ -7,10 +10,45 @@ import {
   statisticX3WeekTool,
   statisticX3YesterdayTool,
 } from "./tools";
+import { debtCollectionRateTool } from "./tools/lark-tool";
 
 async function setupTest() {
   ensureEnvVariables();
-  const sendMessageToLark = getEnv().SEND_MESSAGE_TO_LARK;
+  const compnay = "BGG";
+  const month = "12/2025";
+  const res = await debtCollectionRateTool({ sendMessageToLark: false, companyName: compnay, monthDate: month });
+  console.log(res);
+
+  // const fieldNames = REPORT_COLUMS.map((item) => item.field_name);
+  // try {
+  //   const records = await fetchReportRecords({ view_id: "vewI23hNnx", field_names: fieldNames });
+  //   // console.log(records);
+
+  //   // Find by name first
+  //   const companyNames = records.map((item) => (item.fields["CÔNG TY"] || "") as string);
+  //   // console.log(companyNames);
+  //   const searchTems = "DENKO";
+  //   const monthTerm = "12";
+  //   const foundName = searchLike(searchTems, companyNames)[0];
+  //   const companiesMatchName = records.filter((item) => item.fields["CÔNG TY"] === foundName);
+  //   // console.log(companiesMatchName);
+
+  //   // Find by pay period
+  //   const foundMonth = searchLike(
+  //     monthTerm,
+  //     companiesMatchName.map((item) => item.fields["KỲ LƯƠNG"])
+  //   )[0];
+  //   console.log(foundMonth);
+
+  //   const finalCompnay = companiesMatchName.find((item) => item.fields["KỲ LƯƠNG"] === foundMonth);
+  //   console.log(finalCompnay);
+
+  //   // console.log(companyCandidates);
+  // } catch (e) {
+  //   console.log(e);
+  // }
+
+  // const sendMessageToLark = getEnv().SEND_MESSAGE_TO_LARK;
 
   // const reuslt = await statisticX3TodayTool({ sendMessageToLark: false });
   // console.log(reuslt);

package/src/tools/lark-tool.ts

@@ -0,0 +1,60 @@
+import { CallToolResult } from "@modelcontextprotocol/sdk/types";
+import { extractMonth } from "../helpers";
+import { searchLike } from "../helpers/search-like";
+import { REPORT_COLUMS } from "../services/constants/report-columns";
+import { fetchReportRecords } from "../services/lark-base.service";
+import { sendReportOpsGroup } from "../services/lark.service";
+
+const companyKey = "CÔNG TY";
+const preiodKey = "KỲ LƯƠNG";
+const debtCollectionRateKey = "TỈ LỆ THU NỢ";
+
+export async function debtCollectionRateTool(options: {
+  sendMessageToLark: boolean;
+  companyName: string;
+  monthDate: string;
+}): Promise<CallToolResult> {
+  const { sendMessageToLark, companyName, monthDate } = options;
+  const monthParts = extractMonth(monthDate);
+  if (monthParts === null) {
+    return { content: [{ type: "text", text: `${monthDate} is invalid` }] };
+  }
+  const { month, year } = monthParts;
+  const fieldNames = REPORT_COLUMS.map((item) => item.field_name);
+  try {
+    const records = await fetchReportRecords({ view_id: "vewI23hNnx", field_names: fieldNames });
+    // console.log(records);
+
+    // Find by name first
+    const companyNames = records.map((item) => (item.fields["CÔNG TY"] || "") as string);
+    // console.log(companyNames);
+    // const searchTems = "DENKO";
+    // const monthTerm = "12";
+    const foundName = searchLike(companyName, companyNames)[0];
+    const companiesMatchName = records.filter((item) => item.fields["CÔNG TY"] === foundName);
+    // console.log(companiesMatchName);
+
+    // Find by pay period
+    const foundMonth = searchLike(
+      month.toString(),
+      companiesMatchName.map((item) => item.fields["KỲ LƯƠNG"])
+    )[0];
+    console.log(foundMonth);
+    const finalCompnay = companiesMatchName.find((item) => item.fields["KỲ LƯƠNG"] === foundMonth);
+
+    if (sendMessageToLark) {
+      await sendReportOpsGroup({
+        title: "Tỉ lệ thu nợ",
+        message_content: [
+          `Công ty: ${finalCompnay?.fields[companyKey] || "N/A"}`,
+          `Kỳ lương: ${finalCompnay?.fields[preiodKey] || "N/A"}`,
+          `Tỷ lệ thu nợ: ${Number(finalCompnay?.fields[debtCollectionRateKey]) * 100}%`,
+        ].join("<hr>"),
+      });
+    }
+    return { content: [{ type: "text", text: JSON.stringify(finalCompnay) }] };
+    // console.log(finalCompnay);
+  } catch (e) {
+    return { content: [{ type: "text", text: JSON.stringify(e) }] };
+  }
+}