I put Menu in charge to know which links will be used

Author: marcodearaujo

app/components/LinksApp/LinksApp.js

@@ -1,27 +1,17 @@
 import React from 'react';
-import { Router, Link } from 'react-router'
+import { Link } from 'react-router'
 
 class LinksApp extends React.Component
 {
     constructor(props, context) {
         super(props, context);
         this.generate = this.generate.bind(this);
-        this.handleLogout = this.handleLogout.bind(this);
         this.handleClick = this.handleClick.bind(this);
         this.state = {
-            links: [
-                [ '/', 'Home'],
-                ['/client', 'Client'],
-                ['/area', 'Area']
-            ]
+            links: this.props.links || []
         };
     }
 
-    handleLogout() {
-        delete localStorage.token;
-        this.context.router.push("/");
-    }
-
     handleClick(e) {
         if (this.context.onClick) {
             this.context.onClick();
@@ -46,16 +36,16 @@ 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 = {
+    links: React.PropTypes.array.isRequired
+};
+
 LinksApp.contextTypes = {
-    router: React.PropTypes.object.isRequired,
     onClick: React.PropTypes.func
 };
 

app/components/Menu/Menu.js

@@ -1,4 +1,5 @@
 import React from 'react';
+import { Router } from 'react-router'
 
 import Nav from 'components/Nav/Nav'
 import LinksApp from 'components/LinksApp/LinksApp'
@@ -7,13 +8,31 @@ import LinksApp from 'components/LinksApp/LinksApp'
 class Menu extends React.Component{
     constructor(props, context) {
         super(props, context);
+        this.handleLogout = this.handleLogout.bind(this);
         this.handleView = this.handleView.bind(this);
+        this.state = {
+            links: [
+                [ '/', 'Home'],
+                ['/client', 'Client'],
+                ['/area', 'Area']
+            ]
+        };
+    }
+
+    handleLogout() {
+        delete localStorage.token;
+        this.context.router.push("/");
     }
 
     handleView() {
         return (
             <Nav>
-                <LinksApp />
+                <LinksApp
+                    links={this.state.links}
+                />
+                <span className="nav-item">
+                    <a className="button" onClick={this.handleLogout}>Logout</a>
+                </span>
             </Nav>
         );
     }
@@ -30,5 +49,9 @@ class Menu extends React.Component{
     }
 }
 
+Menu.contextTypes = {
+    router: React.PropTypes.object.isRequired
+};
+
 export default Menu;