sticker-convert 2.8.12__py3-none-any.whl → 2.17.0.0__py3-none-any.whl

sticker_convert/utils/process.py

@@ -0,0 +1,231 @@
+#!/usr/bin/env python3
+import os
+import platform
+import shutil
+import subprocess
+import time
+from getpass import getpass
+from io import BytesIO
+from typing import Callable, Optional, Union
+from zipfile import ZipFile
+
+import psutil
+import requests
+
+from sticker_convert.definitions import CONFIG_DIR, ROOT_DIR
+
+
+def killall(name: str) -> bool:
+    result = False
+
+    for proc in psutil.process_iter():  # type: ignore
+        if name in proc.name().lower():
+            proc.kill()
+            result = True
+
+    return result
+
+
+def find_pid_by_name(name: str) -> Optional[int]:
+    for proc in psutil.process_iter():  # type: ignore
+        if name in proc.name().lower():
+            return proc.pid
+
+    return None
+
+
+def check_admin() -> bool:
+    return True
+
+
+if platform.system() == "Windows":
+
+    def check_admin() -> bool:
+        username = os.getenv("username")
+        if username is None:
+            return False
+
+        s = subprocess.run(
+            ["net", "user", username],
+            capture_output=True,
+            text=True,
+        ).stdout
+
+        return True if "*Administrators" in s else False
+
+elif platform.system() == "Linux":
+
+    def check_admin() -> bool:
+        if shutil.which("sudo") is None:
+            return True
+
+        s = subprocess.run(
+            ["sudo", "-l"],
+            capture_output=True,
+            text=True,
+        ).stdout
+
+        return True if "may run the following commands" in s else False
+
+
+def _get_mem_win_procdump(
+    pid: Union[str, int],
+    pw_func: Optional[Callable[..., str]] = None,
+    is_wine: bool = False,
+) -> Optional[bytes]:
+    procdump_dir = CONFIG_DIR / "ProcDump"
+    procdump_bin = procdump_dir / "procdump.exe"
+    if procdump_bin.is_file() is False:
+        r = requests.get("https://download.sysinternals.com/files/Procdump.zip")
+        if r.ok is False:
+            return None
+        with ZipFile(BytesIO(r.content)) as zf:
+            zf.extractall(procdump_dir)
+
+    if procdump_bin.is_file() is False:
+        return None
+
+    if is_wine:
+        cmd = ["wine"]
+        dump_fpath = f"/tmp/memdump.{pid}.dmp"
+    else:
+        cmd = []
+        dump_fpath = os.path.expandvars(f"%temp%/memdump.{pid}.dmp")
+    cmd += [str(procdump_bin), "-accepteula", "-ma", str(pid), dump_fpath]
+
+    subprocess.run(cmd)
+
+    with open(dump_fpath, "rb") as f:
+        s = f.read()
+    os.remove(dump_fpath)
+
+    return s
+
+
+def _get_mem_win_builtin(
+    pid: Union[str, int], pw_func: Optional[Callable[..., str]] = None
+) -> Optional[bytes]:
+    from pathlib import WindowsPath
+
+    memdump_ps_path = str(WindowsPath(ROOT_DIR / "resources/memdump_windows.ps1"))
+    arglist = f'-NoProfile -ExecutionPolicy Bypass -File "{memdump_ps_path}" {pid}'
+    dump_fpath = os.path.expandvars(f"%temp%/memdump.{pid}.dmp")
+
+    cmd = [
+        "powershell",
+        "-NoProfile",
+        "-ExecutionPolicy",
+        "Bypass",
+        "-Command",
+        f"Start-Process -Verb RunAs powershell -ArgumentList '{arglist}'",
+    ]
+
+    subprocess.run(cmd, capture_output=True, text=True)
+
+    while True:
+        try:
+            with open(dump_fpath, "rb") as f:
+                s = f.read()
+            if len(s) != 0:
+                break
+            time.sleep(1)
+        except (FileNotFoundError, PermissionError):
+            pass
+
+    while True:
+        try:
+            os.remove(dump_fpath)
+            break
+        except PermissionError:
+            pass
+
+    return s
+
+
+def _get_mem_darwin(
+    pid: Union[str, int], pw_func: Optional[Callable[..., str]] = None
+) -> Optional[bytes]:
+    subprocess.run(
+        [
+            "lldb",
+            "--attach-pid",
+            str(pid),
+            "-o",
+            f"process save-core /tmp/memdump.{pid}.dmp",
+            "-o",
+            "quit",
+        ],
+        stdout=subprocess.DEVNULL,
+        stderr=subprocess.DEVNULL,
+    )
+
+    with open(f"/tmp/memdump.{pid}.dmp", "rb") as f:
+        s = f.read()
+
+    os.remove(f"/tmp/memdump.{pid}.dmp")
+
+    return s
+
+
+def _get_mem_linux(
+    pid: Union[str, int], pw_func: Optional[Callable[..., str]] = None
+) -> Optional[bytes]:
+    memdump_sh_path = (ROOT_DIR / "resources/memdump_linux.sh").as_posix()
+
+    s = subprocess.run(
+        [
+            memdump_sh_path,
+            str(pid),
+        ],
+        capture_output=True,
+    ).stdout
+
+    if len(s) > 1000:
+        pass
+    elif shutil.which("pkexec") and os.getenv("DISPLAY"):
+        s = subprocess.run(
+            [
+                "pkexec",
+                memdump_sh_path,
+                str(pid),
+            ],
+            capture_output=True,
+        ).stdout
+    else:
+        prompt = "Enter sudo password: "
+        if pw_func:
+            sudo_password = pw_func(prompt)
+        else:
+            sudo_password = getpass(prompt)
+        sudo_password_pipe = subprocess.Popen(
+            ("echo", sudo_password), stdout=subprocess.PIPE
+        )
+        s = subprocess.run(
+            [
+                "sudo",
+                "-S",
+                memdump_sh_path,
+                str(pid),
+            ],
+            capture_output=True,
+            stdin=sudo_password_pipe.stdout,
+        ).stdout
+
+    return s
+
+
+def get_mem(
+    pid: Union[str, int],
+    pw_func: Optional[Callable[..., str]] = None,
+    is_wine: bool = False,
+) -> Optional[bytes]:
+    if is_wine or platform.system() == "Windows":
+        dump = _get_mem_win_procdump(pid, pw_func, is_wine)
+        if dump is not None or is_wine:
+            return dump
+        else:
+            return _get_mem_win_builtin(pid, pw_func)
+    elif platform.system() == "Darwin":
+        return _get_mem_darwin(pid, pw_func)
+    else:
+        return _get_mem_linux(pid, pw_func)

