Make Prompt use the new API

Author: timdorr

packages/react-router/modules/Prompt.js

@@ -1,33 +1,31 @@
 import React from "react";
 import PropTypes from "prop-types";
 import invariant from "invariant";
+import RouterContext from "./RouterContext";
 
 /**
  * The public API for prompting the user before navigating away
  * from a screen with a component.
  */
-class Prompt extends React.Component {
+class InnerPrompt extends React.Component {
   static propTypes = {
     when: PropTypes.bool,
-    message: PropTypes.oneOfType([PropTypes.func, PropTypes.string]).isRequired
-  };
-
-  static defaultProps = {
-    when: true
-  };
-
-  static contextTypes = {
+    message: PropTypes.oneOfType([PropTypes.func, PropTypes.string]).isRequired,
     router: PropTypes.shape({
       history: PropTypes.shape({
         block: PropTypes.func.isRequired
       }).isRequired
     }).isRequired
   };
 
+  static defaultProps = {
+    when: true
+  };
+
   enable(message) {
     if (this.unblock) this.unblock();
 
-    this.unblock = this.context.router.history.block(message);
+    this.unblock = this.props.router.history.block(message);
   }
 
   disable() {
@@ -39,7 +37,7 @@ class Prompt extends React.Component {
 
   componentWillMount() {
     invariant(
-      this.context.router,
+      this.props.router,
       "You should not use <Prompt> outside a <Router>"
     );
 
@@ -64,4 +62,9 @@ class Prompt extends React.Component {
   }
 }
 
+const Prompt = props =>
+  RouterContext.consume(({ router }) => (
+    <InnerPrompt {...props} router={router} />
+  ));
+
 export default Prompt;