sweet-moon 0.0.1

data/components/api.rb

@@ -0,0 +1,83 @@
+require 'ffi'
+
+module Component
+  API = {
+    open!: ->(shared_objects) {
+      api = Module.new
+
+      api.extend FFI::Library
+
+      # TODO: Lua Constants
+      # attach_variable
+
+      if shared_objects.size > 1
+        # TODO: Dangerous
+        api.ffi_lib_flags(:global, :now)
+      else
+        api.ffi_lib_flags(:local, :now)
+      end
+
+      api.ffi_lib(*shared_objects.map { |o| o[:path] })
+
+      api
+    },
+
+    inject_callbacks!: ->(api, reference, injections) {
+      injections[:callbacks].each do |callback|
+        api.callback(*callback[:ffi])
+        reference[:signatures][:functions].each do |signature|
+          next unless callback[:overwrite][signature[:ffi].first]
+
+          signature[:ffi] = [
+            signature[:ffi].first, *callback[:overwrite][signature[:ffi].first]
+          ]
+        end
+      end
+    },
+
+    attach!: ->(api, reference, injections) {
+      API[:inject_callbacks!].(api, reference, injections)
+
+      signatures = {}
+
+      reference[:signatures][:functions].each do |signature|
+        exists = false
+
+        api.ffi_libraries.each do |ffi_library|
+          exists = true if ffi_library.find_function(signature[:ffi].first.to_s)
+        end
+
+        next unless exists
+
+        api.attach_function(*signature[:ffi])
+        signatures[signature[:ffi].first] = {
+          source: signature[:source],
+          input: signature[:ffi][1], output: signature[:ffi][2]
+        }
+      end
+
+      API[:inject_macros!].(api, signatures, reference, injections)
+    },
+
+    inject_macros!: ->(api, signatures, reference, injections) {
+      reference[:signatures][:macros].each do |macro_signature|
+        macro = injections[:macros][macro_signature[:name].to_sym]
+        next unless macro
+
+        missing = macro[:requires].detect { |function| !signatures[function] }
+
+        next if missing
+
+        api.define_singleton_method macro_signature[:name], macro[:injection]
+        api.define_method macro_signature[:name], macro[:injection]
+
+        signatures[macro_signature[:name].to_sym] = {
+          source: macro_signature[:source], macro: true,
+          requires: macro[:requires].map { |f| signatures[f] }
+        }
+      end
+
+      { api: api, signatures: signatures }
+    }
+  }
+end

data/components/injections/injections_503.rb

@@ -0,0 +1,21 @@
+module Component
+  module V503
+    Injections = {
+      callbacks: [
+        {
+          ffi: [:cfunction, [:pointer], :int],
+          overwrite: {
+            lua_pushcclosure: [%i[pointer cfunction int], :void],
+            lua_tocfunction: [%i[pointer int], :cfunction]
+          }
+        }
+      ],
+      macros: {
+        lua_pop: {
+          requires: [:lua_settop],
+          injection: ->(l, n = 1) { lua_settop(l, -n - 1) }
+        }
+      }
+    }
+  end
+end

data/components/injections/injections_514.rb

@@ -0,0 +1,29 @@
+module Component
+  module V514
+    Injections = {
+      callbacks: [
+        {
+          ffi: [:cfunction, [:pointer], :int],
+          overwrite: {
+            lua_pushcclosure: [%i[pointer cfunction int], :void],
+            lua_tocfunction: [%i[pointer int], :cfunction]
+          }
+        }
+      ],
+      macros: {
+        lua_pop: {
+          requires: [:lua_settop],
+          injection: ->(l, n = 1) { lua_settop(l, -n - 1) }
+        },
+        lua_tostring: {
+          requires: [:lua_tolstring],
+          injection: ->(l, i) { lua_tolstring(l, i, nil) }
+        },
+        luaL_typename: {
+          requires: %i[lua_typename lua_type],
+          injection: ->(l, i) { lua_typename(l, lua_type(l, i)) }
+        }
+      }
+    }
+  end
+end

data/components/injections/injections_542.rb

