Org-roam. Экспорт в Hugo

#!/usr/bin/env python3
"""
Преобразование org-свойств для многоязычных page bundle.
Ищет блоки :PROPERTIES:, содержащие EXPORT_HUGO_SECTION и EXPORT_FILE_NAME,
и преобразует их в формат page bundle с языковыми суффиксами.
Все ключи свойств приводятся к нижнему регистру.
"""

import os
import re
import sys
from pathlib import Path

# ====== НАСТРОЙКИ ======
root_dir = "/путь/к/вашим/org-файлам"   # <--- ИЗМЕНИТЕ ЭТО
dry_run = True                          # При True только просмотр, без записи
# =======================

# Регулярка для поиска блоков свойств
prop_block_re = re.compile(
    r'^(\s*):PROPERTIES:\s*\n(.*?)^\s*:END:\s*$',
    re.MULTILINE | re.DOTALL | re.IGNORECASE
)

# Регулярка для строк свойств
prop_line_re = re.compile(r'^\s*:([^:]+?):\s*(.*?)\s*$', re.MULTILINE)

def transform_properties(prop_text):
    """
    Преобразует текст внутри блока свойств.
    Возвращает новый текст или исходный, если преобразование не требуется.
    """
    # Собираем все текущие свойства, ключи приводим к нижнему регистру
    props = {}
    for match in prop_line_re.finditer(prop_text):
        key = match.group(1).strip()
        value = match.group(2).strip()
        props[key.lower()] = value

    # Проверяем наличие необходимых свойств
    if 'export_hugo_section' not in props or 'export_file_name' not in props:
        return prop_text

    section = props['export_hugo_section']
    old_filename = props['export_file_name']

    # Определяем язык по префиксу ru/ или en/
    lang = None
    if section.lower().startswith('ru/'):
        lang = 'ru'
        new_section = section[3:]   # убираем 'ru/'
    elif section.lower().startswith('en/'):
        lang = 'en'
        new_section = section[3:]   # убираем 'en/'
    else:
        # Нет языкового префикса – пропускаем
        return prop_text

    # Обновляем свойства
    props['export_hugo_section'] = new_section
    props['export_hugo_bundle'] = old_filename
    props['export_file_name'] = f"index.{lang}.md"
    props['export_language'] = lang

    # Сортируем ключи для стабильности порядка
    sorted_keys = sorted(props.keys())

    # Формируем текст свойств
    lines = [f":{key}: {props[key]}" for key in sorted_keys]
    return "\n".join(lines)

def process_file(filepath):
    """Обрабатывает один .org файл."""
    with open(filepath, 'r', encoding='utf-8') as f:
        content = f.read()

    new_content = content
    modified = False

    # Ищем все блоки свойств и обрабатываем каждый
    for match in prop_block_re.finditer(content):
        old_block = match.group(0)
        indent = match.group(1)      # отступ перед :PROPERTIES:
        prop_text = match.group(2)

        new_prop_text = transform_properties(prop_text)
        if new_prop_text != prop_text:
            # Восстанавливаем блок с тем же отступом
            new_block = f"{indent}:PROPERTIES:\n{new_prop_text}\n{indent}:END:"
            new_content = new_content.replace(old_block, new_block)
            modified = True
            # Показываем фрагмент старого блока для информации
            print(f"    Будет изменён блок: {old_block[:80]}...")

    if modified:
        if dry_run:
            print(f"[DRY RUN] Будет изменён: {filepath}")
        else:
            # Создаём резервную копию с расширением .bak
            backup = filepath.with_suffix(filepath.suffix + '.bak')
            if not backup.exists():
                os.rename(filepath, backup)
            with open(filepath, 'w', encoding='utf-8') as f:
                f.write(new_content)
            print(f"Изменён: {filepath}")
    else:
        if dry_run:
            print(f"[DRY RUN] Без изменений: {filepath}")

def main():
    if dry_run:
        print("*** РЕЖИМ ПРОСМОТРА – файлы не будут изменены ***\n")
    else:
        print("*** РЕЖИМ ЗАПИСИ – файлы будут изменены ***\n")

    base = Path(root_dir).expanduser().resolve()
    if not base.is_dir():
        print(f"Ошибка: директория {base} не существует.")
        sys.exit(1)

    org_files = list(base.rglob("*.org"))
    print(f"Найдено .org файлов: {len(org_files)}")
    for org_file in org_files:
        process_file(org_file)

    if dry_run:
        print("\n*** ПРОСМОТР ЗАВЕРШЁН. Для применения изменений установите dry_run=False. ***")

if __name__ == "__main__":
    main()