minne/html-router/src/routes/scratchpad/handlers.rs

use axum::{
    extract::{Path, Query, State},
    http::{HeaderValue, StatusCode},
    response::{IntoResponse, Response},
    Form,
};
use axum_htmx::{HxBoosted, HxRequest, HX_TRIGGER};
use chrono::{DateTime, Utc};
use serde::{Deserialize, Serialize};

use crate::html_state::HtmlState;
use crate::middlewares::{
    auth_middleware::RequireUser,
    response_middleware::{
        template_with_headers, HtmlError, TemplateResponse, TemplateResult, ResponseResult,
    },
};
use common::storage::types::{
    ingestion_payload::IngestionPayload, ingestion_task::IngestionTask, scratchpad::Scratchpad,
};

#[derive(Serialize)]
pub struct ScratchpadPageData {
    scratchpads: Vec<ScratchpadListItem>,
    archived_scratchpads: Vec<ScratchpadArchiveItem>,
    #[serde(skip_serializing_if = "Option::is_none")]
    new_scratchpad: Option<ScratchpadDetail>,
}

#[derive(Serialize)]
pub struct ScratchpadListItem {
    id: String,
    title: String,
    content: String,
    last_saved_at: DateTime<Utc>,
}

#[derive(Serialize)]
pub struct ScratchpadDetailData {
    scratchpad: ScratchpadDetail,
    is_editing_title: bool,
}

#[derive(Serialize)]
pub struct ScratchpadArchiveItem {
    id: String,
    title: String,
    archived_at: Option<DateTime<Utc>>,
    #[serde(skip_serializing_if = "Option::is_none")]
    ingested_at: Option<DateTime<Utc>>,
}

#[derive(Serialize)]
pub struct ScratchpadDetail {
    id: String,
    title: String,
    content: String,
    created_at: DateTime<Utc>,
    updated_at: DateTime<Utc>,
    last_saved_at: DateTime<Utc>,
    is_dirty: bool,
}

#[derive(Serialize)]
pub struct AutoSaveResponse {
    success: bool,
    last_saved_at_display: String,
    last_saved_at_iso: String,
}

impl From<&Scratchpad> for ScratchpadListItem {
    fn from(value: &Scratchpad) -> Self {
        Self {
            id: value.id.clone(),
            title: value.title.clone(),
            content: value.content.clone(),
            last_saved_at: value.last_saved_at,
        }
    }
}

impl From<&Scratchpad> for ScratchpadArchiveItem {
    fn from(value: &Scratchpad) -> Self {
        Self {
            id: value.id.clone(),
            title: value.title.clone(),
            archived_at: value.archived_at,
            ingested_at: value.ingested_at,
        }
    }
}

impl From<&Scratchpad> for ScratchpadDetail {
    fn from(value: &Scratchpad) -> Self {
        Self {
            id: value.id.clone(),
            title: value.title.clone(),
            content: value.content.clone(),
            created_at: value.created_at,
            updated_at: value.updated_at,
            last_saved_at: value.last_saved_at,
            is_dirty: value.is_dirty,
        }
    }
}

#[derive(Deserialize)]
pub struct CreateScratchpadForm {
    title: String,
}

#[derive(Deserialize)]
pub struct UpdateScratchpadForm {
    content: String,
}

#[derive(Deserialize)]
pub struct UpdateTitleForm {
    title: String,
}

#[derive(Deserialize)]
pub struct EditTitleQuery {
    edit_title: Option<bool>,
}

pub async fn show_scratchpad_page(
    RequireUser(user): RequireUser,
    HxRequest(is_htmx): HxRequest,
    HxBoosted(is_boosted): HxBoosted,
    State(state): State<HtmlState>,
) -> TemplateResult {
    let scratchpads = Scratchpad::get_by_user(&user.id, &state.db).await?;
    let archived_scratchpads = Scratchpad::get_archived_by_user(&user.id, &state.db).await?;

    let scratchpad_list: Vec<ScratchpadListItem> =
        scratchpads.iter().map(ScratchpadListItem::from).collect();
    let archived_list: Vec<ScratchpadArchiveItem> = archived_scratchpads
        .iter()
        .map(ScratchpadArchiveItem::from)
        .collect();

    if is_htmx && !is_boosted {
        Ok(TemplateResponse::new_partial(
            "scratchpad/base.html",
            "main",
            ScratchpadPageData {
                scratchpads: scratchpad_list,
                archived_scratchpads: archived_list,
                new_scratchpad: None,
            },
        ))
    } else {
        Ok(TemplateResponse::new_template(
            "scratchpad/base.html",
            ScratchpadPageData {
                scratchpads: scratchpad_list,
                archived_scratchpads: archived_list,
                new_scratchpad: None,
            },
        ))
    }
}