@@ -0,0 +1,49 @@
+module Component
+  module V542
+    Injections = {
+      callbacks: [
+        {
+          ffi: [:cfunction, [:pointer], :int],
+          overwrite: {
+            lua_pushcclosure: [%i[pointer cfunction int], :void],
+            lua_tocfunction: [%i[pointer int], :cfunction]
+          }
+        }
+      ],
+      macros: {
+        lua_pop: {
+          requires: [:lua_settop],
+          injection: ->(l, n = 1) { lua_settop(l, -n - 1) }
+        },
+        lua_pcall: {
+          requires: [:lua_pcallk],
+          injection: ->(l, n, r, f) { lua_pcallk(l, n, r, f, 0, -1) }
+        },
+        luaL_loadfile: {
+          requires: [:luaL_loadfilex],
+          injection: ->(l, f) { luaL_loadfilex(l, f, nil) }
+        },
+        lua_tonumber: {
+          requires: [:lua_tonumberx],
+          injection: ->(l, i) { lua_tonumberx(l, i, nil) }
+        },
+        lua_tointeger: {
+          requires: [:lua_tointegerx],
+          injection: ->(l, i) { lua_tointegerx(l, i, nil) }
+        },
+        lua_tostring: {
+          requires: [:lua_tolstring],
+          injection: ->(l, i) { lua_tolstring(l, i, nil) }
+        },
+        luaL_typename: {
+          requires: %i[lua_typename lua_type],
+          injection: ->(l, i) { lua_typename(l, lua_type(l, i)) }
+        },
+        lua_insert: {
+          requires: %i[lua_rotate],
+          injection: ->(l, idx) { lua_rotate(l, idx, 1) }
+        }
+      }
+    }
+  end
+end

data/components/injections.rb

@@ -0,0 +1,11 @@
+require_relative 'injections/injections_503'
+require_relative 'injections/injections_514'
+require_relative 'injections/injections_542'
+
+module Component
+  Injections = {
+    '5.0.3' => { version: '5.0.3', injections: V503::Injections },
+    '5.1.4' => { version: '5.1.4', injections: V514::Injections },
+    '5.4.2' => { version: '5.4.2', injections: V542::Injections }
+  }
+end

data/components/interpreters/50/function.rb

@@ -0,0 +1,52 @@
+require_relative '../../../logic/interpreters/interpreter_50'
+require_relative '../../../dsl/errors'
+
+require_relative 'writer'
+require_relative 'reader'
+
+module Component
+  module V50
+    Function = {
+      push!: ->(api, state, closure) {
+        handler = ->(current_state) {
+          input = Reader[:read_all!].(api, current_state)
+          result = closure.(*input)
+          Writer[:push!].(api, current_state, result)
+          return 1
+        }
+
+        api.lua_pushcclosure(state, handler, 0)
+      },
+
+      read!: ->(api, state, _stack_index) {
+        reference = api.luaL_ref(
+          state, Logic::V50::Interpreter[:LUA_REGISTRYINDEX]
+        )
+
+        { value: ->(input = [], output = 1) {
+          api.lua_rawgeti(
+            state, Logic::V50::Interpreter[:LUA_REGISTRYINDEX], reference
+          )
+
+          input.each do |value|
+            Writer[:push!].(api, state, value)
+          end
+
+          result = Interpreter[:call!].(api, state, input.size, output)
+
+          if result[:error]
+            raise SweetMoon::Errors::SweetMoonErrorHelper.for(
+              result[:error][:status]
+            ), result[:error][:value]
+          end
+
+          result = Reader[:read_all!].(api, state)
+
+          return result.first if output == 1
+
+          result
+        }, pop: false }
+      }
+    }
+  end
+end

data/components/interpreters/50/interpreter.rb

@@ -0,0 +1,105 @@
+require_relative '../../../logic/interpreters/interpreter_50'
+
+require_relative 'reader'
+require_relative 'writer'
+require_relative 'table'
+
+module Component
+  module V50
+    Interpreter = {
+      version: Logic::V50::Interpreter[:version],
+
+      create_state!: ->(api) {
+        state = api.lua_open
+        { state: state, error: state ? nil : :MemoryAllocation }
+      },
+
+      open_standard_libraries!: ->(api, state) {
+        api.luaopen_base(state)
+        api.luaopen_table(state)
+        api.luaopen_io(state)
+        api.luaopen_string(state)
+        api.luaopen_math(state)
+
+        api.lua_settop(state, -7 - 1)
+
+        { state: state }
+      },
+
+      load_file_and_push_chunck!: ->(api, state, path) {
+        result = api.luaL_loadfile(state, path)
+        { state: state, error: Interpreter[:_error].(api, state, result, pull: true) }
+      },
+
+      push_chunk!: ->(api, state, value) {
+        result = api.luaL_loadbuffer(state, value, value.size, value)
+
+        { state: state, error: Interpreter[:_error].(api, state, result, pull: true) }
+      },
+
+      set_table!: ->(api, state, variable, value) {
+        Table[:set!].(api, state, variable, value)
+      },
+
+      push_value!: ->(api, state, value) {
+        Writer[:push!].(api, state, value)
+        { state: state }
+      },
+
+      pop_and_set_as!: ->(api, state, variable) {
+        api.lua_pushstring(state, variable)
+        api.lua_insert(state, -2)
+        api.lua_settable(state, Logic::V50::Interpreter[:LUA_GLOBALSINDEX])
+        { state: state }
+      },
+
+      get_variable_and_push!: ->(api, state, variable, key = nil) {
+        api.lua_pushstring(state, variable.to_s)
+        api.lua_gettable(state, Logic::V50::Interpreter[:LUA_GLOBALSINDEX])
+
+        Table[:read_field_and_push!].(api, state, key, -1) unless key.nil?
+
+        { state: state }
+      },
+
+      call!: ->(api, state, inputs = 0, outputs = 1) {
+        result = api.lua_pcall(state, inputs, outputs, 0)
+        { state: state, error: Interpreter[:_error].(api, state, result, pull: true) }
+      },
+
+      read_and_pop!: ->(api, state, stack_index = -1, extra_pop: false) {
+        result = Component::V50::Reader[:read!].(api, state, stack_index)
+
+        api.lua_settop(state, -2) if result[:pop]
+        api.lua_settop(state, -2) if extra_pop
+
+        { state: state, output: result[:value] }
+      },
+
+      read_all!: ->(api, state) {
+        result = Reader[:read_all!].(api, state)
+
+        { state: state, output: result }
+      },
+
+      destroy_state!: ->(api, state) {
+        result = api.lua_close(state)
+
+        { state: nil, error: Interpreter[:_error].(api, state, result) }
+      },
+
+      _error: ->(api, state, code, options = {}) {
+        status = Logic::V50::Interpreter[:error][
+          Logic::V50::Interpreter[:status][code]
+        ] || :error
+
+        if code.is_a?(Numeric) && code >= 1
+          return { status: status } unless options[:pull]
+
+          { status: status,
+            value: Interpreter[:read_and_pop!].(api, state, -1)[:output] }
+        end
+      }
+    }
+  end
+end

