xiaozuoassistant 0.2.48 → 0.2.49

package/dist/server/plugins/lark-skills/src/lark-doc.js

@@ -1,5 +1,39 @@
 import { BaseSkill } from '../../../skills/base-skill.js';
 import * as lark from '@larksuiteoapi/node-sdk';
+// Helper function to resolve document token, especially handling /wiki/ URLs
+async function resolveDocumentToken(client, urlOrToken) {
+    let token = urlOrToken;
+    let isWiki = false;
+    if (urlOrToken.includes('/docx/')) {
+        token = urlOrToken.split('/docx/')[1].split(/[?#]/)[0];
+        return { token, type: 'docx' };
+    }
+    else if (urlOrToken.includes('/doc/')) {
+        token = urlOrToken.split('/doc/')[1].split(/[?#]/)[0];
+        return { token, type: 'doc' };
+    }
+    else if (urlOrToken.includes('/wiki/')) {
+        token = urlOrToken.split('/wiki/')[1].split(/[?#]/)[0];
+        isWiki = true;
+    }
+    else if (urlOrToken.startsWith('wikcn')) {
+        isWiki = true;
+    }
+    if (isWiki) {
+        const res = await client.wiki.space.getNode({
+            params: { token }
+        });
+        if (res.code !== 0) {
+            throw new Error(`Failed to resolve wiki token. Code: ${res.code}, Msg: ${res.msg}`);
+        }
+        return {
+            token: res.data?.node?.obj_token || token,
+            type: res.data?.node?.obj_type || 'docx'
+        };
+    }
+    // Fallback assuming it's a docx token if not wiki
+    return { token, type: 'docx' };
+}
 export class ReadLarkDocSkill extends BaseSkill {
     constructor() {
         super(...arguments);
@@ -26,18 +60,6 @@ export class ReadLarkDocSkill extends BaseSkill {
     }
     async execute(args, ctx) {
         try {
-            // Extract document token from URL
-            // Typical Lark doc URL formats:
-            // https://domain.feishu.cn/docx/TOKEN
-            // https://domain.feishu.cn/wiki/WIKITOKEN
-            let documentToken = args.document_url;
-            // Basic extraction logic
-            if (args.document_url.includes('/docx/')) {
-                documentToken = args.document_url.split('/docx/')[1].split(/[?#]/)[0];
-            }
-            else if (args.document_url.includes('/doc/')) {
-                documentToken = args.document_url.split('/doc/')[1].split(/[?#]/)[0];
-            }
             // Fetch config to get app_id and app_secret if not provided
             const configLoader = require('../../../config/loader.js').config;
             let appId = args.app_id;
@@ -67,6 +89,11 @@ export class ReadLarkDocSkill extends BaseSkill {
                 appId: appId,
                 appSecret: appSecret,
             });
+            // Extract and resolve document token from URL
+            const { token: documentToken, type: objType } = await resolveDocumentToken(client, args.document_url);
+            if (objType !== 'docx' && objType !== 'doc') {
+                return { error: `Unsupported document type: ${objType}. Only docx and doc are supported for reading raw text content.` };
+            }
             // We use the drive/v1/export API or docx/v1/documents API to read content
             // Let's try the newer DOCX API first to get raw content
             const res = await client.docx.document.rawContent({
@@ -185,13 +212,6 @@ export class AppendLarkDocSkill extends BaseSkill {
     }
     async execute(args, ctx) {
         try {
-            let documentToken = args.document_url;
-            if (args.document_url.includes('/docx/')) {
-                documentToken = args.document_url.split('/docx/')[1].split(/[?#]/)[0];
-            }
-            else if (args.document_url.includes('/doc/')) {
-                documentToken = args.document_url.split('/doc/')[1].split(/[?#]/)[0];
-            }
             const configLoader = require('../../../config/loader.js').config;
             let appId = args.app_id;
             let appSecret = args.app_secret;
@@ -215,6 +235,10 @@ export class AppendLarkDocSkill extends BaseSkill {
                 }
             }
             const client = new lark.Client({ appId: appId, appSecret: appSecret });
+            const { token: documentToken, type: objType } = await resolveDocumentToken(client, args.document_url);
+            if (objType !== 'docx') {
+                return { error: `Unsupported document type: ${objType}. Append operation currently only supports docx documents.` };
+            }
             // Create blocks from content
             const lines = args.content.split('\n').filter(l => l.trim() !== '');
             const children = lines.map(line => ({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "xiaozuoassistant",
-  "version": "0.2.48",
+  "version": "0.2.49",
   "description": "A local-first personal AI assistant with multi-channel support and enhanced memory.",
   "author": "mantle.lau",
   "license": "MIT",