pub async fn show_scratchpad_modal(
    RequireUser(user): RequireUser,
    State(state): State<HtmlState>,
    Path(scratchpad_id): Path<String>,
    Query(query): Query<EditTitleQuery>,
) -> TemplateResult {
    let scratchpad = Scratchpad::get_by_id(&scratchpad_id, &user.id, &state.db).await?;

    let scratchpad_detail = ScratchpadDetail::from(&scratchpad);

    // Handle edit_title query parameter
    let is_editing_title = query.edit_title.unwrap_or(false);

    Ok(TemplateResponse::new_template(
        "scratchpad/editor_modal.html",
        ScratchpadDetailData {
            scratchpad: scratchpad_detail,
            is_editing_title,
        },
    ))
}

pub async fn create_scratchpad(
    RequireUser(user): RequireUser,
    State(state): State<HtmlState>,
    Form(form): Form<CreateScratchpadForm>,
) -> TemplateResult {
    let user_id = user.id.clone();
    let scratchpad = Scratchpad::new(user_id.clone(), form.title);
    let _stored = state.db.store_item(scratchpad.clone()).await?;

    let scratchpads = Scratchpad::get_by_user(&user.id, &state.db).await?;
    let archived_scratchpads = Scratchpad::get_archived_by_user(&user.id, &state.db).await?;

    let scratchpad_list: Vec<ScratchpadListItem> =
        scratchpads.iter().map(ScratchpadListItem::from).collect();
    let archived_list: Vec<ScratchpadArchiveItem> = archived_scratchpads
        .iter()
        .map(ScratchpadArchiveItem::from)
        .collect();

    Ok(TemplateResponse::new_partial(
        "scratchpad/base.html",
        "main",
        ScratchpadPageData {
            scratchpads: scratchpad_list,
            archived_scratchpads: archived_list,
            new_scratchpad: Some(ScratchpadDetail::from(&scratchpad)),
        },
    ))
}

pub async fn auto_save_scratchpad(
    RequireUser(user): RequireUser,
    State(state): State<HtmlState>,
    Path(scratchpad_id): Path<String>,
    Form(form): Form<UpdateScratchpadForm>,
) -> ResponseResult {
    let updated =
        Scratchpad::update_content(&scratchpad_id, &user.id, &form.content, &state.db).await?;

    // Return a success indicator for auto-save
    Ok(axum::Json(AutoSaveResponse {
        success: true,
        last_saved_at_display: updated
            .last_saved_at
            .format("%Y-%m-%d %H:%M:%S")
            .to_string(),
        last_saved_at_iso: updated.last_saved_at.to_rfc3339(),
    })
    .into_response())
}

pub async fn update_scratchpad_title(
    RequireUser(user): RequireUser,
    State(state): State<HtmlState>,
    Path(scratchpad_id): Path<String>,
    Form(form): Form<UpdateTitleForm>,
) -> TemplateResult {
    Scratchpad::update_title(&scratchpad_id, &user.id, &form.title, &state.db).await?;

    let scratchpad = Scratchpad::get_by_id(&scratchpad_id, &user.id, &state.db).await?;

    Ok(TemplateResponse::new_template(
        "scratchpad/editor_modal.html",
        ScratchpadDetailData {
            scratchpad: ScratchpadDetail::from(&scratchpad),
            is_editing_title: false,
        },
    ))
}

pub async fn delete_scratchpad(
    RequireUser(user): RequireUser,
    State(state): State<HtmlState>,
    Path(scratchpad_id): Path<String>,
) -> TemplateResult {
    Scratchpad::delete(&scratchpad_id, &user.id, &state.db).await?;

    // Return the updated main section content
    let scratchpads = Scratchpad::get_by_user(&user.id, &state.db).await?;
    let archived_scratchpads = Scratchpad::get_archived_by_user(&user.id, &state.db).await?;

    let scratchpad_list: Vec<ScratchpadListItem> =
        scratchpads.iter().map(ScratchpadListItem::from).collect();
    let archived_list: Vec<ScratchpadArchiveItem> = archived_scratchpads
        .iter()
        .map(ScratchpadArchiveItem::from)
        .collect();

    Ok(TemplateResponse::new_partial(
        "scratchpad/base.html",
        "main",
        ScratchpadPageData {
            scratchpads: scratchpad_list,
            archived_scratchpads: archived_list,
            new_scratchpad: None,
        },
    ))
}

