From a5ddde1f7fd797bb4535fa9310f6c84b43b5fa39 Mon Sep 17 00:00:00 2001
From: EnixCoda <enixcoda@gmail.com>
Date: Sun, 17 Nov 2019 15:50:20 +0800
Subject: [PATCH] feat: handle search result page type

---
 src/utils/DOMHelper.ts | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/utils/DOMHelper.ts b/src/utils/DOMHelper.ts
index 613f5c4..b37c3db 100644
--- a/src/utils/DOMHelper.ts
+++ b/src/utils/DOMHelper.ts
@@ -143,6 +143,7 @@ export function loadWithPJAX(URL: string) {
 const PAGE_TYPES = {
   RAW_TEXT: 'raw_text',
   RENDERED: 'rendered',
+  SEARCH: 'search',
   // PREVIEW: 'preview',
   OTHERS: 'others',
 }
@@ -158,7 +159,9 @@ const PAGE_TYPES = {
 export function getCurrentPageType() {
   const blobWrapperSelector = '.repository-content .blob-wrapper table'
   const readmeSelector = '.repository-content .readme'
+  const searchResultSelector = '.codesearch-results'
   return (
+    $(searchResultSelector, () => PAGE_TYPES.SEARCH) ||
     $(blobWrapperSelector, () => PAGE_TYPES.RAW_TEXT) ||
     $(readmeSelector, () => PAGE_TYPES.RENDERED) ||
     PAGE_TYPES.OTHERS