#!/usr/bin/env python3
from pathlib import Path
import sys

app_path = Path(sys.argv[1]) if len(sys.argv) > 1 else Path('/opt/virtuality/web/app.py')
if not app_path.exists():
    raise SystemExit(f'app.py not found: {app_path}')

app_dir = app_path.resolve().parent
template_path = app_dir / 'templates' / 'vm_detail.html'
changed = []

text = app_path.read_text()

helpers = r'''

def vm_state(name: str) -> str:
    result = run_cmd(["virsh", "domstate", name], timeout=8)
    return (result.get("stdout") or "unknown").strip().lower()


def current_vm_iso(name: str) -> str:
    result = run_cmd(["virsh", "domblklist", name, "--details"], timeout=10)
    if not result.get("ok"):
        return ""
    for line in (result.get("stdout") or "").splitlines():
        if ".iso" not in line.lower():
            continue
        parts = line.split()
        if parts:
            return parts[-1]
    return ""


def detach_vm_iso(name: str) -> tuple[bool, str]:
    if not valid_vm_name(name) or not vm_exists(name):
        return False, "VM не найдена."
    xml = run_cmd(["virsh", "dumpxml", name], timeout=12)
    cdrom_targets: list[str] = []
    if xml.get("ok"):
        try:
            root = ET.fromstring(xml.get("stdout") or "")
            devices = root.find("devices")
            if devices is not None:
                for disk in devices.findall("disk"):
                    if disk.attrib.get("device") != "cdrom":
                        continue
                    target = disk.find("target")
                    dev = target.attrib.get("dev") if target is not None else ""
                    if dev:
                        cdrom_targets.append(dev)
        except Exception:
            pass
    if not cdrom_targets:
        cdrom_targets = ["sda", "hda", "sdb", "hdc"]

    running = vm_state(name) == "running"
    errors = []
    changed_any = False
    for target in cdrom_targets:
        commands = []
        if running:
            commands.append(["virsh", "detach-disk", name, target, "--live"])
        commands.append(["virsh", "detach-disk", name, target, "--config"])
        for cmd in commands:
            result = run_cmd(cmd, timeout=30)
            if result.get("ok"):
                changed_any = True
            elif result.get("stderr"):
                errors.append(result.get("stderr"))
    if changed_any:
        return True, "ISO был отмонтирован."
    return False, errors[-1] if errors else "Подключенный ISO не найден."


def mount_vm_iso(name: str, iso_path: str) -> tuple[bool, str]:
    if not valid_vm_name(name) or not vm_exists(name):
        return False, "VM не найдена."
    iso = Path(iso_path or "").resolve()
    try:
        iso_root = ISO_DIR.resolve()
    except Exception:
        iso_root = Path("/var/lib/virtuality/iso")
    if iso_root not in iso.parents or iso.suffix.lower() != ".iso" or not iso.exists():
        return False, "ISO должен быть существующим .iso файлом из /var/lib/virtuality/iso."

    detach_vm_iso(name)
    running = vm_state(name) == "running"
    base = ["virsh", "attach-disk", name, str(iso), "sda", "--type", "cdrom", "--mode", "readonly"]
    if running:
        live = run_cmd(base + ["--live"], timeout=30)
        if not live.get("ok"):
            return False, live.get("stderr") or "Не удалось подключить ISO к запущенной VM."
    config = run_cmd(base + ["--config"], timeout=30)
    if not config.get("ok"):
        return False, config.get("stderr") or "Не удалось сохранить ISO в конфигурации VM."
    return True, "ISO был смонтирован в VM. Если гостевая ОС его не увидела сразу, перезагрузи VM или обнови устройства внутри гостевой ОС."
'''

if 'def mount_vm_iso(name: str, iso_path: str)' not in text:
    marker = '\n\ndef vm_details(name: str) -> dict[str, Any]:'
    if marker in text:
        text = text.replace(marker, helpers + marker, 1)
        changed.append('existing VM ISO mount helpers added')
    else:
        changed.append('ISO helpers skipped: vm_details marker not found')
else:
    changed.append('existing VM ISO mount helpers already present')

