From b2725dcad2d414c1beae76a93bab021853ea20a5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=81=8D=E5=85=AE=E6=83=9A=E5=85=AE?=
 <101191390+HIllya51@users.noreply.github.com>
Date: Mon, 7 Oct 2024 17:02:07 +0800
Subject: [PATCH] lightvn

---
 LunaHook/engine64/LightVN.cpp | 30 ++++++++++++++----------------
 LunaHook/engine64/LightVN.h   |  9 +++++++--
 2 files changed, 21 insertions(+), 18 deletions(-)

diff --git a/LunaHook/engine64/LightVN.cpp b/LunaHook/engine64/LightVN.cpp
index 1524d44..2eba8cb 100644
--- a/LunaHook/engine64/LightVN.cpp
+++ b/LunaHook/engine64/LightVN.cpp
@@ -58,10 +58,7 @@ namespace
   }
   bool _2()
   {
-    // https://vndb.org/r86006
-    // ファーストキス（体験版）
-    // https://vndb.org/r85992
-    // フサの大正女中ぐらし
+    // 有太多乱的输出了，而且基本不需要它，所以先放到后面。
 
     BYTE sig[] = {
         //clang-format off
@@ -97,14 +94,21 @@ namespace
         if (str.find(_) != str.npos)
           return false;
       str = std::regex_replace(str, std::wregex(L"\\[(.*?)\\]<(.*?)>"), L"$1");
-      write_string_overwrite(data, len, str);
-      return true;
+      return write_string_overwrite(data, len, str);
     };
     return NewHook(hp, "LightVN2");
   }
 }
 namespace
 {
+  bool commonfilter(LPVOID data, size_t *size, HookParam *)
+  {
+    auto str = std::wstring((wchar_t *)data, *size / 2);
+    std::wregex pattern(L"-{2,}");
+    str = std::regex_replace(str, pattern, L"");
+    str = std::regex_replace(str, std::wregex(L"\\[(.*?)\\]<(.*?)>"), L"$1");
+    return write_string_overwrite(data, size, str);
+  }
   bool lightvnparsestring()
   {
     BYTE sig[] = {
@@ -136,13 +140,7 @@ namespace
         *split = 1;
       }
     };
-    hp.filter_fun = [](LPVOID data, size_t *size, HookParam *)
-    {
-      auto str = std::wstring((wchar_t *)data, *size / 2);
-      std::wregex pattern(L"-{2,}");
-      str = std::regex_replace(str, pattern, L"");
-      return write_string_overwrite(data, size, str);
-    };
+    hp.filter_fun = commonfilter;
     hp.newlineseperator = L"\\n";
     return NewHook(hp, "Light.VN.16");
   }
@@ -153,7 +151,7 @@ namespace
     // ver12 找不到上面的函数
     auto checkstrings = {
         L"backlog voice already exists at line: {}",
-        L"attempting to log to backlog when backlog showing. likely you faded it out."};
+        L"attempting to log to backlog when backlog showing"}; //. likely you faded it out."};
     auto succ = false;
     for (auto str : checkstrings)
     {
@@ -181,6 +179,7 @@ namespace
           *data = (uintptr_t)((TextUnionW *)stack->rdx)->getText();
           *len = ((TextUnionW *)stack->rdx)->size * 2;
         };
+        hp.filter_fun = commonfilter;
         succ |= NewHook(hp, "Light.VN.12");
       }
     }
@@ -190,8 +189,7 @@ namespace
 bool LightVN::attach_function()
 {
   bool ok = _1();
-  ok = _2() || ok;
   ok |= lightvnparsestring();
   ok |= xreflightvnparsestring();
-  return ok;
+  return ok || _2();
 }
\ No newline at end of file
diff --git a/LunaHook/engine64/LightVN.h b/LunaHook/engine64/LightVN.h
index cfd67b6..2a9b991 100644
--- a/LunaHook/engine64/LightVN.h
+++ b/LunaHook/engine64/LightVN.h
@@ -6,9 +6,14 @@ public:
     LightVN()
     {
 
-        check_by = CHECK_BY::FILE_ANY;
+        check_by = CHECK_BY::CUSTOM;
         is_engine_certain = false;
-        check_by_target = check_by_list{L"Data/Scripts/title.txt", L"Data/data*.vndat", L"Scripts/000_title.txt"};
+        check_by_target = []()
+        {
+            auto s = check_by_list{L"Data/Scripts/title.txt", L"Data/data*.vndat", L"Scripts/000_title.txt"};
+            auto s2 = check_by_list{L"LightTests.exe", L"BugTrap.dll", L"libGLESv2.dll", L"libEGL.dll"};
+            return std::any_of(s.begin(), s.end(), Util::CheckFile) || std::all_of(s2.begin(), s2.end(), Util::CheckFile);
+        };
     };
     bool attach_function();
 };