From 6043fe79ec3e2cc594c01e000dc7f82b654a7471 Mon Sep 17 00:00:00 2001 From: EnixCoda Date: Sat, 20 Jun 2020 19:16:28 +0800 Subject: [PATCH] fix: update DOM selectors --- src/platforms/GitHub/DOMHelper.ts | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/platforms/GitHub/DOMHelper.ts b/src/platforms/GitHub/DOMHelper.ts index c80dc69..1bff0c4 100644 --- a/src/platforms/GitHub/DOMHelper.ts +++ b/src/platforms/GitHub/DOMHelper.ts @@ -6,12 +6,15 @@ import { renderReact } from 'utils/general' import { CopyFileButton, copyFileButtonClassName } from './CopyFileButton' export function isInCodePage() { - const branchListSelector = '#branch-select-menu.branch-select-menu' + const branchListSelector = ['#branch-select-menu', '.branch-select-menu'].join() return Boolean($(branchListSelector)) } export function getCurrentBranch() { - const selectedBranchButtonSelector = '.repository-content .branch-select-menu summary' + const selectedBranchButtonSelector = [ + '.repository-content #branch-select-menu summary', + '.repository-content .branch-select-menu summary', + ].join() const branchButtonElement: HTMLElement = $(selectedBranchButtonSelector) if (branchButtonElement) { const branchNameSpanElement = branchButtonElement.querySelector('span')