98 lines
3.8 KiB
Python
98 lines
3.8 KiB
Python
#!/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 PyQt5 import QtCore
|
|
from PyQt5.QtCore import QUrl, QFileSystemWatcher, QTimer
|
|
from retrying import retry
|
|
from core.webengine import BrowserBuffer
|
|
from core.utils import get_free_port, message_to_emacs, eval_in_emacs, get_emacs_var
|
|
from urllib.error import URLError
|
|
from urllib.request import urlopen
|
|
from urllib.parse import urlencode
|
|
import os
|
|
import platform
|
|
import subprocess
|
|
import tempfile
|
|
|
|
class AppBuffer(BrowserBuffer):
|
|
|
|
def __init__(self, buffer_id, url, arguments):
|
|
BrowserBuffer.__init__(self, buffer_id, url, arguments, False)
|
|
|
|
self.url = url
|
|
self.preview_file = tempfile.mkstemp(prefix='eaf-', suffix='.html', text=True)[1]
|
|
self.render_js = os.path.join(os.path.dirname(__file__), "render.js")
|
|
self.server_port = get_free_port()
|
|
self.dark_mode = "false"
|
|
dark_mode = get_emacs_var("eaf-markdown-dark-mode")
|
|
if (dark_mode == "force" or \
|
|
dark_mode == True or \
|
|
(dark_mode == "follow" and self.theme_mode == "dark")):
|
|
self.dark_mode = "true"
|
|
|
|
self.buffer_widget.dark_mode_js = open(os.path.join(os.path.dirname(__file__),
|
|
"node_modules",
|
|
"darkreader",
|
|
"darkreader.js")).read()
|
|
|
|
self.draw_progressbar = True
|
|
|
|
# Check puml code and Java is installed.
|
|
with open(url) as f:
|
|
import shutil
|
|
if "```puml" in f.read() and shutil.which("java") is None:
|
|
message_to_emacs("Have PlantUML code in file '{}', you need to install Java to preview normally.".format(os.path.basename(url)))
|
|
|
|
self.run_render_server()
|
|
self.render()
|
|
|
|
self.file_watcher = QFileSystemWatcher()
|
|
self.file_watcher.fileChanged.connect(self.on_file_changed)
|
|
self.file_watcher.addPath(url)
|
|
|
|
def run_render_server(self):
|
|
args = ["node", self.render_js, str(self.server_port)]
|
|
subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, shell=False)
|
|
|
|
def on_file_changed(self, *args):
|
|
self.render()
|
|
|
|
def retry_if_connection_refused(ex):
|
|
return isinstance(ex, URLError) and isinstance(ex.reason, ConnectionRefusedError)
|
|
|
|
@retry(wait_fixed=500, stop_max_attempt_number=10, retry_on_exception=retry_if_connection_refused)
|
|
def render(self):
|
|
params = {
|
|
"input_file": self.url,
|
|
"output_file": self.preview_file,
|
|
"dark_mode": self.dark_mode
|
|
}
|
|
url = 'http://localhost:{}?{}'.format(self.server_port, urlencode(params))
|
|
with urlopen(url) as f:
|
|
resp = f.read().decode("utf-8")
|
|
if resp == "ok":
|
|
self.buffer_widget.load(QUrl.fromLocalFile(self.preview_file))
|
|
if platform.system() == "Windows":
|
|
eval_in_emacs('eaf-activate-emacs-window', [])
|
|
else:
|
|
message_to_emacs("preview failed: {}".format(resp))
|