feat: support github file tree view (#146)

homerchen19 · Mar 21, 2023 · f251986 · f251986
1 parent 0420e57
commit f251986
Showing 1 changed file with 19 additions and 3 deletions.
diff --git a/src/ts/content.ts b/src/ts/content.ts
@@ -29,10 +29,16 @@ const fonts = [
   { name: 'Octicons Regular', path: 'fonts/octicons.woff2' },
 ];
 
+const isGithubPage = () => /.*github.*/.test(window.location.hostname);
+
+const isGithubTreeViewPage = () => {
+  if (!isGithubPage()) return false;
+
+  return !!document.querySelector('#repos-file-tree');
+};
+
 const isGithubFilesPage = () => {
-  if (!/.*github.*/.test(window.location.hostname)) {
-    return false;
-  }
+  if (!isGithubPage()) return false;
 
   const pathname = window.location.pathname;
   const filesPageUrlPattern = new RegExp(/^\/.+\/.+\/pull\/\d+\/files$/);
@@ -53,6 +59,16 @@ const getSelector = (hostname: string) => {
         };
       }
 
+      if (isGithubTreeViewPage()) {
+        return {
+          filenameSelector:
+            'div.PRIVATE_TreeView-item-content > span.PRIVATE_TreeView-item-content-text > span',
+          iconSelector:
+            'div.PRIVATE_TreeView-item-content > div.PRIVATE_TreeView-item-visual',
+          host: Host.GitHub,
+        };
+      }
+
       return {
         filenameSelector:
           'tr.js-navigation-item > td.content > span, .files-list > a.list-item, div.js-navigation-item > div[role="rowheader"] > span',