Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions .changeset/custom-renderer-meta.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
---
"streamdown": minor
---

Add meta prop to custom renderer props to expose code fence metastring

54 changes: 54 additions & 0 deletions add_tests.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,54 @@
content = open('packages/streamdown/__tests__/custom-renderer.test.tsx').read()
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

A Python scaffolding script (add_tests.py) used to generate test code was accidentally committed to the repository.

Fix on Vercel


# The triple backtick strings - use chr(96)*3 to avoid issues
bt = chr(96) * 3

addition = f"""
it("passes meta prop to custom renderer when metastring is present", async () => {{
const MetaRenderer = ({{ meta }}: CustomRendererProps) => (
<div data-meta={{meta ?? ""}} data-testid="meta-renderer" />
);
const {{ container }} = render(
<Streamdown
plugins={{{{
renderers: [{{ language: "rust", component: MetaRenderer }}],
}}}}
>
{{'{bt}rust {{1}} title="foo"\\nlet x = 1;\\n{bt}'}}
</Streamdown>
);

await waitFor(() => {{
const el = container.querySelector('[data-testid="meta-renderer"]');
expect(el?.getAttribute("data-meta")).toBe('{{1}} title="foo"');
}});
}});

it("passes undefined meta when no metastring present", async () => {{
const MetaRenderer = ({{ meta }}: CustomRendererProps) => (
<div
data-has-meta={{meta !== undefined ? "true" : "false"}}
data-testid="meta-renderer"
/>
);
const {{ container }} = render(
<Streamdown
plugins={{{{
renderers: [{{ language: "rust", component: MetaRenderer }}],
}}}}
>
{{"{bt}rust\\nlet x = 1;\\n{bt}"}}
</Streamdown>
);

await waitFor(() => {{
const el = container.querySelector('[data-testid="meta-renderer"]');
expect(el?.getAttribute("data-has-meta")).toBe("false");
}});
}});
"""

idx = content.rfind('});')
new_content = content[:idx] + addition + content[idx:]
open('packages/streamdown/__tests__/custom-renderer.test.tsx', 'w').write(new_content)
print('done')
38 changes: 38 additions & 0 deletions packages/streamdown/__tests__/custom-renderer.test.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -7,11 +7,13 @@ const VegaRenderer = ({
code,
language,
isIncomplete,
meta,
}: CustomRendererProps) => (
<div
data-code={code}
data-incomplete={isIncomplete}
data-language={language}
data-meta={meta}
data-testid="vega-renderer"
>
Vega Chart: {code}
Expand Down Expand Up @@ -131,6 +133,42 @@ describe("Custom Renderers", () => {
});
});

it("passes meta prop when metastring is present", async () => {
const { container } = render(
<Streamdown
plugins={{
renderers: [{ language: "vega-lite", component: VegaRenderer }],
}}
>
{"```vega-lite startLine=5\ntest-code\n```"}
</Streamdown>
);

await waitFor(() => {
const renderer = container.querySelector('[data-testid="vega-renderer"]');
expect(renderer).toBeTruthy();
expect(renderer?.getAttribute("data-meta")).toBe("startLine=5");
});
});

it("passes meta as undefined when no metastring is present", async () => {
const { container } = render(
<Streamdown
plugins={{
renderers: [{ language: "vega-lite", component: VegaRenderer }],
}}
>
{"```vega-lite\ntest-code\n```"}
</Streamdown>
);

await waitFor(() => {
const renderer = container.querySelector('[data-testid="vega-renderer"]');
expect(renderer).toBeTruthy();
expect(renderer?.getAttribute("data-meta")).toBeNull();
});
});

it("does not interfere with default code blocks when no renderers configured", async () => {
const { container } = render(
<Streamdown>{"```javascript\nconst x = 1;\n```"}</Streamdown>
Expand Down
1 change: 1 addition & 0 deletions packages/streamdown/lib/components.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -843,6 +843,7 @@ const CodeComponent = ({
code={code}
isIncomplete={isBlockIncomplete}
language={language}
meta={metastring}
/>
</Suspense>
);
Expand Down
1 change: 1 addition & 0 deletions packages/streamdown/lib/plugin-types.ts
Original file line number Diff line number Diff line change
Expand Up @@ -149,6 +149,7 @@ export interface CustomRendererProps {
code: string;
isIncomplete: boolean;
language: string;
meta?: string;
}

export interface CustomRenderer {
Expand Down