115 lines
4.7 KiB
Python
115 lines
4.7 KiB
Python
import hashlib
|
||
import os
|
||
from pathlib import Path
|
||
|
||
from scripts.context import Context
|
||
from scripts.task import Task
|
||
from utils import FileUtils
|
||
|
||
|
||
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)
|
||
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"
|
||
|
||
|
||
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 execute(self):
|
||
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, 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))
|
||
pass
|