Ver código fonte

Merge branch 'test' of https://git.yishihui.com/DeNet/de-net into test

zhangwei 3 anos atrás
pai
commit
60e25ae63d
1 arquivos alterados com 10 adições e 2 exclusões
  1. 10 2
      src/logic/twitter.js

+ 10 - 2
src/logic/twitter.js

@@ -219,8 +219,16 @@ let bindTwitterArt = {
  */
 function _publishTweetEvent(contentStr, cb) {
     setTimeout(() => {
-        let publishTweetBtn = document.querySelector('div[role="dialog"]').querySelector('div[data-testid="tweetButton"]');
-        publishTweetBtn.addEventListener('click', function () {
+        let publishTweetBtn;
+        let dialog = document.querySelector('div[role="dialog"]');
+        if(dialog) {
+            publishTweetBtn = dialog.querySelector('div[data-testid="tweetButton"]');
+        } else {
+            let domMain =  document.querySelector('main[role="main"]');
+            publishTweetBtn = domMain && domMain.querySelector('div[data-testid="tweetButton"]');
+        }
+        
+        publishTweetBtn && publishTweetBtn.addEventListener('click', function () {
             bindTwitterArt.needBind = true;
             bindTwitterArt.postId = contentStr;
             cb && cb()