Move router from Menu to LinksApp

Author: marcodearaujo

app/components/LinksApp/LinksApp.js

@@ -6,6 +6,7 @@ class LinksApp extends React.Component
     constructor(props, context) {
         super(props, context);
         this.generate = this.generate.bind(this);
+        this.handleLogout = this.handleLogout.bind(this);
         this.state = {
             links: [
                 [ '/', 'Home'],
@@ -15,6 +16,12 @@ class LinksApp extends React.Component
         };
     }
 
+    handleLogout() {
+        console.log('Marco', this.context, this.context.router);
+        delete localStorage.token;
+        this.context.router.push("/");
+    }
+
     generate() {
         return this.state.links.map((link, index) => (
             <Link to={link[0]}
@@ -31,13 +38,20 @@ class LinksApp extends React.Component
         return (
             <span className="nav-right nav-menu is-active">
                 {this.generate()}
+                <span className="nav-item">
+                    <a className="button" onClick={this.handleLogout}>Logout</a>
+                </span>
             </span>
         );
     }
 }
 
 LinksApp.propTypes = {
-    hide: React.PropTypes.func.isRequired,
+    hide: React.PropTypes.func.isRequired
+};
+
+LinksApp.contextTypes = {
+    router: React.PropTypes.object.isRequired
 };
 
 export default LinksApp;

app/components/Menu/Menu.js

@@ -1,27 +1,16 @@
 import React from 'react';
-import { Router, Link } from 'react-router'
 
 import Nav from 'components/Nav/Nav'
 
 class Menu extends React.Component{
     constructor(props, context) {
         super(props, context);
-        this.handleLogout = this.handleLogout.bind(this);
         this.handleView = this.handleView.bind(this);
     }
 
-    handleLogout() {
-        delete localStorage.token;
-        this.context.router.push("/");
-    }
-
     handleView() {
         return (
-            <Nav>
-                <span className="nav-item">
-                    <a className="button" onClick={this.handleLogout}>Logout</a>
-                </span>
-            </Nav>
+            <Nav />
         );
     }
 
@@ -37,9 +26,5 @@ class Menu extends React.Component{
     }
 }
 
-Menu.contextTypes = {
-    router: React.PropTypes.object.isRequired
-};
-
 export default Menu;