(refs #1)Show tree of selected branch

Author: tomoki1207

scripts/components/Root.js

@@ -8,20 +8,31 @@ export default class Root extends Directory {
     // FIXME: exists altanarive way?
     const baseUrl = document.querySelector('header.main-header a.logo').getAttribute('href');
     const relPath = document.location.pathname.replace(baseUrl, '');
+    let rootPath;
+    let branch;
+    if (['tree', 'blob'].some(s => relPath.includes(`/${s}/`))) {
+      const m = relPath.match(/(^\/?[^/]+\/[^/]+)\/(?:tree|blob)\/([^/]+)/);
+      rootPath = m[1];
+      branch = m[2];
+    } else {
+      rootPath = relPath.match(/^\/?[^/]+\/[^/]+/);
+      branch = '';
+    }
     this.setState({
-      rootPath: `${baseUrl}${relPath.match(/^\/?[^/]+\/[^/]+/)}`,
+      rootPath: `${baseUrl}${rootPath}`,
+      branch,
     });
   }
 
   getLocalStorageKey() {
-    return this.state.rootPath;
+    return `${this.state.rootPath}/${this.state.branch}`;
   }
 
   render() {
     const arrow = this.state.expanded ? 'octicon octicon-chevron-down' : 'octicon octicon-chevron-right';
     return (
       <div className="tree-node">
-        <button className="root-expander" onClick={() => this.toggleFolder(`${this.state.rootPath}/explore/`)} >
+        <button className="root-expander" onClick={() => this.toggleFolder(`${this.state.rootPath}/explore/${this.state.branch}`)} >
           <i className={arrow} />
         </button>
         <a href={this.state.rootPath} className="submenu-files" >