huh whuh

1 files changed, 102 insertions, 0 deletions

diff --git a/src/main.rs b/src/main.rs
new file mode 100644
index 0000000..f0683ae
--- /dev/null
+++ b/src/main.rs
@@ -0,0 +1,102 @@
+use std::path::PathBuf;
+
+use clap::Parser;
+use color_eyre::Result;
+use error::{LuaXError, ReluaxError};
+
+mod error;
+mod luax;
+mod server;
+
+fn table_to_html<W: std::io::Write>(table: rlua::Table, f: &mut W) -> Result<()> {
+    let type_name: Option<String> = table.get("tag").unwrap();
+
+    if type_name.is_none() {
+        return Err(ReluaxError::LuaX(LuaXError::MissingField("tag".to_string())).into());
+    }
+
+    let type_name = type_name.unwrap();
+
+    write!(f, "<{}", type_name)?;
+    let mut children = None;
+    for pair in table.pairs::<String, rlua::Value>() {
+        let (key, value) = pair?;
+        if key == "tag" {
+            continue;
+        }
+        if key == "children" {
+            children = Some(value);
+            continue;
+        }
+        match value {
+            rlua::Value::Nil => write!(f, " {}=\"\"", key)?,
+            rlua::Value::Boolean(b) => write!(f, " {}=\"{}\"", key, b)?,
+            rlua::Value::Integer(i) => write!(f, " {}=\"{}\"", key, i)?,
+            rlua::Value::Number(n) => write!(f, " {}=\"{}\"", key, n)?,
+            rlua::Value::String(s) => write!(f, " {}=\"{}\"", key, s.to_str()?)?,
+            _ => return Err(ReluaxError::LuaX(LuaXError::WrongFieldType(key)).into()),
+        }
+    }
+    write!(f, ">")?;
+
+    if let Some(children) = children {
+        if let rlua::Value::Table(children) = children {
+            for child in children.sequence_values::<rlua::Value>() {
+                match child? {
+                    rlua::Value::Table(child) => table_to_html(child, f)?,
+                    rlua::Value::String(s) => write!(f, "{}", s.to_str()?)?,
+                    _ => return Err(ReluaxError::LuaX(LuaXError::NonTableChildren).into()),
+                }
+            }
+        } else {
+            return Err(ReluaxError::LuaX(LuaXError::NonTableChildren).into());
+        }
+    }
+
+    write!(f, "</{}>", type_name)?;
+
+    Ok(())
+}
+
+#[derive(Debug, Clone, clap::Parser)]
+struct Args {
+    #[clap(short = 'C', long = "current-dir", default_value = ".")]
+    current_dir: std::path::PathBuf,
+    #[clap(short = 'p', long = "port", default_value = "4310")]
+    port: u16,
+}
+
+#[tokio::main]
+async fn main() -> color_eyre::Result<()> {
+    color_eyre::install().unwrap();
+    let args = Args::parse();
+
+    let current_dir = args.current_dir;
+
+    if !current_dir.is_dir() {
+        return Err(std::io::Error::new(
+            std::io::ErrorKind::NotFound,
+            format!("{} is not a directory", current_dir.display()),
+        )
+        .into());
+    }
+
+    std::env::set_current_dir(current_dir)?;
+
+    luax::preprocess_dir(std::env::current_dir()?.as_path())?;
+
+    println!("⛱️  Reluax files preprocessed!");
+
+    let entry = PathBuf::from("reluax.lua");
+
+    if !entry.is_file() {
+        return Err(std::io::Error::new(
+            std::io::ErrorKind::NotFound,
+            format!("{} is not a file", entry.display()),
+        ).into());
+    }
+
+    server::Server::serve(args.port).await?;
+
+    Ok(())
+}