All files / dingtalk-openclaw-connector/src/services/media file.ts

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
/**
 * 文件处理模块
 * 支持文档解析（docx, pdf, txt 等）和文件上传
 */
 
import * as fs from 'fs';
import type { Logger } from 'openclaw/plugin-sdk';
import type { DingtalkConfig } from '../../types/index.ts';
import { FILE_MARKER_PATTERN, toLocalPath, uploadMediaToDingTalk, TEXT_FILE_EXTENSIONS } from './common.ts';
 
/**
 * 解析文档文件，提取文本内容
 */
export async function parseDocumentFile(filePath: string, log?: Logger): Promise<string | null> {
  try {
    const ext = '.' + filePath.split('.').pop()?.toLowerCase();
    
    if (!TEXT_FILE_EXTENSIONS.has(ext)) {
      log?.warn?.(`[DingTalk][File] 不支持的文件类型：${ext}`);
      return null;
    }
 
    const content = fs.readFileSync(filePath, 'utf-8');
    log?.info?.(`[DingTalk][File] 解析文件成功：${filePath}, 长度=${content.length}`);
    return content;
  } catch (err: any) {
    log?.error?.(`[DingTalk][File] 解析文件失败：${err.message}`);
    return null;
  }
}
 
/**
 * 提取文件标记并发送文件消息
 */
export async function processFileMarkers(
  content: string,
  sessionWebhook: string,
  config: DingtalkConfig,
  oapiToken: string | null,
  log?: Logger,
  useProactiveApi: boolean = false,
  target?: any,
): Promise<string> {
  const logPrefix = useProactiveApi ? '[DingTalk][File][Proactive]' : '[DingTalk][File]';
 
  if (!oapiToken) {
    log?.warn?.(`${logPrefix} 无 oapiToken，跳过文件处理`);
    return content;
  }
 
  const matches = [...content.matchAll(FILE_MARKER_PATTERN)];
  const filePaths: string[] = [];
 
  for (const match of matches) {
    try {
      const fileData = JSON.parse(match[1]);
      const rawPath = fileData.path;
      const absPath = toLocalPath(rawPath);
      filePaths.push(absPath);
    } catch (err) {
      log?.warn?.(`${logPrefix} 解析文件标记失败：${match[1]}`);
    }
  }
 
  if (filePaths.length === 0) {
    return content;
  }
 
  log?.info?.(`${logPrefix} 检测到 ${filePaths.length} 个文件，开始上传...`);
 
  let result = content;
  for (const filePath of filePaths) {
    const mediaId = await uploadMediaToDingTalk(filePath, 'file', oapiToken, 20 * 1024 * 1024, log);
    if (mediaId) {
      result = result.replace(
        `[DINGTALK_FILE]${JSON.stringify({ path: filePath })}[/DINGTALK_FILE]`,
        `[文件已上传：${mediaId}]`,
      );
    }
  }
 
  return result;
}