pi-yank 0.1.0 → 0.1.1

package/extensions/yank-core.js

@@ -14,6 +14,23 @@
  * @returns {CodeBlock[]}
  */
 export function extractCodeBlocks(markdown) {
+	const topLevelBlocks = extractTopLevelCodeBlocks(markdown);
+	if (topLevelBlocks.length === 1) {
+		const [wrapper] = topLevelBlocks;
+		const innerBlocks = extractTopLevelCodeBlocks(wrapper.content);
+		if (innerBlocks.length > 1) {
+			return withPreviews(innerBlocks);
+		}
+	}
+
+	return withPreviews(topLevelBlocks);
+}
+
+/**
+ * @param {string} markdown
+ * @returns {CodeBlock[]}
+ */
+function extractTopLevelCodeBlocks(markdown) {
 	/** @type {CodeBlock[]} */
 	const blocks = [];
 	/** @type {OpenCodeBlock[]} */
@@ -59,9 +76,8 @@ export function extractCodeBlocks(markdown) {
 
 			const finishedBlock = openBlocks.pop();
 			if (finishedBlock && openBlocks.length === 0) {
-				const content = finishedBlock.lines.join("\n");
 				blocks.push({
-					content,
+					content: finishedBlock.lines.join("\n"),
 					preview: "",
 					startLine: finishedBlock.startLine,
 					infoString: finishedBlock.infoString,
@@ -82,16 +98,18 @@ export function extractCodeBlocks(markdown) {
 		});
 	}
 
-	return blocks
-		.sort((left, right) => left.startLine - right.startLine)
-		.map((block, index) => ({
-			...block,
-			preview: buildCodeBlockPreview(
-				block.content,
-				index + 1,
-				block.infoString,
-			),
-		}));
+	return blocks.sort((left, right) => left.startLine - right.startLine);
+}
+
+/**
+ * @param {CodeBlock[]} blocks
+ * @returns {CodeBlock[]}
+ */
+function withPreviews(blocks) {
+	return blocks.map((block, index) => ({
+		...block,
+		preview: buildCodeBlockPreview(block.content, index + 1, block.infoString),
+	}));
 }
 
 /**

package/package.json

@@ -1,7 +1,7 @@
 {
 	"name": "pi-yank",
 	"private": false,
-	"version": "0.1.0",
+	"version": "0.1.1",
 	"type": "module",
 	"description": "Lightweight /yank extension for pi that copies the last assistant message or a selected code block.",
 	"keywords": [

package/tests/yank-core.test.mjs

@@ -5,7 +5,7 @@ import {
 	extractCodeBlocks,
 } from "../extensions/yank-core.js";
 
-test("extracts a single outer markdown block when nested fences appear as literal examples", () => {
+test("prefers inner sibling code blocks when a single outer markdown wrapper encloses them", () => {
 	const text = `我来为你创建一个 Markdown 格式的测试文本：
 
 \`\`\`markdown
@@ -32,10 +32,11 @@ def hello():
 说明文字`;
 
 	const blocks = extractCodeBlocks(text);
-	assert.equal(blocks.length, 1);
+	assert.equal(blocks.length, 2);
 	assert.match(blocks[0].content, /function hello\(\)/);
-	assert.match(blocks[0].content, /def hello\(\):/);
-	assert.equal(blocks[0].preview, "markdown: # Markdown 格式测试文档");
+	assert.match(blocks[1].content, /def hello\(\):/);
+	assert.equal(blocks[0].preview, "javascript: function hello() {");
+	assert.equal(blocks[1].preview, "python: def hello():");
 });
 
 test("extracts sibling top-level code blocks independently", () => {
@@ -59,6 +60,35 @@ after`;
 	assert.equal(blocks[1].preview, 'python: print("b")');
 });
 
+test("unwraps a single outer markdown wrapper around filename and script blocks", () => {
+	const text = `下面给你一套 pi / pi-mono 迁移的一键备份 +恢复脚本（macOS/Linux）。
+
+保存为：
+
+\`\`\`bash
+backup-pi.sh
+\`\`\`
+
+内容如下：
+
+\`\`\`bash
+#!/usr/bin/env bash
+set -euo pipefail
+TIMESTAMP="$(date +%Y%m%d-%H%M%S)"
+\`\`\``;
+
+	const wrapped = `\`\`\`markdown
+${text}
+\`\`\``;
+	const blocks = extractCodeBlocks(wrapped);
+
+	assert.equal(blocks.length, 2);
+	assert.equal(blocks[0].content, "backup-pi.sh");
+	assert.match(blocks[1].content, /set -euo pipefail/);
+	assert.equal(blocks[0].preview, "bash: backup-pi.sh");
+	assert.equal(blocks[1].preview, "bash: #!/usr/bin/env bash");
+});
+
 test("supports tilde fences", () => {
 	const text = `~~~sql
 select 1;