[Refactor] refactor use codesnap library

.cargo/config.toml

@@ -9,3 +9,9 @@ rustflags = [
   "-C", "link-arg=-undefined",
   "-C", "link-arg=dynamic_lookup",
 ]
+
+[target.x86_64-unknown-linux-musl]
+rustflags = ["-C", "target-feature=-crt-static"]
+
+[target.aarch64-unknown-linux-musl]
+rustflags = ["-C", "target-feature=-crt-static"]

generator/Cargo.toml

@@ -3,17 +3,11 @@ name = "generator"
 version = "0.1.0"
 edition = "2021"
 
-[dependencies]
-nvim-oxi = { features = ["neovim-0-9", "libuv"], version = "0.5.1" }
-tiny-skia = "0.11.4"
-syntect = "5.2.0"
-cosmic-text = "0.12.0"
-serde = "1.0.204"
-arboard = { features = ["wayland-data-control"], version = "3.4.0" }
-thiserror = "1.0.63"
-regex = "1.10.5"
-two-face = "0.4.0"
-cached = "0.53.1"
-
 [lib]
 crate-type = ["cdylib"]
+
+[dependencies]
+codesnap = "0.7.1"
+mlua = { version = "0.10.0", features = ["module", "luajit", "serialize"] }
+serde = { version = "1.0.204", features = ["derive"] }
+

generator/src/lib.rs

@@ -1,30 +1,62 @@
-mod code;
-mod color;
-mod components;
-mod config;
-mod copy;
-mod copy_ascii;
-mod edges;
-mod highlight;
-mod path;
-mod save;
-mod snapshot;
-mod text;
-
-use config::TakeSnapshotParams;
-use copy::copy_into_clipboard;
-use copy_ascii::copy_ascii;
-use nvim_oxi::{api, Dictionary, Function};
-use save::save_snapshot;
-
-#[nvim_oxi::plugin]
-fn generator() -> nvim_oxi::Result<Dictionary> {
-    let copy_into_clipboard: Function<TakeSnapshotParams, Result<(), api::Error>> =
-        Function::from_fn(copy_into_clipboard);
-
-    Ok(Dictionary::from_iter([
-        ("copy_into_clipboard", copy_into_clipboard),
-        ("save_snapshot", Function::from_fn(save_snapshot)),
-        ("copy_ascii", Function::from_fn(copy_ascii)),
-    ]))
+mod snapshot_config;
+
+use codesnap::snapshot::{image_snapshot::ImageSnapshot, snapshot_data::SnapshotData};
+use mlua::prelude::*;
+use snapshot_config::SnapshotConfigLua;
+
+enum SnapshotType {
+    Png,
+    Svg,
+    Html,
+}
+
+impl From<String> for SnapshotType {
+    fn from(value: String) -> Self {
+        match value.as_str() {
+            "png" => SnapshotType::Png,
+            "svg" => SnapshotType::Svg,
+            "html" => SnapshotType::Html,
+            _ => SnapshotType::Png,
+        }
+    }
+}
+
+impl SnapshotType {
+    fn snapshot_data(&self, image_snapshot: ImageSnapshot) -> LuaResult<SnapshotData> {
+        let data = match self {
+            SnapshotType::Png => image_snapshot.png_data(),
+            SnapshotType::Svg => todo!(),
+            SnapshotType::Html => todo!(),
+        }
+        .map_err(|_| mlua::Error::RuntimeError("Failed to generate snapshot data".to_string()))?;
+
+        Ok(data)
+    }
+}
+
+fn save(
+    _: &Lua,
+    (snapshot_type, path, config): (String, String, SnapshotConfigLua),
+) -> LuaResult<()> {
+    let image_snapshot = config
+        .0
+        .create_snapshot()
+        .map_err(|_| mlua::Error::RuntimeError("Failed to create snapshot".to_string()))?;
+    let snapshot_type: SnapshotType = snapshot_type.into();
+
+    snapshot_type
+        .snapshot_data(image_snapshot)?
+        .save(&path)
+        .map_err(|_| mlua::Error::RuntimeError(format!("Failed to save snapshot data to {}", path)))
+}
+
+fn copy_to_clipboard(_: &Lua, snapshot_type: SnapshotType) {}
+
+#[mlua::lua_module(skip_memory_check)]
+fn codesnap_generator(lua: &Lua) -> LuaResult<LuaTable> {
+    let exports = lua.create_table()?;
+
+    exports.set("save", lua.create_function(save)?)?;
+
+    Ok(exports)
 }

generator/src/snapshot_config.rs

@@ -0,0 +1,12 @@
+use codesnap::config::SnapshotConfig;
+use mlua::{FromLua, LuaSerdeExt};
+
+pub struct SnapshotConfigLua(pub SnapshotConfig);
+
+impl FromLua for SnapshotConfigLua {
+    fn from_lua(value: mlua::Value, lua: &mlua::Lua) -> mlua::Result<Self> {
+        let config: SnapshotConfig = lua.from_value::<SnapshotConfig>(value)?;
+
+        Ok(SnapshotConfigLua(config))
+    }
+}