bundle lua for upstream PR (#18)

* bundle code via build.rs, much better dx for edit lua files

* remove scratch, no longer needed

Author: tjdevries

crates/vim9-gen/Cargo.toml

@@ -21,3 +21,7 @@ rmpv = "1.0.0"
 [dev-dependencies]
 insta = "1.19.0"
 pretty_assertions = "1.3.0"
+
+[build-dependencies]
+stylua = "0.14.3"
+anyhow = "1.0.62"

crates/vim9-gen/build.rs

@@ -0,0 +1,107 @@
+use std::{fmt::Write, fs, path::Path};
+
+fn get_stylua_config() -> stylua_lib::Config {
+    stylua_lib::Config::new()
+        .with_indent_type(stylua_lib::IndentType::Spaces)
+        .with_indent_width(2)
+        .with_column_width(120)
+}
+
+fn main() -> anyhow::Result<()> {
+    println!("cargo:rerun-if-changed=src/lua/");
+
+    let luadir = Path::new(env!("CARGO_MANIFEST_DIR"))
+        .parent()
+        .expect("crates_dir")
+        .parent()
+        .expect("root_dir")
+        .join("lua");
+
+    if !luadir.is_dir() {
+        std::fs::create_dir_all(&luadir).expect("to create lua directory");
+    }
+
+    let mut file = String::new();
+
+    writeln!(
+        &mut file,
+        r#"
+-------------------------------------------------------------------------------
+-- This file is auto generated by vim9jit. Do not edit by hand.
+--  All content is in the source repository.
+--  Bugs should be reported to: github.com/tjdevries/vim9jit
+--
+-- In addition, this file is considered "private" by neovim. You should
+-- not expect any of the APIs, functions, etc to be stable. They are subject
+-- to change at any time.
+-------------------------------------------------------------------------------
+
+"#
+    )?;
+
+    let source_lua = Path::new("src/lua");
+    let mut entries = Vec::new();
+    for entry in source_lua.read_dir().unwrap() {
+        if let Ok(entry) = entry {
+            entries.push(entry.path());
+        }
+    }
+
+    // always sort `init` to the top of the list
+    entries.sort_by(|a, b| {
+        if a.file_stem().unwrap() == "init" {
+            return std::cmp::Ordering::Less;
+        }
+
+        if b.file_stem().unwrap() == "init" {
+            return std::cmp::Ordering::Greater;
+        }
+
+        a.cmp(b)
+    });
+
+    // Iterate over each of the entries and write them to the new file
+    for entry in entries {
+        // get entry without the extension
+        let name = entry
+            .file_stem()
+            .expect("file_stem")
+            .to_str()
+            .expect("to_str");
+
+        let contents = fs::read_to_string(&entry)?;
+        let contents = contents.replace("require('_vim9script')", "NVIM9");
+        let contents = contents.replace("require(\"_vim9script\")", "NVIM9");
+        let contents = contents.replace("require \"_vim9script\"", "NVIM9");
+
+        if name == "_" {
+            writeln!(&mut file, "{}\n", contents)?;
+        } else if name == "init" {
+            // init.lua declares our base module and it's guaranteed to be first.
+            writeln!(&mut file, "local NVIM9 = (function() {} end)()\n", contents)?;
+        } else {
+            writeln!(
+                &mut file,
+                "NVIM9['{}'] = (function() {} end)()",
+                name, contents
+            )?;
+        }
+    }
+
+    writeln!(&mut file, "")?;
+    writeln!(&mut file, "return NVIM9")?;
+
+    // Format the final lib
+    let file = stylua_lib::format_code(
+        &file,
+        get_stylua_config(),
+        None,
+        stylua_lib::OutputVerification::None,
+    )
+    .expect(&format!("to format code: {}", file));
+
+    let init_lua = luadir.join("_vim9script.lua");
+    fs::write(init_lua, file)?;
+
+    Ok(())
+}

crates/vim9-gen/src/lua/_.lua

@@ -0,0 +1,15 @@
+vim.cmd [[
+function! _Vim9ScriptFn(name, args) abort
+  try
+    let ret = function(a:name, a:args)()
+  catch
+    echo "Failed..."
+    echo a:name
+    echo a:args
+
+    throw v:errmsg
+  endtry
+
+  return [ret, a:args]
+endfunction
+]]

crates/vim9-gen/src/lua/autoload.lua

@@ -0,0 +1,3 @@
+return function(path)
+  return loadfile(path)()
+end

crates/vim9-gen/src/lua/bool.lua

@@ -0,0 +1,3 @@
+return function(...)
+  return require("_vim9script").convert.to_vim_bool(...)
+end

crates/vim9-gen/src/lua/convert.lua

@@ -0,0 +1,51 @@
+local M = {}
+
+M.decl_bool = function(val)
+  if type(val) == "boolean" then
+    return val
+  elseif type(val) == "number" then
+    if val == 0 then
+      return false
+    elseif val == 1 then
+      return true
+    else
+      error(string.format("bad number passed to bool declaration: %s", val))
+    end
+  end
+
+  error(string.format("invalid bool declaration: %s", vim.inspect(val)))
+end
+
+M.decl_dict = function(val)
+  if type(val) == "nil" then
+    return vim.empty_dict()
+  elseif type(val) == "table" then
+    if vim.tbl_isempty(val) then
+      return vim.empty_dict()
+    elseif vim.tbl_islist(val) then
+      error(string.format("Cannot pass list to dictionary? %s", vim.inspect(val)))
+    else
+      return val
+    end
+  end
+
+  error(string.format("invalid dict declaration: %s", vim.inspect(val)))
+end
+
+M.to_vim_bool = function(val)
+  if type(val) == "boolean" then
+    return val
+  elseif type(val) == "number" then
+    return val ~= 0
+  elseif type(val) == "string" then
+    return string.len(val) ~= 0
+  elseif type(val) == "table" then
+    return not vim.tbl_isempty(val)
+  elseif val == nil then
+    return false
+  end
+
+  error("unhandled type: " .. vim.inspect(val))
+end
+
+return M