aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--README.md2
-rw-r--r--src/lauxlib.js2
-rw-r--r--src/lobject.js4
-rw-r--r--src/lstrlib.js14
-rw-r--r--tests/lstrlib.js102
5 files changed, 121 insertions, 3 deletions
diff --git a/README.md b/README.md
index 84195cf..e8257cc 100644
--- a/README.md
+++ b/README.md
@@ -104,6 +104,7 @@
- [x] string.lower
- [x] string.rep
- [x] string.reverse
+ - [x] string.sub
- [x] string.upper
- [ ] string.dump
- [ ] string.find
@@ -112,7 +113,6 @@
- [ ] string.match
- [ ] string.pack
- [ ] string.packsize
- - [ ] string.sub
- [ ] string.unpack
- [ ] Package
- [ ] os
diff --git a/src/lauxlib.js b/src/lauxlib.js
index f77fe5e..3ee04a4 100644
--- a/src/lauxlib.js
+++ b/src/lauxlib.js
@@ -54,7 +54,7 @@ const findfield = function(L, objidx, level) {
const pushglobalfuncname = function(L, ar) {
let top = lapi.lua_gettop(L);
ldebug.lua_getinfo(L, 'f', ar); /* push function */
- lapi.lua_getfield(L, lua.LUA_REGISTRYINDEX, lua.LUA_LOADED_TABLE);
+ lapi.lua_getfield(L, lua.LUA_REGISTRYINDEX, LUA_LOADED_TABLE);
if (findfield(L, top + 1, 2)) {
let name = lapi.lua_tostring(L, -1);
if (name.startsWith("_G.")) {
diff --git a/src/lobject.js b/src/lobject.js
index 719d7a6..e1c3e97 100644
--- a/src/lobject.js
+++ b/src/lobject.js
@@ -452,8 +452,10 @@ module.exports.LocVar = LocVar;
module.exports.TValue = TValue;
module.exports.Table = Table;
module.exports.UTF8BUFFSZ = UTF8BUFFSZ;
+module.exports.intarith = intarith;
module.exports.luaO_chunkid = luaO_chunkid;
module.exports.luaO_hexavalue = luaO_hexavalue;
module.exports.luaO_int2fb = luaO_int2fb;
module.exports.luaO_str2num = luaO_str2num;
-module.exports.luaO_utf8desc = luaO_utf8desc; \ No newline at end of file
+module.exports.luaO_utf8desc = luaO_utf8desc;
+module.exports.numarith = numarith; \ No newline at end of file
diff --git a/src/lstrlib.js b/src/lstrlib.js
index b3d5aea..1d3a7cc 100644
--- a/src/lstrlib.js
+++ b/src/lstrlib.js
@@ -18,6 +18,19 @@ const posrelat = function(pos, len) {
else return len + pos + 1;
};
+const str_sub = function(L) {
+ let s = lauxlib.luaL_checkstring(L, 1);
+ let l = s.length;
+ let start = posrelat(lauxlib.luaL_checkinteger(L, 2), l);
+ let end = posrelat(lauxlib.luaL_optinteger(L, 3, -1), l);
+ if (start < 1) start = 1;
+ if (end > l) end = l;
+ if (start <= end)
+ lapi.lua_pushstring(L, s.slice(start - 1, (start - 1) + (end - start + 1)));
+ else lapi.lua_pushliteral(L, "");
+ return 1;
+};
+
const str_len = function(L) {
lapi.lua_pushinteger(L, lauxlib.luaL_checkstring(L, 1).length);
return 1;
@@ -371,6 +384,7 @@ const strlib = {
"lower": str_lower,
"rep": str_rep,
"reverse": str_reverse,
+ "sub": str_sub,
"upper": str_upper
};
diff --git a/tests/lstrlib.js b/tests/lstrlib.js
index 0475a42..4df4283 100644
--- a/tests/lstrlib.js
+++ b/tests/lstrlib.js
@@ -284,6 +284,7 @@ test('string.format', function (t) {
);
});
+
test('string.format', function (t) {
let luaCode = `
return string.format("%q", 'a string with "quotes" and \\n new line')
@@ -312,4 +313,105 @@ test('string.format', function (t) {
'"a string with \\"quotes\\" and \\\n new line"',
"Correct element(s) on the stack"
);
+});
+
+
+test('string.sub', function (t) {
+ let luaCode = `
+ return string.sub("123456789",2,4), -- "234"
+ string.sub("123456789",7), -- "789"
+ string.sub("123456789",7,6), -- ""
+ string.sub("123456789",7,7), -- "7"
+ string.sub("123456789",0,0), -- ""
+ string.sub("123456789",-10,10), -- "123456789"
+ string.sub("123456789",1,9), -- "123456789"
+ string.sub("123456789",-10,-20), -- ""
+ string.sub("123456789",-1), -- "9"
+ string.sub("123456789",-4), -- "6789"
+ string.sub("123456789",-6, -4) -- "456"
+ `, L;
+
+ t.plan(13);
+
+ t.doesNotThrow(function () {
+
+ L = lauxlib.luaL_newstate();
+
+ linit.luaL_openlibs(L);
+
+ lauxlib.luaL_loadstring(L, luaCode);
+
+ }, "Lua program loaded without error");
+
+ t.doesNotThrow(function () {
+
+ lapi.lua_call(L, 0, -1);
+
+ }, "Lua program ran without error");
+
+ t.strictEqual(
+ lapi.lua_tostring(L, -11),
+ "234",
+ "Correct element(s) on the stack"
+ );
+
+ t.strictEqual(
+ lapi.lua_tostring(L, -10),
+ "789",
+ "Correct element(s) on the stack"
+ );
+
+ t.strictEqual(
+ lapi.lua_tostring(L, -9),
+ "",
+ "Correct element(s) on the stack"
+ );
+
+ t.strictEqual(
+ lapi.lua_tostring(L, -8),
+ "7",
+ "Correct element(s) on the stack"
+ );
+
+ t.strictEqual(
+ lapi.lua_tostring(L, -7),
+ "",
+ "Correct element(s) on the stack"
+ );
+
+ t.strictEqual(
+ lapi.lua_tostring(L, -6),
+ "123456789",
+ "Correct element(s) on the stack"
+ );
+
+ t.strictEqual(
+ lapi.lua_tostring(L, -5),
+ "123456789",
+ "Correct element(s) on the stack"
+ );
+
+ t.strictEqual(
+ lapi.lua_tostring(L, -4),
+ "",
+ "Correct element(s) on the stack"
+ );
+
+ t.strictEqual(
+ lapi.lua_tostring(L, -3),
+ "9",
+ "Correct element(s) on the stack"
+ );
+
+ t.strictEqual(
+ lapi.lua_tostring(L, -2),
+ "6789",
+ "Correct element(s) on the stack"
+ );
+
+ t.strictEqual(
+ lapi.lua_tostring(L, -1),
+ "456",
+ "Correct element(s) on the stack"
+ );
}); \ No newline at end of file