diff --git a/packages/web-ui/src/index.ts b/packages/web-ui/src/index.ts index 334da2e3..26311589 100644 --- a/packages/web-ui/src/index.ts +++ b/packages/web-ui/src/index.ts @@ -66,6 +66,7 @@ export { ArtifactPill } from "./tools/artifacts/ArtifactPill.js"; export { type Artifact, ArtifactsPanel, type ArtifactsParams } from "./tools/artifacts/artifacts.js"; export { ArtifactsToolRenderer } from "./tools/artifacts/artifacts-tool-renderer.js"; export { HtmlArtifact } from "./tools/artifacts/HtmlArtifact.js"; +export { ImageArtifact } from "./tools/artifacts/ImageArtifact.js"; export { MarkdownArtifact } from "./tools/artifacts/MarkdownArtifact.js"; export { SvgArtifact } from "./tools/artifacts/SvgArtifact.js"; export { TextArtifact } from "./tools/artifacts/TextArtifact.js"; diff --git a/packages/web-ui/src/tools/artifacts/ImageArtifact.ts b/packages/web-ui/src/tools/artifacts/ImageArtifact.ts new file mode 100644 index 00000000..8bc2cc99 --- /dev/null +++ b/packages/web-ui/src/tools/artifacts/ImageArtifact.ts @@ -0,0 +1,116 @@ +import { DownloadButton } from "@mariozechner/mini-lit"; +import { html, type TemplateResult } from "lit"; +import { customElement, property } from "lit/decorators.js"; +import { i18n } from "../../utils/i18n.js"; +import { ArtifactElement } from "./ArtifactElement.js"; + +@customElement("image-artifact") +export class ImageArtifact extends ArtifactElement { + @property({ type: String }) private _content = ""; + + get content(): string { + return this._content; + } + + set content(value: string) { + this._content = value; + this.requestUpdate(); + } + + protected override createRenderRoot(): HTMLElement | DocumentFragment { + return this; + } + + override connectedCallback(): void { + super.connectedCallback(); + this.style.display = "block"; + this.style.height = "100%"; + } + + private getMimeType(): string { + const ext = this.filename.split(".").pop()?.toLowerCase(); + if (ext === "jpg" || ext === "jpeg") return "image/jpeg"; + if (ext === "gif") return "image/gif"; + if (ext === "webp") return "image/webp"; + if (ext === "svg") return "image/svg+xml"; + if (ext === "bmp") return "image/bmp"; + if (ext === "ico") return "image/x-icon"; + return "image/png"; + } + + private getImageUrl(): string { + // If content is already a data URL, use it directly + if (this._content.startsWith("data:")) { + return this._content; + } + // Otherwise assume it's base64 and construct data URL + return `data:${this.getMimeType()};base64,${this._content}`; + } + + private decodeBase64(): Uint8Array { + let base64Data: string; + + // If content is a data URL, extract the base64 part + if (this._content.startsWith("data:")) { + const base64Match = this._content.match(/base64,(.+)/); + if (base64Match) { + base64Data = base64Match[1]; + } else { + // Not a base64 data URL, return empty + return new Uint8Array(0); + } + } else { + // Otherwise use content as-is + base64Data = this._content; + } + + // Decode base64 to binary string + const binaryString = atob(base64Data); + + // Convert binary string to Uint8Array + const bytes = new Uint8Array(binaryString.length); + for (let i = 0; i < binaryString.length; i++) { + bytes[i] = binaryString.charCodeAt(i); + } + + return bytes; + } + + public getHeaderButtons() { + return html` +