use crate::{
model::Model,
number_format::{get_default_num_fmt_id, get_new_num_fmt_index, get_num_fmt},
types::{Border, CellStyles, CellXfs, Fill, Font, NumFmt, Style, Styles},
};
impl Styles {
fn get_font_index(&self, font: &Font) -> Option<i32> {
for (font_index, item) in self.fonts.iter().enumerate() {
if item == font {
return Some(font_index as i32);
}
}
None
}
fn get_fill_index(&self, fill: &Fill) -> Option<i32> {
for (fill_index, item) in self.fills.iter().enumerate() {
if item == fill {
return Some(fill_index as i32);
}
}
None
}
fn get_border_index(&self, border: &Border) -> Option<i32> {
for (border_index, item) in self.borders.iter().enumerate() {
if item == border {
return Some(border_index as i32);
}
}
None
}
fn get_num_fmt_index(&self, format_code: &str) -> Option<i32> {
if let Some(index) = get_default_num_fmt_id(format_code) {
return Some(index);
}
for item in self.num_fmts.iter() {
if item.format_code == format_code {
return Some(item.num_fmt_id);
}
}
None
}
pub fn create_new_style(&mut self, style: &Style) -> i32 {
let font = &style.font;
let font_id = if let Some(index) = self.get_font_index(font) {
index
} else {
self.fonts.push(font.clone());
self.fonts.len() as i32 - 1
};
let fill = &style.fill;
let fill_id = if let Some(index) = self.get_fill_index(fill) {
index
} else {
self.fills.push(fill.clone());
self.fills.len() as i32 - 1
};
let border = &style.border;
let border_id = if let Some(index) = self.get_border_index(border) {
index
} else {
self.borders.push(border.clone());
self.borders.len() as i32 - 1
};
let num_fmt = &style.num_fmt;
let num_fmt_id;
if let Some(index) = self.get_num_fmt_index(num_fmt) {
num_fmt_id = index;
} else {
num_fmt_id = get_new_num_fmt_index(&self.num_fmts);
self.num_fmts.push(NumFmt {
format_code: num_fmt.to_string(),
num_fmt_id,
});
}
self.cell_xfs.push(CellXfs {
xf_id: 0,
num_fmt_id,
font_id,
fill_id,
border_id,
apply_number_format: false,
apply_border: false,
apply_alignment: false,
apply_protection: false,
apply_font: false,
apply_fill: false,
quote_prefix: style.quote_prefix,
alignment: style.alignment.clone(),
});
self.cell_xfs.len() as i32 - 1
}
pub fn get_style_index(&self, style: &Style) -> Option<i32> {
for (index, cell_xf) in self.cell_xfs.iter().enumerate() {
let border_id = cell_xf.border_id as usize;
let fill_id = cell_xf.fill_id as usize;
let font_id = cell_xf.font_id as usize;
let num_fmt_id = cell_xf.num_fmt_id;
let quote_prefix = cell_xf.quote_prefix;
if style
== &(Style {
alignment: cell_xf.alignment.clone(),
num_fmt: get_num_fmt(num_fmt_id, &self.num_fmts),
fill: self.fills[fill_id].clone(),
font: self.fonts[font_id].clone(),
border: self.borders[border_id].clone(),
quote_prefix,
})
{
return Some(index as i32);
}
}
None
}
pub(crate) fn get_style_index_or_create(&mut self, style: &Style) -> i32 {
if let Some(index) = self.get_style_index(style) {
index
} else {
self.create_new_style(style)
}
}
pub fn add_named_cell_style(
&mut self,
style_name: &str,
style_index: i32,
) -> Result<(), String> {
if self.get_style_index_by_name(style_name).is_ok() {
return Err("A style with that name already exists".to_string());
}
if self.cell_xfs.len() < style_index as usize {
return Err("There is no style with that index".to_string());
}
let cell_style = CellStyles {
name: style_name.to_string(),
xf_id: style_index,
builtin_id: 0,
};
self.cell_styles.push(cell_style);
Ok(())
}
pub fn get_style_index_by_name(&self, style_name: &str) -> Result<i32, String> {
for cell_style in &self.cell_styles {
if cell_style.name == style_name {
return Ok(cell_style.xf_id);
}
}
Err(format!("Style '{style_name}' not found"))
}
pub fn create_named_style(&mut self, style_name: &str, style: &Style) -> Result<(), String> {
let style_index = self.create_new_style(style);
self.add_named_cell_style(style_name, style_index)
}
pub(crate) fn get_style_with_quote_prefix(&mut self, index: i32) -> Result<i32, String> {
let mut style = self.get_style(index)?;
style.quote_prefix = true;
Ok(self.get_style_index_or_create(&style))
}
pub(crate) fn get_style_with_format(
&mut self,
index: i32,
num_fmt: &str,
) -> Result<i32, String> {
let mut style = self.get_style(index)?;
style.num_fmt = num_fmt.to_string();
Ok(self.get_style_index_or_create(&style))
}
pub(crate) fn get_style_without_quote_prefix(&mut self, index: i32) -> Result<i32, String> {
let mut style = self.get_style(index)?;
style.quote_prefix = false;
Ok(self.get_style_index_or_create(&style))
}
pub(crate) fn style_is_quote_prefix(&self, index: i32) -> bool {
let cell_xf = &self.cell_xfs[index as usize];
cell_xf.quote_prefix
}
pub(crate) fn get_style(&self, index: i32) -> Result<Style, String> {
let cell_xf = &self
.cell_xfs
.get(index as usize)
.ok_or("Invalid index provided".to_string())?;
let border_id = cell_xf.border_id as usize;
let fill_id = cell_xf.fill_id as usize;
let font_id = cell_xf.font_id as usize;
let num_fmt_id = cell_xf.num_fmt_id;
let quote_prefix = cell_xf.quote_prefix;
let alignment = cell_xf.alignment.clone();
Ok(Style {
alignment,
num_fmt: get_num_fmt(num_fmt_id, &self.num_fmts),
fill: self.fills[fill_id].clone(),
font: self.fonts[font_id].clone(),
border: self.borders[border_id].clone(),
quote_prefix,
})
}
}
impl<'a> Model<'a> {
pub fn set_cell_style(
&mut self,
sheet: u32,
row: i32,
column: i32,
style: &Style,
) -> Result<(), String> {
let style_index = self.workbook.styles.get_style_index_or_create(style);
self.workbook
.worksheet_mut(sheet)?
.set_cell_style(row, column, style_index)
}
pub fn copy_cell_style(
&mut self,
source_cell: (u32, i32, i32),
destination_cell: (u32, i32, i32),
) -> Result<(), String> {
let source_style_index = self
.workbook
.worksheet(source_cell.0)?
.get_style(source_cell.1, source_cell.2);
self.workbook
.worksheet_mut(destination_cell.0)?
.set_cell_style(destination_cell.1, destination_cell.2, source_style_index)
}
pub fn set_cell_style_by_name(
&mut self,
sheet: u32,
row: i32,
column: i32,
style_name: &str,
) -> Result<(), String> {
let style_index = self.workbook.styles.get_style_index_by_name(style_name)?;
self.workbook
.worksheet_mut(sheet)?
.set_cell_style(row, column, style_index)
}
pub fn set_sheet_style(&mut self, sheet: u32, style_name: &str) -> Result<(), String> {
let style_index = self.workbook.styles.get_style_index_by_name(style_name)?;
self.workbook.worksheet_mut(sheet)?.set_style(style_index)?;
Ok(())
}
pub fn set_sheet_row_style(
&mut self,
sheet: u32,
row: i32,
style_name: &str,
) -> Result<(), String> {
let style_index = self.workbook.styles.get_style_index_by_name(style_name)?;
self.workbook
.worksheet_mut(sheet)?
.set_row_style(row, style_index)?;
Ok(())
}
pub fn set_sheet_column_style(
&mut self,
sheet: u32,
column: i32,
style_name: &str,
) -> Result<(), String> {
let style_index = self.workbook.styles.get_style_index_by_name(style_name)?;
self.workbook
.worksheet_mut(sheet)?
.set_column_style(column, style_index)?;
Ok(())
}
}