logisheets_controller 1.16.1

use gents_derives::TS;
use imbl::HashMap;
use logisheets_base::{BlockCellId, BlockId, SheetId};

/// Appendix is one of the craft extensions.
///
/// To utilize the flexibility of cell coordinates, LogiSheets
/// allows crafts to add custom metadata to cells
#[derive(Debug, Clone, TS)]
#[ts(file_name = "appendix.ts", rename_all = "camelCase")]
pub struct Appendix {
    pub craft_id: String,
    pub tag: u8,
    pub content: String,
}

#[derive(Debug, Clone, TS)]
#[ts(file_name = "appendix_with_cell.ts", rename_all = "camelCase")]
pub struct AppendixWithCell {
    pub appendix: Appendix,
    pub sheet_id: SheetId,
    pub block_id: BlockId,
    pub row_idx: usize,
    pub col_idx: usize,
}

#[derive(Debug, Clone, Default)]
pub struct AppendixManager {
    data: HashMap<(SheetId, BlockCellId), Vec<Appendix>>,
}

impl AppendixManager {
    pub fn new() -> Self {
        Self {
            data: HashMap::new(),
        }
    }

    pub fn get(&self, sheet_id: SheetId, block_cell_id: &BlockCellId) -> Option<Vec<Appendix>> {
        self.data.get(&(sheet_id, *block_cell_id)).cloned()
    }

    /// Every annotated cell on `sheet_id`, with the appendices stacked on it in
    /// the order they were pushed. Iteration follows the hash map, so a caller
    /// writing a file has to sort — the save must not depend on hash order.
    pub fn iter_sheet(
        &self,
        sheet_id: SheetId,
    ) -> impl Iterator<Item = (&BlockCellId, &Vec<Appendix>)> {
        self.data
            .iter()
            .filter_map(move |((sid, cell), list)| (*sid == sheet_id).then_some((cell, list)))
    }

    pub fn push(&mut self, sheet_id: SheetId, block_cell_id: BlockCellId, appendix: Appendix) {
        self.data
            .entry((sheet_id, block_cell_id))
            .or_insert(vec![])
            .push(appendix);
    }

    pub fn remove(&mut self, sheet_id: SheetId, block_cell_id: BlockCellId) {
        self.data.remove(&(sheet_id, block_cell_id));
    }
}