data/components/interpreters/50/reader.rb

@@ -0,0 +1,65 @@
+require 'ffi'
+
+require_relative 'interpreter'
+require_relative 'function'
+require_relative 'table'
+
+module Component
+  module V50
+    Reader = {
+      read_all!: ->(api, state) {
+        (1..api.lua_gettop(state)).map do
+          Interpreter[:read_and_pop!].(api, state)[:output]
+        end.reverse
+      },
+
+      read!: ->(api, state, stack_index = -1) {
+        stack_index = api.lua_gettop(state) if stack_index == -1
+
+        type = api.lua_typename(state, api.lua_type(state, stack_index)).read_string
+
+        case type
+        when 'string'
+          Reader[:read_string!].(api, state, stack_index)
+        when 'number'
+          Reader[:read_number!].(api, state, stack_index)
+        when 'no value'
+          { value: nil, pop: true, type: type }
+        when 'nil'
+          { value: nil, pop: true }
+        when 'boolean'
+          Reader[:read_boolean!].(api, state, stack_index)
+        when 'table'
+          Table[:read!].(api, state, stack_index)
+        when 'function'
+          Function[:read!].(api, state, stack_index)
+        else
+          # none nil boolean lightuserdata number
+          # string table function userdata thread
+          { value: "#{type}: 0x#{api.lua_topointer(state, stack_index).address}",
+            type: type, pop: true }
+        end
+      },
+
+      read_string!: ->(api, state, stack_index) {
+        { value: api.lua_tostring(state, stack_index).read_string,
+          pop: true }
+      },
+
+      read_number!: ->(api, state, stack_index) {
+        if api.respond_to?(:lua_isinteger) &&
+           api.respond_to?(:lua_tointeger) &&
+           api.lua_isinteger(state, stack_index) == 1
+
+          return { value: api.lua_tointeger(state, stack_index), pop: true }
+        end
+
+        { value: api.lua_tonumber(state, stack_index), pop: true }
+      },
+
+      read_boolean!: ->(api, state, stack_index) {
+        { value: api.lua_toboolean(state, stack_index) == 1, pop: true }
+      }
+    }
+  end
+end

data/components/interpreters/50/table.rb

