建档

source/squirrel/SquirrelEx.cpp

@@ -0,0 +1,210 @@
+#include "squirrel/SquirrelEx.h"
+#include <dirent.h>
+#include <string>
+#include <fstream>
+#include "SquirrelEx.h"
+#include "Asset/Asset_Script.h"
+
+SquirrelEx::SquirrelEx()
+{
+}
+SquirrelEx::~SquirrelEx()
+{
+}
+void SquirrelEx::printfunc(HSQUIRRELVM v, const SQChar *s, ...)
+{
+    va_list vl;
+    va_start(vl, s);
+    int len = vsnprintf(NULL, 0, (const char *)s, vl);
+    if (len <= 0)
+    {
+        va_end(vl);
+        return;
+    }
+    char *buf = (char *)malloc(len + 1);
+    if (!buf)
+    {
+        va_end(vl);
+        return;
+    }
+    va_end(vl);
+    va_start(vl, s);
+    vsnprintf(buf, len + 1, (const char *)s, vl);
+    va_end(vl);
+    SDL_Log("%s", buf);
+    va_end(vl);
+}
+
+void SquirrelEx::errorfunc(HSQUIRRELVM v, const SQChar *s, ...)
+{
+    va_list vl;
+    va_start(vl, s);
+    int len = vsnprintf(NULL, 0, (const char *)s, vl);
+    if (len <= 0)
+    {
+        va_end(vl);
+        return;
+    }
+    char *buf = (char *)malloc(len + 1);
+    if (!buf)
+    {
+        va_end(vl);
+        return;
+    }
+    va_end(vl);
+    va_start(vl, s);
+    vsnprintf(buf, len + 1, (const char *)s, vl);
+    va_end(vl);
+    SDL_LogError(0, "%s", buf);
+    va_end(vl);
+}
+
+bool SquirrelEx::Compilebuffer(std::string Path, std::string Code)
+{
+    if (v == nullptr)
+    {
+        SDL_Log("松鼠虚拟机未初始化!");
+        return false;
+    }
+    if (sq_compilebuffer(v, Code.c_str(), Code.length(), Path.c_str(), SQTrue) >= 0)
+    {
+        sq_pushroottable(v);
+        sq_call(v, 1, SQFalse, SQTrue);
+        sq_pop(v, 1);
+        return true;
+    }
+    return false;
+}
+
+void SquirrelEx::Init()
+{
+    v = sq_open(1024); // 栈大小1024
+    sqstd_seterrorhandlers(v);
+    sq_pushroottable(v);
+    sqstd_register_bloblib(v);
+    sqstd_register_iolib(v);
+    sqstd_register_systemlib(v);
+    sqstd_register_mathlib(v);
+    sqstd_register_stringlib(v);
+
+    sq_setprintfunc(v, printfunc, errorfunc);
+
+    // 加载脚本
+    LoadLocalScript();
+}
+
+void SquirrelEx::RequestNetScript(std::string Ip, std::string Port)
+{
+    // char *response = NULL;
+    // size_t response_size = 0;
+    // std::string url = "http://" + Ip + ":" + Port + "/get/getadvertisement?key=Files"; // 确保协议正确
+    // int result = http_get(url.c_str(), &response, &response_size);
+    // if (result == 0 && response != NULL && response_size > 0)
+    // {
+    //     std::string response_str(response, response_size);
+    //     free(response); // 释放内存
+    //     nlohmann::json ex1 = nlohmann::json::parse(response_str);
+    //     for (const auto &[key, value] : ex1.items())
+    //     {
+    //         std::string key_str = key;
+    //         std::string value_str = value;
+    //         bool Flag = Compilebuffer(key_str, value_str);
+    //         if (!Flag)
+    //         {
+    //             SDL_Log("Squirrel Compilebuffer Error! FileName: %s", key_str.c_str());
+    //         }
+    //     }
+    // }
+    // else
+    // {
+    //     if (response)
+    //         free(response);
+    // }
+}
+
+void SquirrelEx::LoadLocalScript()
+{
+
+    std::vector<std::string> lines;
+    std::ifstream file("sqr/SquirrelFileConfig.cfg");
+    if (file.is_open())
+    {
+        std::string line;
+        while (std::getline(file, line))
+        {
+            lines.push_back(line);
+        }
+        file.close();
+    }
+    else
+    {
+        SDL_LogError(0, "配置文件未找到,请检查项目目录 sqr/SquirrelFileConfig.cfg 文件是否存在");
+    }
+
+    for (const auto &line : lines)
+    {
+        std::string clean_line = line;
+        // 移除换行符和回车符
+        clean_line.erase(std::remove(clean_line.begin(), clean_line.end(), '\n'), clean_line.end());
+        clean_line.erase(std::remove(clean_line.begin(), clean_line.end(), '\r'), clean_line.end());
+        if (line.find(".nut") == std::string::npos)
+            continue;
+
+        std::string RegistPath = "sqr/" + clean_line;
+        if (!SQ_SUCCEEDED(sqstd_dofile(v, (SQChar *)(RegistPath.c_str()), SQFalse, SQTrue)))
+        {
+            SDL_LogError(0, "脚本文件未找到,请检查项目目录 %s 文件是否存在", RegistPath.c_str());
+        }
+        {
+            // std::ifstream F;
+            // F.open((RegistPath).c_str(), std::ios::in | std::ios::binary);
+            // if (F.is_open())
+            // {
+            //     std::stringstream ContentStringStream;
+            //     ContentStringStream << F.rdbuf();
+            //     std::string ContentString(ContentStringStream.str());
+
+            //     std::string RealContentString = ContentString;
+            //     if (RealContentString.length() > 0)
+            //     {
+            //         if (SQ_SUCCEEDED(sq_compilebuffer(v, (SQChar *)(RealContentString.c_str()), RealContentString.length(), (SQChar *)(RegistPath.c_str()), true)))
+            //         {
+            //             sq_pushroottable(v);
+            //             sq_call(v, 1, 1, 1);
+            //             sq_pop(v, 1);
+            //         }
+            //     }
+            // }
+            // else
+            // {
+            //     SDL_LogError(0, "脚本文件未找到,请检查项目目录 %s 文件是否存在", RegistPath.c_str());
+            // }
+            // F.close();
+        }
+    }
+}
+
+void SquirrelEx::Run()
+{
+    // 载入main函数
+    SQInteger top = sq_gettop(v); // saves the stack size before the call
+    sq_pushroottable(v);          // pushes the global table
+    sq_pushstring(v, _SC("main"), -1);
+    if (SQ_SUCCEEDED(sq_get(v, -2)))
+    {                                   // gets the field 'foo' from the global table
+        sq_pushroottable(v);            // push the 'this' (in this case is the global table)
+        sq_call(v, 1, SQFalse, SQTrue); // calls the function
+    }
+    sq_settop(v, top); // restores the original stack size
+}
+
+void SquirrelEx::Clean()
+{
+    sq_close(v);
+    v = nullptr;
+}
+
+HSQUIRRELVM SquirrelEx::GetSquirrelVM()
+{
+    return v;
+}