yoctomarkdown 0.0.1 → 0.0.2

package/README.md

@@ -13,6 +13,8 @@ A small, fast, Bun-native Markdown highlighter for terminal output.
 
 ## Installation
 
+[npm package](https://www.npmjs.com/package/yoctomarkdown)
+
 ```bash
 bun add yoctomarkdown
 ```

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "yoctomarkdown",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "module": "src/index.ts",
   "type": "module",
   "devDependencies": {
@@ -30,5 +30,13 @@
     "format": "prettier --write .",
     "format:check": "prettier --check .",
     "test": "bun test"
+  },
+  "types": "./src/index.ts",
+  "exports": {
+    ".": {
+      "types": "./src/index.ts",
+      "import": "./src/index.ts",
+      "default": "./src/index.ts"
+    }
   }
 }

package/src/index.ts

@@ -83,17 +83,17 @@ export function createHighlighter(options?: Options): Highlighter {
     return result;
   }
 
-  function parseLine(line: string, _isLastLine: boolean): string {
+  function parseLine(line: string, isPartial: boolean): string {
     if (inCodeBlock) {
       if (line.trim().startsWith("```")) {
-        inCodeBlock = false;
+        if (!isPartial) inCodeBlock = false;
         return theme.codeBlock(line);
       }
       return theme.codeBlock(line);
     }
 
     if (line.trim().startsWith("```")) {
-      inCodeBlock = true;
+      if (!isPartial) inCodeBlock = true;
       return theme.codeBlock(line);
     }
 
@@ -127,6 +127,8 @@ export function createHighlighter(options?: Options): Highlighter {
     return parseInline(line);
   }
 
+  let hasPartial = false;
+
   return {
     write(chunk: string | Uint8Array): string {
       const text =
@@ -136,16 +138,29 @@ export function createHighlighter(options?: Options): Highlighter {
       const lines = buffer.split("\n");
       buffer = lines.pop() ?? "";
 
-      return (
-        lines.map((line) => parseLine(line, false)).join("\n") +
-        (lines.length > 0 ? "\n" : "")
-      );
+      let out = "";
+      if (hasPartial) {
+        out += "\x1b[2K\r";
+      }
+
+      if (lines.length > 0) {
+        out += `${lines.map((line) => parseLine(line, false)).join("\n")}\n`;
+      }
+
+      if (buffer.length > 0) {
+        out += parseLine(buffer, true);
+        hasPartial = true;
+      } else {
+        hasPartial = false;
+      }
+
+      return out;
     },
     end(): string {
       if (buffer.length > 0) {
-        const out = parseLine(buffer, true);
+        const out = parseLine(buffer, false);
         buffer = "";
-        return out;
+        return (hasPartial ? "\x1b[2K\r" : "") + out;
       }
       return "";
     },

package/tests/index.test.ts

@@ -32,6 +32,12 @@ describe("Programmatic Usage: highlightSync", () => {
   });
 });
 
+function expectStreamMatch(res: string, markdown: string) {
+  const expected = highlightSync(markdown, { theme: "none" });
+  // biome-ignore lint/suspicious/noControlCharactersInRegex: clear line ansi escape
+  expect(res.replace(/[^\n]*\x1b\[2K\r/g, "")).toBe(expected);
+}
+
 describe("Programmatic Usage: createHighlighter (Streaming)", () => {
   test("should handle chunks without breaking output", () => {
     const highlighter = createHighlighter({ theme: "none" });
@@ -55,8 +61,7 @@ describe("Programmatic Usage: createHighlighter (Streaming)", () => {
     }
     res += highlighter.end();
 
-    const expected = highlightSync(markdown, { theme: "none" });
-    expect(res).toBe(expected);
+    expectStreamMatch(res, markdown);
   });
 
   test("should handle random chunk sizes", () => {
@@ -74,7 +79,6 @@ describe("Programmatic Usage: createHighlighter (Streaming)", () => {
     }
     res += highlighter.end();
 
-    const expected = highlightSync(markdown, { theme: "none" });
-    expect(res).toBe(expected);
+    expectStreamMatch(res, markdown);
   });
 });