feat: wrappers + improvements

Author: mdwagner

src/io.cr

@@ -0,0 +1,2 @@
+require "./luajit/wrappers/io"
+alias LuajitIO = Luajit::Wrappers::IO

src/luajit.cr

@@ -6,6 +6,11 @@ require "./luajit/lua_any"
 require "./luajit/*"
 
 module Luajit
+  # :nodoc:
+  # Define Wrappers namespace
+  module Wrappers
+  end
+
   # Same as `LuaState.create`
   def self.new : LuaState
     LuaState.create

src/luajit/iterators.cr

@@ -20,22 +20,24 @@ module Luajit
       result : TablePair? = nil
 
       while @state.next(@index)
-        key = case @state.get_type(KEY)
-              when .string?
-                @state.to_string(KEY)
-              when .number?
-                @state.to_f(KEY)
-              else
-                @state.pop(1)
-                next
-              end
-
-        value = if any_value = @state.to_any?(VALUE)
-                  any_value
-                else
-                  @state.pop(1)
-                  next
-                end
+        key =
+          case @state.get_type(KEY)
+          when .string?
+            @state.to_string(KEY)
+          when .number?
+            @state.to_f(KEY)
+          else
+            @state.pop(1)
+            next
+          end
+
+        value =
+          if any_value = @state.to_any?(VALUE)
+            any_value
+          else
+            @state.pop(1)
+            next
+          end
 
         result = {key, value}
 

src/luajit/lua_any.cr

@@ -52,7 +52,7 @@ module Luajit
       @raw.pretty_print(pp)
     end
 
-    def ==(other : JSON::Any)
+    def ==(other : LuaAny)
       raw == other.raw
     end
 
@@ -61,5 +61,22 @@ module Luajit
     end
 
     def_hash raw
+
+    # Returns a new LuaAny instance with the `raw` value `dup`ed.
+    def dup
+      LuaAny.new(raw.dup)
+    end
+
+    # Returns a new LuaAny instance with the `raw` value `clone`ed.
+    # NOTE: If the `raw` value is a `LuaRef`, it will create a new one.
+    def clone(state : LuaState)
+      case value = raw
+      when LuaRef
+        state.get_ref_value(value)
+        LuaAny.new(state.create_ref)
+      else
+        LuaAny.new(raw.clone)
+      end
+    end
   end
 end

src/luajit/lua_ref.cr

@@ -1,5 +1,12 @@
 module Luajit
   record LuaRef, ref : Int32, type : LuaType do
+    def pcall(state : LuaState, nargs : Int32, nresults : Int32) : LuaStatus
+      raise LuaError.new("Ref is not a function") unless type.function?
+      state.get_ref_value(ref)
+      state.insert(-(nargs + 1)) if nargs > 0
+      state.pcall(nargs, nresults)
+    end
+
     def inspect(io : IO) : Nil
       io << "LuaRef(@type=#{type.to_s}, @ref=#{ref})"
     end

src/luajit/wrappers/io.cr

@@ -0,0 +1,10 @@
+require "../../luajit"
+
+class Luajit::Wrappers::IO < Luajit::LuaObject
+  global_name "__IO__"
+
+  property io : ::IO
+
+  def initialize(@io)
+  end
+end

src/luajit/wrappers/io.lua

@@ -0,0 +1,4 @@
+---@meta _
+
+---@class (exact) __IO__
+__IO__ = {}

src/luajit/wrappers/nil.lua

@@ -0,0 +1,4 @@
+---@meta _
+
+---@class (exact) NIL
+NIL = {}

src/luajit/wrappers/path.cr

@@ -0,0 +1,68 @@
+require "../../luajit"
+
+class Luajit::Wrappers::Path < Luajit::LuaObject
+  global_name "__PATH__"
+
+  # ---@field new fun(path: string?): self
+  def_class_method "new" do |state|
+    _self = (
+      if state.is_none?(1)
+        new(::Path.new)
+      elsif state.is_string?(1)
+        new(::Path.new(state.to_string(1)))
+      else
+        state.raise_type_error!(1, "expected string or nil")
+      end
+    )
+    Luajit.setup_userdata(state, _self, self)
+    1
+  end
+
+  # ---@field home fun(): self
+  def_class_method "home" do |state|
+    _self = new(::Path.home)
+    Luajit.setup_userdata(state, _self, self)
+    1
+  end
+
+  # ---@field is_absolute fun(self): boolean
+  def_instance_method "is_absolute" do |state|
+    state.assert_userdata!(1)
+    _self = Luajit.userdata_value(state, self, 1)
+    state.push _self.path.absolute?
+    1
+  end
+
+  # ---@field anchor fun(self): self?
+  def_instance_method "anchor" do |state|
+    state.assert_userdata!(1)
+    _self = Luajit.userdata_value(state, self, 1)
+
+    new_path = _self.path.anchor
+
+    unless new_path
+      state.push(nil)
+      next 1
+    end
+
+    state.get_global("__PATH__")
+    state.get_field(-1, "new")
+    state.pcall(0, 1)
+    __self = Luajit.userdata_value(state, self, -1)
+    __self.path = new_path
+    1
+  end
+
+  # ---@operator tostring(self): self
+  def_instance_method "__tostring" do |state|
+    state.assert_userdata!(1)
+    _self = Luajit.userdata_value(state, self, 1)
+    state.push _self.path.to_s
+    1
+  end
+
+  property path : ::Path
+
+  def initialize(@path)
+  end
+end

src/luajit/wrappers/path.lua

@@ -0,0 +1,37 @@
+---@meta _
+
+---@class __PATH__DriveAndRoot
+---@field drive string|NIL
+---@field root string|NIL
+
+---@class __PATH__ExpandOptions
+---@field base __PATH__|string?
+---@field home __PATH__|string|boolean?
+---@field expand_base boolean?
+
+---@class (exact) __PATH__
+---@field new fun(path: string?): self
+---@field home fun(): self
+---@field is_absolute fun(self): boolean
+---@field anchor fun(self): self?
+---@field basename fun(self, suffix: string?): string
+---@field dirname fun(self): string
+---@field drive fun(self): self?
+---@field drive_and_root fun(self): __PATH__DriveAndRoot
+---@field each_parent fun(self, cb: fun(path: __PATH__))
+---@field each_part fun(self, cb: fun(str: string))
+---@field ends_with_separator fun(self): boolean
+---@field expand fun(self, options: __PATH__ExpandOptions?)
+---@field extension fun(self): string
+---@field join fun(self, parts: string[]|string): self
+---@field normalize fun(self, remove_final_separator: boolean?)
+---@field parent fun(self): self
+---@field parents fun(self): self[]
+---@field parts fun(self): string[]
+---@field relative_to fun(self, base: self|string): self
+---@field is_relative_to fun(self, base: self|string): self?
+---@field root fun(self): self?
+---@field sibling fun(self, name: self|string): self
+---@field stem fun(self): string
+---@operator concat(): string
+__PATH__ = {}