@@ -0,0 +1,99 @@
+require_relative '../../../logic/tables'
+
+require_relative 'writer'
+require_relative 'reader'
+
+module Component
+  module V50
+    Table = {
+      push!: ->(api, state, list, stack_index = -1) {
+        stack_index = api.lua_gettop(state) if stack_index == -1
+
+        api.lua_newtable(state)
+
+        if list.is_a? Hash
+          list.each_key do |key|
+            Writer[:push!].(api, state, key)
+            Writer[:push!].(api, state, list[key])
+            api.lua_settable(state, stack_index + 1)
+          end
+        else
+          list.each_with_index do |value, index|
+            Writer[:push!].(api, state, index + 1)
+            Writer[:push!].(api, state, value)
+            api.lua_settable(state, stack_index + 1)
+          end
+        end
+      },
+
+      read!: ->(api, state, stack_index) {
+        stack_index = api.lua_gettop(state) if stack_index == -1
+
+        type = api.lua_typename(state, api.lua_type(state, stack_index)).read_string
+
+        api.lua_pushnil(state)
+
+        return nil if type != 'table'
+
+        tuples = []
+
+        while api.lua_next(state, stack_index).positive?
+          value = Reader[:read!].(api, state, stack_index + 2)
+          key = Reader[:read!].(api, state, stack_index + 1)
+          api.lua_settop(state, -2) if value[:pop]
+
+          tuples << [key[:value], value[:value]]
+
+          break if value[:type] == 'no value'
+        end
+
+        { value: Logic::Tables[:to_hash_or_array].(tuples), pop: true }
+      },
+
+      read_field_and_push!: ->(api, state, expected_key, stack_index) {
+        stack_index = api.lua_gettop(state) if stack_index == -1
+
+        type = api.lua_typename(state, api.lua_type(state, stack_index)).read_string
+
+        api.lua_pushnil(state)
+
+        return nil if type != 'table'
+
+        result = nil
+
+        while api.lua_next(state, stack_index).positive?
+          value = Reader[:read!].(api, state, stack_index + 2)
+          key = Reader[:read!].(api, state, stack_index + 1)
+
+          api.lua_settop(state, -2) if value[:pop]
+
+          key_type = api.lua_typename(
+            state, api.lua_type(state, stack_index + 1)
+          ).read_string
+
+          if Table[:is_same_key].(key_type, key[:value], expected_key)
+            result = value[:value]
+            break
+          end
+
+          break if value[:type] == 'no value'
+        end
+
+        api.lua_settop(state, -2)
+
+        Writer[:push!].(api, state, result)
+      },
+
+      is_same_key: ->(lua_key_type, lua_key, ruby_key) {
+        lua_key == case lua_key_type
+                   when 'string'
+                     ruby_key.to_s
+                   when 'number'
+                     ruby_key.to_f
+                   else
+                     ruby_key
+                   end
+      }
+    }
+  end
+end

data/components/interpreters/50/writer.rb

@@ -0,0 +1,45 @@
+require_relative 'function'
+require_relative 'table'
+
+module Component
+  module V50
+    Writer = {
+      push!: ->(api, state, value) {
+        case Writer[:_to_lua_type].(value)
+        when 'string'
+          api.lua_pushstring(state, value.to_s)
+        when 'number'
+          api.lua_pushnumber(state, value)
+        when 'integer'
+          if api.respond_to? :lua_pushinteger
+            api.lua_pushinteger(state, value)
+          else
+            api.lua_pushnumber(state, value)
+          end
+        when 'nil'
+          api.lua_pushnil(state)
+        when 'boolean'
+          api.lua_pushboolean(state, value ? 1 : 0)
+        when 'table'
+          Table[:push!].(api, state, value)
+        when 'function'
+          Function[:push!].(api, state, value)
+        else
+          api.lua_pushstring(
+            state, "#<#{value.class}:0x#{format('%016x', value.object_id)}>"
+          )
+        end
+      },
+
+      _to_lua_type: ->(value) {
+        return 'nil' if value.nil?
+        return 'function' if value.is_a?(Proc)
+        return 'integer' if value.is_a? Integer
+        return 'number' if value.is_a? Float
+        return 'table' if value.is_a?(Hash) || value.is_a?(Array)
+        return 'string' if value.is_a?(String) || value.instance_of?(Symbol)
+        return 'boolean' if [true, false].include? value
+      }
+    }
+  end
+end

data/components/interpreters/51/function.rb

@@ -0,0 +1,52 @@
+require_relative '../../../logic/interpreters/interpreter_51'
+require_relative '../../../dsl/errors'
+
+require_relative 'writer'
+require_relative 'reader'
+
+module Component
+  module V51
+    Function = {
+      push!: ->(api, state, closure) {
+        handler = ->(current_state) {
+          input = Reader[:read_all!].(api, current_state)
+          result = closure.(*input)
+          Writer[:push!].(api, current_state, result)
+          return 1
+        }
+
+        api.lua_pushcclosure(state, handler, 0)
+      },
+
+      read!: ->(api, state, _stack_index) {
+        reference = api.luaL_ref(
+          state, Logic::V51::Interpreter[:LUA_REGISTRYINDEX]
+        )
+
+        { value: ->(input = [], output = 1) {
+          api.lua_rawgeti(
+            state, Logic::V51::Interpreter[:LUA_REGISTRYINDEX], reference
+          )
+
+          input.each do |value|
+            Writer[:push!].(api, state, value)
+          end
+
+          result = Interpreter[:call!].(api, state, input.size, output)
+
+          if result[:error]
+            raise SweetMoon::Errors::SweetMoonErrorHelper.for(
+              result[:error][:status]
+            ), result[:error][:value]
+          end
+
+          result = Reader[:read_all!].(api, state)
+
+          return result.first if output == 1
+
+          result
+        }, pop: false }
+      }
+    }
+  end
+end