src/lobject.js: Give functions and objects ids

2 files changed, 10 insertions, 0 deletions

diff --git a/src/lobject.js b/src/lobject.js
index 04a08c0..a5219e5 100644
--- a/src/lobject.js
+++ b/src/lobject.js
@@ -576,6 +576,15 @@ const luaO_pushvfstring = function(L, fmt, argp) {
                     v instanceof CClosure ||
                     v instanceof lfunc.UpVal) {
                     pushstr(L, defs.to_luastring("0x"+v.id.toString(16)));
+                } else if (v === null) { /* handle null before checking for typeof == object */
+                    pushstr(L, defs.to_luastring("null"));
+                } else if (typeof v === "function" || typeof v === "object") {
+                    let id = L.l_G.ids.get(v);
+                    if (!id) {
+                        id = L.l_G.id_counter++;
+                        L.l_G.ids.set(v, id);
+                    }
+                    pushstr(L, defs.to_luastring("0x"+id.toString(16)));
                 } else if (typeof v === "number") { /* before check object as null is an object */
                     pushstr(L, defs.to_luastring("Number("+v+")"));
                 } else if (typeof v === "string") { /* before check object as null is an object */
diff --git a/src/lstate.js b/src/lstate.js
index d09bebd..db046dc 100644
--- a/src/lstate.js
+++ b/src/lstate.js
@@ -74,6 +74,7 @@ class global_State {
 
     constructor() {
         this.id_counter = 0; /* used to give objects unique ids */
+        this.ids = new WeakMap();
 
         this.mainthread = null;
         this.l_registry = new lobject.TValue(CT.LUA_TNIL, null);