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

233 lines
5.5 KiB
TypeScript
Raw Normal View History

import { escapeToBuffer } from '../../utils/html'
import type { StringBuffer, HtmlEscaped, HtmlEscapedString } from '../../utils/html'
2022-06-07 10:03:42 +00:00
2022-11-30 22:49:28 +00:00
// eslint-disable-next-line @typescript-eslint/no-explicit-any
type Props = Record<string, any>
2022-11-30 22:49:28 +00:00
2022-06-07 10:03:42 +00:00
declare global {
// eslint-disable-next-line @typescript-eslint/no-namespace
namespace JSX {
type Element = HtmlEscapedString
interface ElementChildrenAttribute {
children: Child
}
2022-06-07 10:03:42 +00:00
interface IntrinsicElements {
[tagName: string]: Props
2022-06-07 10:03:42 +00:00
}
}
}
const emptyTags = [
'area',
'base',
'br',
'col',
'embed',
'hr',
'img',
'input',
'keygen',
'link',
'meta',
'param',
'source',
'track',
'wbr',
]
const booleanAttributes = [
'allowfullscreen',
'async',
'autofocus',
'autoplay',
'checked',
'controls',
'default',
'defer',
'disabled',
'formnovalidate',
'hidden',
'inert',
'ismap',
'itemscope',
'loop',
'multiple',
'muted',
'nomodule',
'novalidate',
'open',
'playsinline',
'readonly',
'required',
'reversed',
'selected',
]
const childrenToStringToBuffer = (children: Child[], buffer: StringBuffer): void => {
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) {
continue
} else if (child instanceof JSXNode) {
child.toStringToBuffer(buffer)
} else if (
typeof child === 'number' ||
(child as unknown as { isEscaped: boolean }).isEscaped
) {
buffer[0] += child
} else {
// `child` type is `Child[]`, so stringify recursively
childrenToStringToBuffer(child, buffer)
}
}
}
type Child = string | number | JSXNode | Child[]
export class JSXNode implements HtmlEscaped {
tag: string | Function
2022-11-30 22:49:28 +00:00
props: Props
children: Child[]
2023-05-02 09:45:21 +00:00
isEscaped: true = true as const
2022-11-30 22:49:28 +00:00
constructor(tag: string | Function, props: Props, children: Child[]) {
this.tag = tag
this.props = props
this.children = children
2022-06-07 10:03:42 +00:00
}
toString(): string {
const buffer: StringBuffer = ['']
this.toStringToBuffer(buffer)
return buffer[0]
}
toStringToBuffer(buffer: StringBuffer): void {
const tag = this.tag as string
const props = this.props
let { children } = this
buffer[0] += `<${tag}`
const propsKeys = Object.keys(props || {})
for (let i = 0, len = propsKeys.length; i < len; i++) {
const v = props[propsKeys[i]]
if (typeof v === 'string') {
buffer[0] += ` ${propsKeys[i]}="`
escapeToBuffer(v, buffer)
buffer[0] += '"'
} else if (typeof v === 'number') {
buffer[0] += ` ${propsKeys[i]}="${v}"`
} else if (v === null || v === undefined) {
// Do nothing
} else if (typeof v === 'boolean' && booleanAttributes.includes(propsKeys[i])) {
if (v) {
buffer[0] += ` ${propsKeys[i]}=""`
}
} else if (propsKeys[i] === 'dangerouslySetInnerHTML') {
if (children.length > 0) {
throw 'Can only set one of `children` or `props.dangerouslySetInnerHTML`.'
}
const escapedString = new String(v.__html) as HtmlEscapedString
escapedString.isEscaped = true
children = [escapedString]
} else {
buffer[0] += ` ${propsKeys[i]}="`
escapeToBuffer(v.toString(), buffer)
buffer[0] += '"'
}
}
if (emptyTags.includes(tag as string)) {
buffer[0] += '/>'
return
}
buffer[0] += '>'
childrenToStringToBuffer(children, buffer)
2022-06-07 10:03:42 +00:00
buffer[0] += `</${tag}>`
}
}
class JSXFunctionNode extends JSXNode {
toStringToBuffer(buffer: StringBuffer): void {
const { children } = this
const res = (this.tag as Function).call(null, {
...this.props,
children: children.length <= 1 ? children[0] : children,
})
if (res instanceof JSXNode) {
res.toStringToBuffer(buffer)
} else if (typeof res === 'number' || (res as HtmlEscaped).isEscaped) {
buffer[0] += res
} else {
escapeToBuffer(res, buffer)
}
}
}
class JSXFragmentNode extends JSXNode {
toStringToBuffer(buffer: StringBuffer): void {
childrenToStringToBuffer(this.children, buffer)
}
}
export { jsxFn as jsx }
const jsxFn = (
tag: string | Function,
2022-11-30 22:49:28 +00:00
props: Props,
...children: (string | HtmlEscapedString)[]
): JSXNode => {
if (typeof tag === 'function') {
return new JSXFunctionNode(tag, props, children)
} else {
return new JSXNode(tag, props, children)
}
2022-06-07 10:03:42 +00:00
}
2022-11-30 22:49:28 +00:00
type FC<T = Props> = (props: T) => HtmlEscapedString
2022-11-30 22:49:28 +00:00
const shallowEqual = (a: Props, b: Props): boolean => {
if (a === b) {
return true
}
const aKeys = Object.keys(a)
const bKeys = Object.keys(b)
if (aKeys.length !== bKeys.length) {
return false
}
for (let i = 0, len = aKeys.length; i < len; i++) {
if (a[aKeys[i]] !== b[aKeys[i]]) {
return false
}
}
return true
}
export const memo = <T>(
component: FC<T>,
propsAreEqual: (prevProps: Readonly<T>, nextProps: Readonly<T>) => boolean = shallowEqual
): FC<T> => {
let computed = undefined
let prevProps: T | undefined = undefined
return ((props: T): HtmlEscapedString => {
if (prevProps && !propsAreEqual(prevProps, props)) {
computed = undefined
}
prevProps = props
return (computed ||= component(props))
}) as FC<T>
}
2022-11-30 22:49:28 +00:00
export const Fragment = (props: { key?: string; children?: Child[] }): JSXNode => {
return new JSXFragmentNode('', {}, props.children || [])
}