mt5-manager/deploy_service.py

'''
File: deploy_service.py
Description: Разворачивает сервис ServiceStateWriter по всем терминалам из
config.json - обновляет исходный код через git, компилирует его через
MetaEditor (ME) и раскладывает config/services.ini.
'''

__version__ = '0.1.0'

import argparse
import os
import shutil
import subprocess
import sys
from pathlib import Path

from config import Config
from mt5_control import MT5_Control

SCRIPT_DIR = Path(__file__).resolve().parent
SHARED_REL = Path('MQL5/Shared Projects/mt5-manager')
SERVICE_NAME = 'ServiceStateWriter'
SERVICE_MQ5 = f'{SERVICE_NAME}.mq5'
SERVICE_EX5_REL = Path('MQL5/Services/Shared Projects/mt5-manager') / f'{SERVICE_NAME}.ex5'
SERVICES_INI_REL = Path('static/services.ini')


def run(args: list, cwd: Path | None = None) -> subprocess.CompletedProcess:
    return subprocess.run(args, cwd=cwd, capture_output=True, text=True)


def git_output(args: list, cwd: Path) -> str:
    return run(['git', *args], cwd=cwd).stdout.strip()


def current_repo_info() -> tuple[str, Path, str]:
    '''Определяет ветку, корень и URL репозитория по текущему рабочему дереву,
    из которого запущен сам скрипт - это и есть источник для развёртывания.
    '''
    branch = git_output(['rev-parse', '--abbrev-ref', 'HEAD'], SCRIPT_DIR)
    repo_root = Path(git_output(['rev-parse', '--show-toplevel'], SCRIPT_DIR))
    remote_url = git_output(['config', '--get', 'remote.origin.url'], SCRIPT_DIR)
    return branch, repo_root, remote_url


def update_source(shared_dir: Path, repo_root: Path, repo_url: str, branch: str) -> bool:
    '''Обновляет исходный код сервиса в shared_dir до указанной ветки.

    Возвращает False, если что-то пошло не так и терминал стоит пропустить.
    Исходник в repo_root не трогаем - это и есть текущее рабочее дерево.
    '''
    if shared_dir.resolve() == repo_root.resolve():
        print(f'  {shared_dir}: это текущий репозиторий, git не трогаем')
        return True

    if not (shared_dir / '.git').exists():
        print(f'  Клонирую {repo_url} (ветка {branch})')
        shared_dir.parent.mkdir(parents=True, exist_ok=True)
        result = run(['git', 'clone', '--branch', branch, repo_url, str(shared_dir)])
        if result.returncode != 0:
            print(f'  ОШИБКА клонирования: {result.stderr.strip()}')
            return False
        return True

    # Незакоммиченные изменения не трогаем - мало ли, зачем они там оказались
    if git_output(['status', '--porcelain'], shared_dir):
        print(f'  {shared_dir}: есть незакоммиченные изменения, git pull пропущен')
        return True

    run(['git', 'fetch', 'origin'], shared_dir)

    checkout = run(['git', 'checkout', branch], shared_dir)
    if checkout.returncode != 0:
        print(f'  ОШИБКА checkout {branch}: {checkout.stderr.strip()}')
        return False

    pull = run(['git', 'pull', '--ff-only', 'origin', branch], shared_dir)
    if pull.returncode != 0:
        print(f'  ОШИБКА pull: {pull.stderr.strip()}')
        return False

    commit = git_output(['rev-parse', '--short', 'HEAD'], shared_dir)
    print(f'  Обновлено до {branch} ({commit})')
    return True


def read_log(log_path: Path) -> str:
    if not log_path.exists():
        return ''
    for encoding in ('utf-16', 'utf-8', 'cp1251'):
        try:
            return log_path.read_text(encoding=encoding)
        except (UnicodeError, UnicodeDecodeError):
            continue
    return log_path.read_text(errors='replace')