sticker_convert/utils/translate.py

@@ -0,0 +1,108 @@
+#!/usr/bin/env python3
+import gettext
+import locale
+import platform
+import sys
+from json import JSONDecodeError
+from multiprocessing import current_process
+from typing import Any, Callable, Dict, cast
+
+from sticker_convert.definitions import CONFIG_DIR, ROOT_DIR, RUNTIME_STATE
+from sticker_convert.utils.files.json_manager import JsonManager
+
+LANG_DICT = {
+    "en_US": ("en_US",),
+    "ja_JP": ("ja_jp",),
+    "zh_CN": ("zh_chs", "zh_cn", "zh_sg"),
+    "zh_TW": ("zh_tw", "zh_hk", "zh_mo", "zh_cht"),
+}
+
+
+def get_default_lang() -> str:
+    lang = locale.getlocale()[0]
+    winlang = ""
+    if platform.system() == "Windows":
+        import ctypes
+
+        windll = ctypes.windll.kernel32  # type: ignore
+        winlang_id = cast(int, windll.GetUserDefaultUILanguage())  # type: ignore
+        winlang = cast(str, locale.windows_locale[winlang_id])  # type: ignore
+
+    for k, v in LANG_DICT.items():
+        for i in v:
+            if (
+                lang is not None and lang.lower().startswith(i)
+            ) or winlang.lower().startswith(i):
+                return k
+
+    return "en_US"
+
+
+def get_lang() -> str:
+    # Priority 1: From --lang CLI flag
+    is_lang = False
+    lang = None
+    for i in sys.argv:
+        if is_lang:
+            return i
+        if i == "--lang":
+            is_lang = True
+
+    # Priority 2: From config
+    settings_path = CONFIG_DIR / "config.json"
+    if settings_path.is_file():
+        try:
+            settings: Dict[Any, Any] = JsonManager.load_json(settings_path)
+            lang = settings.get("lang")
+            if lang:
+                if lang in LANG_DICT:
+                    return lang
+                elif lang != "auto":
+                    return "en_US"
+        except JSONDecodeError:
+            pass
+
+    # Priority 3: Default locale
+    lang = get_default_lang()
+    return lang
+
+
+LANG = RUNTIME_STATE.get("LANG", get_lang())
+RUNTIME_STATE["LANG"] = LANG
+
+
+def gettext_dummy(x: str) -> str:
+    return x
+
+
+def get_translator() -> Callable[[str], str]:
+    if RUNTIME_STATE.get("TRANS") is not None:
+        return RUNTIME_STATE["TRANS"]
+
+    try:
+        trans: Callable[[str], str] = gettext.translation(
+            "base", ROOT_DIR / "locales", languages=[RUNTIME_STATE["LANG"]]
+        ).gettext
+    except FileNotFoundError:
+        if current_process().name == "MainProcess":
+            print(f"Warning: Cannot find locales/{LANG}/LC_MESSAGES/base.mo")
+            print(
+                "Please generate by running scripts/update-locales.sh in root of repo"
+            )
+            print("Fallback to using English...")
+        trans = gettext_dummy
+    RUNTIME_STATE["TRANS"] = trans
+    return trans
+
+
+def I(x: str) -> str:  # noqa: E743
+    return get_translator()(x)
+
+
+SUPPORTED_LANG = {
+    I("Auto"): "auto",
+    "English": "en_US",
+    "繁體中文": "zh_TW",
+    "简体中文": "zh_CN",
+    "日本語": "ja_JP",
+}

