mirror of
https://github.com/ikunshare/Onekey.git
synced 2026-01-13 00:27:32 +08:00
101 lines
4.3 KiB
Python
101 lines
4.3 KiB
Python
import os
|
|
from aiohttp import ClientSession, ConnectionTimeoutError
|
|
|
|
from common.checkcn import checkcn
|
|
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 KeyboardInterrupt:
|
|
log.info("\n👋 程序已退出")
|
|
except Exception as e:
|
|
log.error(f'⚠ 获取信息失败: {stack_error(e)}')
|
|
return None
|
|
except ConnectionTimeoutError 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]
|
|
|
|
checkcn()
|
|
|
|
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
|