From b2daa4c3b5f3b215fe2bc2f9a7479b4684d421bc Mon Sep 17 00:00:00 2001
From: EnixCoda <enixcoda@gmail.com>
Date: Fri, 22 Feb 2019 12:22:58 +0800
Subject: [PATCH] feat: resolve git modules on or not on github

---
 src/components/Node.tsx         |  2 +-
 src/content.less                |  7 +++++++
 src/driver/core/FileExplorer.ts | 32 +++++++++++++++++++++++++++++---
 3 files changed, 37 insertions(+), 4 deletions(-)

diff --git a/src/components/Node.tsx b/src/components/Node.tsx
index f6c8522..f023836 100644
--- a/src/components/Node.tsx
+++ b/src/components/Node.tsx
@@ -53,7 +53,7 @@ export default class Node extends React.PureComponent<Props> {
       )
     }
     return (
-      <div className={cx(`node-item-row`, { focused })} title={path}>
+      <div className={cx(`node-item-row`, { focused, disabled: node.accessDenied })} title={path}>
         <a href={node.url} onClick={this.onClick}>
           <div
             className={cx('node-item', { expanded })}
diff --git a/src/content.less b/src/content.less
index 9a83372..22cad40 100644
--- a/src/content.less
+++ b/src/content.less
@@ -303,6 +303,13 @@
             background: #f0f0f6;
           }
 
+          &.disabled {
+            .node-item {
+              pointer-events: none;
+              color: #999999;
+            }
+          }
+
           .node-item {
             display: flex;
             justify-content: space-between;
diff --git a/src/driver/core/FileExplorer.ts b/src/driver/core/FileExplorer.ts
index 2469fa1..0353eef 100644
--- a/src/driver/core/FileExplorer.ts
+++ b/src/driver/core/FileExplorer.ts
@@ -8,6 +8,7 @@ import { findNode } from 'utils/general'
 import { MethodCreator } from 'driver/connect'
 import { Props } from 'components/FileExplorer'
 import Node from 'components/Node'
+import { raiseError } from 'analytics'
 
 export type ConnectorState = {
   stateText: string
@@ -50,16 +51,39 @@ let visibleNodesGenerator: VisibleNodesGenerator
 const init: MethodCreator<Props, ConnectorState> = dispatch => () =>
   dispatch.call(setStateText, 'Fetching File List...')
 
+const githubSubModuleURLRegex = {
+  HTTP: /^https:\/\/github.com\/.*?\/.*?\.git$/,
+  git: /^git@github.com:(.*?)\/(.*?)\.git$/,
+}
+
+function transformModuleGitURL(node: TreeNode, URL: string) {
+  const matched = URL.match(githubSubModuleURLRegex.git)
+  if (!matched) return
+  const [_, userName, repoName] = matched
+  return `https://github.com/${userName}/${repoName}/tree/${node.sha}`
+}
+
+function transformModuleHTTPURL(node: TreeNode, URL: string) {
+  return URL.replace(/\.git/, `/tree/${node.sha}`)
+}
 function resolveGitModules(root: TreeNode, blobData: BlobData) {
   if (blobData) {
     if (blobData.encoding === 'base64' && blobData.content && Array.isArray(root.contents)) {
       const content = atob(blobData.content)
       const parsed = ini.parse(content)
-      Object.values(parsed).map((value: { url: string; path: string }) => {
+      Object.values(parsed).forEach((value: { url: string; path: string }) => {
         const { url, path } = value
         const node = findNode(root, path.split('/'))
         if (node) {
-          node.url = url
+          if (githubSubModuleURLRegex.HTTP.test(url)) {
+            node.url = transformModuleHTTPURL(node, url)
+          } else if (githubSubModuleURLRegex.git.test(url)) {
+            node.url = transformModuleGitURL(node, url)
+          } else {
+            node.accessDenied = true
+          }
+        } else {
+          raiseError(Error(`Sub-module node ${path} not found`))
         }
       })
     }
@@ -318,7 +342,9 @@ const onNodeClick: MethodCreator<Props, ConnectorState, [TreeNode]> = dispatch =
     dispatch.call(focusNode, node, true)
     if (node.url) DOMHelper.loadWithPJAX(node.url)
   } else if (node.type === 'commit') {
-    window.open(node.url, '_blank')
+    if (node.url) {
+      window.open(node.url, '_blank')
+    }
   }
 }