-
Notifications
You must be signed in to change notification settings - Fork 231
/
webengine.py
executable file
·1758 lines (1421 loc) · 67.1 KB
/
webengine.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright (C) 2018 Andy Stewart
#
# Author: Andy Stewart <lazycat.manatee@gmail.com>
# Maintainer: Andy Stewart <lazycat.manatee@gmail.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from PyQt6 import QtCore
from PyQt6.QtCore import QUrl, Qt, QEvent, QEventLoop, QTimer, QFile, QPointF, QPoint, pyqtSlot
from PyQt6.QtWebChannel import QWebChannel
from PyQt6.QtWebEngineWidgets import QWebEngineView
from PyQt6.QtWebEngineCore import QWebEnginePage, QWebEngineScript, QWebEngineProfile, QWebEngineSettings
from PyQt6.QtWidgets import QApplication, QWidget
from core.buffer import Buffer
from core.utils import (touch, string_to_base64, popen_and_call,
call_and_check_code, interactive,
eval_in_emacs, message_to_emacs, clear_emacs_message,
open_url_in_background_tab, duplicate_page_in_new_tab,
open_url_in_new_tab, open_url_in_new_tab_other_window,
focus_emacs_buffer, atomic_edit, get_emacs_config_dir,
to_camel_case, get_emacs_vars, PostGui)
from urllib.parse import urlparse, parse_qs
import base64
import os
import platform
import pathlib
MOUSE_LEFT_BUTTON = 1
MOUSE_WHEEL_BUTTON = 4
MOUSE_BACK_BUTTON = 8
MOUSE_FORWARD_BUTTON = 16
class BrowserView(QWebEngineView):
translate_selected_text = QtCore.pyqtSignal(str)
def __init__(self, buffer_id):
super(QWebEngineView, self).__init__()
self.installEventFilter(self)
self.buffer_id = buffer_id
self.is_button_press = False
self.web_page = BrowserPage()
self.setPage(self.web_page)
self.url_hovered = ""
self.page().linkHovered.connect(self.link_hovered)
self.selectionChanged.connect(self.select_text_change)
# Cookie init.
self.cookies_manager = CookiesManager(self)
self.search_term = ""
self.buildin_js_dir = os.path.join(os.path.dirname(__file__), "js")
self.marker_js_raw = None
self.get_focus_text_js = None
self.get_next_page_url_js = None
self.set_focus_text_raw = None
self.clear_focus_js = None
self.dark_mode_js = None
self.select_input_text_js = None
self.get_selection_text_js = None
self.focus_input_js = None
self.simulated_wheel_event = False
(self.default_zoom, self.zoom_step,
self.show_hover_link, self.marker_letters,
self.marker_fontsize, self.scroll_step) = get_emacs_vars(
["eaf-webengine-default-zoom",
"eaf-webengine-zoom-step",
"eaf-webengine-show-hover-link",
"eaf-marker-letters",
"eaf-marker-fontsize",
"eaf-webengine-scroll-step"])
def delete_all_cookies(self):
self.cookies_manager.delete_all_cookies()
def delete_cookie(self):
self.cookies_manager.delete_cookie()
def load_css(self, path, name):
path = QFile(path)
if not path.open(QFile.OpenModeFlag.ReadOnly | QFile.OpenModeFlag.Text):
return
css = path.readAll().data().decode("utf-8")
SCRIPT = """
(function() {
try {
css = document.createElement('style');
css.type = 'text/css';
css.id = "%s";
document.head.appendChild(css);
css.innerText = `%s`;
} catch(e) {}
})()
""" % (name, css)
script = QWebEngineScript()
self.web_page.runJavaScript(SCRIPT, QWebEngineScript.ScriptWorldId.ApplicationWorld)
script.setName(name)
script.setSourceCode(SCRIPT)
script.setInjectionPoint(QWebEngineScript.InjectionPoint.DocumentReady)
script.setRunsOnSubFrames(True)
script.setWorldId(QWebEngineScript.ScriptWorldId.ApplicationWorld)
self.web_page.scripts().insert(script)
def remove_css(self, name, immediately):
SCRIPT = """
(function() {
var element = document.getElementById('%s');
element.outerHTML = '';
delete element;
})()
""" % (name)
if immediately:
self.web_page.runJavaScript(SCRIPT, QWebEngineScript.ScriptWorldId.ApplicationWorld)
script = self.web_page.scripts().findScript(name)
self.web_page.scripts().remove(script)
def read_js_content(self, js_file):
''' Read content of JavaScript(js) files.'''
return open(os.path.join(self.buildin_js_dir, js_file), "r").read()
def filter_url(self, url):
''' Filter url, avoid duplicate url for the same keyword.'''
parsed = urlparse(url)
qd = parse_qs(parsed.query, keep_blank_values=True)
if parsed.netloc.startswith("www.google.com"):
# Only save search parameters for Google, avoid duplicate url for same keyword.
filtered = dict((k, v) for k, v in qd.items() if k.startswith("q"))
else:
filtered = dict((k, v) for k, v in qd.items())
from urllib.parse import urlunparse, urlencode
return urlunparse([
parsed.scheme,
parsed.netloc,
parsed.path,
parsed.params,
urlencode(filtered, doseq=True), # query string
parsed.fragment
])
def filter_title(self, title):
"If title is google url, we should drop this history, it's a temp redirect url."
try:
parsed = urlparse(title)
qd = parse_qs(parsed.query, keep_blank_values=True)
if parsed.netloc.startswith("www.google.com"):
return ""
else:
return title
except Exception:
return title
def callback_text_search(self, found):
''' Callback of text search'''
if self.search_term != "" and (not found):
message_to_emacs("No ocurrence of \"" + self.search_term + "\" is found.", False)
def clean_search_and_select(self):
self.search_term = ""
self.web_page.findText("")
QTimer().singleShot(0, lambda : self.triggerPageAction(self.web_page.WebAction.Unselect))
def clean_search(self):
self.search_term = ""
self.web_page.findText("")
def _search_text(self, text, is_backward = False):
self.search_term = text
if is_backward:
self.web_page.findText(self.search_term, self.web_page.FindFlag.FindBackward,
self.callback_text_search)
else:
self.web_page.findText(self.search_term, resultCallback = self.callback_text_search)
@interactive
def search_text_forward(self):
''' Forward Search Text.'''
self.buffer.send_input_message("Forward Search Text: ", "search_text_forward", "search")
@interactive
def search_text_backward(self):
''' Backward Search Text.'''
self.buffer.send_input_message("Backward Search Text: ", "search_text_backward", "search")
@interactive
def action_quit(self):
''' Quit action.'''
# Clean search selections if search text is not empty.
self.clean_search_and_select()
if self.buffer.caret_browsing_mode:
if self.buffer.caret_browsing_mark_activated:
self.buffer.caret_toggle_mark()
else:
self.buffer.caret_exit()
def select_text_change(self):
''' Change selected text.'''
# Not translate text if just exit caret mode.
if self.buffer.caret_browsing_exit_flag:
self.buffer.caret_browsing_exit_flag = False
return
# Only translate text when not in caret mode.
if not self.buffer.caret_browsing_mode:
modifiers = QApplication.keyboardModifiers()
if modifiers == Qt.KeyboardModifier.ControlModifier and self.selectedText().strip() != "":
self.translate_selected_text.emit(self.selectedText())
def create_new_window(self):
''' Create new browser window.'''
pass
def createWindow(self, window_type):
''' Create new browser window.'''
return self.create_new_window()
def event(self, event):
''' Catch event.'''
if event.type() == QEvent.Type.ChildAdded:
obj = event.child()
if isinstance(obj, QWidget):
obj.installEventFilter(self)
return QWebEngineView.event(self, event)
def eventFilter(self, obj, event):
''' Handle event.'''
# Debug event.
# if event.type() != 1:
# import time
# print(time.time(), event.type(), self.rect())
if event.type() in [QEvent.Type.MouseButtonPress]:
self.is_button_press = True
elif event.type() in [QEvent.Type.MouseButtonRelease]:
self.is_button_press = False
# Focus emacs buffer when user click view.
event_type = [QEvent.Type.MouseButtonPress, QEvent.Type.MouseButtonRelease, QEvent.Type.MouseButtonDblClick]
if platform.system() != "Darwin":
event_type += [QEvent.Type.Wheel]
if event.type() == QEvent.Type.MouseButtonRelease:
self.buffer.is_focus()
if event.type() in event_type:
if self.simulated_wheel_event:
self.simulated_wheel_event = False
else:
focus_emacs_buffer(self.buffer_id)
if event.type() == QEvent.Type.MouseButtonPress:
if platform.system() == "Darwin":
eval_in_emacs('eaf-activate-emacs-window', [])
if event.button() == MOUSE_FORWARD_BUTTON:
modifiers = QApplication.keyboardModifiers()
if modifiers == Qt.KeyboardModifier.ControlModifier:
self.switch_to_next_webpage_buffer()
else:
self.forward()
event.accept()
return True
elif event.button() == MOUSE_BACK_BUTTON:
modifiers = QApplication.keyboardModifiers()
if modifiers == Qt.KeyboardModifier.ControlModifier:
self.switch_to_previous_webpage_buffer()
else:
self.back()
event.accept()
return True
elif event.button() == MOUSE_LEFT_BUTTON:
modifiers = QApplication.keyboardModifiers()
if modifiers == Qt.KeyboardModifier.ControlModifier and self.url_hovered:
self.open_url_background_buffer(self.url_hovered)
return True
elif event.button() == MOUSE_WHEEL_BUTTON:
if self.url_hovered:
self.open_url_new_buffer_other_window(self.url_hovered)
return True
if event.type() == QEvent.Type.Wheel:
modifiers = QApplication.keyboardModifiers()
if modifiers == Qt.KeyboardModifier.ControlModifier:
if event.angleDelta().y() > 0:
self.zoom_in()
else:
self.zoom_out()
return super(QWebEngineView, self).eventFilter(obj, event)
def link_hovered(self, url):
self.url_hovered = url
if self.show_hover_link:
if url:
message_to_emacs(url, True, False)
else:
clear_emacs_message()
return True
def open_url(self, url):
''' Configure current url.'''
self.setUrl(QUrl(url))
eval_in_emacs('eaf-activate-emacs-window', [])
def open_url_new_buffer(self, url):
''' Open url in a new tab.'''
open_url_in_new_tab(url)
eval_in_emacs('eaf-activate-emacs-window', [])
def open_url_new_buffer_other_window(self, url):
''' Open url in a new tab.'''
open_url_in_new_tab_other_window(url)
eval_in_emacs('eaf-activate-emacs-window', [])
def open_url_background_buffer(self, url):
''' Open url in background tab.'''
open_url_in_background_tab(url)
eval_in_emacs('eaf-activate-emacs-window', [])
def switch_to_next_webpage_buffer(self):
''' Switch to next web page buffer.'''
eval_in_emacs('eaf-next-buffer-same-app', [])
eval_in_emacs('eaf-activate-emacs-window', [])
def switch_to_previous_webpage_buffer(self):
''' Switch to previous web page buffer.'''
eval_in_emacs('eaf-previous-buffer-same-app', [])
eval_in_emacs('eaf-activate-emacs-window', [])
@interactive(insert_or_do=True)
def zoom_in(self):
''' Zoom in.'''
self.setZoomFactor(min(5, self.zoomFactor() + self.zoom_step))
if self.default_zoom == self.zoomFactor():
self.buffer.zoom_data.delete_entry(urlparse(self.buffer.current_url).hostname)
else:
self.buffer.zoom_data.add_entry(urlparse(self.buffer.current_url).hostname, self.zoomFactor())
@interactive(insert_or_do=True)
def zoom_out(self):
''' Zoom out.'''
self.setZoomFactor(max(0.25, self.zoomFactor() - self.zoom_step))
if self.default_zoom == self.zoomFactor():
self.buffer.zoom_data.delete_entry(urlparse(self.buffer.current_url).hostname)
else:
self.buffer.zoom_data.add_entry(urlparse(self.buffer.current_url).hostname, self.zoomFactor())
@interactive(insert_or_do=True)
def zoom_reset(self):
''' Reset the magnification.'''
self.setZoomFactor(float(self.default_zoom))
def eval_js(self, js):
''' Run JavaScript.'''
self.web_page.runJavaScript(js)
def eval_js_file(self, js_file):
''' Run JavaScript from JS file.'''
self.eval_js(self.read_js_content(js_file))
def execute_js(self, js):
''' Execute JavaScript and get result.
NOTE:
Please use eval_js instead if JavaScript function haven't result return.
Otherwise, execute_js will block EAF!!!
'''
return self.web_page.execute_javascript(js)
def eval_js_function(self, *args):
import json
function_name = args[0]
function_args = args[1:]
format_string = ""
for index, arg in enumerate(function_args):
if type(arg) == str:
format_string += '\"{}\"'.format(arg)
else:
format_string += '{}'.format(json.dumps(arg))
if index != len(function_args) - 1:
format_string += ","
format_string = function_name + "(" + format_string + ");"
try:
self.web_page.runJavaScript(format_string)
except:
import traceback
traceback.print_exc()
def scroll_wheel(self, x_offset, y_offset):
from PyQt6.QtGui import QWheelEvent
self.simulated_wheel_event = True
pos = self.rect().center()
global_pos = self.mapToGlobal(pos)
event = QWheelEvent(
QPointF(pos),
QPointF(global_pos),
QPoint(x_offset, y_offset),
QPoint(x_offset, y_offset),
Qt.MouseButton.NoButton,
Qt.KeyboardModifier.NoModifier,
Qt.ScrollPhase.NoScrollPhase,
False
)
for widget in self.buffer.get_key_event_widgets():
QApplication.postEvent(widget, event)
@interactive(insert_or_do=True)
def scroll_left(self):
''' Scroll to left side.'''
self.scroll_wheel(-35, 0)
@interactive(insert_or_do=True)
def scroll_right(self):
''' Scroll to right side.'''
self.scroll_wheel(35, 0)
@interactive(insert_or_do=True)
def scroll_up(self):
''' Scroll up.'''
self.scroll_wheel(0, -self.scroll_step)
@interactive(insert_or_do=True)
def scroll_down(self):
''' Scroll down.'''
self.scroll_wheel(0, self.scroll_step)
@interactive
def scroll_up_page(self):
''' Scroll page up.'''
self.scroll_wheel(0, -self.rect().height())
@interactive(insert_or_do=True)
def scroll_down_page(self):
''' Scroll down a page.'''
self.scroll_wheel(0, self.rect().height())
@interactive(insert_or_do=True)
def scroll_to_begin(self):
''' Scroll to the beginning.'''
self.scroll_wheel(0, 1000000)
@interactive(insert_or_do=True)
def scroll_to_bottom(self):
''' Scroll to the bottom.'''
self.scroll_wheel(0, -1000000)
@interactive
def insert_or_scroll_up_page(self):
'''If input is focus send space key to insert space.
If browser is fullscreen, send space key to play/pause video.
Otherwise, scroll page up.
'''
if self.buffer.is_focus() or self.buffer.is_fullscreen:
self.buffer.send_key(self.buffer.current_event_string)
else:
self.scroll_up_page()
# Try scroll to next page if reach bottom, such as, access google.com
if self.get_next_page_url_js == None:
self.get_next_page_url_js = self.read_js_content("get_next_page_url.js")
next_page_url = self.execute_js(self.get_next_page_url_js)
if next_page_url and next_page_url != "":
self.buffer.change_url(next_page_url)
@interactive
def get_selection_text(self):
''' Get the selected text.'''
if self.get_selection_text_js == None:
self.get_selection_text_js = self.read_js_content("get_selection_text.js")
return self.execute_js(self.get_selection_text_js)
@interactive(insert_or_do=True)
def refresh_page(self):
''' Refresh the page.'''
self.reload()
if platform.system() == "Windows":
eval_in_emacs('eaf-activate-emacs-window', [])
@interactive(insert_or_do=True)
def copy_text(self):
''' Copy selected text.'''
self.triggerPageAction(self.web_page.WebAction.Copy)
if self.buffer.caret_browsing_mode and self.buffer.caret_browsing_mark_activated:
self.buffer.caret_exit()
@interactive()
def yank_text(self):
''' Paste selected text.'''
self.triggerPageAction(self.web_page.WebAction.Paste)
message_to_emacs("Yank selected text.")
@interactive()
def kill_text(self):
''' Cut selected text.'''
self.triggerPageAction(self.web_page.WebAction.Cut)
message_to_emacs("Kill selected text.")
@interactive
def undo_action(self):
''' Undo action.'''
self.triggerPageAction(self.web_page.WebAction.Undo)
@interactive
def redo_action(self):
''' Redo action.'''
self.triggerPageAction(self.web_page.WebAction.Redo)
@interactive
def exit_fullscreen(self):
''' Exit full screen.'''
self.triggerPageAction(self.web_page.WebAction.ExitFullScreen)
@interactive(insert_or_do=True)
def view_source(self):
''' View source.'''
self.triggerPageAction(self.web_page.WebAction.ViewSource)
def select_all(self):
''' Select all text.'''
# We need window focus before select all text.
self.eval_js("window.focus()")
self.triggerPageAction(self.web_page.WebAction.SelectAll)
def select_input_text(self):
''' Select input text.'''
if self.select_input_text_js == None:
self.select_input_text_js = self.read_js_content("select_input_text.js")
self.eval_js(self.select_input_text_js)
@interactive
def get_url(self):
''' Get current url.'''
return self.url().toString().replace(" ", "%20")
def load_marker_file(self):
if self.marker_js_raw == None:
self.marker_js_raw = self.read_js_content("marker.js")
self.eval_js(self.marker_js_raw
.replace("%{marker_letters}", self.marker_letters)
.replace("%{marker_offset_x}", str(self.buffer.marker_offset_x()))
.replace("%{marker_offset_y}", str(self.buffer.marker_offset_y()))
.replace("%{marker_fontsize}", str(self.marker_fontsize)))
def cleanup_links_dom(self):
''' Clean up links.'''
self.load_marker_file()
self.eval_js("Marker.cleanupLinks();")
def get_link_markers(self):
''' Get link markers.'''
self.load_marker_file()
self.eval_js("Marker.generateMarker(Marker.generateClickMarkerList());")
def get_text_markers(self):
''' Get visiable text markers.'''
self.load_marker_file()
self.eval_js("Marker.generateMarker(Marker.generateTextMarkerList());");
def get_marker_link(self, marker):
''' Get marker's link.'''
# print(self.execute_js("Marker.getMarkerSelector('%s')" % str(marker)))
self.load_marker_file()
link = self.execute_js("Marker.gotoMarker('%s', Marker.getMarkerAction)" % str(marker))
self.cleanup_links_dom()
if link is None or type(link).__name__ == "QVariant" or link.startswith("eaf::"):
return False
else:
return link
def _open_link(self, marker):
''' Jump to link according to marker.'''
link = self.get_marker_link(marker)
if link: self.open_url(link)
def _open_link_new_buffer(self, marker):
''' Open the link at the marker in a new buffer.'''
link = self.get_marker_link(marker)
if link: self.open_url_new_buffer(link)
def _open_link_new_buffer_other_window(self, marker):
''' Open the link at the marker in a new buffer in other window.'''
link = self.get_marker_link(marker)
if link: self.open_url_new_buffer_other_window(link)
def _open_link_background_buffer(self, marker):
''' Open link at the marker in the background.'''
link = self.get_marker_link(marker)
if link: self.open_url_background_buffer(link)
def _copy_link(self, marker):
''' Copy the link.'''
link = self.get_marker_link(marker)
if link:
self.buffer.set_clipboard_text(link)
message_to_emacs("Copied " + link)
def get_code_markers(self):
''' Get the code markers.'''
self.load_marker_file()
self.eval_js("Marker.generateMarker(document.querySelectorAll('pre'))")
def get_code_content(self, marker):
''' Get the code content according to marker.'''
self.load_marker_file()
content = self.execute_js("Marker.gotoMarker('%s', (e)=> e.textContent)" % str(marker))
self.cleanup_links_dom()
return content
def _caret_at_line(self, marker):
'''Enable caret by marker'''
self.load_marker_file()
self.eval_js("Marker.gotoMarker('%s', (e) => window.getSelection().collapse(e, 0))" % str(marker))
self.cleanup_links_dom()
markEnabled = self.execute_js("CaretBrowsing.markEnabled")
# reset to clear caret state so the next sentence can be marked
if markEnabled:
self.eval_js("CaretBrowsing.shutdown();")
self.buffer.caret_enable_mark()
self.buffer.caret_next_sentence()
self.buffer.caret_browsing_mode = True
eval_in_emacs('eaf--toggle-caret-browsing', ["'t"])
def copy_code_content(self, marker):
''' Copy the code content according to marker.'''
content = self.get_code_content(marker)
if content != "":
self.buffer.set_clipboard_text(content)
message_to_emacs("Copied code block!")
def get_focus_text(self):
''' Get the focus text.'''
if self.get_focus_text_js == None:
self.get_focus_text_js = self.read_js_content("get_focus_text.js")
return self.execute_js(self.get_focus_text_js)
@interactive
def set_focus_text(self, new_text):
''' Set the focus text.'''
new_text = base64.b64decode(new_text).decode("utf-8")
if self.set_focus_text_raw == None:
self.set_focus_text_raw = self.read_js_content("set_focus_text.js")
self.set_focus_text_js = self.set_focus_text_raw.replace("%{new_text_base64}", string_to_base64(new_text));
self.eval_js(self.set_focus_text_js)
@interactive(insert_or_do=True)
def focus_input(self):
''' input in focus.'''
if self.focus_input_js == None:
self.focus_input_js = self.read_js_content("focus_input.js")
self.eval_js(self.focus_input_js)
eval_in_emacs('eaf-update-focus-state', [self.buffer_id, "'t"])
@interactive
def clear_focus(self):
''' Clear the focus.'''
if self.clear_focus_js == None:
self.clear_focus_js = self.read_js_content("clear_focus.js")
self.eval_js(self.clear_focus_js)
eval_in_emacs('eaf-update-focus-state', [self.buffer_id, "'nil"])
def init_dark_mode_js(self, module_path, selection_color="auto", dark_mode_theme="dark"):
self.dark_mode_js = open(os.path.join(os.path.dirname(module_path), "node_modules", "darkreader", "darkreader.js")).read()
if selection_color != "auto":
self.dark_mode_js = self.dark_mode_js.replace("selectionColor: 'auto'", "selectionColor: '" + selection_color + "'")
self.dark_mode_js += """DarkReader.setFetchMethod(window.fetch);"""
if dark_mode_theme == "dark":
self.dark_mode_js += """DarkReader.enable({brightness: 100, contrast: 90, sepia: 10, mode: 1});"""
else: # light theme
self.dark_mode_js += """DarkReader.enable({brightness: 100, contrast: 90, sepia: 10, mode: 0});"""
class BrowserPage(QWebEnginePage):
def __init__(self):
QWebEnginePage.__init__(self)
def execute_javascript(self, script_src):
''' Execute JavaScript.'''
if hasattr(self, "loop") and self.loop.isRunning():
# NOTE:
#
# Just return None is QEventLoop is busy, such as press 'j' key not release on webpage.
# Otherwise will got error 'RecursionError: maximum recursion depth exceeded while calling a Python object'.
#
# And don't warry, API 'execute_javascript' is works well for programming purpse since we just call this interface occasionally.
return None
else:
# Build event loop.
self.loop = QEventLoop()
# Run JavaScript code.
self.runJavaScript(script_src, self.callback_js)
# Execute event loop, and wait event loop quit.
self.loop.exec()
# Return JavaScript function result.
return self.result
def callback_js(self, result):
''' Callback of JavaScript, call loop.quit to jump code after loop.exec.'''
self.result = result
self.loop.quit()
class BrowserBuffer(Buffer):
close_page = QtCore.pyqtSignal(str)
open_devtools_tab = QtCore.pyqtSignal(object)
def __init__(self, buffer_id, url, arguments, fit_to_view):
Buffer.__init__(self, buffer_id, url, arguments, fit_to_view)
self.add_widget(BrowserView(buffer_id))
self.url = url
self.config_dir = get_emacs_config_dir()
self.page_closed = False
self.zoom_data = ZoomSizeDb(os.path.join(os.path.dirname(self.config_dir), "browser", "zoom_data.db"))
(self.pc_user_agent,
self.phone_user_agent,
self.font_family,
self.fixed_font_family,
self.serif_font_family,
self.font_size,
self.fixed_font_size,
self.enable_plugin,
self.enable_javascript,
self.enable_javascript_access_clipboard,
self.enable_scrollbar,
self.unknown_url_scheme_policy,
self.download_path,
self.default_zoom,
self.zoom_step) = get_emacs_vars(
["eaf-webengine-pc-user-agent",
"eaf-webengine-phone-user-agent",
"eaf-webengine-font-family",
"eaf-webengine-fixed-font-family",
"eaf-webengine-serif-font-family",
"eaf-webengine-font-size",
"eaf-webengine-fixed-font-size",
"eaf-webengine-enable-plugin",
"eaf-webengine-enable-javascript",
"eaf-webengine-enable-javascript-access-clipboard",
"eaf-webengine-enable-scrollbar",
"eaf-webengine-unknown-url-scheme-policy",
"eaf-webengine-download-path",
"eaf-webengine-default-zoom",
"eaf-webengine-zoom-step"])
# self.profile = QWebEngineProfile(self.buffer_widget)
# self.profile.defaultProfile() == QWebEngineProfile.defaultProfile()
# The default profile can be accessed by defaultProfile(). It is a built-in profile that all web pages not specifically created with another profile belong to.
self.profile = QWebEngineProfile.defaultProfile()
self.profile.defaultProfile().setHttpUserAgent(self.pc_user_agent)
self.caret_js_ready = False
self.caret_browsing_mode = False
self.caret_browsing_exit_flag = True
self.caret_browsing_mark_activated = False
self.caret_browsing_search_text = ""
self.is_dark_mode_enabled = self.dark_mode_is_enabled()
self.current_url = ""
self.request_url = ""
self.buffer_widget.web_page.windowCloseRequested.connect(self.close_buffer)
self.buffer_widget.web_page.fullScreenRequested.connect(self.handle_fullscreen_request)
self.buffer_widget.web_page.pdfPrintingFinished.connect(self.notify_print_message)
self.buffer_widget.web_page.featurePermissionRequested.connect(self.permission_requested) # enable camera permission
self.profile.defaultProfile().downloadRequested.connect(self.handle_download_request)
self.settings = self.buffer_widget.settings()
try:
if self.font_family:
for ff in (
self.settings.FontFamily.StandardFont,
self.settings.FontFamily.CursiveFont,
self.settings.FontFamily.FantasyFont,
self.settings.FontFamily.PictographFont):
self.settings.setFontFamily(ff, self.font_family)
if self.fixed_font_family:
self.settings.setFontFamily(self.settings.FontFamily.FixedFont, self.fixed_font_family)
if self.serif_font_family:
for ff in (
self.settings.FontFamily.SerifFont,
self.settings.FontFamily.SansSerifFont):
self.settings.setFontFamily(ff, self.serif_font_family)
self.settings.setFontSize(QWebEngineSettings.FontSize.DefaultFontSize, self.font_size)
self.settings.setFontSize(QWebEngineSettings.FontSize.DefaultFixedFontSize, self.fixed_font_size)
self.settings.setAttribute(QWebEngineSettings.WebAttribute.FullScreenSupportEnabled, True)
self.settings.setAttribute(QWebEngineSettings.WebAttribute.DnsPrefetchEnabled, True)
self.settings.setAttribute(QWebEngineSettings.WebAttribute.FocusOnNavigationEnabled, True)
self.settings.setAttribute(QWebEngineSettings.WebAttribute.PlaybackRequiresUserGesture, False)
self.settings.setAttribute(QWebEngineSettings.WebAttribute.PluginsEnabled, self.enable_plugin)
self.settings.setAttribute(QWebEngineSettings.WebAttribute.JavascriptEnabled, self.enable_javascript)
self.settings.setAttribute(QWebEngineSettings.WebAttribute.JavascriptCanAccessClipboard, self.enable_javascript_access_clipboard)
self.settings.setAttribute(QWebEngineSettings.WebAttribute.ShowScrollBars, self.enable_scrollbar)
if self.unknown_url_scheme_policy == "DisallowUnknownUrlSchemes":
self.settings.setUnknownUrlSchemePolicy(self.settings.UnknownUrlSchemePolicy.DisallowUnknownUrlSchemes)
elif self.unknown_url_scheme_policy == "AllowUnknownUrlSchemesFromUserInteraction":
self.settings.setUnknownUrlSchemePolicy(self.settings.UnknownUrlSchemePolicy.AllowUnknownUrlSchemesFromUserInteraction)
elif self.unknown_url_scheme_policy == "AllowAllUnknownUrlSchemes":
self.settings.setUnknownUrlSchemePolicy(self.settings.UnknownUrlSchemePolicy.AllowAllUnknownUrlSchemes)
except Exception:
import traceback
traceback.print_exc()
self.build_all_methods(self.buffer_widget)
self.build_all_methods(self)
# Reset with HiDPI.
self.buffer_widget.zoom_reset()
# Build webchannel object.
QtCore.qInstallMessageHandler(self.filter_instant_message)
self.channel = QWebChannel()
self.channel.registerObject("pyobject", self)
self.buffer_widget.web_page.setWebChannel(self.channel)
def add_widget(self, widget):
super(BrowserBuffer, self).add_widget(widget)
# Set web page background.
if hasattr(self.buffer_widget, "web_page"):
self.init_web_page_background()
def fetch_marker_callback(self):
try:
return list(map(lambda x: x.lower(),
self.buffer_widget.execute_js("Array.from(document.getElementsByClassName(\"eaf-marker\")).map(function(e) { return e.id });")))
except Exception:
return None
def filter_instant_message(self, *args):
# Disable QWebChannel warnings.
if not args[-1].endswith('value updates in HTML will be broken!'):
print("".join(list(map(str, args[2:]))))
def permission_requested(self, frame, feature):
self.buffer_widget.web_page.setFeaturePermission(frame, feature, QWebEnginePage.PermissionPolicy.PermissionGrantedByUser)
def notify_print_message(self, file_path, success):
''' Notify the print as pdf message.'''
if success:
# Try to rename pdf file with title.
# Use host name if title include invalid file char.
title_path = os.path.join(os.path.expanduser(self.download_path), "{}.pdf".format(self.title))
try:
os.rename(file_path, title_path)
message_to_emacs("Successfully saved current webpage as '{}'.".format(title_path))
except Exception:
message_to_emacs("Successfully saved current webpage as '{}'.".format(file_path))
else:
message_to_emacs("Failed to save current webpage as '{}'.".format(file_path))
def notify_monolith_message(self, download_path, file_path, title, retcode):
''' Notify the save as html message.'''
if retcode == 0:
title_path = os.path.join(os.path.expanduser(download_path), "{}.html".format(title))
try:
os.rename(file_path, title_path)
message_to_emacs("Successfully saved current webpage as '{}'.".format(title_path))
except Exception:
message_to_emacs("Successfully saved current webpage as '{}'.".format(file_path))
else:
message_to_emacs("Failed to save current page as single file.")
def dark_mode_is_enabled(self):
''' Return bool of whether dark mode is enabled.'''
return False
def dark_mode_js_load(self, progress):
# is_dark_mode_enabled use for toggle dark mode.
if self.is_dark_mode_enabled and self.dark_mode_is_enabled() and self.buffer_widget.dark_mode_js != None and progress < 100:
self.buffer_widget.eval_js(self.buffer_widget.dark_mode_js)
def handle_fullscreen_request(self, request):
''' Handle fullscreen request.'''
if request.toggleOn():
self.enter_fullscreen_request.emit()
else:
self.exit_fullscreen_request.emit()
request.accept()
def should_skip_download_item(self, download_item):
return download_item.page() != self.buffer_widget.web_page
def handle_download_request(self, download_item):
''' Handle download request.'''
if self.should_skip_download_item(download_item):
return
download_data = download_item.url().toString()
if download_data.startswith("data:image/"):
image_path = os.path.join(os.path.expanduser(self.download_path), "image.png")
touch(image_path)
with open(image_path, "wb") as f:
b64bytes = download_data.split(",")[1].encode("utf-8")
f.write(base64.b64decode(b64bytes))
message_to_emacs("Save image: " + image_path)
else:
if hasattr(self, "try_start_aria2_daemon"):
self.try_start_aria2_daemon()
from core.pyaria2 import Jsonrpc
download_url = download_item.url().toString()
jsonrpc = Jsonrpc('localhost', 6800)
resp = jsonrpc.addUris(download_url)
message_to_emacs("Downloading: " + download_url)
def _save_as_pdf(self):
parsed = urlparse(self.url)
qd = parse_qs(parsed.query, keep_blank_values=True)
pdf_path = os.path.join(os.path.expanduser(self.download_path), "{}.pdf".format(parsed.netloc))
message_to_emacs("Saving as pdf...")
self.buffer_widget.web_page.printToPdf(pdf_path)
@interactive(insert_or_do=True)