mirror of
https://github.com/ikunshare/Onekey.git
synced 2026-01-13 00:27:32 +08:00
93 lines
4.0 KiB
Python
93 lines
4.0 KiB
Python
import os
|
||
from aiohttp import ClientSession
|
||
|
||
from common.config import config
|
||
from common.dkey_merge import depotkey_merge
|
||
from common.migration import migrate
|
||
from common.st_unlock import stool_add
|
||
from common.gl_unlock import greenluma_add
|
||
from common.get_manifest_info import get_manifest
|
||
from common.check import check_github_api_rate_limit
|
||
from common.log import log
|
||
from common.get_steam_path import steam_path
|
||
from common.stack_error import stack_error
|
||
|
||
isGreenLuma = any((steam_path / dll).exists() for dll in ['GreenLuma_2024_x86.dll', 'GreenLuma_2024_x64.dll', 'User32.dll'])
|
||
isSteamTools = (steam_path / 'config' / 'stUI').is_dir()
|
||
|
||
async def fetch_branch_info(session, url, headers):
|
||
try:
|
||
async with session.get(url, headers=headers, ssl=False) as response:
|
||
return await response.json()
|
||
except Exception as e:
|
||
log.error(f'⚠ 获取信息失败: {stack_error(e)}')
|
||
return None
|
||
|
||
async def get_latest_repo_info(session, repos, app_id, headers):
|
||
latest_date = None
|
||
selected_repo = None
|
||
|
||
for repo in repos:
|
||
url = f'https://api.github.com/repos/{repo}/branches/{app_id}'
|
||
r_json = await fetch_branch_info(session, url, headers)
|
||
if r_json and 'commit' in r_json:
|
||
date = r_json['commit']['commit']['author']['date']
|
||
if latest_date is None or date > latest_date:
|
||
latest_date = date
|
||
selected_repo = repo
|
||
|
||
return selected_repo, latest_date
|
||
|
||
async def main(app_id: str, repos: list) -> bool:
|
||
app_id_list = list(filter(str.isdecimal, app_id.strip().split('-')))
|
||
if not app_id_list:
|
||
log.error(f'⚠ App ID无效')
|
||
return False
|
||
app_id = app_id_list[0]
|
||
|
||
async with ClientSession() as session:
|
||
github_token = config.get("Github_Personal_Token", "")
|
||
headers = {'Authorization': f'Bearer {github_token}'} if github_token else None
|
||
await check_github_api_rate_limit(headers, session)
|
||
|
||
selected_repo, latest_date = await get_latest_repo_info(session, repos, app_id, headers)
|
||
|
||
if selected_repo:
|
||
log.info(f'🔄 选择清单仓库：{selected_repo}')
|
||
url = f'https://api.github.com/repos/{selected_repo}/branches/{app_id}'
|
||
r_json = await fetch_branch_info(session, url, headers)
|
||
|
||
if r_json and 'commit' in r_json:
|
||
sha = r_json['commit']['sha']
|
||
url = r_json['commit']['commit']['tree']['url']
|
||
r2_json = await fetch_branch_info(session, url, headers)
|
||
|
||
if r2_json and 'tree' in r2_json:
|
||
collected_depots = []
|
||
for item in r2_json['tree']:
|
||
result = await get_manifest(sha, item['path'], steam_path, selected_repo, session)
|
||
collected_depots.extend(result)
|
||
|
||
if collected_depots:
|
||
if isSteamTools:
|
||
await migrate(st_use=True, session=session)
|
||
await stool_add(collected_depots, app_id)
|
||
log.info('✅ 找到SteamTools，已添加解锁文件')
|
||
|
||
if isGreenLuma:
|
||
await migrate(st_use=False, session=session)
|
||
await greenluma_add([app_id])
|
||
depot_config = {'depots': {depot_id: {'DecryptionKey': depot_key} for depot_id, depot_key in collected_depots}}
|
||
await depotkey_merge(steam_path / 'config' / 'config.vdf', depot_config)
|
||
if await greenluma_add([int(i) for i in depot_config['depots'] if i.isdecimal()]):
|
||
log.info('✅ 找到GreenLuma，已添加解锁文件')
|
||
|
||
log.info(f'✅ 清单最后更新时间：{latest_date}')
|
||
log.info(f'✅ 入库成功: {app_id}')
|
||
os.system('pause')
|
||
return True
|
||
|
||
log.error(f'⚠ 清单下载或生成失败: {app_id}')
|
||
os.system('pause')
|
||
return False
|