if '"current_iso": current_vm_iso(name)' not in text:
    replacements = [
        (
            '"resource_message": request.query_params.get("resource_message", ""), "resource_error": request.query_params.get("resource_error", "")})',
            '"resource_message": request.query_params.get("resource_message", ""), "resource_error": request.query_params.get("resource_error", ""), "isos": list_iso_files(), "current_iso": current_vm_iso(name), "iso_message": request.query_params.get("iso_message", ""), "iso_error": request.query_params.get("iso_error", "")})',
        ),
        (
            '"boot_message": request.query_params.get("boot_message", ""), "boot_error": request.query_params.get("boot_error", "")})',
            '"boot_message": request.query_params.get("boot_message", ""), "boot_error": request.query_params.get("boot_error", ""), "isos": list_iso_files(), "current_iso": current_vm_iso(name), "iso_message": request.query_params.get("iso_message", ""), "iso_error": request.query_params.get("iso_error", "")})',
        ),
        (
            '"host_ip": system_summary()["ip"]})',
            '"host_ip": system_summary()["ip"], "isos": list_iso_files(), "current_iso": current_vm_iso(name), "iso_message": request.query_params.get("iso_message", ""), "iso_error": request.query_params.get("iso_error", "")})',
        ),
    ]
    for old, new in replacements:
        if old in text:
            text = text.replace(old, new, 1)
            changed.append('vm detail context gets ISO mount data')
            break
    else:
        changed.append('ISO context skipped: vm detail context marker not found')
else:
    changed.append('vm detail context already has ISO mount data')

routes = r'''

@app.post("/vm/{name}/iso/mount")
def vm_iso_mount_apply(request: Request, name: str, iso_path: str = Form(...)):
    auth_redirect = require_auth(request)
    if auth_redirect:
        return auth_redirect
    ok, message = mount_vm_iso(name, iso_path)
    if ok:
        return RedirectResponse(url=f"/vm/{name}?iso_message={message}", status_code=303)
    return RedirectResponse(url=f"/vm/{name}?iso_error={message}", status_code=303)


@app.post("/vm/{name}/iso/unmount")
def vm_iso_unmount_apply(request: Request, name: str):
    auth_redirect = require_auth(request)
    if auth_redirect:
        return auth_redirect
    ok, message = detach_vm_iso(name)
    if ok:
        return RedirectResponse(url=f"/vm/{name}?iso_message={message}", status_code=303)
    return RedirectResponse(url=f"/vm/{name}?iso_error={message}", status_code=303)
'''

if '@app.post("/vm/{name}/iso/mount")' not in text:
    markers = ['\n\n@app.post("/vm/{name}/{action}")', '\n\n@app.get("/vm/create"', '\n\n@app.get("/operations"']
    for marker in markers:
        if marker in text:
            text = text.replace(marker, routes + marker, 1)
            changed.append('existing VM ISO mount routes added')
            break
    else:
        changed.append('ISO routes skipped: safe route marker not found')
else:
    changed.append('existing VM ISO mount routes already present')

app_path.write_text(text)

iso_card = r'''
      <article class="card">
        <div class="card-head">
          <h2>ISO-привод</h2>
          <span class="pill">cdrom</span>
        </div>
        {% if current_iso %}
          <div class="notice">Сейчас подключен ISO: <b>{{ current_iso }}</b></div>
        {% else %}
          <div class="muted small-note">ISO сейчас не подключён.</div>
        {% endif %}
        <form method="post" action="/vm/{{ vm.name }}/iso/mount" class="form-grid">
          <label>
            <span>ISO образ</span>
            <select name="iso_path" required>
              {% for iso in isos %}
                <option value="{{ iso.path }}" {% if current_iso == iso.path %}selected{% endif %}>{{ iso.name }} — {{ iso.size }}</option>
              {% endfor %}
            </select>
          </label>
          <button class="primary wide" type="submit" {% if not isos %}disabled{% endif %}>Смонтировать ISO в VM</button>
        </form>
        <form method="post" action="/vm/{{ vm.name }}/iso/unmount" class="form-grid" onsubmit="return confirm('Отмонтировать ISO из VM {{ vm.name }}?');">
          <button type="submit" class="ghost wide">Отмонтировать ISO</button>
        </form>
        {% if not isos %}
          <div class="alert danger">ISO-образов пока нет. Сначала загрузи .iso в разделе ISO.</div>
        {% endif %}
        <p class="muted small-note">Для запущенной VM ISO подключается live и сохраняется в конфигурации. Для выключенной VM ISO будет доступен при следующем старте.</p>
      </article>
'''

