Fix error in non-html page (#174)

sheepzh · sheepzh · commit 393c3740c9ae · 2022-12-20T21:56:32.000+08:00
diff --git a/src/content-script/limit.ts b/src/content-script/limit.ts
@@ -52,6 +52,9 @@ class _Modal {
     }
 
     showModal(showDelay: boolean) {
+        if (!document.body) {
+            return
+        }
         const _thisUrl = this.url
         if (showDelay && this.mask.childElementCount === 1) {
             this.delayContainer = document.createElement('p')
@@ -84,7 +87,7 @@ class _Modal {
     }
 
     hideModal() {
-        if (!this.visible) {
+        if (!this.visible || document.body) {
             return
         }
         this.mask.remove()

Original file line number	Diff line number	Diff line change
`@@ -52,6 +52,9 @@ class _Modal {`
`52`	`52`	`}`
`53`	`53`
`54`	`54`	`showModal(showDelay: boolean) {`
	`55`	`+ if (!document.body) {`
	`56`	`+ return`
	`57`	`+ }`
`55`	`58`	`const _thisUrl = this.url`
`56`	`59`	`if (showDelay && this.mask.childElementCount === 1) {`
`57`	`60`	`this.delayContainer = document.createElement('p')`
`@@ -84,7 +87,7 @@ class _Modal {`
`84`	`87`	`}`
`85`	`88`
`86`	`89`	`hideModal() {`
`87`		`- if (!this.visible) {`
	`90`	`+ if (!this.visible \|\| document.body) {`
`88`	`91`	`return`
`89`	`92`	`}`
`90`	`93`	`this.mask.remove()`