0
0
mirror of https://github.com/honojs/hono.git synced 2024-12-01 10:51:01 +00:00
hono/deno_dist/helper/html/index.ts

56 lines
1.7 KiB
TypeScript
Raw Normal View History

import { escapeToBuffer, stringBufferToString } from '../../utils/html.ts'
import type {
StringBuffer,
HtmlEscaped,
HtmlEscapedString,
HtmlEscapedCallback,
} from '../../utils/html.ts'
2022-07-02 06:09:45 +00:00
export const raw = (value: unknown, callbacks?: HtmlEscapedCallback[]): HtmlEscapedString => {
2022-07-02 06:09:45 +00:00
const escapedString = new String(value) as HtmlEscapedString
escapedString.isEscaped = true
escapedString.callbacks = callbacks
2022-07-02 06:09:45 +00:00
return escapedString
}
export const html = (
strings: TemplateStringsArray,
...values: unknown[]
): HtmlEscapedString | Promise<HtmlEscapedString> => {
const buffer: StringBuffer = ['']
2022-07-02 06:09:45 +00:00
for (let i = 0, len = strings.length - 1; i < len; i++) {
buffer[0] += strings[i]
2022-07-02 06:09:45 +00:00
const children =
values[i] instanceof Array ? (values[i] as Array<unknown>).flat(Infinity) : [values[i]]
2022-07-02 06:09:45 +00:00
for (let i = 0, len = children.length; i < len; i++) {
// eslint-disable-next-line @typescript-eslint/no-explicit-any
const child = children[i] as any
if (typeof child === 'string') {
escapeToBuffer(child, buffer)
} else if (typeof child === 'boolean' || child === null || child === undefined) {
2022-07-02 06:09:45 +00:00
continue
} else if (
(typeof child === 'object' && (child as HtmlEscaped).isEscaped) ||
typeof child === 'number'
) {
const tmp = child.toString()
if (tmp instanceof Promise) {
buffer.unshift('', tmp)
} else {
buffer[0] += tmp
}
} else if (child instanceof Promise) {
buffer.unshift('', child)
2022-07-02 06:09:45 +00:00
} else {
escapeToBuffer(child.toString(), buffer)
2022-07-02 06:09:45 +00:00
}
}
}
buffer[0] += strings[strings.length - 1]
2022-07-02 06:09:45 +00:00
return buffer.length === 1 ? raw(buffer[0]) : stringBufferToString(buffer)
2022-07-02 06:09:45 +00:00
}