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

38 lines
1.2 KiB
TypeScript
Raw Normal View History

import { escapeToBuffer } from '../../utils/html.ts'
import type { StringBuffer, HtmlEscaped, HtmlEscapedString } from '../../utils/html.ts'
2022-07-02 06:09:45 +00:00
export const raw = (value: unknown): HtmlEscapedString => {
2022-07-02 06:09:45 +00:00
const escapedString = new String(value) as HtmlEscapedString
escapedString.isEscaped = true
return escapedString
}
export const html = (strings: TemplateStringsArray, ...values: unknown[]): 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++) {
const child = children[i]
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'
) {
buffer[0] += 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 raw(buffer[0])
2022-07-02 06:09:45 +00:00
}