{% extends "base.html" %}
{#
============================================================
Slurm 用户关联页
路由: GET /cluster-users
============================================================
#}
{% block title %}Slurm 用户关联{% endblock %}
{% block breadcrumb_title %}Slurm 用户关联{% endblock %}

{% block content %}
<!-- 页面标题行 -->
<div class="flex justify-between items-center flex-wrap gap-3" style="margin-bottom: 24px">
    <div>
        <h2 style="font-size: 28px; font-weight: 700">Slurm 用户关联</h2>
        <p style="color: #6b7280; margin-top: 6px">
            使用 sacctmgr 维护账户关联（Association）
        </p>
    </div>
    <div class="flex items-center flex-wrap gap-3">
    <a class="btn btn-secondary" href="/slurm-users">修改用户默认账户</a>
    <button class="btn btn-primary" onclick="showCreateAssocModal()">
        + 添加关联
    </button>
    </div>
</div>

<!-- 过滤与统计 -->
<div class="card" style="margin-bottom: 16px">
    <div class="flex items-center gap-3 flex-wrap">
        <label for="assocAccountFilter" style="font-weight: 600; color: #374151">账户筛选</label>
        <select id="assocAccountFilter" style="min-width: 220px">
            <option value="">使用配置的筛选账户</option>
        </select>
        <button class="btn btn-secondary" onclick="loadAssociations()">刷新</button>
        <span style="color: #6b7280; font-size: 14px; margin-left: auto">
            总计: <strong id="assocCount">—</strong> 条关联
        </span>
    </div>
</div>

<!-- 关联列表 -->
<div class="card">
    <div class="data-table-scroll" role="region" aria-label="Slurm 用户关联列表" tabindex="0">
        <table class="table data-table data-table--wide has-actions" id="associationsTable">
            <thead>
                <tr>
                    <th>用户</th>
                    <th>账户</th>
                    <th>集群</th>
                    <th>分区</th>
                    <th>分区限制</th>
                    <th>QOS</th>
                    <th>默认 QOS</th>
                    <th class="col-number">核时限额 (h)</th>
                    <th class="col-number">卡时限额 (h)</th>
                    <th>Comment</th>
                    <th class="col-status">默认关联</th>
                    <th class="col-actions">操作</th>
                </tr>
            </thead>
            <tbody id="associationsTableBody">
                <tr>
                    <td colspan="12" class="text-center py-8 text-gray-500">
                        加载中…
                    </td>
                </tr>
            </tbody>
        </table>
    </div>
</div>

<!-- 操作说明卡片 -->
<div class="card" style="margin-top: 20px; background-color: #eff6ff; border-left: 4px solid #3b82f6">
    <h3 style="font-size: 16px; font-weight: 600; margin-bottom: 8px; color: #1e40af">
        <span aria-hidden="true" style="margin-right: 8px">ⓘ</span>关联管理说明
    </h3>
    <ul style="list-style: disc; margin-left: 24px; color: #1e40af; font-size: 14px; line-height: 1.8">
        <li>“默认关联”展示 Slurm 返回的默认标记；修改用户默认账户，请点击上方“修改用户默认账户”，在用户行中选择“修改默认账户”</li>
        <li>“账户筛选”只改变列表显示；“使用配置的筛选账户”使用服务端配置，不会修改任何用户的默认账户。“默认 QOS”是另一项作业调度设置</li>
        <li>选择账户后点击“添加关联”，为用户分配可使用的 Slurm 账户和集群</li>
        <li>全局关联允许用户使用该账户提交到所有分区；只要全局关联仍存在，分区限制就不会生效</li>
        <li>限制分区时，先创建一个或多个分区关联，确认后再删除同一用户/账户的全局关联</li>
        <li>分区属于关联标识，创建后不可编辑；更换分区需新建关联并删除旧关联</li>
        <li>删除全局关联可能取消该关联下正在运行或排队的作业，请先检查作业队列</li>
        <li>核时/卡时限额会参与调度校验，修改后对后续作业生效</li>
    </ul>
</div>
{% endblock %}

{% block modals %}
<!-- 创建关联弹窗 -->
<div id="createAssocModal" class="modal" style="display: none">
    <div class="modal-content">
        <div class="modal-header">
            <h3>添加账户关联</h3>
            <button class="modal-close" onclick="closeCreateAssocModal()">
                &times;
            </button>
        </div>
        <form id="createAssocForm" onsubmit="handleCreateAssoc(event)">
            <div class="form-group">
                <label for="create_assoc_user">用户名</label>
                <input
                    type="text"
                    id="create_assoc_user"
                    name="username"
                    required
                    autocomplete="off"
                />
            </div>
            <div class="form-group">
                <label for="create_assoc_account">账户</label>
                <select id="create_assoc_account" name="account" required></select>
            </div>
            <div class="form-group">
                <label for="create_assoc_partition">分区</label>
                <select id="create_assoc_partition" name="partition">
                    <option value="">全局（不限制分区）</option>
                </select>
                <div style="font-size:12px;color:#9ca3af;margin-top:5px">
                    限制多个分区时，请为每个允许的分区分别添加一条关联；最后删除全局关联。
                </div>
            </div>
            <div class="form-group">
                <label for="create_assoc_qos">允许使用的 QoS（可多选）</label>
                <select id="create_assoc_qos" name="qos" multiple size="4"></select>
                <div style="font-size:12px;color:#9ca3af;margin-top:5px">
                    用户通过 --qos 选择；按住 Ctrl/Command 可多选。留空表示继承账户允许的 QoS。
                </div>
            </div>
            <div class="form-group">
                <label for="create_assoc_default_qos">默认 QoS（未指定 --qos 时）</label>
                <select id="create_assoc_default_qos" name="default_qos"></select>
                <div style="font-size:12px;color:#9ca3af;margin-top:5px">
                    默认 QoS 必须属于上面的允许范围；未选择时继承账户默认 QoS。
                </div>
            </div>
            <div class="modal-footer">
                <button
                    type="button"
                    class="btn btn-secondary"
                    onclick="closeCreateAssocModal()"
                >
                    取消
                </button>
                <button type="submit" class="btn btn-primary">创建</button>
            </div>
        </form>
    </div>
</div>

<!-- 编辑关联弹窗 -->
<div id="editAssocModal" class="modal" style="display: none">
    <div class="modal-content">
        <div class="modal-header">
            <h3>编辑关联</h3>
            <button class="modal-close" onclick="closeEditAssocModal()">
                &times;
            </button>
        </div>
        <form id="editAssocForm" onsubmit="handleEditAssoc(event)">
            <input type="hidden" id="edit_assoc_account" name="account" />
            <input type="hidden" id="edit_assoc_user" name="username" />
            <input type="hidden" id="edit_assoc_partition" name="partition" />
            <div class="form-group">
                <label>用户 / 账户</label>
                <input
                    type="text"
                    id="edit_assoc_identity"
                    disabled
                    style="background-color: #f3f4f6; cursor: not-allowed"
                />
            </div>
            <div class="form-group">
                <label for="edit_assoc_partition_display">分区（关联标识，不可修改）</label>
                <input
                    type="text"
                    id="edit_assoc_partition_display"
                    disabled
                    style="background-color: #f3f4f6; cursor: not-allowed"
                />
            </div>
            <div class="form-group">
                <label for="edit_assoc_qos">允许使用的 QoS（可多选）</label>
                <select id="edit_assoc_qos" name="qos" multiple size="4"></select>
                <div style="font-size:12px;color:#9ca3af;margin-top:5px">
                    用户通过 --qos 选择；按住 Ctrl/Command 可多选。
                </div>
            </div>
            <div class="form-group">
                <label for="edit_assoc_default_qos">默认 QoS（未指定 --qos 时）</label>
                <select id="edit_assoc_default_qos" name="default_qos"></select>
                <div style="font-size:12px;color:#9ca3af;margin-top:5px">
                    默认 QoS 必须属于上面的允许范围；未选择时继承账户默认 QoS。
                </div>
            </div>
            <div class="modal-footer">
                <button
                    type="button"
                    class="btn btn-secondary"
                    onclick="closeEditAssocModal()"
                >
                    取消
                </button>
                <button type="submit" class="btn btn-primary">保存</button>
            </div>
        </form>
    </div>
</div>

<!-- 核时/卡时拨付弹窗 -->
<div id="tresMinutesModal" class="modal" style="display: none">
    <div class="modal-content" style="max-width: 520px">
        <div class="modal-header">
            <h3 id="tresMinutesModalTitle">核时/卡时拨付</h3>
            <button class="modal-close" onclick="closeTRESMinutesModal()">
                &times;
            </button>
        </div>
        <form id="tresMinutesForm" onsubmit="handleTRESMinutesSubmit(event)">
            <input type="hidden" id="tres_account" name="account" />
            <input type="hidden" id="tres_username" name="username" />
            <input type="hidden" id="tres_partition" name="partition" />
            <div class="form-group">
                <label for="tres_cpu_hours">调整 CPU 核时（负数表示扣除）</label>
                <input
                    type="number"
                    id="tres_cpu_hours"
                    name="cpu_hours"
                    min="-1000000"
                    max="1000000"
                    step="0.1"
                    placeholder="如 120"
                />
            </div>
            <div class="form-group">
                <label for="tres_gpu_hours">调整 GPU 卡时（负数表示扣除）</label>
                <input
                    type="number"
                    id="tres_gpu_hours"
                    name="gpu_hours"
                    min="-1000000"
                    max="1000000"
                    step="0.1"
                    placeholder="如 24"
                />
            </div>
            <div style="font-size: 12px; color: #6b7280; margin-top: -6px; margin-bottom: 12px;">
                正数拨付，负数表示扣除；扣除后不会低于历史已用量。
            </div>
            <div id="tresCurrentSummary" style="margin-bottom: 16px; padding: 10px 12px; background: #f8fafc; border: 1px solid #e5e7eb; border-radius: 6px; color: #475569; font-size: 13px;"></div>
            <div class="form-group">
                <label for="tres_comment">Comment（拨付说明）</label>
                <textarea id="tres_comment" name="comment" class="credit-comment-textarea"
                          rows="5" maxlength="478" required wrap="soft"
                          placeholder="填写拨付原因、项目或工单号，便于追溯"></textarea>
                <div class="credit-comment-hint">
                    最多 478 个字符，长内容会自动换行显示；请勿输入换行。
                </div>
            </div>
            <div class="modal-footer">
                <button
                    type="button"
                    class="btn btn-secondary"
                    onclick="closeTRESMinutesModal()"
                >
                    取消
                </button>
                <button type="submit" class="btn btn-primary">确认拨付</button>
            </div>
        </form>
    </div>
</div>
{% endblock %}

{% block page_scripts %}
<script>
    /* ── 数据加载 ─────────────────────────────────────────── */

    function escapeHtml(value) {
        const node = document.createElement("span");
        node.textContent = String(value ?? "");
        return node.innerHTML
            .replaceAll('"', "&quot;")
            .replaceAll("'", "&#39;");
    }

    async function loadAccountOptions() {
        const accounts = await fetchAccounts();
        const filterSelect = document.getElementById("assocAccountFilter");
        const createSelect = document.getElementById("create_assoc_account");

        const options = accounts
            .map((a) => `<option value="${a.name}">${a.name}</option>`)
            .join("");

        if (createSelect) {
            createSelect.innerHTML = options || "";
        }

        if (filterSelect) {
            const current = filterSelect.value;
            filterSelect.innerHTML =
                '<option value="">使用配置的筛选账户</option>' + options;
            if (current) filterSelect.value = current;
        }
    }

    async function loadQosOptions() {
        const qos = await fetchQos();
        const options = qos
            .map((item) => `<option value="${escapeHtml(item.name)}">${escapeHtml(item.name)}</option>`)
            .join("");
        ["create_assoc_qos", "edit_assoc_qos"].forEach((id) => {
            const select = document.getElementById(id);
            if (select) {
                select.innerHTML = options;
                if (!select.dataset.defaultQosLinked) {
                    const defaultId = id.replace("_qos", "_default_qos");
                    select.addEventListener("change", () => {
                        const defaultSelect = document.getElementById(defaultId);
                        syncDefaultQosOptions(id, defaultId, defaultSelect?.value || "");
                    });
                    select.dataset.defaultQosLinked = "true";
                }
            }
        });
        syncDefaultQosOptions("create_assoc_qos", "create_assoc_default_qos");
        syncDefaultQosOptions("edit_assoc_qos", "edit_assoc_default_qos");
    }

    function syncDefaultQosOptions(qosSelectId, defaultSelectId, preferredValue = "") {
        const qosSelect = document.getElementById(qosSelectId);
        const defaultSelect = document.getElementById(defaultSelectId);
        if (!qosSelect || !defaultSelect) return;

        const allowed = Array.from(qosSelect.selectedOptions).map((option) => option.value);
        const options = allowed
            .map((name) => `<option value="${escapeHtml(name)}">${escapeHtml(name)}</option>`)
            .join("");
        defaultSelect.innerHTML = '<option value="">继承账户默认 QoS</option>' + options;
        defaultSelect.disabled = allowed.length === 0;
        defaultSelect.value = allowed.includes(preferredValue) ? preferredValue : "";
    }

    async function loadPartitionOptions() {
        const partitions = await fetchPartitions();
        const select = document.getElementById("create_assoc_partition");
        const options = partitions
            .map((item) => `<option value="${escapeHtml(item.name)}">${escapeHtml(item.name)}</option>`)
            .join("");
        if (select) {
            select.innerHTML = '<option value="">全局（不限制分区）</option>' + options;
        }
    }

    function renderPartitionAccessStatus(assoc) {
        const labels = {
            global: '<span class="badge badge-warning">全局开放</span>',
            overridden_by_global: '<span class="badge badge-warning">限制未生效（存在全局关联）</span>',
            restricted: '<span class="badge badge-success">限制已生效</span>',
            account_level: '<span style="color:#9ca3af">—</span>',
        };
        return labels[assoc.partition_access_status] || '<span style="color:#9ca3af">未知</span>';
    }

    function setSelectedQos(selectId, values) {
        const select = document.getElementById(selectId);
        const selected = new Set(values || []);
        if (select) Array.from(select.options).forEach((option) => {
            option.selected = selected.has(option.value);
        });
    }

    async function loadAssociations() {
        const tbody = document.getElementById("associationsTableBody");
        const accountFilter =
            document.getElementById("assocAccountFilter").value || null;
        const assocs = await fetchAssociations(accountFilter);

        if (!assocs || assocs.length === 0) {
            tbody.innerHTML =
                '<tr><td colspan="12" class="text-center py-8 text-gray-500">暂无关联数据</td></tr>';
            document.getElementById("assocCount").textContent = "0";
            return;
        }

        tbody.innerHTML = assocs
            .map((assoc) => {
                const qos = assoc.qos && assoc.qos.length > 0 ? assoc.qos.join(", ") : "—";
                const isAccountLevel = !assoc.user;
                const userLabel = isAccountLevel ? "（账户级）" : assoc.user;
                const actionButtons = isAccountLevel
                    ? "<span style=\"color:#9ca3af\">—</span>"
                    : `
                        <button class="btn btn-secondary"
                                style="padding: 4px 12px; font-size: 12px; margin-right: 4px;"
                                onclick="showTRESMinutesModal('${assoc.account}', '${assoc.user}', '${assoc.partition || ""}')">拨付</button>
                        <button class="btn btn-secondary"
                                style="padding: 4px 12px; font-size: 12px; margin-right: 4px;"
                                onclick="showEditAssocModal('${assoc.account}', '${assoc.user}', '${assoc.partition || ""}')">编辑</button>
                        <button class="btn"
                                style="padding: 4px 12px; font-size: 12px; background-color: #dc3023; color: white;"
                                onclick="deleteAssocConfirm('${assoc.account}', '${assoc.user}', '${assoc.partition || ""}')">删除</button>
                      `;
                return `
                <tr>
                    <td><strong>${userLabel || "—"}</strong></td>
                    <td>${assoc.account || "—"}</td>
                    <td>${assoc.cluster || "—"}</td>
                    <td>${assoc.partition || "全局"}</td>
                    <td>${renderPartitionAccessStatus(assoc)}</td>
                    <td style="font-size: 13px; color: #6b7280;">${qos}</td>
                    <td>${assoc.default_qos || "—"}</td>
                    <td class="col-number">${assoc.cpu_minutes == null ? "无限" : (assoc.cpu_minutes / 60).toFixed(2)}</td>
                    <td class="col-number">${assoc.gpu_minutes == null ? "无限" : (assoc.gpu_minutes / 60).toFixed(2)}</td>
                    <td title="${escapeHtml(assoc.comment || "")}">${escapeHtml(assoc.comment || "—")}</td>
                    <td class="col-status">${assoc.is_default ? "是" : "否"}</td>
                    <td class="col-actions"><div class="data-table-actions">${actionButtons}</div></td>
                </tr>`;
            })
            .join("");

        document.getElementById("assocCount").textContent = assocs.length;
    }

    /* ── 创建关联 ───────────────────────────────────────────── */

    function showCreateAssocModal() {
        document.getElementById("createAssocForm").reset();
        setSelectedQos("create_assoc_qos", []);
        syncDefaultQosOptions("create_assoc_qos", "create_assoc_default_qos");
        document.getElementById("createAssocModal").style.display = "flex";
    }

    function closeCreateAssocModal() {
        document.getElementById("createAssocModal").style.display = "none";
    }

    async function handleCreateAssoc(event) {
        event.preventDefault();
        const fd = new FormData(event.target);
        const payload = {
            username: fd.get("username"),
            account: fd.get("account"),
            partition: fd.get("partition") || null,
            qos: Array.from(document.getElementById("create_assoc_qos").selectedOptions).map((option) => option.value).join(",") || null,
            default_qos: fd.get("default_qos") || null,
        };

        const success = await createAssociation(payload);
        if (success) {
            closeCreateAssocModal();
            loadAssociations();
        }
    }

    /* ── 编辑关联 ───────────────────────────────────────────── */

    let editingAssociation = null;

    async function showEditAssocModal(account, username, partition) {
        const assocs = await fetchAssociations(account);
        const target = assocs.find(
            (a) =>
                a.account === account &&
                a.user === username &&
                (a.partition || "") === (partition || ""),
        );
        if (!target) {
            showToast("获取关联信息失败", "error");
            return;
        }

        editingAssociation = target;
        document.getElementById("edit_assoc_account").value = account;
        document.getElementById("edit_assoc_user").value = username;
        document.getElementById("edit_assoc_identity").value = `${username} / ${account}`;
        document.getElementById("edit_assoc_partition").value = target.partition || "";
        document.getElementById("edit_assoc_partition_display").value =
            target.partition || "全局";
        setSelectedQos("edit_assoc_qos", target.qos || []);
        syncDefaultQosOptions(
            "edit_assoc_qos",
            "edit_assoc_default_qos",
            target.default_qos || "",
        );
        document.getElementById("editAssocModal").style.display = "flex";
    }

    function closeEditAssocModal() {
        document.getElementById("editAssocModal").style.display = "none";
        editingAssociation = null;
    }

    async function handleEditAssoc(event) {
        event.preventDefault();
        const fd = new FormData(event.target);
        const account = fd.get("account");
        const username = fd.get("username");
        const selectedQos = Array.from(
            document.getElementById("edit_assoc_qos").selectedOptions,
        ).map((option) => option.value);
        const defaultQos = fd.get("default_qos") || "";
        const payload = { partition: fd.get("partition") || "" };
        const originalQos = (editingAssociation?.qos || []).slice().sort().join(",");
        if (selectedQos.slice().sort().join(",") !== originalQos) {
            payload.qos = selectedQos.join(",");
        }
        if (defaultQos !== (editingAssociation?.default_qos || "")) {
            payload.default_qos = defaultQos;
        }

        const success = await updateAssociation(account, username, payload);
        if (success) {
            closeEditAssocModal();
            loadAssociations();
        }
    }

    /* ── 删除关联 ───────────────────────────────────────────── */

    async function deleteAssocConfirm(account, username, partition) {
        const warning = partition
            ? `确定要删除分区关联 "${username}/${account}/${partition}" 吗？此操作无法撤销！`
            : `确定要删除全局关联 "${username}/${account}" 吗？\n\n删除全局关联可能取消该关联下正在运行或排队的作业，请先检查作业队列。`;
        if (!confirm(warning))
            return;
        const success = await deleteAssociation(account, username, partition || null);
        if (success) loadAssociations();
    }

    /* ── 核时/卡时拨付 ───────────────────────────────────── */

    async function showTRESMinutesModal(account, username, partition) {
        const assocs = await fetchAssociations(account);
        const target = assocs.find(
            (a) =>
                a.account === account &&
                a.user === username &&
                (a.partition || "") === (partition || ""),
        );
        if (!target) {
            showToast("获取关联信息失败", "error");
            return;
        }

        document.getElementById("tresMinutesModalTitle").textContent = `核时/卡时拨付：${username} / ${account}`;
        document.getElementById("tres_account").value = account;
        document.getElementById("tres_username").value = username;
        document.getElementById("tres_partition").value = partition || "";
        document.getElementById("tres_cpu_hours").value = "";
        document.getElementById("tres_gpu_hours").value = "";
        const cpuLimit = target.cpu_minutes == null ? "无限" : `${(target.cpu_minutes / 60).toFixed(2)} h`;
        const gpuLimit = target.gpu_minutes == null ? "无限" : `${(target.gpu_minutes / 60).toFixed(2)} h`;
        const previousComment = target.comment || "无";
        document.getElementById("tresCurrentSummary").textContent =
            `当前限额：CPU ${cpuLimit}，GPU ${gpuLimit}；上一条 Comment：${previousComment}`;
        document.getElementById("tres_comment").value = "";
        document.getElementById("tresMinutesModal").style.display = "flex";
    }

    function closeTRESMinutesModal() {
        document.getElementById("tresMinutesModal").style.display = "none";
    }

    async function handleTRESMinutesSubmit(event) {
        event.preventDefault();
        const fd = new FormData(event.target);
        const account = fd.get("account");
        const username = fd.get("username");
        const partition = fd.get("partition");
        const cpuRaw = fd.get("cpu_hours");
        const gpuRaw = fd.get("gpu_hours");
        const comment = (fd.get("comment") || "").trim();

        const payload = { account, partition: partition || null, comment };
        if (cpuRaw) payload.cpu_hours = parseFloat(cpuRaw);
        if (gpuRaw) payload.gpu_hours = parseFloat(gpuRaw);
        if (payload.cpu_hours === undefined && payload.gpu_hours === undefined) {
            showToast("请至少填写核时或卡时", "warning");
            return;
        }
        if (!comment) {
            showToast("请填写 Comment（拨付说明）", "warning");
            return;
        }

        const submitButton = event.submitter;
        if (submitButton) submitButton.disabled = true;
        const success = await grantUserCredits(username, payload);
        if (submitButton) submitButton.disabled = false;
        if (success) {
            closeTRESMinutesModal();
            loadAssociations();
        }
    }

    /* ── 页面初始化 ───────────────────────────────────────── */

    document.addEventListener("DOMContentLoaded", async () => {
        await loadAccountOptions();
        await loadQosOptions();
        await loadPartitionOptions();
        await loadAssociations();
        enableTableSorting("associationsTable");
        const filter = document.getElementById("assocAccountFilter");
        filter.addEventListener("change", loadAssociations);
    });
</script>
{% endblock %}