# encoding:utf-8
# export_plc_src.py
# Экспорт POU / GVL / DUT из открытого проекта CODESYS в текстовые файлы
# для хранения в git и работы с Claude Code.
#
# Запуск:
#   1) Внутри CODESYS: Tools -> Scripting -> Execute Script File -> выбрать этот файл
#   2) Из командной строки (без открытия GUI):
#      "C:\Program Files (x86)\CODESYS 3.5.17.30\CODESYS\Common\CODESYS.exe" ^
#        --Profile="CODESYS V3.5 SP17 Patch 3" --runscript="export_plc_src.py" ^
#        --project="D:\path\to\ТвойПроект.project"
#
# Проверено для установки: CODESYS V3.5 SP17 Patch 3 (32-bit)
#
# По умолчанию папка plc_src создаётся автоматически рядом с файлом .project.
# Если авто-определение не сработает, впиши путь вручную в MANUAL_OUTPUT_ROOT ниже.

from __future__ import print_function
import os
import codecs
import re

PRAGMA_RE = re.compile(r"^\s*\{[^}]*\}\s*")


def strip_leading_pragmas(text):
    """Убирает ведущие блоки вида {attribute 'qualified_only'} перед проверкой ключевого слова."""
    while True:
        new_text = PRAGMA_RE.sub("", text, 1)
        if new_text == text:
            return text
        text = new_text


def get_project_file_path(proj):
    """Пытается получить полный путь к .project файлу разными способами,
    так как имя атрибута может отличаться между версиями CODESYS."""
    for attr in ("path", "get_path", "project_path", "full_path"):
        try:
            val = getattr(proj, attr)
            val = val() if callable(val) else val
            if val:
                return val
        except Exception:
            pass
    # Фолбэк: вытащить путь из текстового представления объекта,
    # например "Project(Project=0, stPath=C:\...\*.project)"
    try:
        s = str(proj)
        m = re.search(r"stPath=(.*?)\)", s)
        if m:
            return m.group(1)
    except Exception:
        pass
    return None


def safe_name(obj):
    try:
        return obj.get_name(False)
    except Exception:
        return "unnamed"


def get_declaration_text(obj):
    try:
        if hasattr(obj, "textual_declaration") and obj.textual_declaration is not None:
            return obj.textual_declaration.text
    except Exception:
        return None
    return None


def get_text(obj, declaration_text):
    """Собирает декларацию + реализацию (если есть) в единый текст."""
    text = declaration_text or ""
    try:
        if hasattr(obj, "has_textual_implementation") and obj.has_textual_implementation:
            text += "\r\n(*----- IMPLEMENTATION -----*)\r\n"
            text += obj.textual_implementation.text
    except Exception as e:
        text += "(* ошибка чтения implementation: %s *)\r\n" % e
    return text


def classify(declaration_text):
    """Определяет тип объекта по первому ключевому слову декларации."""
    if not declaration_text:
        return None
    stripped = strip_leading_pragmas(declaration_text.strip()).strip().upper()
    if stripped.startswith(("PROGRAM", "FUNCTION_BLOCK", "FUNCTION", "INTERFACE", "METHOD", "ACTION", "PROPERTY")):
        return "pou"
    if stripped.startswith("VAR_GLOBAL"):
        return "gvl"
    if stripped.startswith("TYPE"):
        return "dut"
    return None


def write_file(folder, name, text):
    path = os.path.join(folder, name + ".st")
    with codecs.open(path, "w", encoding="utf-8") as f:
        f.write(text)
    print("  -> %s" % path)


def sanitize_folder_name(name):
    """Убирает символы, недопустимые в имени папки Windows."""
    invalid = '\\/:*?"<>|'
    for ch in invalid:
        name = name.replace(ch, "_")
    name = name.strip()
    return name if name else "device"


def make_dirs(base):
    d = {
        "pou": os.path.join(base, "POU"),
        "gvl": os.path.join(base, "GVL"),
        "dut": os.path.join(base, "DUT"),
    }
    for p in d.values():
        if not os.path.exists(p):
            os.makedirs(p)
    return d


def is_device_node(obj):
    try:
        return bool(getattr(obj, "is_device", False))
    except Exception:
        return False


def walk(node, dirs, exported):
    for child in node.get_children(False):
        name = safe_name(child)
        decl = get_declaration_text(child)
        kind = classify(decl)

        if kind == "pou":
            print("POU: %s" % name)
            write_file(dirs["pou"], name, get_text(child, decl))
            exported["pou"] += 1
        elif kind == "gvl":
            print("GVL: %s" % name)
            write_file(dirs["gvl"], name, get_text(child, decl))
            exported["gvl"] += 1
        elif kind == "dut":
            print("DUT: %s" % name)
            write_file(dirs["dut"], name, get_text(child, decl))
            exported["dut"] += 1
        else:
            exported["skipped"] += 1

        # рекурсивно обходим детей (для FB с методами, папок и т.д.)
        walk(child, dirs, exported)


# ==== НАСТРОЙКА ====
# Оставь пустой строкой "" чтобы папка plc_src создавалась автоматически
# рядом с файлом .project. Укажи путь вручную только если авто-определение
# не сработает (см. вывод скрипта при запуске).
MANUAL_OUTPUT_ROOT = r""
# =====================================================

proj = projects.primary

if proj is None:
    print("ОШИБКА: нет открытого проекта. Открой .project перед запуском скрипта.")
else:
    if MANUAL_OUTPUT_ROOT:
        OUTPUT_ROOT = MANUAL_OUTPUT_ROOT
        print("Папка вывода задана вручную: %s" % OUTPUT_ROOT)
    else:
        project_file = get_project_file_path(proj)
        if project_file:
            project_dir = os.path.dirname(project_file)
            OUTPUT_ROOT = os.path.join(project_dir, "plc_src")
            print("Проект найден: %s" % project_file)
            print("Папка вывода определена автоматически: %s" % OUTPUT_ROOT)
        else:
            OUTPUT_ROOT = None
            print("ОШИБКА: не удалось определить путь к .project автоматически.")
            print("Задай путь вручную в переменной MANUAL_OUTPUT_ROOT и запусти снова.")

    if OUTPUT_ROOT:
        exported = {"pou": 0, "gvl": 0, "dut": 0, "skipped": 0}

        print("--- Экспорт в %s ---" % OUTPUT_ROOT)

        top_children = proj.get_children(False)
        device_nodes = [c for c in top_children if is_device_node(c)]

        if device_nodes:
            for device in device_nodes:
                dev_name = sanitize_folder_name(safe_name(device))
                print("=== Устройство: %s ===" % dev_name)
                dev_dirs = make_dirs(os.path.join(OUTPUT_ROOT, dev_name))
                walk(device, dev_dirs, exported)

            # на случай текстовых объектов вне дерева устройств (редко, но на всякий случай)
            other_dirs = make_dirs(os.path.join(OUTPUT_ROOT, "_common"))
            for child in top_children:
                if not is_device_node(child):
                    walk(child, other_dirs, exported)
        else:
            # устройств не нашлось (или is_device не сработал) -- работаем как раньше, без разбивки
            print("Устройства верхнего уровня не найдены, экспорт без разбивки по папкам.")
            dirs = make_dirs(OUTPUT_ROOT)
            walk(proj, dirs, exported)

        print("--- Готово: POU=%d, GVL=%d, DUT=%d, пропущено=%d ---" % (
            exported["pou"], exported["gvl"], exported["dut"], exported["skipped"]
        ))