eaf-camera/buffer.py at master · emacs-eaf/eaf-camera · GitHub

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
#!/usr/bin/env python3
# -*- coding: utf-8 -*-

# Copyright (C) 2018 Andy Stewart
#
# Author:     Andy Stewart <[email protected]>
# Maintainer: Andy Stewart <[email protected]>
#
# 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
from core.webengine import BrowserBuffer
from core.utils import get_emacs_var, message_to_emacs
import base64
import os

class AppBuffer(BrowserBuffer):
    def __init__(self, buffer_id, url, arguments):
        BrowserBuffer.__init__(self, buffer_id, url, arguments, False)

        self.download_path = get_emacs_var("eaf-webengine-download-path")

        # Set WebEngine permissions
        self.buffer_widget.web_page.setFeaturePermission(
            QUrl("file://"),
            self.buffer_widget.web_page.Feature.MediaVideoCapture,
            self.buffer_widget.web_page.PermissionPolicy.PermissionGrantedByUser
        )

        # Print WebEngine version
        from PyQt6.QtWebEngineCore import QWebEngineProfile
        print(f"WebEngine version: {QWebEngineProfile.defaultProfile().httpUserAgent()}")

        self.load_index_html(__file__)

    @QtCore.pyqtSlot(str)
    def save_screenshot(self, base64_string):
        path = os.path.expanduser(os.path.join(self.download_path, "screenshot.png"))
        self.save_base64_image(base64_string, path)

        message_to_emacs(f"Save screenshot to: {path}")

    def save_base64_image(self, base64_string, file_path):
        # Remove "data:image/png;base64," if it's included in the string
        if base64_string.startswith("data:image/png;base64,"):
            base64_string = base64_string[len("data:image/png;base64,"):]

        # Decode the Base64 string to bytes
        image_data = base64.b64decode(base64_string)

        with open(file_path, "wb") as f:
            f.write(image_data)

    @QtCore.pyqtSlot(result=bool)
    def request_camera_permission(self):
        """Request camera permission, may trigger system permission dialog"""
        self.buffer_widget.reload()
        return True