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