1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
use mlua::{FromLuaMulti, Lua, RegistryKey};
use super::{
cell::Cell,
register::{Register, SheetId},
};
pub enum Tablized {
Vector(Vec<Vec<Cell>>),
Table(RegistryKey),
}
impl Tablized {
pub fn to_sheet(&self, lua: &Lua) -> mlua::Result<SheetId> {
match self {
Tablized::Vector(v) => {
let rows = v.len();
let columns = v
.iter()
.max_by_key(|v| v.len())
.unwrap_or(&Vec::new())
.len();
let id = Register::create(columns, rows);
let sheet = Register::get(id).unwrap();
for row in 0..rows {
for column in 0..columns {
if let Some(value) = v.get(row).and_then(|c| c.get(column)) {
sheet.set_cell(row, column, value.clone())
}
}
}
Ok(id)
}
Tablized::Table(key) => {
let table = lua.registry_value(key)?;
Register::create_from_table(table)
},
}
}
}
impl<'lua> FromLuaMulti<'lua> for Tablized {
fn from_lua_multi(
values: mlua::prelude::LuaMultiValue<'lua>,
lua: &'lua mlua::prelude::Lua,
) -> mlua::prelude::LuaResult<Self> {
if let Some(value) = values.get(0) {
if value.is_table() {
let key = lua.create_registry_value(value)?;
return Ok(Self::Table(key));
}
}
Err(mlua::Error::runtime("table needed"))
}
}
|
