105 lines
3.1 KiB
Python
105 lines
3.1 KiB
Python
import os
|
|
import platform
|
|
import re
|
|
import subprocess
|
|
|
|
from pathlib import Path
|
|
from typing import Iterator
|
|
from PyQt6 import QtWidgets
|
|
|
|
from gotify_tray import gotify
|
|
from gotify_tray.database import Downloader
|
|
|
|
|
|
def verify_server(force_new: bool = False, enable_import: bool = True) -> bool:
|
|
from gotify_tray.gui import ServerInfoDialog
|
|
from gotify_tray.database import Settings
|
|
|
|
settings = Settings("gotify-tray")
|
|
|
|
url = settings.value("Server/url", type=str)
|
|
token = settings.value("Server/client_token", type=str)
|
|
|
|
if not url or not token or force_new:
|
|
dialog = ServerInfoDialog(url, token, enable_import)
|
|
if dialog.exec():
|
|
settings.setValue("Server/url", dialog.line_url.text())
|
|
settings.setValue("Server/client_token", dialog.line_token.text())
|
|
return True
|
|
else:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
|
|
def process_messages(messages: list[gotify.GotifyMessageModel]) -> Iterator[gotify.GotifyMessageModel]:
|
|
downloader = Downloader()
|
|
for message in messages:
|
|
if image_url := get_image(message.message):
|
|
downloader.get_filename(image_url)
|
|
yield message
|
|
|
|
|
|
def convert_links(text):
|
|
_link = re.compile(
|
|
r'(?:(https://|http://)|(www\.))(\S+\b/?)([!"#$%&\'()*+,\-./:;<=>?@[\\\]^_`{|}~]*)(\s|$)',
|
|
re.I,
|
|
)
|
|
|
|
def replace(match):
|
|
groups = match.groups()
|
|
protocol = groups[0] or "" # may be None
|
|
www_lead = groups[1] or "" # may be None
|
|
return '<a href="http://{1}{2}" rel="nofollow">{0}{1}{2}</a>{3}{4}'.format(
|
|
protocol, www_lead, *groups[2:]
|
|
)
|
|
|
|
return _link.sub(replace, text)
|
|
|
|
|
|
def get_image(s: str) -> str | None:
|
|
"""If `s` contains only an image URL, this function returns that URL.
|
|
This function also extracts a URL in the `` markdown image format.
|
|
"""
|
|
s = s.strip()
|
|
|
|
# Return True if 's' is a url and has an image extension
|
|
RE = r'(?:(https://|http://)|(www\.))(\S+\b/?)([!"#$%&\'()*+,\-./:;<=>?@[\\\]^_`{|}~]*).(jpg|jpeg|png|bmp|gif)(\s|$)'
|
|
if re.compile(RE, re.I).fullmatch(s) is not None:
|
|
return s
|
|
|
|
# Return True if 's' has the markdown image format
|
|
RE = r'!\[[^\]]*\]\((.*?)\s*("(?:.*[^"])")?\s*\)'
|
|
if re.compile(RE, re.I).fullmatch(s) is not None:
|
|
return re.compile(RE, re.I).findall(s)[0][0]
|
|
|
|
return None
|
|
|
|
|
|
def get_abs_path(s) -> str:
|
|
h = Path(__file__).parent.parent
|
|
p = Path(s)
|
|
return os.path.join(h, p).replace("\\", "/")
|
|
|
|
|
|
def open_file(filename: str):
|
|
if platform.system() == "Linux":
|
|
subprocess.call(["xdg-open", filename])
|
|
elif platform.system() == "Windows":
|
|
os.startfile(filename)
|
|
elif platform.system() == "Darwin":
|
|
subprocess.call(["open", filename])
|
|
|
|
|
|
def get_icon(name: str) -> str:
|
|
if platform.system() == "Darwin":
|
|
name += "-macos"
|
|
|
|
return get_abs_path(f"gotify_tray/gui/images/{name}.png")
|
|
|
|
def update_widget_property(widget: QtWidgets.QWidget, property: str, value: str):
|
|
widget.setProperty(property, value)
|
|
widget.style().unpolish(widget)
|
|
widget.style().polish(widget)
|
|
widget.update()
|