From 0c146fc87d8f13c5a6f3180e71bbded5cc51b40f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Michal=20=C4=8Caplygin?= <myfonj@gmail.com>
Date: Mon, 29 Apr 2019 03:45:58 +0200
Subject: [PATCH] Remove entire html@class if there wasn't any.

Prior applyTransitionPatch
---
 content/apply.js | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/content/apply.js b/content/apply.js
index cd30cd1c..bc5fa4c2 100644
--- a/content/apply.js
+++ b/content/apply.js
@@ -348,6 +348,7 @@ const APPLY = (() => {
     // the browsers, especially Firefox, may apply all transitions on page load
     const className = chrome.runtime.id + '-transition-bug-fix';
     const docId = document.documentElement.id ? '#' + document.documentElement.id : '';
+    const docHadClass = document.documentElement.hasAttribute('class');
     document.documentElement.classList.add(className);
     const el = styleInjector.createStyle('transition-patch');
     // FIXME: this will trigger docRootObserver and cause a resort. We should
@@ -361,7 +362,11 @@ const APPLY = (() => {
       .then(() => {
         setTimeout(() => {
           el.remove();
-          document.documentElement.classList.remove(className);
+          if (docHadClass) {
+            document.documentElement.classList.remove(className);
+          } else {
+            document.documentElement.removeAttribute('class');
+          }
         });
       });
   }