pub async fn ingest_scratchpad(
    RequireUser(user): RequireUser,
    State(state): State<HtmlState>,
    Path(scratchpad_id): Path<String>,
) -> ResponseResult {
    let scratchpad = Scratchpad::get_by_id(&scratchpad_id, &user.id, &state.db).await?;

    if scratchpad.content.trim().is_empty() {
        let trigger_payload = serde_json::json!({
            "toast": {
                "title": "Ingestion skipped",
                "description": "Cannot ingest an empty scratchpad.",
                "type": "warning"
            }
        });
        let trigger_value = serde_json::to_string(&trigger_payload).unwrap_or_else(|_| {
            r#"{"toast":{"title":"Ingestion skipped","description":"Cannot ingest an empty scratchpad.","type":"warning"}}"#.to_string()
        });

        let mut response = Response::builder()
            .status(StatusCode::BAD_REQUEST)
            .body(axum::body::Body::empty())
            .unwrap_or_else(|_| Response::new(axum::body::Body::empty()));

        if let Ok(header_value) = HeaderValue::from_str(&trigger_value) {
            response.headers_mut().insert(HX_TRIGGER, header_value);
        }

        return Ok(response);
    }

    // Create ingestion task

    let payload = IngestionPayload::Text {
        text: scratchpad.content.clone(),
        context: format!("Scratchpad: {}", scratchpad.title),
        category: "scratchpad".to_string(),
        user_id: user.id.clone(),
    };

    let task = IngestionTask::new(payload, user.id.clone());
    state.db.store_item(task).await?;

    // Archive the scratchpad once queued for ingestion
    Scratchpad::archive(&scratchpad_id, &user.id, &state.db, true).await?;

    let scratchpads = Scratchpad::get_by_user(&user.id, &state.db).await?;
    let archived_scratchpads = Scratchpad::get_archived_by_user(&user.id, &state.db).await?;

    let scratchpad_list: Vec<ScratchpadListItem> =
        scratchpads.iter().map(ScratchpadListItem::from).collect();
    let archived_list: Vec<ScratchpadArchiveItem> = archived_scratchpads
        .iter()
        .map(ScratchpadArchiveItem::from)
        .collect();

    let trigger_payload = serde_json::json!({
        "toast": {
            "title": "Ingestion queued",
            "description": format!("\"{}\" archived and added to the ingestion queue.", scratchpad.title),
            "type": "success"
        }
    });
    let trigger_value = serde_json::to_string(&trigger_payload).unwrap_or_else(|_| {
        r#"{"toast":{"title":"Ingestion queued","description":"Scratchpad archived and added to the ingestion queue.","type":"success"}}"#.to_string()
    });

    Ok(template_with_headers(
        TemplateResponse::new_partial(
            "scratchpad/base.html",
            "main",
            ScratchpadPageData {
                scratchpads: scratchpad_list,
                archived_scratchpads: archived_list,
                new_scratchpad: None,
            },
        ),
        |headers| {
            if let Ok(header_value) = HeaderValue::from_str(&trigger_value) {
                headers.insert(HX_TRIGGER, header_value);
            }
        },
    ))
}

pub async fn archive_scratchpad(
    RequireUser(user): RequireUser,
    State(state): State<HtmlState>,
    Path(scratchpad_id): Path<String>,
) -> TemplateResult {
    Scratchpad::archive(&scratchpad_id, &user.id, &state.db, false).await?;

    let scratchpads = Scratchpad::get_by_user(&user.id, &state.db).await?;
    let archived_scratchpads = Scratchpad::get_archived_by_user(&user.id, &state.db).await?;

    let scratchpad_list: Vec<ScratchpadListItem> =
        scratchpads.iter().map(ScratchpadListItem::from).collect();
    let archived_list: Vec<ScratchpadArchiveItem> = archived_scratchpads
        .iter()
        .map(ScratchpadArchiveItem::from)
        .collect();

    Ok(TemplateResponse::new_template(
        "scratchpad/base.html",
        ScratchpadPageData {
            scratchpads: scratchpad_list,
            archived_scratchpads: archived_list,
            new_scratchpad: None,
        },
    ))
}

