"use client";

import { Fragment, useEffect, useMemo, useState } from "react";
import { api } from "@/lib/api";
import { useAuth } from "@/hooks/useAuth";
import { Check, Lock, Save } from "lucide-react";

interface PermissionDef {
  key: string;
  label: string;
  group: string;
}

type Matrix = Record<string, Record<string, boolean>>;

const ROLE_LABELS: Record<string, string> = {
  admin: "Admin",
  moderator: "Moderator",
  writer: "Writer",
};

function buildMatrix(
  roles: string[],
  perms: PermissionDef[],
  granted: Record<string, string[]>
): Matrix {
  const matrix: Matrix = {};
  for (const role of roles) {
    matrix[role] = {};
    const set = new Set(granted[role] ?? []);
    for (const perm of perms) {
      matrix[role][perm.key] = set.has(perm.key);
    }
  }
  return matrix;
}

export default function RolesPage() {
  const { can } = useAuth();
  const allowed = can("roles.edit_permissions");

  const [permissions, setPermissions] = useState<PermissionDef[]>([]);
  const [roles, setRoles] = useState<string[]>([]);
  const [matrix, setMatrix] = useState<Matrix>({});
  const [original, setOriginal] = useState<Matrix>({});
  const [loading, setLoading] = useState(true);
  const [error, setError] = useState("");
  const [saving, setSaving] = useState(false);
  const [notice, setNotice] = useState("");

  const load = async () => {
    setLoading(true);
    try {
      const [registry, current] = await Promise.all([
        api.getPermissionRegistry(),
        api.getRolePermissions(),
      ]);
      const granted: Record<string, string[]> = {};
      for (const r of current.roles) granted[r.role] = r.permissions;
      const built = buildMatrix(registry.roles, registry.permissions, granted);
      setPermissions(registry.permissions);
      setRoles(registry.roles);
      setMatrix(built);
      setOriginal(built);
    } catch (err: any) {
      setError(err.message);
    } finally {
      setLoading(false);
    }
  };

  useEffect(() => {
    if (allowed) load();
    else setLoading(false);
  }, [allowed]);

  const groups = useMemo(() => {
    const order: string[] = [];
    const byGroup: Record<string, PermissionDef[]> = {};
    for (const perm of permissions) {
      if (!byGroup[perm.group]) {
        byGroup[perm.group] = [];
        order.push(perm.group);
      }
      byGroup[perm.group].push(perm);
    }
    return order.map((group) => ({ group, perms: byGroup[group] }));
  }, [permissions]);

  const dirtyRoles = useMemo(() => {
    return roles.filter((role) =>
      permissions.some((perm) => matrix[role]?.[perm.key] !== original[role]?.[perm.key])
    );
  }, [roles, permissions, matrix, original]);

  const toggle = (role: string, key: string) => {
    setNotice("");
    setMatrix((prev) => ({
      ...prev,
      [role]: { ...prev[role], [key]: !prev[role]?.[key] },
    }));
  };

  const saveAll = async () => {
    if (dirtyRoles.length === 0) return;
    setSaving(true);
    setError("");
    setNotice("");
    try {
      for (const role of dirtyRoles) {
        const keys = permissions.filter((p) => matrix[role]?.[p.key]).map((p) => p.key);
        await api.updateRolePermissions(role, keys);
      }
      setOriginal(() => {
        const next: Matrix = {};
        for (const role of roles) next[role] = { ...matrix[role] };
        return next;
      });
      setNotice("Permissions saved.");
    } catch (err: any) {
      setError(err.message);
    } finally {
      setSaving(false);
    }
  };

  if (!allowed) {
    return (
      <div className="flex items-center justify-center min-h-[60vh]">
        <div className="text-on-surface/50">
          You do not have permission to manage roles.
        </div>
      </div>
    );
  }

  if (loading) {
    return (
      <div className="flex items-center justify-center min-h-[60vh]">
        <div className="text-on-surface/50">Loading roles...</div>
      </div>
    );
  }

  return (
    <div className="space-y-6">
      <div>
        <h1 className="text-2xl font-bold text-on-surface">Roles and Permissions</h1>
        <p className="text-on-surface/60 text-sm mt-1">
          Toggle what each role can do. SuperAdmin always has every permission and
          cannot be changed.
        </p>
      </div>

      {error && <p className="text-error text-sm">{error}</p>}
      {notice && <p className="text-primary text-sm">{notice}</p>}

      <div className="bg-surface-container-low rounded-xl overflow-x-auto">
        <table className="w-full text-sm border-collapse">
          <thead>
            <tr className="text-left">
              <th className="sticky left-0 bg-surface-container-low p-4 font-semibold text-on-surface/70">
                Permission
              </th>
              {roles.map((role) => (
                <th key={role} className="p-4 text-center font-semibold text-on-surface/70">
                  {ROLE_LABELS[role] ?? role}
                </th>
              ))}
              <th className="p-4 text-center font-semibold text-on-surface/70">
                <span className="inline-flex items-center gap-1">
                  <Lock size={12} />
                  SuperAdmin
                </span>
              </th>
            </tr>
          </thead>
          <tbody>
            {groups.map(({ group, perms }) => (
              <Fragment key={group}>
                <tr>
                  <td
                    colSpan={roles.length + 2}
                    className="bg-surface-container px-4 py-2 text-xs font-bold uppercase tracking-wide text-on-surface/50"
                  >
                    {group}
                  </td>
                </tr>
                {perms.map((perm) => (
                  <tr key={perm.key} className="border-b border-surface-container-high/40">
                    <td className="sticky left-0 bg-surface-container-low p-4">
                      <div className="text-on-surface">{perm.label}</div>
                      <div className="text-on-surface/40 text-xs font-mono">{perm.key}</div>
                    </td>
                    {roles.map((role) => (
                      <td key={role} className="p-4 text-center">
                        <input
                          type="checkbox"
                          checked={!!matrix[role]?.[perm.key]}
                          onChange={() => toggle(role, perm.key)}
                          className="h-4 w-4 accent-primary cursor-pointer"
                          aria-label={`${ROLE_LABELS[role] ?? role}: ${perm.label}`}
                        />
                      </td>
                    ))}
                    <td className="p-4 text-center text-primary/70">
                      <Check size={16} className="inline" aria-label="Always granted" />
                    </td>
                  </tr>
                ))}
              </Fragment>
            ))}
          </tbody>
        </table>
      </div>

      <div className="flex flex-wrap items-center gap-3">
        <button
          onClick={saveAll}
          disabled={dirtyRoles.length === 0 || saving}
          className="flex items-center gap-2 px-4 py-2 rounded-lg bg-gradient-to-r from-primary to-primary-container text-on-primary font-semibold text-sm hover:opacity-90 transition-opacity disabled:opacity-50"
        >
          <Save size={16} />
          {saving ? "Saving..." : "Save changes"}
        </button>
        {dirtyRoles.length > 0 && !saving && (
          <span className="text-on-surface/50 text-sm">
            Unsaved changes to {dirtyRoles.map((r) => ROLE_LABELS[r] ?? r).join(", ")}
          </span>
        )}
      </div>
    </div>
  );
}
