wdpe 0.4.2

use std::{collections::HashMap, ops::Index};

use scraper::ElementRef;

use super::{
    FromSapTable, SapTableDef, SapTableHeader, cell::SapTableCellDefWrapper,
    property::SapTableRowType, row::SapTableRow,
};
use crate::element::parser::ElementParser;
use crate::{
    element::definition::ElementDefinition,
    error::{ElementError, WebDynproError},
};

/// [`SapTable`](super::SapTable) 내부 테이블
#[derive(Clone, Debug)]
#[allow(unused)]
pub struct SapTableBody {
    table_def: SapTableDef,
    header: Option<SapTableHeader>,
    rows: Vec<SapTableRow>,
}

impl<'a> SapTableBody {
    /// TODO: Implement unit test for rowspan/colspan handling
    pub(super) fn new(
        table_def: SapTableDef,
        elem_ref: ElementRef<'a>,
    ) -> Result<SapTableBody, ElementError> {
        let ref_iter = elem_ref.children().filter_map(ElementRef::wrap);
        let mut header_iter = ref_iter
            .clone()
            .filter_map(|row_ref| SapTableHeader::new(table_def.clone(), row_ref).ok());
        let header = header_iter.next();
        if header.is_some() && header_iter.next().is_some() {
            return Err(ElementError::InvalidContent {
                element: table_def.id().to_owned(),
                content: "Multiple header in table".to_owned(),
            });
        }
        let mut rows: Vec<SapTableRow> = Vec::new();
        // Def, rowsize, colsize
        type CellSpanInfo = (SapTableCellDefWrapper, u32, u32);
        let mut spans: HashMap<u32, CellSpanInfo> = HashMap::new();
        for row_ref in ref_iter.clone() {
            let row_type = row_ref
                .value()
                .attr("rt")
                .map(|s| s.into())
                .unwrap_or(SapTableRowType::default());
            let row_count = row_ref
                .value()
                .attr("rr")
                .and_then(|s| s.parse::<u32>().ok());
            if matches!(row_type, SapTableRowType::Header) {
                continue;
            }

            // If it meets empty row(zero-index), instantly terminate repetition
            if row_count.is_some_and(|c| c == 0) {
                break;
            }
            let subct_selector = scraper::Selector::parse("[subct]").unwrap();
            let subcts = row_ref.select(&subct_selector);
            let mut cells: Vec<SapTableCellDefWrapper> = Vec::new();
            let mut col_counter: u32 = 0;
            for cell_ref in subcts {
                let cell = SapTableCellDefWrapper::from_ref(table_def.clone(), cell_ref);
                if let Some(cell) = cell {
                    if spans.contains_key(&col_counter) {
                        let spanned_cell = spans.remove(&col_counter).unwrap();
                        if spanned_cell.1 - 1 > 1 {
                            spans.insert(
                                col_counter,
                                (spanned_cell.0.clone(), spanned_cell.1 - 1, spanned_cell.2),
                            );
                        }
                        for _ in 0..spanned_cell.2 {
                            col_counter += 1;
                            cells.push(spanned_cell.0.clone());
                        }
                    }
                    let cell_value = cell_ref.value();
                    let rowspan = cell_value
                        .attr("rowspan")
                        .and_then(|str| str.parse::<u32>().ok())
                        .unwrap_or(1);
                    let colspan = cell_value
                        .attr("colspan")
                        .and_then(|str| str.parse::<u32>().ok())
                        .unwrap_or(1);
                    if rowspan > 1 {
                        spans.insert(col_counter, (cell.clone(), rowspan, colspan));
                    }
                    for _ in 0..colspan {
                        cells.push(cell.clone());
                        col_counter += 1;
                    }
                }
            }
            // checks if spanning lasts after last cell
            if spans.contains_key(&col_counter) {
                let spanned_cell = spans.remove(&col_counter).unwrap();
                if spanned_cell.1 - 1 > 1 {
                    spans.insert(
                        col_counter,
                        (spanned_cell.0.clone(), spanned_cell.1 - 1, spanned_cell.2),
                    );
                }
                #[allow(unused_assignments)]
                for _ in 0..spanned_cell.2 {
                    col_counter += 1;
                    cells.push(spanned_cell.0.clone());
                }
            }
            if let Ok(row) = SapTableRow::new(table_def.clone(), row_ref, cells) {
                rows.push(row);
            }
        }
        Ok(SapTableBody {
            table_def,
            header,
            rows,
        })
    }

    /// 헤더 행을 제외한 행의 갯수를 반환합니다.
    pub fn len(&self) -> usize {
        self.rows.len()
    }

    /// 헤더 행을 제외한 행이 비어있는지 확인합니다.
    #[must_use]
    pub fn is_empty(&self) -> bool {
        self.len() == 0
    }

    /// 내부 행의 Iterator를 반환합니다.
    pub fn iter(&'a self) -> impl ExactSizeIterator<Item = &'a SapTableRow> {
        self.rows.iter()
    }

    /// 내부 행에 헤더 행을 포함한 튜플의 Iterator를 반환합니다.
    ///
    /// 헤더가 없는 경우 `None`을 반환합니다.
    pub fn zip_header(
        &'a self,
    ) -> Option<impl ExactSizeIterator<Item = (&'a SapTableHeader, &'a SapTableRow)>> {
        let header = self.header.as_ref()?;
        Some(self.rows.iter().map(move |row| (header, row)))
    }

    /// 이 테이블의 원본 [`SapTableDef`]를 반환합니다.
    pub fn table_def(&self) -> SapTableDef {
        self.table_def.clone()
    }

    /// 헤더 행을 반환합니다.
    pub fn header(&self) -> Option<&SapTableHeader> {
        self.header.as_ref()
    }

    /// 테이블을 [`FromSapTable`]을 구현하는 형의 [`Vec`]으로 변환합니다.
    pub fn try_table_into<T: FromSapTable<'a>>(
        &'a self,
        parser: &'a ElementParser,
    ) -> Result<Vec<T>, WebDynproError> {
        let header = self.header.as_ref();
        self.iter()
            .map(|row| T::from_table(header, row, parser))
            .collect::<Result<Vec<T>, WebDynproError>>()
    }
}

impl Index<usize> for SapTableBody {
    type Output = SapTableRow;

    fn index(&self, index: usize) -> &Self::Output {
        &self.rows[index]
    }
}