imlement native csv loader

6 files changed, 65 insertions, 8 deletions

diff --git a/Cargo.lock b/Cargo.lock
index 584660b..5a7ea4e 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -117,6 +117,27 @@ dependencies = [
 ]
 
 [[package]]
+name = "csv"
+version = "1.3.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ac574ff4d437a7b5ad237ef331c17ccca63c46479e5b5453eb8e10bb99a759fe"
+dependencies = [
+ "csv-core",
+ "itoa",
+ "ryu",
+ "serde",
+]
+
+[[package]]
+name = "csv-core"
+version = "0.1.11"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5efa2b3d7902f4b634a20cae3c9c4e6209dc4779feb6863329607560143efa70"
+dependencies = [
+ "memchr",
+]
+
+[[package]]
 name = "dirs"
 version = "5.0.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
@@ -278,6 +299,7 @@ dependencies = [
 name = "neosheet"
 version = "0.1.0"
 dependencies = [
+ "csv",
  "dirs",
  "mlua",
  "once_cell",
diff --git a/Cargo.toml b/Cargo.toml
index c046f4e..3fd65ea 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -4,6 +4,7 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
+csv = "1.3.0"
 dirs = "5.0.1"
 mlua = { version = "0.9.9", features = ["luajit"] }
 once_cell = "1.19.0"
diff --git a/src/sheet/cell.rs b/src/sheet/cell.rs
index d723732..6973b96 100644
--- a/src/sheet/cell.rs
+++ b/src/sheet/cell.rs
@@ -57,7 +57,10 @@ impl From<String> for Cell {
 
 impl From<&str> for Cell {
     fn from(value: &str) -> Self {
-        Cell::String(value.to_string())
+        match value.parse::<f64>() {
+            Ok(n) => Cell::Number(n),
+            Err(_) => Cell::String(value.to_string()),
+        }
     }
 }
 
diff --git a/src/sheet/loader.rs b/src/sheet/loader/mod.rs
index 35b6da2..0e083fe 100644
--- a/src/sheet/loader.rs
+++ b/src/sheet/loader/mod.rs
@@ -5,8 +5,12 @@ use once_cell::sync::Lazy;
 
 use crate::lua::{ownedfunction::OwnedFunction, runnable::Runnable};
 
+use self::native::csv::load_csv;
+
 use super::{register::SheetId, tablized::Tablized};
 
+pub mod native;
+
 #[derive(Default)]
 pub struct Loader {
     loaders: HashMap<String, Box<dyn Runnable<String, Tablized>>>,
@@ -14,18 +18,24 @@ pub struct Loader {
 
 impl Loader {
     pub fn new() -> Self {
-        Self {
-            loaders: HashMap::new(),
-        }
+        let mut loaders: HashMap<String, Box<dyn Runnable<String, Tablized>>> = HashMap::new();
+
+        loaders.insert("csv".to_string(), Box::new(load_csv));
+
+        Self { loaders }
     }
 
     pub fn load_sheet(&self, path: String, lua: &Lua) -> Option<SheetId> {
         let p = Path::new(&path);
-        let loader = self.loaders.get(p.extension()?.to_str()?)?;
-        match loader.run(path, lua) {
-            Ok(table) => table.to_sheet(lua).ok(),
-            Err(_) => None,
+        if p.is_file() {
+            let loader = self.loaders.get(p.extension()?.to_str()?)?;
+            return match loader.run(path, lua) {
+                Ok(table) => table.to_sheet(lua).ok(),
+                Err(_) => None,
+            };
         }
+
+        return None;
     }
 
     pub fn add(&mut self, extension: String, func: Box<dyn Runnable<String, Tablized>>) {
diff --git a/src/sheet/loader/native/csv.rs b/src/sheet/loader/native/csv.rs
new file mode 100644
index 0000000..4a8f05d
--- /dev/null
+++ b/src/sheet/loader/native/csv.rs
@@ -0,0 +1,19 @@
+use std::{fs::File, io::BufReader};
+
+use crate::sheet::{cell::Cell, tablized::Tablized};
+
+pub fn load_csv(path: String) -> Tablized {
+    let file = File::open(path).unwrap();
+    let reader = BufReader::new(file);
+    let mut csv_reader = csv::Reader::from_reader(reader);
+
+    let mut table = Vec::new();
+
+    for result in csv_reader.records() {
+        if let Ok(record) = result {
+            table.push(record.iter().map(|s| Cell::from(s)).collect::<Vec<Cell>>())
+        }
+    }
+
+    Tablized::Vector(table)
+}
diff --git a/src/sheet/loader/native/mod.rs b/src/sheet/loader/native/mod.rs
new file mode 100644
index 0000000..41ec797
--- /dev/null
+++ b/src/sheet/loader/native/mod.rs
@@ -0,0 +1,2 @@
+
+pub mod csv;