pub async fn restore_scratchpad(
    RequireUser(user): RequireUser,
    State(state): State<HtmlState>,
    Path(scratchpad_id): Path<String>,
) -> ResponseResult {
    Scratchpad::restore(&scratchpad_id, &user.id, &state.db).await?;

    let scratchpads = Scratchpad::get_by_user(&user.id, &state.db).await?;
    let archived_scratchpads = Scratchpad::get_archived_by_user(&user.id, &state.db).await?;

    let scratchpad_list: Vec<ScratchpadListItem> =
        scratchpads.iter().map(ScratchpadListItem::from).collect();
    let archived_list: Vec<ScratchpadArchiveItem> = archived_scratchpads
        .iter()
        .map(ScratchpadArchiveItem::from)
        .collect();

    let trigger_payload = serde_json::json!({
        "toast": {
            "title": "Scratchpad restored",
            "description": "The scratchpad is back in your active list.",
            "type": "info"
        }
    });
    let trigger_value = serde_json::to_string(&trigger_payload).unwrap_or_else(|_| {
        r#"{"toast":{"title":"Scratchpad restored","description":"The scratchpad is back in your active list.","type":"info"}}"#.to_string()
    });

    Ok(template_with_headers(
        TemplateResponse::new_partial(
            "scratchpad/base.html",
            "main",
            ScratchpadPageData {
                scratchpads: scratchpad_list,
                archived_scratchpads: archived_list,
                new_scratchpad: None,
            },
        ),
        |headers| {
            if let Ok(header_value) = HeaderValue::from_str(&trigger_value) {
                headers.insert(HX_TRIGGER, header_value);
            }
        },
    ))
}

#[cfg(test)]
mod tests {
    use super::*;
    use chrono::Utc;

    #[test]
    fn test_scratchpad_list_item_conversion() {
        // Create a test scratchpad with datetime values
        let now = Utc::now();
        let mut scratchpad = common::storage::types::scratchpad::Scratchpad::new(
            "test_user".to_string(),
            "Test Scratchpad".to_string(),
        );

        // Override the timestamps with known values for testing
        scratchpad.last_saved_at = now;

        // Test conversion to ScratchpadListItem
        let list_item = ScratchpadListItem::from(&scratchpad);

        assert_eq!(list_item.id, scratchpad.id);
        assert_eq!(list_item.title, scratchpad.title);
        assert_eq!(list_item.content, scratchpad.content);
        assert_eq!(list_item.last_saved_at, scratchpad.last_saved_at);
    }

    #[test]
    fn test_scratchpad_detail_conversion() {
        // Create a test scratchpad with datetime values
        let now = Utc::now();
        let mut scratchpad = common::storage::types::scratchpad::Scratchpad::new(
            "test_user".to_string(),
            "Test Scratchpad".to_string(),
        );

        // Override the timestamps with known values for testing
        scratchpad.last_saved_at = now;

        // Test conversion to ScratchpadDetail
        let detail = ScratchpadDetail::from(&scratchpad);

        assert_eq!(detail.id, scratchpad.id);
        assert_eq!(detail.title, scratchpad.title);
        assert_eq!(detail.content, scratchpad.content);
        assert_eq!(detail.created_at, scratchpad.created_at);
        assert_eq!(detail.updated_at, scratchpad.updated_at);
        assert_eq!(detail.last_saved_at, scratchpad.last_saved_at);
        assert_eq!(detail.is_dirty, scratchpad.is_dirty);
    }

    #[test]
    fn test_scratchpad_archive_item_conversion() {
        // Create a test scratchpad with optional datetime values
        let now = Utc::now();
        let mut scratchpad = common::storage::types::scratchpad::Scratchpad::new(
            "test_user".to_string(),
            "Test Scratchpad".to_string(),
        );

        // Set optional datetime fields
        scratchpad.archived_at = Some(now);
        scratchpad.ingested_at = Some(now);

        // Test conversion to ScratchpadArchiveItem
        let archive_item = ScratchpadArchiveItem::from(&scratchpad);

        assert_eq!(archive_item.id, scratchpad.id);
        assert_eq!(archive_item.title, scratchpad.title);
        assert_eq!(archive_item.archived_at, scratchpad.archived_at);
        assert_eq!(archive_item.ingested_at, scratchpad.ingested_at);
    }

    #[test]
    fn test_scratchpad_archive_item_conversion_with_none_values() {
        // Create a test scratchpad without optional datetime values
        let scratchpad = common::storage::types::scratchpad::Scratchpad::new(
            "test_user".to_string(),
            "Test Scratchpad".to_string(),
        );

        // Test conversion to ScratchpadArchiveItem
        let archive_item = ScratchpadArchiveItem::from(&scratchpad);

        assert_eq!(archive_item.id, scratchpad.id);
        assert_eq!(archive_item.title, scratchpad.title);
        assert_eq!(archive_item.archived_at, None);
        assert_eq!(archive_item.ingested_at, None);
    }
}