table.rs

//! docx の表
//!
//! 表は行の並び、行はセルの並び、セルは段落や表の並びである。セルの中に表を置けるので、入れ子になる。

use crate::document::Block;
use crate::error::Error;
use crate::paragraph::{self, Paragraph};
use crate::xml::{Event, Reader};


/// セル 1 つ
#[derive(Debug, Clone, Default, PartialEq)]
pub struct Cell {
    /// 横につながっているセルの数。1 ならつながっていない
    pub span: u32,
    /// 中身。セルには必ず段落が 1 つ以上入る
    pub block: Vec<Block>,
}


impl Cell {
    /// 文字 1 行だけのセル
    pub fn new(text: &str) -> Cell {
        Cell {span: 1, block: vec![Block::Paragraph(Paragraph::new(text))]}
    }

    /// 中の文字を、段落ごとに改行でつないで返す
    pub fn text(&self) -> String {
        let mut result = String::new();

        for one in &self.block {
            if !result.is_empty() {
                result.push('\n');
            }

            result.push_str(&one.text());
        }

        result
    }
}


/// 行 1 つ
#[derive(Debug, Clone, Default, PartialEq)]
pub struct Row {
    pub cell: Vec<Cell>,
}


impl Row {
    /// 文字を並べた 1 行
    pub fn new(text: &[&str]) -> Row {
        Row {cell: text.iter().map(|one| Cell::new(one)).collect()}
    }
}


/// 表
#[derive(Debug, Clone, Default, PartialEq)]
pub struct Table {
    /// 表の書式の名前 (`<w:tblStyle>`)。ないときは空
    pub style: String,
    /// 表そのものに枠線の指定 (`<w:tblBorders>`) があるか
    ///
    /// 書式から来る枠線はここに出ない。Word の「表 (格子)」のような書式は
    /// 枠線を書式のほうに持つので、読むと false になる。
    pub border: bool,
    pub row: Vec<Row>,
}


impl Table {
    /// 枠線のある表
    pub fn new(row: Vec<Row>) -> Table {
        Table {style: String::new(), border: true, row}
    }

    /// いちばん多い行の列数
    pub fn width(&self) -> usize {
        self.row.iter().map(|one| one.cell.iter().map(|cell| cell.span.max(1) as usize).sum()).max().unwrap_or(0)
    }
}


/// `<w:tbl>` を読んだ直後から `</w:tbl>` まで読む
pub fn read(source: &mut Reader) -> Result<Table, Error> {
    let mut result = Table::default();

    while let Some(event) = source.next()? {
        if event == Event::Close {
            if source.name() == "tbl" {
                return Ok(result);
            }

            continue;
        }

        if event == Event::Text {
            continue;
        }

        match source.name() {
            "tblStyle" => result.style = source.attribute("val").unwrap_or("").to_string(),
            "tblBorders" => result.border = true,
            "tr" if event == Event::Open => result.row.push(read_row(source)?),
            _ => {}
        }
    }

    Ok(result)
}


/// `<w:tr>` を読んだ直後から `</w:tr>` まで読む
fn read_row(source: &mut Reader) -> Result<Row, Error> {
    let mut result = Row::default();

    while let Some(event) = source.next()? {
        if event == Event::Close {
            if source.name() == "tr" {
                return Ok(result);
            }

            continue;
        }

        if event == Event::Open && source.name() == "tc" {
            result.cell.push(read_cell(source)?);
        }
    }

    Ok(result)
}


/// `<w:tc>` を読んだ直後から `</w:tc>` まで読む
fn read_cell(source: &mut Reader) -> Result<Cell, Error> {
    let mut result = Cell {span: 1, block: Vec::new()};

    while let Some(event) = source.next()? {
        if event == Event::Close {
            if source.name() == "tc" {
                return Ok(result);
            }

            continue;
        }

        if event == Event::Text {
            continue;
        }

        match source.name() {
            "gridSpan" => {
                let value = source.attribute("val").unwrap_or("1");
                result.span = value.parse().unwrap_or(1);
            }

            "p" => {
                if event == Event::Empty {
                    result.block.push(Block::Paragraph(Paragraph::default()));
                } else {
                    result.block.push(Block::Paragraph(paragraph::read(source)?));
                }
            }

            "tbl" if event == Event::Open => result.block.push(Block::Table(read(source)?)),
            _ => {}
        }
    }

    Ok(result)
}


/// 表を XML にして足す
pub fn write_into(result: &mut String, one: &Table) {
    // docx は tblPr の中の順番が決まっている。tblStyle、tblW、tblBorders の順
    result.push_str("<w:tbl><w:tblPr>");

    if !one.style.is_empty() {
        result.push_str("<w:tblStyle w:val=\"");
        crate::xml::escape_into(&one.style, result);
        result.push_str("\"/>");
    }

    result.push_str("<w:tblW w:w=\"0\" w:type=\"auto\"/>");

    if one.border {
        result.push_str("<w:tblBorders>");

        for side in ["top", "left", "bottom", "right", "insideH", "insideV"] {
            result.push_str(&format!(
                "<w:{side} w:val=\"single\" w:sz=\"4\" w:space=\"0\" w:color=\"auto\"/>"
            ));
        }

        result.push_str("</w:tblBorders>");
    }

    result.push_str("</w:tblPr><w:tblGrid>");

    for _ in 0..one.width() {
        result.push_str("<w:gridCol w:w=\"2000\"/>");
    }

    result.push_str("</w:tblGrid>");

    for row in &one.row {
        result.push_str("<w:tr>");

        for cell in &row.cell {
            result.push_str("<w:tc><w:tcPr><w:tcW w:w=\"0\" w:type=\"auto\"/>");

            if cell.span > 1 {
                result.push_str(&format!("<w:gridSpan w:val=\"{}\"/>", cell.span));
            }

            result.push_str("</w:tcPr>");

            // セルには段落が 1 つ以上必要である
            if cell.block.is_empty() {
                result.push_str("<w:p/>");
            }

            for block in &cell.block {
                crate::document::block_into(result, block);
            }

            result.push_str("</w:tc>");
        }

        result.push_str("</w:tr>");
    }

    result.push_str("</w:tbl>");
}