396 lines
11 KiB
Python
Executable File
396 lines
11 KiB
Python
Executable File
#!/usr/bin/python3
|
|
|
|
import gi, json, os, shutil
|
|
|
|
|
|
gi.require_version("Gtk", "3.0")
|
|
gi.require_version("WebKit2", "4.1")
|
|
gi.require_version("Keybinder", "3.0")
|
|
gi.require_version('Notify', '0.7') # gir1.2-notify-0.7
|
|
|
|
from gi.repository import Gtk, Gdk, WebKit2, GLib, Gio, Keybinder, Notify
|
|
from gi.repository.GdkPixbuf import Pixbuf
|
|
|
|
from notes.utils import *
|
|
from notes.mimetypes import get_mimetype
|
|
|
|
# 优先尝试使用指示器, 没有再使用 Gtk.StatusIcon
|
|
try:
|
|
gi.require_version('AyatanaAppIndicator3', '0.1')
|
|
# 需要安装这个包 gir1.2-ayatanaappindicator3-0.1
|
|
from gi.repository import AyatanaAppIndicator3 as AppIndicator3
|
|
except (ValueError, ImportError):
|
|
AppIndicator3 = None
|
|
|
|
# 初始化 Keybinder
|
|
Keybinder.init()
|
|
Notify.init(Notify.get_app_name() or 'webapp')
|
|
|
|
# im_context = Gtk.IMContext()
|
|
|
|
|
|
class WebKitWindow(Gtk.Window):
|
|
def __init__(self):
|
|
|
|
Gtk.Window.__init__(self, title="WebKit Example")
|
|
|
|
self.set_default_size(800, 600)
|
|
|
|
settings = WebKit2.Settings()
|
|
settings.set_enable_page_cache(True)
|
|
settings.set_enable_offline_web_application_cache(True)
|
|
|
|
settings.set_enable_developer_extras(True)
|
|
# settings.set_disable_web_security(True)
|
|
|
|
|
|
|
|
|
|
settings.set_enable_html5_database(True)
|
|
settings.set_enable_html5_local_storage(True)
|
|
|
|
settings.set_javascript_can_access_clipboard(True)
|
|
settings.set_javascript_can_open_windows_automatically(True)
|
|
|
|
settings.set_user_agent("Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/605.1.15 (KHTML, like Gecko) weapp/1.0.0 Version/16.4 Safari/605.1.15")
|
|
|
|
|
|
manager = WebKit2.UserContentManager()
|
|
|
|
script = open(self.file_path('./inject.js'), 'r').read()
|
|
frame = WebKit2.UserContentInjectedFrames.ALL_FRAMES
|
|
time = WebKit2.UserScriptInjectionTime.END
|
|
script = WebKit2.UserScript(script, frame, time, None, None)
|
|
|
|
manager.add_script(script)
|
|
|
|
manager.connect('script-message-received::app', self.on_script_message)
|
|
manager.register_script_message_handler('app')
|
|
|
|
self.webview = WebKit2.WebView.new_with_user_content_manager(manager)
|
|
self.webview.set_settings(settings)
|
|
|
|
|
|
context = self.webview.get_context()
|
|
context.register_uri_scheme('app', self.resource_request_callback)
|
|
|
|
|
|
im = self.webview.get_input_method_context()
|
|
im.set_enable_preedit(True)
|
|
|
|
# self.webview.load_uri("http://127.0.0.1:10086/index.html")
|
|
self.webview.load_uri("app:///index.html")
|
|
# self.webview.load_uri("https://benchmark.wkit.fun")
|
|
|
|
|
|
im.connect('preedit-changed', self.on_preedit_changed)
|
|
|
|
|
|
|
|
self.clipboard = Gtk.Clipboard.get(Gdk.SELECTION_CLIPBOARD)
|
|
|
|
|
|
self.add(self.webview)
|
|
|
|
|
|
def on_preedit_changed(self, im):
|
|
print(p)
|
|
im.notify_focus_in()
|
|
im.notify_cursor_area(100, 200, 0,0)
|
|
|
|
|
|
|
|
def resource_request_callback(self, req):
|
|
|
|
schema = req.get_scheme()
|
|
pathname = req.get_path()
|
|
ext = pathname.split('.')[-1]
|
|
|
|
if schema == 'app':
|
|
data = open(self.file_path('./webview' + pathname)).read()
|
|
data = Gio.MemoryInputStream.new_from_data(data.encode())
|
|
req.finish(data, -1, get_mimetype(ext))
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def create_tray(self):
|
|
|
|
if AppIndicator3 :
|
|
indicator = AppIndicator3.Indicator.new(
|
|
"youtube",
|
|
"youtube",
|
|
AppIndicator3.IndicatorCategory.APPLICATION_STATUS
|
|
)
|
|
|
|
# indicator.set_title('alacritty 6666')
|
|
# indicator.set_label('alacritty 8888', '')
|
|
# indicator.set_icon_full('alacritty','alacritty')
|
|
# indicator.set_attention_icon_full('alacritty', 'alacritty')
|
|
# indicator.set_ordering_index(99)
|
|
indicator.set_status(AppIndicator3.IndicatorStatus.ACTIVE)
|
|
# indicator.set_status(AppIndicator3.IndicatorStatus.ATTENTION)
|
|
menu = Gtk.Menu()
|
|
menu_items = Gtk.MenuItem.new_with_label("Toggle Floater")
|
|
menu.append(menu_items)
|
|
menu_items.connect("activate", self.toggle_visible)
|
|
|
|
menu.show_all()
|
|
indicator.set_menu(menu)
|
|
indicator.set_secondary_activate_target(menu_items)
|
|
else:
|
|
# windows 和 macos 必须传二进制图标, linux可传图标名称(自会去主题中找)
|
|
# indicator = Gtk.StatusIcon.new_from_pixbuf(get_logo(32))
|
|
# linux
|
|
# indicator = Gtk.StatusIcon.new_from_icon_name('youtube')
|
|
|
|
# return indicator
|
|
indicator = Gtk.StatusIcon.new_from_icon_name('youtube')
|
|
indicator.connect('activate', self.toggle_visible)
|
|
return indicator
|
|
|
|
|
|
def toggle_visible(self, icon):
|
|
|
|
if self.is_visible():
|
|
self.hide()
|
|
else:
|
|
self.present()
|
|
|
|
|
|
def call_js(self, method, data = None):
|
|
scripts = 'native.$emit("' + method + '",' + json.dumps(data) + ')'
|
|
self.webview.evaluate_javascript(scripts, -1)
|
|
|
|
|
|
|
|
def file_path(self, filepath):
|
|
root = os.path.dirname(os.path.realpath(__file__))
|
|
return os.path.join(root, filepath)
|
|
|
|
|
|
def on_script_message(self, webview, message):
|
|
|
|
data = json.loads(message.get_js_value().to_json(0))
|
|
|
|
event = data.get('event')
|
|
callback = data.get('callback')
|
|
params = data.get('data')
|
|
output = None
|
|
|
|
|
|
match event:
|
|
case 'fs':
|
|
filepath = params['filepath']
|
|
if params['action'] == 'read':
|
|
file = open(filepath)
|
|
output = file.read()
|
|
|
|
if params['action'] == 'write':
|
|
try:
|
|
file = open(filepath, 'a' if params['append'] else 'w')
|
|
output = file.write(params['content'])
|
|
finally:
|
|
if file:
|
|
file.close()
|
|
|
|
if params['action'] == 'exists':
|
|
output = os.path.exists(filepath)
|
|
|
|
if params['action'] == 'list':
|
|
file = open(filepath)
|
|
output = file.read()
|
|
|
|
if params['action'] == 'remove':
|
|
if os.path.isfile(filepath):
|
|
output = os.remove(filepath)
|
|
elif os.path.isdir(filepath):
|
|
output = os.removedirs(filename)
|
|
|
|
if params['action'] == 'rename':
|
|
if os.path.exists(filepath):
|
|
output = shutil.move(filepath, params['target'])
|
|
|
|
if params['action'] == 'copy':
|
|
if os.path.exists(filepath):
|
|
output = shutil.copy2(filepath, params['target'])
|
|
|
|
if params['action'] == 'isfile':
|
|
output = os.path.isfile(filepath)
|
|
|
|
if params['action'] == 'isdir':
|
|
output = os.path.isdir(filepath)
|
|
|
|
|
|
|
|
|
|
case 'clipboard':
|
|
# 读文本
|
|
if params['action'] == 'wait_for_text':
|
|
output = self.clipboard.wait_for_text()
|
|
|
|
# 写文本
|
|
elif params['action'] == 'set_text':
|
|
self.clipboard.set_text(params['value'], -1)
|
|
|
|
# 写图片
|
|
elif params['action'] == 'set_image':
|
|
image = params['value']
|
|
# 前端传进来的值, 如果是路径的话, 直接读取
|
|
if type(image) == str:
|
|
image = Pixbuf.new_from_file(image)
|
|
else:
|
|
image = dict_to_pixbuf(image)
|
|
|
|
self.clipboard.set_image(image)
|
|
self.clipboard.store()
|
|
|
|
# 读图片
|
|
elif params['action'] == 'wait_for_image':
|
|
output = self.clipboard.wait_for_image()
|
|
output = pixbuf_to_dict(output, 'noname.png')
|
|
|
|
# 清除剪切板
|
|
elif params['action'] == 'clear':
|
|
self.clipboard.clear()
|
|
|
|
|
|
# 退出app
|
|
case 'quit':
|
|
all_quit(self)
|
|
|
|
# 读取图片, 返回图片像素数据
|
|
case 'image':
|
|
filename = params['value']
|
|
pixbuf = Pixbuf.new_from_file(filename)
|
|
output = pixbuf_to_dict(pixbuf, filename)
|
|
|
|
|
|
case 'monitor':
|
|
if params['action'] == 'get-all':
|
|
display = Gdk.Display.get_default()
|
|
monitor_num = display.get_n_monitors()
|
|
monitors = [display.get_monitor(i) for i in range(monitor_num)]
|
|
output = [get_monitor_info(m) for m in monitors]
|
|
|
|
elif params['action'] == 'get-primary':
|
|
display = Gdk.Display.get_default()
|
|
monitor = display.get_primary_monitor()
|
|
output = get_monitor_info(monitor)
|
|
|
|
case 'keybinder':
|
|
keymap = params.get('value')
|
|
shortcut_callback = params.get('shortcut_callback') or ''
|
|
|
|
if params['action'] == 'register':
|
|
# 绑定之前, 先解绑, 避免被重复绑定
|
|
Keybinder.unbind(keymap)
|
|
output = Keybinder.bind(
|
|
keymap,
|
|
lambda km : self.call_js(shortcut_callback)
|
|
)
|
|
|
|
elif params['action'] == 'unregister':
|
|
Keybinder.unbind(keymap)
|
|
output = True
|
|
|
|
elif params['action'] == 'supported':
|
|
output = Keybinder.supported()
|
|
|
|
|
|
case 'tray':
|
|
if params['action'] == 'create':
|
|
pass
|
|
|
|
elif params['action'] == 'remove':
|
|
pass
|
|
|
|
|
|
case 'window':
|
|
if params['action'] == 'fullscreen':
|
|
self.fullscreen()
|
|
|
|
elif params['action'] == 'unfullscreen':
|
|
self.unfullscreen()
|
|
|
|
elif params['action'] == 'maximize':
|
|
self.maximize()
|
|
|
|
elif params['action'] == 'unmaximize':
|
|
self.unmaximize()
|
|
|
|
elif params['action'] == 'set_title':
|
|
self.set_title(params['value'] or '')
|
|
|
|
elif params['action'] == 'resize':
|
|
self.resize(params['value'].get('width'), params['value'].get('height'))
|
|
|
|
elif params['action'] == 'set_opacity':
|
|
self.set_opacity(params['value'])
|
|
|
|
elif params['action'] == 'set_keep_above':
|
|
self.set_keep_above(params['value'])
|
|
|
|
elif params['action'] == 'set_keep_below':
|
|
self.set_keep_below(params['value'])
|
|
|
|
elif params['action'] == 'move':
|
|
self.move(params['value'].get('x'), params['value'].get('y'))
|
|
|
|
elif params['action'] == 'toggle_visible':
|
|
if self.is_visible():
|
|
self.hide()
|
|
else:
|
|
self.present()
|
|
|
|
elif params['action'] == 'hide':
|
|
self.hide()
|
|
|
|
elif params['action'] == 'show':
|
|
self.present()
|
|
|
|
elif params['action'] == 'is_visible':
|
|
output = self.is_visible()
|
|
|
|
case 'notify':
|
|
title = params.get('title')
|
|
summary = params.get('summary')
|
|
icon = params.get('icon')
|
|
progress = params.get('progress')
|
|
urgency = params.get('urgency')
|
|
|
|
notify = Notify.Notification.new(title, summary, icon)
|
|
|
|
if progress:
|
|
notify.set_hint('value', progress)
|
|
|
|
notify.set_urgency(urgency)
|
|
notify.add_action("click", "Click Me!", my_callback_func)
|
|
|
|
notify.show()
|
|
|
|
# 有回调则返回结果
|
|
if callback :
|
|
self.call_js(callback, output)
|
|
|
|
|
|
def all_quit(win):
|
|
print('朕要休息了~~~')
|
|
Notify.uninit()
|
|
Gtk.main_quit()
|
|
|
|
def my_callback_func(notification, action_name, user_data):
|
|
print("Button clicked! Action name: %s" % action_name)
|
|
notification.close()
|
|
|
|
|
|
|
|
win = WebKitWindow()
|
|
win.connect("destroy", all_quit)
|
|
win.show_all()
|
|
|
|
|
|
|
|
tray = win.create_tray()
|
|
|
|
Gtk.main() |
Python
60.1%
JavaScript
37.6%
HTML
2.3%