lark-docx2md 0.2.0 → 0.2.1-beta.2

package/dist/{converter-ByfbJV0N.js → converter-FwY1m1jm.js}

@@ -90,6 +90,37 @@ function createClient(appId, appSecret, loggerLevel = LoggerLevel.warn) {
 		await resp.writeFile(filename);
 		return filename;
 	}
+	async function getSpreadsheetInfo(token) {
+		const data = await call("getSpreadsheetInfo", () => client.sheets.v3.spreadsheet.get({ path: { spreadsheet_token: token } }));
+		const spreadsheet = data.spreadsheet ?? data;
+		return {
+			title: spreadsheet.title ?? "",
+			url: spreadsheet.url
+		};
+	}
+	async function listSheets(token) {
+		return (await call("listSheets", () => client.sheets.v3.spreadsheetSheet.query({ path: { spreadsheet_token: token } }))).sheets ?? [];
+	}
+	async function getSheetMeta(token, sheetId) {
+		const data = await call("getSheetMeta", () => client.sheets.v3.spreadsheetSheet.get({ path: {
+			spreadsheet_token: token,
+			sheet_id: sheetId
+		} }));
+		return data.sheet ?? data;
+	}
+	async function readSheetValues(token, range) {
+		const resp = await client.request({
+			url: `https://open.feishu.cn/open-apis/sheets/v2/spreadsheets/${token}/values/${encodeURIComponent(range)}`,
+			method: "GET",
+			params: {
+				valueRenderOption: "UnformattedValue",
+				dateTimeRenderOption: "FormattedString"
+			},
+			headers: { "Content-Type": "application/json; charset=utf-8" }
+		});
+		if (resp?.code !== 0) throw new Error(`readSheetValues failed: [${resp?.code}] ${resp?.msg}`);
+		return resp.data?.valueRange?.values ?? [];
+	}
 	return {
 		getWikiNodeInfo,
 		getDocxDocument,
@@ -97,7 +128,11 @@ function createClient(appId, appSecret, loggerLevel = LoggerLevel.warn) {
 		downloadImage,
 		batchGetTmpDownloadUrl,
 		getWhiteboardNodes,
-		downloadWhiteboardAsImage
+		downloadWhiteboardAsImage,
+		getSpreadsheetInfo,
+		listSheets,
+		getSheetMeta,
+		readSheetValues
 	};
 }
 //#endregion