sticker_convert/utils/url_detect.py

@@ -1,33 +1,40 @@
-#!/usr/bin/env python3
-import string
-from typing import Optional
-from urllib.parse import urlparse
-
-
-class UrlDetect:
-    @staticmethod
-    def detect(url: str) -> Optional[str]:
-        domain = urlparse(url).netloc
-
-        if domain == "signal.art":
-            return "signal"
-
-        if domain in ("telegram.me", "t.me"):
-            return "telegram"
-
-        if (
-            domain in ("store.line.me", "line.me")
-            or url.startswith("line://shop/detail/")
-            or (len(url) == 24 and all(c in string.hexdigits for c in url))
-        ):
-            return "line"
-
-        if domain in ("e.kakao.com", "emoticon.kakao.com") or url.startswith(
-            "kakaotalk://store/emoticon/"
-        ):
-            return "kakao"
-
-        if domain == "stickers.viber.com":
-            return "viber"
-
-        return None
+#!/usr/bin/env python3
+import string
+from typing import Optional
+from urllib.parse import urlparse
+
+
+class UrlDetect:
+    @staticmethod
+    def detect(url: str) -> Optional[str]:
+        domain = urlparse(url).netloc
+
+        if domain == "signal.art" or url.startswith("sgnl://addstickers/"):
+            return "signal"
+
+        if domain in ("telegram.me", "t.me"):
+            return "telegram"
+
+        if (
+            domain in ("store.line.me", "line.me")
+            or url.startswith("line://shop/detail/")
+            or (len(url) == 24 and all(c in string.hexdigits for c in url))
+        ):
+            return "line"
+
+        if domain in ("e.kakao.com", "emoticon.kakao.com"):
+            return "kakao"
+
+        if domain == "www.band.us":
+            return "band"
+
+        if domain == "ogqmarket.naver.com":
+            return "ogq"
+
+        if domain == "stickers.viber.com":
+            return "viber"
+
+        if domain == "discord.com":
+            return "discord"
+
+        return None

sticker_convert/version.py

@@ -1,3 +1,3 @@
 #!/usr/bin/env python3
 
-__version__ = "2.8.12"
+__version__ = "2.17.0.0"