+ bin/ruby + shebang support

Author: palkan

src/templates/default/bin/ruby

@@ -0,0 +1,28 @@
+#!/usr/bin/env -S node --disable-warning=ExperimentalWarning
+
+import initVM from "../lib/rails.js";
+import util from "node:util";
+import fs from "node:fs";
+
+const args = process.argv.slice(2);
+
+if (args[0] && fs.existsSync(process.cwd() + '/' + args[0])) {
+  const firstLine = fs.readFileSync(process.cwd() + '/' + args[0], 'utf8').split('\n')[0];
+  // Check if called via shebang
+  if (firstLine.includes('#!/usr/bin/env ruby')) {
+    const path = args.shift();
+    const vm = await initVM({ env: { "HOME": "/rails-vm" } });
+
+    await vm.evalAsync(`
+      args = ${util.inspect(args)}
+      ARGV.replace(args)
+
+      begin
+        load "./${path}"
+      rescue SystemExit
+      end
+    `);
+  }
+} else {
+  await initVM({ env: { "HOME": "/rails-vm" }, args, skipRails: true });
+}

src/templates/default/lib/rails.js

@@ -4,11 +4,12 @@ import fs from "fs/promises";
 import { setupDatabase } from "./database.js";
 
 const rubyWasm = new URL("../node_modules/@rails-tutorial/wasm/dist/rails.wasm", import.meta.url).pathname;
+
 const railsRootDir = new URL("../workspace/store", import.meta.url).pathname;
 const pgDataDir = new URL("../pgdata", import.meta.url).pathname;
 
 export default async function initVM(vmopts = {}) {
-  const { env, args } = vmopts;
+  const { env, args, skipRails } = vmopts;
   const binary = await fs.readFile(rubyWasm);
   const module = await WebAssembly.compile(binary);
 
@@ -19,6 +20,8 @@ export default async function initVM(vmopts = {}) {
     `/workspace${process.cwd().slice(workspaceDir.length)}` :
     "";
 
+  const cliArgs = args?.length ? ['ruby.wasm'].concat(args) : undefined;
+
   const wasi = new WASI(
     {
       env: {"RUBYOPT": "-EUTF-8 -W0", ...env},
@@ -27,26 +30,31 @@ export default async function initVM(vmopts = {}) {
       preopens: {
         "/workspace": workspaceDir
       },
-      args: args || [] // FIXME: doesn't work
+      args: cliArgs
     }
   );
+
   const { vm } = await RubyVM.instantiateModule({
-    module, wasip1: wasi
+    module,
+    wasip1: wasi,
+    args: cliArgs
   });
 
-  try {
-    await fs.readdir(railsRootDir);
-    await setupDatabase(pgDataDir);
-  } catch (error) {
-    // not database directory — skip it
-  }
+  if (!skipRails) {
+    try {
+      await fs.readdir(railsRootDir);
+      await setupDatabase(pgDataDir);
+    } catch (error) {
+      // not database directory — skip it
+    }
 
-  vm.eval(`
-    Dir.chdir("${workdir}") unless "${workdir}".empty?
-    require "/rails-vm/boot"
+    vm.eval(`
+      Dir.chdir("${workdir}") unless "${workdir}".empty?
+      require "/rails-vm/boot"
 
-    require "js"
-  `)
+      require "js"
+    `)
+  }
 
   return vm;
 }

src/templates/rails-new/workspace/store/bin/rails

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+APP_PATH = File.expand_path("../config/application", __dir__)
+require_relative "../config/boot"
+require "rails/commands"