197 lines
8.0 KiB
Python
197 lines
8.0 KiB
Python
import hashlib
|
||
import os
|
||
from pathlib import Path
|
||
import re
|
||
|
||
from scripts.context import Context
|
||
from scripts.task import Task
|
||
from utils import FileUtils
|
||
|
||
|
||
def extract_launcher_ids(xml_content):
|
||
"""
|
||
修复版：提取所有以 "launcher_" 开头的 android:id 值
|
||
"""
|
||
pattern = r'android:id\s*=\s*"@\+id/(launcher_\w+)"'
|
||
matches = re.findall(pattern, xml_content)
|
||
return matches
|
||
|
||
|
||
# def extract_launcher_ids(xml_content):
|
||
# """提取所有 android:id 值"""
|
||
# pattern = r'android:id\s*=\s*"@\+id/([^"]+)"'
|
||
# return re.findall(pattern, xml_content)
|
||
|
||
|
||
def string_to_md5(text):
|
||
# 将字符串编码为UTF-8字节
|
||
text_bytes = text.encode('utf-8')
|
||
# 创建MD5哈希对象并更新字节数据
|
||
md5_hash = hashlib.md5(text_bytes)
|
||
# 返回十六进制哈希字符串
|
||
return md5_hash.hexdigest()
|
||
|
||
|
||
def generate_encryption_key(key: str, s_len: int = -1, target_package_name: str = "") -> str:
|
||
# if game_editor == "Cocos":
|
||
# return key
|
||
handle_key = target_package_name + key + target_package_name
|
||
processed_key = string_to_md5(handle_key)
|
||
while processed_key[0].isdigit():
|
||
processed_key = processed_key[1:] # 移除首字符
|
||
|
||
# 计算目标长度
|
||
if s_len > 0:
|
||
target_length = s_len
|
||
else:
|
||
target_length = len(key)
|
||
|
||
# 取前N位（根据原始key长度），不足则全部保留
|
||
# 如果处理后的key为空（极小概率），则返回空字符串
|
||
return processed_key[:target_length] if processed_key else ""
|
||
|
||
|
||
layout_path = "LauncherCode/res/layout".replace("/", os.sep)
|
||
values_path = "LauncherCode/res/values".replace("/", os.sep)
|
||
drawable_xxhdpi_path = "LauncherCode/res/drawable-xxhdpi".replace("/", os.sep)
|
||
drawable_path = "LauncherCode/res/drawable".replace("/", os.sep)
|
||
code_path = "LauncherCode/src".replace("/", os.sep)
|
||
|
||
android_manifest_path = "lawnchair/AndroidManifest.xml".replace("/", os.sep)
|
||
xml_path = "lawnchair/res/xml".replace("/", os.sep)
|
||
launcher_code_path = "lawnchair/src".replace("/", os.sep)
|
||
src_code_path = "src".replace("/", os.sep)
|
||
launchercode = "launchercode"
|
||
|
||
lawn_layout = "res/layout".replace("/", os.sep)
|
||
|
||
qucikset_manifest = "quickstep/AndroidManifest-launcher.xml".replace("/", os.sep)
|
||
lawn_manifest = "AndroidManifest.xml".replace("/", os.sep)
|
||
|
||
pag_file = "lawnchair/assets/pag_wallpaper_slide.pag".replace("/", os.sep)
|
||
|
||
|
||
class ProjectProguard(Task):
|
||
def __init__(self, context: Context):
|
||
super().__init__(context)
|
||
self.context.proguard_dict = {
|
||
launchercode: generate_encryption_key(launchercode, target_package_name=self.context.package_name)}
|
||
|
||
def add_proguard_key(self, key: str) -> str:
|
||
value = self.context.proguard_dict.get(key)
|
||
if value:
|
||
return value
|
||
value = generate_encryption_key(key, target_package_name=self.context.package_name)
|
||
self.context.proguard_dict[key] = value
|
||
return value
|
||
|
||
def rename_files_in_dir(self, dir_path: str):
|
||
for root, dirs, files in os.walk(dir_path):
|
||
for file in files:
|
||
file_path = os.path.join(root, file)
|
||
name = Path(file).stem
|
||
add_proguard_key = self.add_proguard_key(name)
|
||
FileUtils.move(file_path,
|
||
os.path.join(root, file.replace(name, add_proguard_key))
|
||
.replace(launchercode, self.context.proguard_dict.get(launchercode)),
|
||
overwrite=True)
|
||
|
||
def update(self, file_path: str):
|
||
text = open(file_path, "r", encoding="UTF-8").read()
|
||
for key, value in self.context.proguard_dict.items():
|
||
text = text.replace(key, value)
|
||
pass
|
||
open(file_path, "w", encoding="UTF-8").write(text)
|
||
|
||
def update_proguard(self, dir_path: str):
|
||
if os.path.isdir(dir_path):
|
||
for root, dirs, files in os.walk(dir_path):
|
||
for file in files:
|
||
if file.endswith(".xml") or file.endswith(".java") or file.endswith(".kt"):
|
||
file_path = os.path.join(root, file)
|
||
self.update(file_path)
|
||
else:
|
||
self.update(dir_path)
|
||
pass
|
||
|
||
def update_id(self, file_path: str):
|
||
for root, dirs, files in os.walk(file_path):
|
||
for file in files:
|
||
xml_file = os.path.join(root, file)
|
||
text = open(xml_file, "r", encoding="UTF-8").read()
|
||
ids = extract_launcher_ids(text)
|
||
for id in ids:
|
||
self.add_proguard_key(id)
|
||
|
||
def write_proguard(self):
|
||
file_path = os.path.join(self.context.temp_project_path, "proguard.pro")
|
||
lines = open(file_path, "r", encoding="UTF-8").readlines()
|
||
lines.append(f"""
|
||
|
||
-repackageclasses '{self.context.package_name}' # 将所有类移动到 '{self.context.package_name}' 包下
|
||
-flattenpackagehierarchy '{self.context.package_name}' # 扁平化包结构
|
||
|
||
""")
|
||
|
||
open(file_path, "w", encoding="UTF-8").writelines(lines)
|
||
pass
|
||
|
||
def execute(self):
|
||
self.write_proguard()
|
||
self.add_proguard_key("LauncherMyNoActionBar")
|
||
self.add_proguard_key("LauncherIntroTheme")
|
||
self.add_proguard_key("LauncherGameIntroTheme")
|
||
self.add_proguard_key("LauncherTransparentTheme")
|
||
|
||
self.add_proguard_key("launchergamead")
|
||
self.add_proguard_key("launcher_help_detail")
|
||
self.add_proguard_key("launcher_help")
|
||
self.add_proguard_key("launcher_uninstall")
|
||
self.add_proguard_key("launcher_onboard")
|
||
self.add_proguard_key("pag_wallpaper_slide")
|
||
|
||
self.update_id(os.path.join(self.context.temp_project_path, lawn_layout))
|
||
self.update_id(os.path.join(self.context.temp_project_path, layout_path))
|
||
|
||
self.rename_files_in_dir(os.path.join(self.context.temp_project_path, layout_path))
|
||
self.rename_files_in_dir(os.path.join(self.context.temp_project_path, drawable_xxhdpi_path))
|
||
self.rename_files_in_dir(os.path.join(self.context.temp_project_path, drawable_path))
|
||
self.rename_files_in_dir(os.path.join(self.context.temp_project_path, code_path))
|
||
|
||
self.context.proguard_dict = {
|
||
k: v for k, v in sorted(
|
||
self.context.proguard_dict.items(),
|
||
key=lambda item: len(item[0]),
|
||
reverse=True
|
||
)
|
||
}
|
||
self.update_proguard(os.path.join(self.context.temp_project_path, layout_path))
|
||
self.update_proguard(os.path.join(self.context.temp_project_path, values_path))
|
||
self.update_proguard(os.path.join(self.context.temp_project_path, drawable_xxhdpi_path))
|
||
self.update_proguard(os.path.join(self.context.temp_project_path, drawable_path))
|
||
self.update_proguard(os.path.join(self.context.temp_project_path, code_path))
|
||
self.update_proguard(os.path.join(self.context.temp_project_path, android_manifest_path))
|
||
self.update_proguard(os.path.join(self.context.temp_project_path, xml_path))
|
||
self.update_proguard(os.path.join(self.context.temp_project_path, launcher_code_path))
|
||
self.update_proguard(os.path.join(self.context.temp_project_path, src_code_path))
|
||
self.update_proguard(os.path.join(self.context.temp_project_path, qucikset_manifest))
|
||
self.update_proguard(os.path.join(self.context.temp_project_path, lawn_manifest))
|
||
|
||
self.update_proguard(os.path.join(self.context.temp_project_path, lawn_layout))
|
||
|
||
target_pag_file = os.path.join(self.context.temp_project_path, pag_file)
|
||
FileUtils.move(target_pag_file,
|
||
target_pag_file.replace("pag_wallpaper_slide",
|
||
self.context.proguard_dict.get("pag_wallpaper_slide")))
|
||
pass
|
||
|
||
|
||
if __name__ == "__main__":
|
||
file = "../project/com_punch_prankster_puzzle_fwaes_free_launcher_fgsaw223908/LauncherCode/res/layout/a60034437eae19e7b00.xml".replace(
|
||
"/", os.sep)
|
||
text = open(file, "r", encoding="UTF-8").read()
|
||
print(text)
|
||
temp = extract_launcher_ids(text)
|
||
print(temp)
|
||
pass
|