Wait dom on chat frame

fiahfy · fiahfy · commit 4a20e6c3fde7 · 2021-02-03T00:06:53.000+09:00
diff --git a/src/content-script.ts b/src/content-script.ts
@@ -56,8 +56,8 @@ const updateMenuButton = () => {
   }
 }
 
-const addMenuButton = () => {
-  const header = document.querySelector(
+const addMenuButton = async () => {
+  const header = await querySelectorAsync(
     '#chat-messages > yt-live-chat-header-renderer'
   )
   const refIconButton = header && header.querySelector('yt-icon-button')
@@ -302,7 +302,7 @@ document.addEventListener('DOMContentLoaded', async () => {
   enabled = data.enabled
   settings = data.settings
   updateRoot()
-  addMenuButton()
   updateItems()
+  await addMenuButton()
   await observe()
 })

Original file line number	Diff line number	Diff line change
`@@ -56,8 +56,8 @@ const updateMenuButton = () => {`
`56`	`56`	`}`
`57`	`57`	`}`
`58`	`58`
`59`		`-const addMenuButton = () => {`
`60`		`- const header = document.querySelector(`
	`59`	`+const addMenuButton = async () => {`
	`60`	`+ const header = await querySelectorAsync(`
`61`	`61`	`'#chat-messages > yt-live-chat-header-renderer'`
`62`	`62`	`)`
`63`	`63`	`const refIconButton = header && header.querySelector('yt-icon-button')`
`@@ -302,7 +302,7 @@ document.addEventListener('DOMContentLoaded', async () => {`
`302`	`302`	`enabled = data.enabled`
`303`	`303`	`settings = data.settings`
`304`	`304`	`updateRoot()`
`305`		`- addMenuButton()`
`306`	`305`	`updateItems()`
	`306`	`+ await addMenuButton()`
`307`	`307`	`await observe()`
`308`	`308`	`})`