gm-mcp 1.0.10 → 1.1.0

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

@@ -13,7 +13,7 @@ const larkClient = new lark.Client({
   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 +68,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",
@@ -98,7 +98,7 @@ export async function sendStatisticX3Message(data: { date: string; content: stri
 
 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,46 @@
+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";
+
+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);
+    return { content: [{ type: "text", text: JSON.stringify(finalCompnay) }] };
+    console.log(finalCompnay);
+    if (sendMessageToLark) {
+    }
+  } catch (e) {
+    return { content: [{ type: "text", text: JSON.stringify(e) }] };
+  }
+}