boot_card = r'''
      <article class="card">
        <div class="card-head">
          <h2>Порядок загрузки</h2>
          <span class="pill">drag boot</span>
        </div>
        <form method="post" action="/vm/{{ vm.name }}/boot-order" class="form-grid boot-order-form">
          <input type="hidden" name="boot_order" id="boot-order-value" value="{{ current_boot_order }}">
          <div class="boot-order-list" id="boot-order-list" data-current="{{ current_boot_order }}">
            <div class="boot-order-item" draggable="true" data-device="cdrom"><span class="drag-handle">☰</span><div><b>ISO / CD-ROM</b><small>Установщик или rescue-образ</small></div></div>
            <div class="boot-order-item" draggable="true" data-device="hd"><span class="drag-handle">☰</span><div><b>Диск</b><small>Основной qcow2/raw диск VM</small></div></div>
            <div class="boot-order-item" draggable="true" data-device="network"><span class="drag-handle">☰</span><div><b>Сеть / PXE</b><small>Загрузка по сети</small></div></div>
          </div>
          <button class="primary wide" type="submit">Применить порядок загрузки</button>
        </form>
        <p class="muted small-note">Перетащи нужный источник выше. Для загрузки с ISO поставь ISO / CD-ROM первым. Изменение сработает после перезапуска VM.</p>
        <script>
          (function(){
            const list = document.getElementById('boot-order-list');
            const input = document.getElementById('boot-order-value');
            if (!list || !input) return;
            const initialMap = {'cdrom_disk':['cdrom','hd','network'],'disk_cdrom':['hd','cdrom','network'],'network_disk':['network','hd','cdrom'],'disk':['hd','cdrom','network'],'auto':['hd','cdrom','network']};
            const order = initialMap[list.dataset.current || 'auto'] || initialMap.auto;
            const nodes = Array.from(list.querySelectorAll('.boot-order-item'));
            order.forEach(device => { const node = nodes.find(item => item.dataset.device === device); if (node) list.appendChild(node); });
            function updateValue(){
              const devices = Array.from(list.querySelectorAll('.boot-order-item')).map(item => item.dataset.device);
              const first = devices[0]; const second = devices[1];
              if (first === 'cdrom' && second === 'hd') input.value = 'cdrom_disk';
              else if (first === 'hd' && second === 'cdrom') input.value = 'disk_cdrom';
              else if (first === 'network' && second === 'hd') input.value = 'network_disk';
              else if (first === 'hd') input.value = 'disk';
              else input.value = 'auto';
            }
            let dragged = null;
            list.addEventListener('dragstart', event => { dragged = event.target.closest('.boot-order-item'); if (!dragged) return; dragged.classList.add('dragging'); event.dataTransfer.effectAllowed = 'move'; });
            list.addEventListener('dragend', () => { if (dragged) dragged.classList.remove('dragging'); dragged = null; updateValue(); });
            list.addEventListener('dragover', event => { event.preventDefault(); const after = Array.from(list.querySelectorAll('.boot-order-item:not(.dragging)')).find(item => { const box = item.getBoundingClientRect(); return event.clientY < box.top + box.height / 2; }); if (!dragged) return; if (after) list.insertBefore(dragged, after); else list.appendChild(dragged); });
            updateValue();
          })();
        </script>
      </article>
'''

if template_path.exists():
    tpl = template_path.read_text()
    original = tpl
    settings_grid = '\n\n    <section class="grid two vm-boot-iso-grid">\n' + iso_card + boot_card + '    </section>\n'

    if 'action="/vm/{{ vm.name }}/iso/mount"' not in tpl and 'vm-boot-iso-grid' not in tpl:
        markers = ['\n\n    <section class="grid three vm-resource-boot-iso-grid">', '\n\n    <section class="grid two">', '\n\n    <section class="card">']
        for marker in markers:
            if marker in tpl:
                if 'vm-resource-boot-iso-grid' in marker:
                    insert_pos = tpl.find(marker) + len(marker)
                    tpl = tpl[:insert_pos] + '\n' + iso_card + boot_card + tpl[insert_pos:]
                    changed.append('ISO and boot cards inserted into three-column grid')
                else:
                    tpl = tpl.replace(marker, settings_grid + marker, 1)
                    changed.append('ISO and boot cards inserted by fallback')
                break
        else:
            changed.append('ISO layout skipped: safe template marker not found')
    elif 'vm-boot-iso-grid' in tpl and 'boot-order-list' not in tpl:
        changed.append('ISO grid exists but draggable boot marker absent; left unchanged safely')
    else:
        changed.append('ISO mount template already present')

    if tpl != original:
        template_path.write_text(tpl)
else:
    changed.append('ISO template skipped: vm_detail.html not found')

print('existing VM ISO mount patch applied:')
for item in changed:
    print(f'- {item}')