feat: lua 添加一些 string 模块常用函数

Author: zeromake

script/analysis_ani.go

@@ -7,7 +7,7 @@ import (
 	"github.com/zeromake/pkg/binaryutil"
 )
 
-func main() {
+func main1() {
 	input, err := os.Open(os.Args[1])
 	if err != nil {
 		log.Fatal(err)

script/analysis_stand.go

@@ -0,0 +1,81 @@
+package main
+
+import (
+	"fmt"
+	"log"
+	"os"
+	"strings"
+
+	"github.com/zeromake/pkg/binaryutil"
+)
+
+func to_string(buf []uint8) string {
+	offset := 32
+	for offset > 0 && buf[offset-1] == 0 {
+		offset--
+	}
+	return string(buf[:offset])
+}
+
+func main() {
+	input, err := os.Open(os.Args[1])
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer input.Close()
+	r := binaryutil.NewDecoder(input)
+	output, err := os.OpenFile(
+		os.Args[2],
+		os.O_CREATE|os.O_WRONLY|os.O_TRUNC,
+		0644,
+	)
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer output.Close()
+	output.WriteString("fill,img1,ani,img2,x,y\n")
+
+	for r.Error() == nil {
+		r.Skip(4)
+		if r.Error() != nil {
+			break
+		}
+		imgBuf := make([]uint8, 32)
+		r.ReadLE(imgBuf)
+		output.WriteString(to_string(imgBuf))
+		output.Write([]byte{','})
+		r.ReadLE(imgBuf)
+		output.WriteString(to_string(imgBuf))
+		output.Write([]byte{','})
+		r.ReadLE(imgBuf)
+		output.WriteString(to_string(imgBuf))
+		r.ReadLE(imgBuf)
+		if imgBuf[0] == 0 {
+			r.Skip(8)
+			output.Write([]byte{'\n'})
+			continue
+		}
+		img := to_string(imgBuf)
+		output.Write([]byte{','})
+		output.WriteString(img)
+		output.Write([]byte{','})
+		x := r.Read32LE()
+		y := r.Read32LE()
+		if strings.HasPrefix(img, "UMI_") {
+			if strings.HasPrefix(img, "UMI_A") {
+				x -= 185
+				y -= 57
+			} else if strings.HasPrefix(img, "UMI_0101") || strings.HasPrefix(img, "UMI_0201") {
+				x -= 237
+				y -= 57
+				// x -= 237
+				// y -= 57
+			} else {
+				x -= 280
+				y -= 66
+			}
+		}
+		output.WriteString(fmt.Sprintf("%d,%d", x, y))
+		output.Write([]byte{'\n'})
+	}
+}

src/LUAHandler.cpp

@@ -994,6 +994,7 @@ static int NSCall(lua_State *state) {
     lh->sh->enterExternalScript(cmd_buf);
     lh->ons->runScript();
     lh->sh->leaveExternalScript();
+    return 0;
 }
 
 #define LUA_FUNC_LUT(s) \
@@ -1115,6 +1116,50 @@ static int fmt_println(lua_State *state) {
     return 0;
 }
 
+static int string_split(lua_State* state) {
+    const char* input = luaL_checkstring(state, 1);
+    size_t sep_len = 0;
+    const char* sep = luaL_checklstring(state, 2, &sep_len);
+    size_t prev = 0;
+    lua_newtable(state);
+    size_t i = 0;
+    while (true) {
+        i++;
+        const char* next = strstr(input, sep);
+        if (next) {
+            lua_pushlstring(state, input, next - input);
+            lua_rawseti(state, -2, i);
+            next += sep_len;
+            input = next;
+            continue;
+        }
+        lua_pushstring(state, input);
+        lua_rawseti(state, -2, i);
+        break;
+    }
+    return 1;
+}
+
+static int string_startswith(lua_State* state) {
+    size_t cmp1_len = 0;
+    const char* cmp1 = luaL_checklstring(state, 1, &cmp1_len);
+    size_t cmp2_len = 0;
+    const char* cmp2 = luaL_checklstring(state, 2, &cmp2_len);
+    bool ok = !strncmp(cmp1, cmp2, cmp2_len);
+    lua_pushboolean(state, ok);
+    return 1;
+}
+
+static int string_endswith(lua_State* state) {
+    size_t cmp1_len = 0;
+    const char* cmp1 = luaL_checklstring(state, 1, &cmp1_len);
+    size_t cmp2_len = 0;
+    const char* cmp2 = luaL_checklstring(state, 2, &cmp2_len);
+    bool ok = !strncmp(cmp1+cmp1_len-cmp2_len, cmp2, cmp2_len);
+    lua_pushboolean(state, ok);
+    return 1;
+}
+
 static const struct luaL_Reg module_nsutf[] = {
     LUA_FUNC_LUT(nsutf_from_ansi), LUA_FUNC_LUT(nsutf_to_ansi), {NULL, NULL}};
 
@@ -1187,6 +1232,14 @@ void LUAHandler::init(
     luaL_register(state, "nsutf", module_nsutf);
     luaL_register(state, "dpshadow", module_dpshadow);
     luaL_register(state, "fmt", module_fmt);
+
+    lua_getglobal(state, "string");
+    lua_pushcfunction(state, string_split);
+    lua_setfield(state, -2, "split");
+    lua_pushcfunction(state, string_startswith);
+    lua_setfield(state, -2, "startswith");
+    lua_pushcfunction(state, string_endswith);
+    lua_setfield(state, -2, "endswith");
 #endif
 
     lua_pushlightuserdata(state, this);

src/ons_cache.h

@@ -10,7 +10,6 @@
 #include <optional>
 #include <string>
 #include <utility>
-#define USE_IMAGE_CACHE
 
 namespace onscache {
 typedef caches::fixed_sized_cache<std::string,

src/private/uitls.cpp

@@ -41,7 +41,7 @@ void split(std::vector<std::string> &output,
     while (true) {
         size_t pos = input.find(sep, prev);
         if (pos != std::string::npos) {
-            output.push_back(std::move(input.substr(prev, pos)));
+            output.push_back(std::move(input.substr(prev, pos - prev)));
             prev = pos + 1;
             continue;
         }