1
0
Fork 0
mirror of https://github.com/terribleplan/next.js.git synced 2024-01-19 02:48:18 +00:00
next.js/packages/next-server/lib/head.js

111 lines
2.8 KiB
JavaScript
Raw Normal View History

2016-10-07 01:57:31 +00:00
import React from 'react'
import PropTypes from 'prop-types'
2016-10-07 01:57:31 +00:00
import sideEffect from './side-effect'
import { HeadManagerContext } from './head-manager-context'
2016-10-07 01:57:31 +00:00
class Head extends React.Component {
static contextType = HeadManagerContext
2016-10-07 01:57:31 +00:00
render () {
return null
}
}
const NEXT_HEAD_IDENTIFIER = 'next-head'
export function defaultHead (className = NEXT_HEAD_IDENTIFIER) {
return [
<meta key='charSet' charSet='utf-8' className={className} />
]
2016-11-17 23:06:54 +00:00
}
2016-10-07 01:57:31 +00:00
function reduceComponents (components) {
return components
.map((component) => React.Children.toArray(component.props.children))
.reduce((a, b) => a.concat(b), [])
.reduce((a, b) => {
if (React.Fragment && b.type === React.Fragment) {
return a.concat(React.Children.toArray(b.props.children))
}
return a.concat(b)
}, [])
.reverse()
.concat(defaultHead(''))
.filter(Boolean)
.filter(unique())
.reverse()
.map((c, i) => {
const className = (c.props && c.props.className ? c.props.className + ' ' : '') + NEXT_HEAD_IDENTIFIER
const key = c.key || i
return React.cloneElement(c, { key, className })
})
2016-10-07 01:57:31 +00:00
}
function mapOnServer (head) {
return head
}
function onStateChange (head) {
if (this.context) {
this.context.updateHead(head)
2016-10-07 01:57:31 +00:00
}
}
const METATYPES = ['name', 'httpEquiv', 'charSet', 'itemProp']
2016-10-07 01:57:31 +00:00
/*
returns a function for filtering head child elements
which shouldn't be duplicated, like <title/>,
except we explicit allow it in ALLOWED_DUPLICATES array
*/
2016-10-07 01:57:31 +00:00
function unique () {
const keys = new Set()
2016-10-07 01:57:31 +00:00
const tags = new Set()
const metaTypes = new Set()
const metaCategories = {}
return (h) => {
if (h.key && h.key.indexOf('.$') === 0) {
if (keys.has(h.key)) return false
keys.add(h.key)
return true
}
2016-10-07 01:57:31 +00:00
switch (h.type) {
case 'title':
case 'base':
if (tags.has(h.type)) return false
tags.add(h.type)
break
case 'meta':
for (let i = 0, len = METATYPES.length; i < len; i++) {
const metatype = METATYPES[i]
if (!h.props.hasOwnProperty(metatype)) continue
if (metatype === 'charSet') {
2016-10-07 01:57:31 +00:00
if (metaTypes.has(metatype)) return false
metaTypes.add(metatype)
} else {
const category = h.props[metatype]
const categories = metaCategories[metatype] || new Set()
if (categories.has(category)) return false
2016-10-07 01:57:31 +00:00
categories.add(category)
metaCategories[metatype] = categories
}
}
break
}
return true
}
}
if (process.env.NODE_ENV === 'development') {
const exact = require('prop-types-exact')
Head.propTypes = exact({
children: PropTypes.node.isRequired
})
}
2016-10-07 01:57:31 +00:00
export default sideEffect(reduceComponents, onStateChange, mapOnServer)(Head)