/*
    This file is part of jellything (https://codeberg.org/metamuffin/jellything)
    which is licensed under the GNU Affero General Public License (version 3); see /COPYING.
    Copyright (C) 2025 metamuffin <metamuffin.org>
*/
use crate::{
    helper::{language::AcceptLanguage, A},
    ui::error::MyResult,
};
use anyhow::Context;
use jellycommon::user::UserPermission;
use jellyimport::is_importing;
use jellylogic::{
    admin::user::{admin_users, delete_user, get_user, update_user_perms, GrantState},
    session::AdminSession,
};
use jellyui::{
    admin::user::{AdminUserPage, AdminUsersPage},
    render_page,
    scaffold::{RenderInfo, SessionInfo},
};
use rocket::{form::Form, get, post, response::content::RawHtml, FromForm, FromFormField};

#[get("/admin/users")]
pub fn r_admin_users(session: A<AdminSession>, lang: AcceptLanguage) -> MyResult<RawHtml<String>> {
    let AcceptLanguage(lang) = lang;
    let r = admin_users(&session.0)?;
    Ok(RawHtml(render_page(
        &AdminUsersPage {
            flash: None,
            lang: &lang,
            users: &r.users,
        },
        RenderInfo {
            importing: is_importing(),
            session: Some(SessionInfo {
                user: session.0 .0.user,
            }),
        },
        lang,
    )))
}

#[get("/admin/user/<name>")]
pub fn r_admin_user<'a>(
    session: A<AdminSession>,
    name: &'a str,
    lang: AcceptLanguage,
) -> MyResult<RawHtml<String>> {
    let AcceptLanguage(lang) = lang;
    let user = get_user(&session.0, name)?;

    Ok(RawHtml(render_page(
        &AdminUserPage {
            flash: None,
            lang: &lang,
            user: &user,
        },
        RenderInfo {
            importing: is_importing(),
            session: Some(SessionInfo {
                user: session.0 .0.user,
            }),
        },
        lang,
    )))
}

#[derive(FromForm)]
pub struct UserPermissionForm {
    permission: String,
    action: UrlGrantState,
}

#[derive(FromFormField)]
pub enum UrlGrantState {
    Grant,
    Revoke,
    Unset,
}

#[post("/admin/user/<name>/update_permission", data = "<form>")]
pub fn r_admin_user_permission(
    session: A<AdminSession>,
    form: Form<UserPermissionForm>,
    name: &str,
    lang: AcceptLanguage,
) -> MyResult<RawHtml<String>> {
    let AcceptLanguage(lang) = lang;
    let perm = serde_json::from_str::<UserPermission>(&form.permission)
        .context("parsing provided permission")?;

    update_user_perms(
        &session.0,
        name,
        perm,
        match form.action {
            UrlGrantState::Grant => GrantState::Grant,
            UrlGrantState::Revoke => GrantState::Revoke,
            UrlGrantState::Unset => GrantState::Unset,
        },
    )?;

    let user = get_user(&session.0, name)?;

    Ok(RawHtml(render_page(
        &AdminUserPage {
            flash: Some(Ok("Permissions updated".to_string())),
            lang: &lang,
            user: &user,
        },
        RenderInfo {
            importing: is_importing(),
            session: Some(SessionInfo {
                user: session.0 .0.user,
            }),
        },
        lang,
    )))
}

#[post("/admin/<name>/remove")]
pub fn r_admin_remove_user(
    session: A<AdminSession>,
    name: &str,
    lang: AcceptLanguage,
) -> MyResult<RawHtml<String>> {
    let AcceptLanguage(lang) = lang;
    delete_user(&session.0, name)?;
    let r = admin_users(&session.0)?;

    Ok(RawHtml(render_page(
        &AdminUsersPage {
            flash: Some(Ok("User removed".to_string())),
            lang: &lang,
            users: &r.users,
        },
        RenderInfo {
            importing: is_importing(),
            session: Some(SessionInfo {
                user: session.0 .0.user,
            }),
        },
        lang,
    )))
}