python3-webengine-gtk3/usr/lib/python3/dist-packages/webengine/gtk3/_webengine.py

#!/usr/bin/env python3
# @author yutent<yutent.io@gmail.com>
# @date 2023/08/08 14:07:26


import gi, os, json, shutil, hashlib

gi.require_version("WebKit2", "4.1")


from gi.repository import GObject, Gtk, Gdk, WebKit2, GLib, Gio, GdkPixbuf


# 优先尝试使用指示器, 没有再使用 Gtk.StatusIcon
try:
  # gir1.2-ayatanaappindicator3-0.1
  gi.require_version('AyatanaAppIndicator3', '0.1')
  from gi.repository import AyatanaAppIndicator3 as AppIndicator3
except:
  AppIndicator3 = None

try:
  # gir1.2-keybinder-3.0
  gi.require_version("Keybinder", "3.0")
  from gi.repository import Keybinder

  # 初始化 Keybinder
  Keybinder.init()
except:
  Keybinder = None





from ._settings import create_setting
from ._protocal import create_protocal
from ._notify import create_notify
from ._inject import Inject
from ._utils import get_monitor_info, pixbuf_to_dict, dict_to_pixbuf



env = {
  "HOME_DIR": os.getenv('HOME'),
  "CONFIG_DIR": os.path.join(os.getenv('HOME'), '.config'),
  "CACHE_DIR": os.path.join(os.getenv('HOME'), '.cache')
}


def noop():
  pass


