xiaozuoassistant 0.2.47 → 0.2.49

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

@@ -1,4 +1,4 @@
-import { ReadLarkDocSkill } from './lark-doc.js';
+import { ReadLarkDocSkill, CreateLarkDocSkill, AppendLarkDocSkill } from './lark-doc.js';
 const plugin = {
     metadata: {
         name: 'lark-skills',
@@ -8,6 +8,8 @@ const plugin = {
     },
     onLoad: (context) => {
         context.registerSkill(new ReadLarkDocSkill());
+        context.registerSkill(new CreateLarkDocSkill());
+        context.registerSkill(new AppendLarkDocSkill());
     }
 };
 export default plugin;

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({
@@ -87,3 +114,162 @@ export class ReadLarkDocSkill extends BaseSkill {
         }
     }
 }
+export class CreateLarkDocSkill extends BaseSkill {
+    constructor() {
+        super(...arguments);
+        this.name = 'create_lark_doc';
+        this.description = 'Create a new Lark/Feishu document (docx)';
+        this.parameters = {
+            type: 'object',
+            properties: {
+                title: {
+                    type: 'string',
+                    description: 'Title of the new document'
+                },
+                folder_token: {
+                    type: 'string',
+                    description: 'Optional folder token to create the document in'
+                },
+                app_id: {
+                    type: 'string',
+                    description: 'The Lark App ID (optional, defaults to configured bot)'
+                },
+                app_secret: {
+                    type: 'string',
+                    description: 'The Lark App Secret (optional)'
+                }
+            },
+            required: ['title']
+        };
+    }
+    async execute(args, ctx) {
+        try {
+            const configLoader = require('../../../config/loader.js').config;
+            let appId = args.app_id;
+            let appSecret = args.app_secret;
+            if (!appId || !appSecret) {
+                const feishuBots = configLoader.channels?.feishu;
+                if (feishuBots && Array.isArray(feishuBots) && feishuBots.length > 0) {
+                    if (ctx?.metadata?.botName) {
+                        const specificBot = feishuBots.find((b) => b.name === ctx.metadata.botName);
+                        if (specificBot) {
+                            appId = specificBot.appId;
+                            appSecret = specificBot.appSecret;
+                        }
+                    }
+                    if (!appId || !appSecret) {
+                        appId = feishuBots[0].appId;
+                        appSecret = feishuBots[0].appSecret;
+                    }
+                }
+                else {
+                    return { error: 'Lark App ID and Secret are required.' };
+                }
+            }
+            const client = new lark.Client({ appId: appId, appSecret: appSecret });
+            const res = await client.docx.document.create({
+                data: {
+                    title: args.title,
+                    folder_token: args.folder_token || ''
+                }
+            });
+            if (res.code !== 0) {
+                return { error: `Failed to create document. Code: ${res.code}, Msg: ${res.msg}` };
+            }
+            const docId = res.data?.document?.document_id;
+            return {
+                success: true,
+                document_id: docId,
+                url: `https://${client.domain || 'feishu.cn'}/docx/${docId}`
+            };
+        }
+        catch (error) {
+            return { error: `Failed to create document: ${error.message}` };
+        }
+    }
+}
+export class AppendLarkDocSkill extends BaseSkill {
+    constructor() {
+        super(...arguments);
+        this.name = 'append_lark_doc';
+        this.description = 'Append text content to an existing Lark/Feishu document (docx)';
+        this.parameters = {
+            type: 'object',
+            properties: {
+                document_url: {
+                    type: 'string',
+                    description: 'The URL or token of the document to append to'
+                },
+                content: {
+                    type: 'string',
+                    description: 'The text content to append'
+                },
+                app_id: { type: 'string' },
+                app_secret: { type: 'string' }
+            },
+            required: ['document_url', 'content']
+        };
+    }
+    async execute(args, ctx) {
+        try {
+            const configLoader = require('../../../config/loader.js').config;
+            let appId = args.app_id;
+            let appSecret = args.app_secret;
+            if (!appId || !appSecret) {
+                const feishuBots = configLoader.channels?.feishu;
+                if (feishuBots && Array.isArray(feishuBots) && feishuBots.length > 0) {
+                    if (ctx?.metadata?.botName) {
+                        const specificBot = feishuBots.find((b) => b.name === ctx.metadata.botName);
+                        if (specificBot) {
+                            appId = specificBot.appId;
+                            appSecret = specificBot.appSecret;
+                        }
+                    }
+                    if (!appId || !appSecret) {
+                        appId = feishuBots[0].appId;
+                        appSecret = feishuBots[0].appSecret;
+                    }
+                }
+                else {
+                    return { error: 'Lark App ID and Secret are required.' };
+                }
+            }
+            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 => ({
+                block_type: 2, // Text block
+                text: {
+                    elements: [
+                        {
+                            text_run: { content: line }
+                        }
+                    ]
+                }
+            }));
+            if (children.length === 0)
+                return { success: true, message: 'Nothing to append' };
+            const res = await client.docx.documentBlockChildren.create({
+                path: {
+                    document_id: documentToken,
+                    block_id: documentToken
+                },
+                data: {
+                    children: children,
+                    index: -1
+                }
+            });
+            if (res.code !== 0) {
+                return { error: `Failed to append to document. Code: ${res.code}, Msg: ${res.msg}` };
+            }
+            return { success: true, message: 'Content appended successfully' };
+        }
+        catch (error) {
+            return { error: `Failed to append to document: ${error.message}` };
+        }
+    }
+}

package/package.json

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