feat: support wasm

Mark24Code · Mark24Code · commit 5014dc8273e2 · 2024-06-29T21:06:15.000+08:00
diff --git a/Rakefile b/Rakefile
@@ -19,15 +19,25 @@ namespace :mruby do
       host_agent.clean_all
       host_agent.mruby_download
     end
-    host_agent.mruby_config
-    if !host_agent.mruby_build_dir.exist?
+
+    config_changed = host_agent.config_changed
+    if config_changed
+      host_agent.mruby_config
+    end
+
+    if !host_agent.mruby_build_dir.exist? || config_changed
       host_agent.mruby_build
     end
   end
 
   task :init_wasm => [:"mruby:init"] do
-    wasm_agent.mruby_config
-    if !wasm_agent.mruby_build_dir.exist?
+
+    config_changed = wasm_agent.config_changed
+    if config_changed
+      wasm_agent.mruby_config
+    end
+
+    if !wasm_agent.mruby_build_dir.exist? || config_changed
       wasm_agent.mruby_build
     end
   end
@@ -48,6 +58,11 @@ task :build => [:"mruby:init"] do
   host_agent.build
 end
 
+desc "run wasm program"
+task :"run:wasm" => [:"mruby:init_wasm"] do
+  wasm_agent.run
+end
+
 desc "build wasm program"
 task :"build:wasm" => [:"mruby:init_wasm"] do
   wasm_agent.build
diff --git a/config/wasm.rb b/config/wasm.rb
@@ -1,5 +1,6 @@
 MRuby::Build.new do |conf|
   toolchain :gcc
+  conf.gem :mgem => 'mruby-os'
   conf.gembox 'default'
 end
 
diff --git a/task/agent/base.rb b/task/agent/base.rb
@@ -2,11 +2,13 @@
 require_relative "./utils"
 
 class BaseAgent
+
   include Utils
 
   attr_accessor :platform
   attr_reader :mruby_dir, :cache_dir, :build_dir, :mruby_build_dir
   attr_reader :mruby, :mrbc
+  attr_reader :custom_build_config_file, :copied_custom_build_config_file
   def initialize(app_name:, debug: false)
     @app_name = app_name
 
@@ -102,6 +104,11 @@ def mruby_config
     shell_cp @custom_build_config_file, @copied_custom_build_config_file
   end
 
+  def config_changed
+    file_content_change? @custom_build_config_file, @copied_custom_build_config_file
+  end
+
+
   def clean_dir(dir_path)
     shell_rm dir_path
   end
@@ -150,12 +157,10 @@ def build_to_c_code
     end
   end
 
-  def wrap_code
-
-  end
 
   def run
-    pack_code(@cache_dir)
+    shell_clean @cache_dir
+    pack_code @cache_dir
     shell "#{@mruby} #{@cache_dir}/main.rb"
   end
 
diff --git a/task/agent/wasm.rb b/task/agent/wasm.rb
@@ -18,34 +18,42 @@ def build
     shell_clean @build_dir
     @host_agent.pack_code(@build_dir)
     @host_agent.build_to_c_code
-    build_code
+    build_code @build_dir
     build_clean
   end
 
-  def web_template
-    File.open("#{@build_dir}/main.c", "w") do |f|
+  def generate_web_template(dir_path)
+    File.open("#{dir_path}/index.html", "w") do |f|
     template = <<-CODE
-    #include <mruby.h>
-    #include <mruby/irep.h>
-    extern const uint8_t #{@code_wrapper_name}[];
-
-    int
-    main(void)
-    {
-      mrb_state *mrb = mrb_open();
-      if (!mrb) { /* handle error */ }
-      mrb_load_irep(mrb, #{@code_wrapper_name});
-      mrb_close(mrb);
-      return 0;
-    }
-
+<!DOCTYPE html>
+<html lang="en">
+<head>
+	<meta charset="UTF-8">
+	<meta name="viewport" content="width=device-width, initial-scale=1.0">
+	<title>Document</title>
+</head>
+<body>
+    <h1>Hello World</h1>
+    <script src="./app.js"></script>
+</body>
+</html>
     CODE
 
-        f.puts template
-        end
+    f.puts template
+    end
+  end
+
+  def serve(dir_path)
+    system("ruby -run -e httpd #{dir_path}")
+  end
+
+  def run
+    build
+    serve @build_dir
   end
 
-  def build_code
-    shell "emcc -s WASM=1 -Os -I #{@mruby_include_dir} #{@build_dir}/*.c #{@mruby_lib_dir}/libmruby.a -lm -o #{@build_dir}/#{@app_name}.js --closure 1"
+  def build_code(dir_path)
+    shell "emcc -s WASM=1 -Os -I #{@mruby_include_dir} #{dir_path}/*.c #{@mruby_lib_dir}/libmruby.a -lm -o #{dir_path}/#{@app_name}.js --closure 1"
+    generate_web_template(dir_path)
   end
 end
