1
0
Fork 0
mirror of https://github.com/terribleplan/next.js.git synced 2024-01-19 02:48:18 +00:00
next.js/client/patch-react.js
Naoyuki Kanezawa 8570d19af6 Handle errors of React lifecycle methods (#661)
* handle errors of lifecycle methods

* handle errors of render method
2017-01-05 09:27:39 -08:00

63 lines
1.4 KiB
JavaScript

// monkeypatch React for fixing https://github.com/facebook/react/issues/2461
// based on https://gist.github.com/Aldredcz/4d63b0a9049b00f54439f8780be7f0d8
import React from 'react'
let patched = false
export default (handleError = () => {}) => {
if (patched) {
throw new Error('React is already monkeypatched')
}
patched = true
const { createElement } = React
React.createElement = function (Component, ...rest) {
if (typeof Component === 'function') {
const { prototype } = Component
if (prototype && prototype.render) {
prototype.render = wrapRender(prototype.render)
} else {
// stateless component
Component = wrapRender(Component)
}
}
return createElement.call(this, Component, ...rest)
}
const { Component: { prototype: componentPrototype } } = React
const { forceUpdate } = componentPrototype
componentPrototype.forceUpdate = function (...args) {
if (this.render) {
this.render = wrapRender(this.render)
}
return forceUpdate.apply(this, args)
}
function wrapRender (render) {
if (render.__wrapped) {
return render.__wrapped
}
const _render = function (...args) {
try {
return render.apply(this, args)
} catch (err) {
handleError(err)
return null
}
}
// copy all properties
Object.assign(_render, render)
render.__wrapped = _render.__wrapped = _render
return _render
}
}