class WebEngine(WebKit2.WebView):

  __gsignals__ = {
    'quit': (GObject.SignalFlags.RUN_FIRST, None, ())
  }

  root = None
  window = None

  def __init__(self, window):

    WebKit2.WebView.__init__(self)

    self.window = window

    self.clipboard = Gtk.Clipboard.get(Gdk.SELECTION_CLIPBOARD)
    self.display = Gdk.Display.get_default()
    
    self.use(create_notify())

    setting = create_setting()
    self.use(setting)


    im = self.get_input_method_context()
    im.set_enable_preedit(True)

    # 解决输入法候选框跟随问题
    im.connect('preedit-started', self.on_preedit_changed)
    im.connect('preedit-changed', self.on_preedit_changed)
    im.connect('preedit-finished', self.on_preedit_changed)

    Inject(self).connect(self.called_by_js)


  def set_root(self, root):
    self.root = root
    return self.use(create_protocal(root))
  
  def use(self, middle_ware = noop, extra = None):
    middle_ware(self, extra)
    return self


  def load(self, url = '/index.html'):
    if self.root is None:
      raise EnvironmentError('web root dir not set!')
    else:
      self.load_uri(f"app://{url}")


  def on_preedit_changed(self, im):
    seat = self.display.get_default_seat()
    p = seat.get_pointer().get_position() # 光标位置
    x, y = self.get_position() # 窗口位置

    im.notify_focus_in()
    im.notify_cursor_area(p.x - x, p.y - y, 0, 0) # 修正输入法跟随


  def call_js(self, method, data = None, err = None):
    if err is not None:
      err = str(err)
    scripts = 'native.$emit("' + method + '", ' + json.dumps(err) + ', ' + json.dumps(data) + ')'
    self.evaluate_javascript(scripts, -1)

  
  def called_by_js(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
    _error = None


    match event:
      case 'init':
        output = env

      case 'fs':
        filepath = params.get('filepath')

        if params['action'] == 'access':
          try:
            with open(filepath, params.get('mode')) as file:
              output = True
          except Exception as err:
            output = False

        elif params['action'] == 'read':
          try:
            with open(filepath, params.get('mode')) as file:
              output = file.read()
              if params.get('mode').find('b') > -1:
                output = list(output)
          except Exception as err:
            _error = err

        elif params['action'] == 'write':
          # 调整以支持二进制数据写入
          try:
            with open(filepath, params.get('mode')) as file:
              buff = params['content']

              if params.get('mode').find('b') > -1:
                buff = bytes(buff)

              output = file.write(buff)
          except Exception as err:
            _error = err

        elif params['action'] == 'exists':
          output = os.path.exists(filepath)

        elif params['action'] == 'list':
          with os.scandir(filepath) as entries:
            output = [{
              "name": it.name,
              "path": os.path.join(filepath, it.name),
              "is_dir": it.is_dir(),
              "size": it.stat().st_size,
              "atime": int(it.stat().st_atime),
              "mtime": int(it.stat().st_mtime),
            } for it in entries]

        elif params['action'] == 'remove':
          if os.path.isfile(filepath):
            output = os.remove(filepath)
          elif os.path.isdir(filepath):
            output = os.removedirs(filename)

        elif params['action'] == 'rename':
          if os.path.exists(filepath):
            output = shutil.move(filepath, params['target'])

        elif params['action'] == 'copy':
          if os.path.exists(filepath):
            output = shutil.copy2(filepath, params['target'])

        elif params['action'] == 'isfile':
          output = os.path.isfile(filepath)
   
        elif 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 = GdkPixbuf.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':
        self.emit('quit')

      # 读取图片, 返回图片像素数据
      case 'image':
        filename = params['value']
        pixbuf = GdkPixbuf.Pixbuf.new_from_file(filename)
        output = pixbuf_to_dict(pixbuf, filename)

      case 'monitor':
        if params['action'] == 'get-all':
          monitor_num = self.display.get_n_monitors()
          monitors = [self.display.get_monitor(i) for i in range(monitor_num)]
          output = [get_monitor_info(m) for m in monitors]
          
        elif params['action'] == 'get-primary':
          monitor = self.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':
          # 绑定之前, 先解绑, 避免被重复绑定
          if Keybinder:
            Keybinder.unbind(keymap)
            output = Keybinder.bind(
              keymap,
              lambda km : self.call_js(shortcut_callback)
            )

        elif params['action'] == 'unregister':
          if Keybinder:
            Keybinder.unbind(keymap)
            output = True

        elif params['action'] == 'supported':
          if Keybinder:
            output = Keybinder.supported()
          else:
            output = False


      case 'tray':
        if params['action'] == 'create':
          pass

        elif params['action'] == 'remove':
          pass

      case 'window':
        if params['action'] == 'fullscreen':
          self.window.fullscreen()

        elif params['action'] == 'unfullscreen':
          self.window.unfullscreen()

        elif params['action'] == 'maximize':
          self.window.maximize()

        elif params['action'] == 'unmaximize':
          self.window.unmaximize()

        elif params['action'] == 'set_title':
          self.window.set_title(params['value'] or '')

        elif params['action'] == 'resize':
          self.window.resize(params['value'].get('width'), params['value'].get('height'))

        elif params['action'] == 'set_opacity':
          self.window.set_opacity(params['value'])

        elif params['action'] == 'set_keep_above':
          self.window.set_keep_above(params['value'])

        elif params['action'] == 'set_keep_below':
          self.window.set_keep_below(params['value'])

        elif params['action'] == 'move':
          self.window.move(params['value'].get('x'), params['value'].get('y'))

        elif params['action'] == 'toggle_visible':
          if self.is_visible():
            self.window.hide()
          else:
            self.window.present()

        elif params['action'] == 'hide':
          self.window.hide()

        elif params['action'] == 'show':
          self.window.present()

        elif params['action'] == 'is_visible':
          output = self.window.is_visible()


      case 'notify':
        title = params.get('title')
        summary = params.get('summary')
        icon = params.get('icon')
        progress = params.get('progress')
        urgency = params.get('urgency')

        self.notify.create(title, summary, icon, progress, urgency, params.get('callback'))

      case 'md5':
        output = hashlib.md5(str(params.get('value'))).hexdigest()

    # 有回调则返回结果
    if callback:
      self.call_js(callback, output, _error)