@@ -194,37 +229,36 @@ function parseElement(e, inline) {
 }
 function parseTextRun(tr) {
 	const s = tr.text_element_style;
-	const textNode = {
+	let node = s?.inline_code ? {
+		type: "inlineCode",
+		content: tr.content
+	} : {
 		type: "text",
 		content: tr.content
 	};
-	if (!s) return textNode;
-	if (s.bold) return {
-		type: "bold",
-		children: [textNode]
+	if (!s) return node;
+	if (s.link) node = {
+		type: "link",
+		url: decodeURIComponent(s.link.url),
+		children: [node]
 	};
-	if (s.italic) return {
-		type: "italic",
-		children: [textNode]
+	if (s.underline) node = {
+		type: "underline",
+		children: [node]
 	};
-	if (s.strikethrough) return {
+	if (s.strikethrough) node = {
 		type: "strikethrough",
-		children: [textNode]
+		children: [node]
 	};
-	if (s.underline) return {
-		type: "underline",
-		children: [textNode]
-	};
-	if (s.inline_code) return {
-		type: "inlineCode",
-		content: tr.content
+	if (s.italic) node = {
+		type: "italic",
+		children: [node]
 	};
-	if (s.link) return {
-		type: "link",
-		url: decodeURIComponent(s.link.url),
-		children: [textNode]
+	if (s.bold) node = {
+		type: "bold",
+		children: [node]
 	};
-	return textNode;
+	return node;
 }
 //#endregion
 //#region src/md-ast/parsers/page.ts
@@ -562,6 +596,17 @@ const whiteboardParser = {
 	}
 };
 //#endregion
+//#region src/md-ast/parsers/sheet.ts
+const sheetBlockParser = {
+	blockType: 30,
+	parse(block, _ctx) {
+		return {
+			type: "sheet",
+			token: block.sheet?.token ?? ""
+		};
+	}
+};
+//#endregion
 //#region src/md-ast/parsers/index.ts
 function registerBuiltinParsers(parser) {
 	parser.register(pageParser);
@@ -588,6 +633,7 @@ function registerBuiltinParsers(parser) {
 	parser.register(tableParser);
 	parser.register(quoteContainerParser);
 	parser.register(whiteboardParser);
+	parser.register(sheetBlockParser);
 }
 //#endregion
 //#region src/md-ast/serializer.ts
@@ -783,6 +829,34 @@ const htmlSerializer = {
 		return node.content;
 	}
 };
+const sheetResolvedSerializer = {
+	type: "sheetResolved",
+	serialize(node) {
+		if (node.type !== "sheetResolved") return "";
+		let out = "";
+		for (const s of node.sheets) {
+			out += `## 工作表：${s.title}\n\n`;
+			if (s.error) {
+				out += `> ${s.error}\n\n`;
+				continue;
+			}
+			if (!s.rows.length) {
+				out += "_（空表）_\n\n";
+				continue;
+			}
+			const [head, ...body] = s.rows;
+			if (!head) {
+				out += "_（空表）_\n\n";
+				continue;
+			}
+			out += `| ${head.join(" | ")} |\n`;
+			out += `| ${head.map(() => "---").join(" | ")} |\n`;
+			for (const r of body) out += `| ${r.join(" | ")} |\n`;
+			out += "\n";
+		}
+		return out;
+	}
+};
 function registerBuiltinSerializers(serializer) {
 	serializer.register(pageSerializer);
 	serializer.register(headingSerializer);
@@ -799,6 +873,7 @@ function registerBuiltinSerializers(serializer) {
 	serializer.register(tableSerializer);
 	serializer.register(gridSerializer);
 	serializer.register(htmlSerializer);
+	serializer.register(sheetResolvedSerializer);
 }
 //#endregion
 //#region src/whiteboard/utils.ts
@@ -2490,6 +2565,65 @@ function collectImageTokens$1(nodes, out) {
 	}
 }
 //#endregion
+//#region src/sheet/index.ts
+const escapeCell = (s) => s.replace(/\|/g, "\\|").replace(/\n/g, "<br>");
+function cellToMd(cell) {
+	if (cell == null) return "";
+	if (typeof cell !== "object") return escapeCell(String(cell));
+	if (Array.isArray(cell)) return cell.map(cellToMd).join("");
+	const o = cell;
+	switch (o.type) {
+		case "text": return escapeCell(String(o.text ?? ""));
+		case "url": return `[${escapeCell(o.text ?? o.link ?? "")}](${o.link ?? ""})`;
+		case "mentionUser": return `@${escapeCell(o.name ?? o.textArr?.join("") ?? "")}`;
+		case "formula": return `\`${escapeCell(o.text ?? "")}\``;
+		default:
+			if (o.text != null) return escapeCell(String(o.text));
+			return "";
+	}
+}
+function expandMerges(rows, merges) {
+	const grid = rows.map((r) => r.slice());
+	for (const m of merges) {
+		const r0 = m.start_row_index ?? 0;
+		const r1 = m.end_row_index ?? r0;
+		const c0 = m.start_column_index ?? 0;
+		const c1 = m.end_column_index ?? c0;
+		const v = grid[r0]?.[c0] ?? "";
+		for (let r = r0; r <= r1; r++) {
+			if (!grid[r]) grid[r] = [];
+			for (let c = c0; c <= c1; c++) {
+				if (r === r0 && c === c0) continue;
+				grid[r][c] = v;
+			}
+		}
+	}
+	return grid;
+}
+const isEmptyCell = (v) => v == null || v === "";
+function trimTrailingEmpty(rows) {
+	let lastRow = -1;
+	for (let r = 0; r < rows.length; r++) if ((rows[r] ?? []).some((c) => !isEmptyCell(c))) lastRow = r;
+	if (lastRow < 0) return [];
+	const trimmedRows = rows.slice(0, lastRow + 1);
+	let lastCol = -1;
+	for (const row of trimmedRows) for (let c = (row?.length ?? 0) - 1; c >= 0; c--) if (!isEmptyCell(row[c])) {
+		if (c > lastCol) lastCol = c;
+		break;
+	}
+	if (lastCol < 0) return trimmedRows.map(() => []);
+	return trimmedRows.map((r) => (r ?? []).slice(0, lastCol + 1));
+}
+function columnIndexToLetter(n) {
+	let result = "";
+	while (n > 0) {
+		n--;
+		result = String.fromCharCode(65 + n % 26) + result;
+		n = Math.floor(n / 26);
+	}
+	return result;
+}
+//#endregion
 //#region src/logger.ts
 const COLORS = {
 	[LoggerLevel.fatal]: "\x1B[35m",
@@ -2540,13 +2674,14 @@ var MdTransformer = class {
 	async transform(ast) {
 		const imageTokens = collectImageTokens(ast);
 		const whiteboardTokens = collectWhiteboardTokens(ast);
-		replaceInAst(ast, await this.resolveImages(imageTokens), await this.resolveWhiteboards(whiteboardTokens));
+		const sheetTokens = collectSheetTokens(ast);
+		replaceInAst(ast, await this.resolveImages(imageTokens), await this.resolveWhiteboards(whiteboardTokens), await this.resolveSheets(sheetTokens));
 	}
 	async resolveImages(tokens) {
 		const map = /* @__PURE__ */ new Map();
 		const uniqueTokens = [...new Set(tokens)];
 		if (uniqueTokens.length === 0) return map;
-		if (this.opts.imageMode === "online" || this.opts.agent) for (let i = 0; i < uniqueTokens.length; i += 5) {
+		if (this.opts.imageMode === "online" || this.opts.agent === true) for (let i = 0; i < uniqueTokens.length; i += 5) {
 			const batch = uniqueTokens.slice(i, i + 5);
 			const urlMap = await this.client.batchGetTmpDownloadUrl(batch);
 			for (const token of batch) {
@@ -2679,6 +2814,70 @@ var MdTransformer = class {
 		}
 		return yamlContent;
 	}
+	async resolveSheets(tokens) {
+		const map = /* @__PURE__ */ new Map();
+		const uniqueTokens = [...new Set(tokens)];
+		if (uniqueTokens.length === 0) return map;
+		for (const token of uniqueTokens) {
+			if (!token) continue;
+			try {
+				const info = await this.client.getSpreadsheetInfo(token);
+				const list = await this.client.listSheets(token);
+				const resolved = [];
+				for (const s of list) {
+					if (s.hidden) continue;
+					if (s.resource_type && s.resource_type !== "sheet") {
+						resolved.push({
+							title: s.title ?? "",
+							kind: "bitable",
+							rows: [],
+							error: `非网格表(${s.resource_type})，已跳过`
+						});
+						continue;
+					}
+					try {
+						const meta = await this.client.getSheetMeta(token, s.sheet_id);
+						const { row_count = 0, column_count = 0 } = meta.grid_properties ?? {};
+						if (!row_count || !column_count) {
+							resolved.push({
+								title: s.title ?? "",
+								kind: "grid",
+								rows: []
+							});
+							continue;
+						}
+						const endCol = columnIndexToLetter(column_count);
+						const trimmed = trimTrailingEmpty(expandMerges((await this.client.readSheetValues(token, `${s.sheet_id}!A1:${endCol}${row_count}`) ?? []).map((row) => row.map(cellToMd)), meta.merges ?? []));
+						resolved.push({
+							title: s.title ?? "",
+							kind: "grid",
+							rows: trimmed
+						});
+					} catch (e) {
+						resolved.push({
+							title: s.title ?? "",
+							kind: "grid",
+							rows: [],
+							error: `读取失败：${e.message}`
+						});
+					}
+				}
+				map.set(token, {
+					type: "sheetResolved",
+					title: info.title ?? "",
+					sheets: resolved
+				});
+			} catch (e) {
+				logger$1.warn(`Failed to render sheet ${token}:`, e.message);
+				map.set(token, {
+					type: "sheetResolved",
+					title: "",
+					sheets: []
+				});
+			}
+		}
+		return map;
+	}
 };
 function collectImageTokens(node) {
 	const tokens = [];
@@ -2694,7 +2893,14 @@ function collectWhiteboardTokens(node) {
 	});
 	return tokens;
 }
-function replaceInAst(node, imageMap, whiteboardMap) {
+function collectSheetTokens(node) {
+	const tokens = [];
+	traverseBlockAst(node, (n) => {
+		if (n.type === "sheet") tokens.push(n.token);
+	});
+	return tokens;
+}
+function replaceInAst(node, imageMap, whiteboardMap, sheetMap) {
 	if (node.type === "image") {
 		const newSrc = imageMap.get(node.src);
 		if (newSrc) node.src = newSrc;
@@ -2711,11 +2917,18 @@ function replaceInAst(node, imageMap, whiteboardMap) {
 				continue;
 			}
 		}
+		if (child.type === "sheet") {
+			const replacement = sheetMap.get(child.token);
+			if (replacement) {
+				children[i] = replacement;
+				continue;
+			}
+		}
 		if (child.type === "image") {
 			const newSrc = imageMap.get(child.src);
 			if (newSrc) child.src = newSrc;
 		}
-		replaceInAst(child, imageMap, whiteboardMap);
+		replaceInAst(child, imageMap, whiteboardMap, sheetMap);
 	}
 }
 function traverseBlockAst(node, visitor) {
@@ -2730,7 +2943,7 @@ function hasBlockChildren(node) {
 //#region src/converter.ts
 const logger = createLogger("converter");
 function parseWikiUrl(url) {
-	const m = url.match(/^https:\/\/[\w.-]+\/(docs|docx|wiki)\/([a-zA-Z0-9]+)/);
+	const m = url.match(/^https:\/\/[\w.-]+\/(docs|docx|wiki|sheets)\/([a-zA-Z0-9]+)/);
 	if (!m) throw new Error("Invalid feishu document URL");
 	return {
 		docType: m[1],
@@ -2743,24 +2956,41 @@ async function convert(opts) {
 	const sdkLoggerLevel = opts.agent ? LoggerLevel.error : LoggerLevel.warn;
 	const client = createClient(opts.appId, opts.appSecret, sdkLoggerLevel);
 	let docToken = rawToken;
+	let objType = docType;
 	if (docType === "wiki") {
-		docToken = (await client.getWikiNodeInfo(docToken)).obj_token;
-		logger.info("Resolved docx token:", docToken);
-	}
-	const doc = await client.getDocxDocument(docToken);
-	const blocks = await client.getDocxBlocks(docToken);
-	logger.info(`Fetched ${blocks.length} blocks`);
-	const parser = new Parser();
-	registerBuiltinParsers(parser);
-	const ast = parser.parse(doc, blocks);
+		const node = await client.getWikiNodeInfo(docToken);
+		docToken = node.obj_token;
+		objType = node.obj_type ?? "docx";
+		logger.info("Resolved wiki node:", objType, docToken);
+	} else if (docType === "sheets") objType = "sheet";
+	let ast;
+	if (objType === "sheet") ast = {
+		type: "page",
+		title: [{
+			type: "text",
+			content: (await client.getSpreadsheetInfo(docToken)).title ?? ""
+		}],
+		children: [{
+			type: "sheet",
+			token: docToken
+		}]
+	};
+	else {
+		const doc = await client.getDocxDocument(docToken);
+		const blocks = await client.getDocxBlocks(docToken);
+		logger.info(`Fetched ${blocks.length} blocks`);
+		const parser = new Parser();
+		registerBuiltinParsers(parser);
+		ast = parser.parse(doc, blocks);
+	}
 	await new MdTransformer(client, opts).transform(ast);
 	const serializer = new MdSerializer();
 	registerBuiltinSerializers(serializer);
 	const markdown = serializer.serialize(ast);
 	let filePath;
-	if (!opts.agent) {
+	if (!opts.agent || opts.agent === "local") {
 		fs.mkdirSync(opts.output, { recursive: true });
-		filePath = path.join(opts.output, `${docToken}.md`);
+		filePath = path.resolve(opts.output, `${docToken}.md`);
 		fs.writeFileSync(filePath, markdown);
 		logger.info("Downloaded markdown file to", filePath);
 	}
@@ -2773,4 +3003,4 @@ async function convert(opts) {
 //#endregion
 export { parseWikiUrl as n, setLogLevel as r, convert as t };
 
-//# sourceMappingURL=converter-ByfbJV0N.js.map
+//# sourceMappingURL=converter-FwY1m1jm.js.map