add ezTrans translator

# -*- coding: utf-8 -*-

from ctypes import c_char_p, c_int, c_wchar_p

from ctypes.wintypes import BOOL

from typing import Literal

from msl.loadlib import Server32

import re

ENGINE_TYPES = Literal['J2K', 'K2J']

class MyServer(Server32):

    def __init__(self, host, port, engine_path, engine_type: ENGINE_TYPES, dat_path):

        super(MyServer, self).__init__(engine_path, 'windll', host, port)

        self.engine = TransEngine(self.lib, engine_type, dat_path)

    def translate(self, src_text: str | list) -> str | list:

        def work(t):

            return encode_text(self.engine.translate(decode_text(t)))

        if type(src_text) == list:

            if len(src_text) == 1:

                return [work(src_text[0])]

            return self.translate(src_text[:-1]) + [work(src_text[-1])]

        elif type(src_text) == str:

            return work(src_text)

class TransEngine:

    def __init__(self, engine, engine_type: ENGINE_TYPES, dat_path):

        self.start = getattr(engine, f"{engine_type}_InitializeEx")

        self.start.argtypes = [c_char_p, c_char_p]

        self.start.restype = BOOL

        self.trans = getattr(engine, f"{engine_type}_TranslateMMNTW")

        self.trans.argtypes = [c_int, c_wchar_p]

        self.trans.restype = c_wchar_p

        self.start_obj = self.start(b"CSUSER123455", dat_path.encode('utf-8'))

    def translate(self, src_text):

        trans_obj = self.trans(0, src_text)

        return trans_obj

def decode_text(txt):

    chars = "↔◁◀▷▶♤♠♡♥♧♣⊙◈▣◐◑▒▤▥▨▧▦▩♨☏☎☜☞↕↗↙↖↘♩♬㉿㈜㏇™㏂㏘＂＇∼ˇ˘˝¡˚˙˛¿ː∏￦℉€㎕㎖㎗ℓ㎘㎣㎤㎥㎦㎙㎚㎛㎟㎠㎢㏊㎍㏏㎈㎉㏈㎧㎨㎰㎱㎲㎳㎴㎵㎶㎷㎸㎀㎁㎂㎃㎄㎺㎻㎼㎽㎾㎿㎐㎑㎒㎓㎔Ω㏀㏁㎊㎋㎌㏖㏅㎭㎮㎯㏛㎩㎪㎫㎬㏝㏐㏓㏃㏉㏜㏆┒┑┚┙┖┕┎┍┞┟┡┢┦┧┪┭┮┵┶┹┺┽┾╀╁╃╄╅╆╇╈╉╊┱┲ⅰⅱⅲⅳⅴⅵⅶⅷⅸⅹ½⅓⅔¼¾⅛⅜⅝⅞ⁿ₁₂₃₄ŊđĦIJĿŁŒŦħıijĸŀłœŧŋʼn㉠㉡㉢㉣㉤㉥㉦㉧㉨㉩㉪㉫㉬㉭㉮㉯㉰㉱㉲㉳㉴㉵㉶㉷㉸㉹㉺㉻㈀㈁㈂㈃㈄㈅㈆㈇㈈㈉㈊㈋㈌㈍㈎㈏㈐㈑㈒㈓㈔㈕㈖㈗㈘㈙㈚㈛ⓐⓑⓒⓓⓔⓕⓖⓗⓘⓙⓚⓛⓜⓝⓞⓟⓠⓡⓢⓣⓤⓥⓦⓧⓨⓩ①②③④⑤⑥⑦⑧⑨⑩⑪⑫⑬⑭⑮⒜⒝⒞⒟⒠⒡⒢⒣⒤⒥⒦⒧⒨⒩⒪⒫⒬⒭⒮⒯⒰⒱⒲⒳⒴⒵⑴⑵⑶⑷⑸⑹⑺⑻⑼⑽⑾⑿⒀⒁⒂"

    for c in chars:

        if c in txt:

            txt = txt.replace(c, "\\u" + str(hex(ord(c)))[2:])

    return txt

def encode_text(txt):

    return re.sub(r'(?i)(?<!\\)(?:\\\\)*\\u([0-9a-f]{4})', lambda m: chr(int(m.group(1), 16)), txt)

from pathlib import Path

import sys

from .base import *

from typing import Literal

from msl.loadlib import Client64

class MyClient(Client64):

    def __init__(self, engine_path, engine_type: Literal['J2K', 'K2J'], dat_path):

        super(MyClient, self).__init__(module32=str(Path(__file__).parent) + '\module_eztrans32.py',

                                       engine_path=engine_path,

                                       engine_type=engine_type,

                                       dat_path=dat_path)

    def translate(self, src_text: str | list):

        return self.request32('translate', src_text)

@register_translator('ezTrans')

class ezTransTranslator(BaseTranslator):

    concate_text = False

    params: Dict = {

        'path_dat': r"C:\Program Files (x86)\ChangShinSoft\ezTrans XP\Dat",

        'path_j2k': r"C:\Program Files (x86)\ChangShinSoft\ezTrans XP\J2KEngine.dll",

        'path_k2j(Optional)': r"C:\Program Files (x86)\ChangShinSoft\ezTrans XP\ehnd-kor.dll"

    }

    def _setup_translator(self):

        self.lang_map['日本語'] = 'j'

        self.lang_map['한국어'] = 'k'

        self.j2k_engine = MyClient(self.params['path_j2k'], "J2K", self.params['path_dat'])

        if self.params['path_k2j(Optional)']:

            self.k2j_engine = MyClient(self.params['path_k2j(Optional)'], "K2J", self.params['path_dat'])

    def _translate(self, src_list: List[str]) -> List[str]:

        source = self.lang_map[self.lang_source]

        target = self.lang_map[self.lang_target]

        if source != target:

            engine: MyClient = getattr(self, f"{source}2{target}_engine")

            return engine.translate(src_list)

        else:

            return src_list

    def updateParam(self, param_key: str, param_content):

        pass

    @property

    def supported_tgt_list(self) -> List[str]:

        return ['한국어', '日本語'] if self.params['path_k2j(Optional)'] else ['한국어']

    @property

    def supported_src_list(self) -> List[str]:

        return ['한국어', '日本語'] if self.params['path_k2j(Optional)'] else ['日本語']

pyobjc-framework-coreml; sys_platform == 'darwin'

pyobjc-framework-quartz; sys_platform == 'darwin'

pyobjc-framework-vision; sys_platform == 'darwin'

msl-loadlib