def compile_reference(reference_dir: Path) -> bool:
    '''Компилирует сервис один раз в reference_dir - терминале, в дереве
    которого физически лежит исходник (repo_root). Дальше получившийся
    .ex5 просто копируется по остальным терминалам.

    Компиляция отдельно на каждом терминале через CLI ME оказалась
    ненадёжной на практике: команда иногда завершается кодом 0, не создавая
    ни лога, ни нового .ex5 - похоже, headless-запуску не хватает
    интерактивной сессии рабочего стола. Поэтому здесь всего один вызов ME
    и явная проверка по времени изменения файла, а не только по коду
    завершения.
    '''
    metaeditor = reference_dir / 'MetaEditor64.exe'
    mq5_path = reference_dir / SHARED_REL / SERVICE_MQ5
    ex5_path = reference_dir / SERVICE_EX5_REL
    log_path = reference_dir / SHARED_REL / 'compile.log'

    if not metaeditor.exists():
        print(f'ОШИБКА: не найден {metaeditor}')
        return False
    if not mq5_path.exists():
        print(f'ОШИБКА: не найден {mq5_path}')
        return False

    before = ex5_path.stat().st_mtime if ex5_path.exists() else None

    # Если .ex5 уже новее исходника (например, его собрали вручную в
    # MetaEditor по F7), повторная компиляция не нужна
    if before is not None and before >= mq5_path.stat().st_mtime:
        print(f'Компиляция пропущена, {ex5_path.name} уже новее исходника')
        return True

    run([str(metaeditor), f'/compile:{mq5_path}', f'/log:{log_path}'])

    log_text = read_log(log_path)
    log_path.unlink(missing_ok=True)

    after = ex5_path.stat().st_mtime if ex5_path.exists() else None
    if after is None or after == before:
        print('ОШИБКА: MetaEditor завершился, но .ex5 не обновился.')
        print('Headless-компиляция в этом окружении ненадёжна - откройте '
              f'{mq5_path} в MetaEditor вручную и нажмите F7, затем '
              'запустите скрипт заново (шаг компиляции пропустится сам,'
              ' раз .ex5 уже свежий).')
        if log_text.strip():
            print('  ' + log_text.replace('\n', '\n  ').strip())
        return False

    print(f'Скомпилировано: {ex5_path}')
    return True


def deploy_ex5(terminal_dir: Path, ex5_path: Path) -> bool:
    dst = terminal_dir / SERVICE_EX5_REL
    dst.parent.mkdir(parents=True, exist_ok=True)
    shutil.copyfile(ex5_path, dst)
    print('  .ex5 скопирован')
    return True


def deploy_services_ini(terminal_dir: Path, src: Path) -> bool:
    if not src.exists():
        print(f'  ОШИБКА: не найден {src}')
        return False

    dst_dir = terminal_dir / 'config'
    dst_dir.mkdir(parents=True, exist_ok=True)
    shutil.copyfile(src, dst_dir / 'services.ini')
    print('  services.ini разложен в config')
    return True


def main():
    branch_default, repo_root, repo_url_default = current_repo_info()

    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument('--config-file', default='config.json',
                        help='Путь к файлу конфигурации менеджера')
    parser.add_argument('--branch', default=branch_default,
                        help=f'Ветка для развёртывания (сейчас: {branch_default})')
    parser.add_argument('--repo-url', default=repo_url_default,
                        help='URL репозитория для клонирования на новых терминалах')
    args = parser.parse_args()

    os.environ['MT5_MANAGER_CONFIG_FILE'] = args.config_file
    config = Config(_env_file=None)

    terminal_dirs = {name: Path(config.mt5_folder) / name for name in config.terminals}

    print('=== Обновление исходного кода ===')
    ok = True
    for name, terminal_dir in terminal_dirs.items():
        print(f'--- {name} ---')
        if not terminal_dir.exists():
            print(f'  ОШИБКА: папка терминала не найдена: {terminal_dir}')
            ok = False
            continue
        if not update_source(terminal_dir / SHARED_REL, repo_root, args.repo_url, args.branch):
            ok = False

    print('\n=== Компиляция ===')
    reference_dir = repo_root.parents[len(SHARED_REL.parts) - 1]
    if not (reference_dir / 'MetaEditor64.exe').exists():
        print(f'ОШИБКА: {reference_dir} не похож на папку терминала (нет MetaEditor64.exe), '
              'компилировать негде')
        sys.exit(1)

    compiled = compile_reference(reference_dir)
    ok = ok and compiled

    print('\n=== Раскладка .ex5 и services.ini ===')
    ex5_path = reference_dir / SERVICE_EX5_REL
    services_ini_src = repo_root / SERVICES_INI_REL
    for name, terminal_dir in terminal_dirs.items():
        if not terminal_dir.exists():
            continue
        print(f'--- {name} ---')
        if compiled and terminal_dir.resolve() != reference_dir.resolve():
            if not deploy_ex5(terminal_dir, ex5_path):
                ok = False
        if not deploy_services_ini(terminal_dir, services_ini_src):
            ok = False

    # Напоминание, какие терминалы сейчас запущены - у них в памяти всё ещё
    # старый .ex5, свежий файл на диске сам по себе не перезагрузится
    control = MT5_Control(config.terminals, config.mt5_folder, config.database, config.mt5_exe)
    running = [n for n, i in control.load_instances().items() if i.get('pid')]

    print()
    if not ok:
        print('Готово с ошибками, см. вывод выше.')
    if running:
        print('Запущены сейчас (нужен перезапуск, чтобы подхватить новый .ex5): '
              + ', '.join(running))
    if ok and not running:
        print('Готово.')

    sys.exit(0 if ok else 1)


if __name__ == '__main__':
    main()