修复构建

feat:优化代码&更清晰的图标

.github/FUNDING.yml

@@ -0,0 +1,15 @@
+# These are supported funding model platforms
+
+github: # Replace with up to 4 GitHub Sponsors-enabled usernames e.g., [user1, user2]
+patreon: # Replace with a single Patreon username
+open_collective: # Replace with a single Open Collective username
+ko_fi: # Replace with a single Ko-fi username
+tidelift: # Replace with a single Tidelift platform-name/package-name e.g., npm/babel
+community_bridge: # Replace with a single Community Bridge project-name e.g., cloud-foundry
+liberapay: # Replace with a single Liberapay username
+issuehunt: # Replace with a single IssueHunt username
+lfx_crowdfunding: # Replace with a single LFX Crowdfunding project-name e.g., cloud-foundry
+polar: # Replace with a single Polar username
+buy_me_a_coffee: # Replace with a single Buy Me a Coffee username
+thanks_dev: # Replace with a single thanks.dev username
+custom: ['https://afdian.com/a/ikun0014']

.github/ISSUE_TEMPLATE/bug反馈.md

@@ -0,0 +1,25 @@
+---
+name: Bug反馈
+about: 反馈一个虫子（确信）
+title: "[BUG]: "
+labels: ''
+assignees: ''
+
+---
+
+**触发Bug后出现了什么症状**
+1.
+
+**如何复现该Bug**
+步骤:
+1. 
+
+**日志**
+复制在这里
+
+**截图**
+对程序的完整截图
+
+**系统:**
+ - OS: Windows
+ - Version: 10

.github/workflows/release.yml

@@ -7,27 +7,28 @@ on:
 
 jobs:
   build:
-    runs-on: windows-2019
+    permissions:
+      contents: write
+    runs-on: windows-latest
     steps:
     - name: Check out git repository
       uses: actions/checkout@v4
 
-    # Push tag to GitHub if package.json version's tag is not tagged
     - name: Get package version
       shell: powershell
       run: |
         $version = (Get-Content package.json | ConvertFrom-Json).version
         echo "PACKAGE_VERSION=$version" >> $env:GITHUB_ENV
   
-
-    - name: Set up Python 3.11
+    - name: Set up Python 3.12
       uses: actions/setup-python@v3
       with:
-        python-version: 3.11.9
+        python-version: 3.12
 
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
+        pip install imageio
         pip install -r requirements.txt
 
     - name: Build
@@ -35,10 +36,10 @@ jobs:
       with:
         nuitka-version: main
         script-name: main.py
-        standalone: true
-        onefile: true
+        mode: onefile
         show-memory: true
         windows-uac-admin: true
+        onefile-tempdir-spec: "%TEMP%\\onekey_%PID%_%TIME%"
         windows-icon-from-ico: icon.jpg
         company-name: ikunshare
         product-name: Onekey
@@ -59,13 +60,40 @@ jobs:
         git_commit_sha: ${{ github.sha }}
         git_tag_prefix: "v"
 
+    - name: Upload Artifact
+      uses: actions/upload-artifact@v4
+      with:
+        name: Onekey_v${{ env.PACKAGE_VERSION }}.exe
+        path: build/Onekey_v${{ env.PACKAGE_VERSION }}.exe
+
     - name: Release
       uses: softprops/action-gh-release@v2
       with:
+        tag_name: v${{ env.PACKAGE_VERSION }}
+        files: build/Onekey---v${{ env.PACKAGE_VERSION }}.exe
         prerelease: false
         draft: false
-        tag_name: v${{ env.PACKAGE_VERSION }}
-        files: |
-          build/Onekey---v${{ env.PACKAGE_VERSION }}.exe
       env:
         GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+
+    - name: Gitee Release
+      uses: nicennnnnnnlee/action-gitee-release@v1.0.5
+      with:
+        gitee_owner: ikun0014
+        gitee_repo: Onekey
+        gitee_token: ${{ secrets.GITEE_TOKEN }}
+        gitee_tag_name: v${{ env.PACKAGE_VERSION }}
+        gitee_release_name: v${{ env.PACKAGE_VERSION }}
+        gitee_release_body: I don't know
+        gitee_target_commitish: main
+        gitee_upload_retry_times:  3
+        gitee_file_name: Onekey---v${{ env.PACKAGE_VERSION }}.exe
+        gitee_file_path: build/Onekey---v${{ env.PACKAGE_VERSION }}.exe
+
+    - name: Upload to Telegram Channel
+      run: |
+        & curl -F "chat_id=${{ secrets.TELEGRAM_TO }}" `
+            -F "document=@build/Onekey---v${{ env.PACKAGE_VERSION }}.exe" `
+            -F "caption=Onekey's New Update ${{ env.PACKAGE_VERSION }}" `
+            -F "parse_mode=Markdown" `
+            "https://api.telegram.org/bot${{ secrets.TELEGRAM_TOKEN }}/sendDocument"

.github/workflows/sync.yml

@@ -0,0 +1,25 @@
+name: Sync to Gitee
+
+on:
+  push:
+    branches:
+      - main
+
+jobs:
+  sync:
+    permissions:
+      contents: write
+    runs-on: ubuntu-latest
+    steps:
+    - name: Sync to Gitee
+      uses: Yikun/hub-mirror-action@master
+      with:
+        src: github/ikunshare
+        dst: gitee/ikun0014
+        dst_key: ${{ secrets.GITEE_PRIVATE_KEY }}
+        dst_token: ${{ secrets.GITEE_TOKEN }}
+        src_account_type: org
+        dst_account_type: user
+        white_list: "Onekey"
+        force_update: true
+        debug: true

.gitignore

@@ -170,3 +170,5 @@ config.json
 *.exe
 *.dll
 /main.dist
+/.vscode
+/.vs

README.md

@@ -11,14 +11,15 @@
 
 
 ## Onekey
- Onekey Steam Depot Manifest Downloader
+Onekey Steam Depot Manifest Downloader
 
 ## 使用方法
- 先去Release下最新发布，然后去steamtools官网下steamtools，日志会有点石介意别用
+去Releases处下载最新的发布，并且安装好SteamTools或者GreenLuma  
+然后打开Onekey输入App ID即可使用
 
 ## 开发
 本程序使用Python编程语言开发   
-要求环境：   
+要求环境:    
 1.Python 3.10及以上   
 2.Windows 10及以上   
 3.使用Git进行版本管理   
@@ -38,7 +39,7 @@ pip install -r requirements.txt
 ## 项目协议
  本项目基于 GPL-3.0 许可证发行，以下协议是对于 GPL-3.0 原协议的补充，如有冲突，以以下协议为准。
 
- 词语约定：“使用者”指签署本协议的使用者;“版权数据”指包括但不限于图像、音频、名字等在内的他人拥有所属版权的数据。
+ 词语约定: “使用者”指签署本协议的使用者;“版权数据”指包括但不限于图像、音频、名字等在内的他人拥有所属版权的数据。
 
  本项目的数据来源原理是从Steam官方的CDN服务器中拉取游戏清单数据，经过对数据简单地筛选与合并后进行展示，因此本项目不对数据的准确性负责。
  使用本项目的过程中可能会产生版权数据，对于这些版权数据，本项目不拥有它们的所有权，为了避免造成侵权，使用者务必在24 小时内清除使用本项目的过程中所产生的版权数据。
@@ -59,3 +60,11 @@ pip install -r requirements.txt
  <a href="https://github.com/ikunshare/Onekey/graphs/contributors">
    <img src="https://contrib.rocks/image?repo=ikunshare/Onekey" />
  </a>
+
+## 常见问题解答（FAQ）
+查看 [FAQ](https://ikunshare.com/d/49) 获取常见问题的解答。
+
+## 社区和支持
+加入我们的社区，参与讨论和支持: 
+- [GitHub Discussions](https://github.com/ikunshare/Onekey/discussions)
+- [Telegram](https://t.me/ikunshare_qun)

common/check.py

@@ -1,26 +0,0 @@
-import os
-import time
-
-from .log import log
-
-async def check_github_api_rate_limit(headers, session):
-    url = 'https://api.github.com/rate_limit'
-
-    async with session.get(url, headers=headers, ssl=False) as r:
-        if not r == None:
-            r_json = await r.json()
-        else:
-            log.error('孩子，你怎么做到的？')
-            os.system('pause')
-
-    if r.status == 200:
-        rate_limit = r_json['rate']
-        remaining_requests = rate_limit['remaining']
-        reset_time = rate_limit['reset']
-        reset_time_formatted = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(reset_time))
-        log.info(f' 🔄 剩余请求次数: {remaining_requests}')
-    else:
-        log.error('Github请求数检查失败')
-
-    if remaining_requests == 0:
-        log.warning(f' ⚠ GitHub API 请求数已用尽，将在 {reset_time_formatted} 重置, 不想等生成一个填配置文件里')

common/config.py

@@ -1,49 +0,0 @@
-import os
-import sys
-import asyncio
-import ujson as json
-import aiofiles
-
-from .stack_error import stack_error
-from .log import log
-
-
-async def gen_config_file():
-    qa1 = "温馨提示：Github_Personal_Token可在Github设置的最底下开发者选项找到，详情看教程"
-    tutorial = "https://ikunshare.com/Onekey_tutorial"
-    default_config ={
-                    "Github_Personal_Token": "",
-                    "Custom_Steam_Path": "",
-                    "QA1": qa1,
-                    "教程": tutorial
-                    }
-    async with aiofiles.open("./config.json",
-                             mode="w",
-                             encoding="utf-8") as f:
-        await f.write(json.dumps(default_config,
-                                 indent=2,
-                                 ensure_ascii=False,
-                                 escape_forward_slashes=False))
-        await f.close()
-    log.info(' 🖱️ 程序可能为第一次启动，请填写配置文件后重新启动程序')
-
-
-async def load_config():
-    if not os.path.exists('./config.json'):
-        await gen_config_file()
-        os.system('pause')
-        sys.exit()
-    else:
-        try:
-            async with aiofiles.open("./config.json",
-                                     mode="r",
-                                     encoding="utf-8") as f:
-                config = json.loads(await f.read())
-                return config
-        except Exception as e:
-            log.error(f"配置文件加载失败，原因: {stack_error(e)}")
-            os.remove("./config.json")
-            os.system('pause')
-
-        
-config = asyncio.run(load_config())

common/dkey_merge.py

@@ -1,27 +0,0 @@
-import asyncio
-import aiofiles
-import vdf
-
-from .log import log
-
-lock = asyncio.Lock()
-
-
-async def depotkey_merge(config_path, depots_config):
-    if not config_path.exists():
-        async with lock:
-            log.error(' 👋 Steam默认配置不存在，可能是没有登录账号')
-        return
-    async with aiofiles.open(config_path, encoding='utf-8') as f:
-        content = await f.read()
-    config = vdf.loads(content)
-    software = config['InstallConfigStore']['Software']
-    valve = software.get('Valve') or software.get('valve')
-    steam = valve.get('Steam') or valve.get('steam')
-    if 'depots' not in steam:
-        steam['depots'] = {}
-    steam['depots'].update(depots_config['depots'])
-    async with aiofiles.open(config_path, mode='w', encoding='utf-8') as f:
-        new_content = vdf.dumps(config, pretty=True)
-        await f.write(new_content)
-    return True

common/get_manifest_info.py

@@ -1,34 +0,0 @@
-from pathlib import Path
-import aiofiles
-import vdf
-
-from .log import log
-from .manifest_down import get
-from .stack_error import stack_error
-
-# 获取清单信息
-async def get_manifest(sha, path, steam_path: Path, repo, session):
-    collected_depots = []
-    try:
-        if path.endswith('.manifest'):
-            depot_cache_path = steam_path / 'depotcache'
-            if not depot_cache_path.exists():
-                depot_cache_path.mkdir(exist_ok=True)
-            save_path = depot_cache_path / path
-            if save_path.exists():
-                log.warning(f'👋已存在清单: {path}')
-                return collected_depots
-            content = await get(sha, path, repo, session)
-            log.info(f' 🔄 清单下载成功: {path}')
-            async with aiofiles.open(save_path, 'wb') as f:
-                await f.write(content)
-        elif path == 'Key.vdf':
-            content = await get(sha, path, repo, session)
-            log.info(f' 🔄 密钥下载成功: {path}')
-            depots_config = vdf.loads(content.decode(encoding='utf-8'))
-            for depot_id, depot_info in depots_config['depots'].items():
-                collected_depots.append((depot_id, depot_info['DecryptionKey']))
-    except Exception as e:
-        log.error(f'处理失败: {path} - {stack_error(e)}')
-        raise
-    return collected_depots

common/get_steam_path.py

@@ -1,23 +0,0 @@
-import os
-
-import winreg
-
-from pathlib import Path
-from .log import log
-from .config import config
-from .stack_error import stack_error
-
-def get_steam_path():
-    try:
-        key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, r'Software\Valve\Steam')
-        steam_path = Path(winreg.QueryValueEx(key, 'SteamPath')[0])
-        custom_steam_path = config["Custom_Steam_Path"]
-        if not custom_steam_path == '':
-            return Path(custom_steam_path)
-        else:
-            return steam_path
-    except Exception as e:
-        log.error(f'Steam路径获取失败, {stack_error(e)}')
-        os.system('pause')
-    
-steam_path = get_steam_path()

common/gl_unlock.py

@@ -1,28 +0,0 @@
-from .get_steam_path import steam_path
-
-async def greenluma_add(depot_id_list):
-    app_list_path = steam_path / 'AppList'
-    if app_list_path.exists() and app_list_path.is_file():
-        app_list_path.unlink(missing_ok=True)
-    if not app_list_path.is_dir():
-        app_list_path.mkdir(parents=True, exist_ok=True)
-    depot_dict = {}
-    for i in app_list_path.iterdir():
-        if i.stem.isdecimal() and i.suffix == '.txt':
-            with i.open('r', encoding='utf-8') as f:
-                app_id_ = f.read().strip()
-                depot_dict[int(i.stem)] = None
-                if app_id_.isdecimal():
-                    depot_dict[int(i.stem)] = int(app_id_)
-    for depot_id in depot_id_list:
-        if int(depot_id) not in depot_dict.values():
-            index = max(depot_dict.keys()) + 1 if depot_dict.keys() else 0
-            if index != 0:
-                for i in range(max(depot_dict.keys())):
-                    if i not in depot_dict.keys():
-                        index = i
-                        break
-            with (app_list_path / f'{index}.txt').open('w', encoding='utf-8') as f:
-                f.write(str(depot_id))
-            depot_dict[index] = int(depot_id)
-    return True

common/init_text.py

@@ -1,15 +0,0 @@
-from .log import log
-
-def init():
-    print('\033[1;32;40m  _____   __   _   _____   _   _    _____  __    __ ' + '\033[0m')
-    print('\033[1;32;40m /  _  \\ |  \\ | | | ____| | | / /  | ____| \\ \\  / /' + '\033[0m')
-    print('\033[1;32;40m | | | | |   \\| | | |__   | |/ /   | |__    \\ \\/ /' + '\033[0m')
-    print('\033[1;32;40m | | | | | |\\   | |  __|  | |\\ \\   |  __|    \\  / ' + '\033[0m')
-    print('\033[1;32;40m | |_| | | | \\  | | |___  | | \\ \\  | |___    / /' + '\033[0m')
-    print('\033[1;32;40m \\_____/ |_|  \\_| |_____| |_|  \\_\\ |_____|  /_/' + '\033[0m')
-    log.info('作者ikun0014')
-    log.info('本项目基于wxy1343/ManifestAutoUpdate进行修改，采用GPL-3.0许可证')
-    log.info('版本：1.2.1')
-    log.info('项目仓库：https://github.com/ikunshare/Onekey')
-    log.info('官网：ikunshare.com')
-    log.warning('本项目完全开源免费，如果你在淘宝，QQ群内通过购买方式获得，赶紧回去骂商家死全家\n交流群组：\nhttps://qm.qq.com/q/d7sWovfAGI\nhttps://t.me/ikunshare_group')

common/log.py

@@ -1,23 +0,0 @@
-import logging
-import colorlog
-
-
-def init_log():
-    logger = logging.getLogger('Onekey')
-    logger.setLevel(logging.DEBUG)
-    stream_handler = logging.StreamHandler()
-    stream_handler.setLevel(logging.DEBUG)
-    fmt_string = '%(log_color)s[%(name)s][%(levelname)s]%(message)s'
-    log_colors = {
-        'INFO': 'cyan',
-        'WARNING': 'yellow',
-        'ERROR': 'red',
-        'CRITICAL': 'purple'
-    }
-    fmt = colorlog.ColoredFormatter(fmt_string, log_colors=log_colors)
-    stream_handler.setFormatter(fmt)
-    logger.addHandler(stream_handler)
-    return logger
-
-
-log = init_log()

common/main_func.py

@@ -1,76 +0,0 @@
-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 main(app_id, repos):
-    app_id_list = list(filter(str.isdecimal, app_id.strip().split('-')))
-    app_id = app_id_list[0]
-    
-    async with ClientSession() as session:
-        github_token = config["Github_Personal_Token"]
-        headers = {'Authorization': f'Bearer {github_token}'} if github_token else None
-        latest_date = None
-        selected_repo = None
-
-        await check_github_api_rate_limit(headers, session)
-
-        for repo in repos:
-            url = f'https://api.github.com/repos/{repo}/branches/{app_id}'
-            try:
-                async with session.get(url, headers=headers, ssl=False) as r:
-                    r_json = await r.json()
-                    if '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
-            except Exception as e:
-                log.error(f' ⚠ 获取分支信息失败: {stack_error(e)}')
-        if selected_repo:
-            log.info(f' 🔄 选择清单仓库：{selected_repo}')
-            url = f'https://api.github.com/repos/{selected_repo}/branches/{app_id}'
-            async with session.get(url, headers=headers, ssl=False) as r:
-                r_json = await r.json()
-                if 'commit' in r_json:
-                    sha = r_json['commit']['sha']
-                    url = r_json['commit']['commit']['tree']['url']
-                    async with session.get(url, headers=headers, ssl=False) as r2:
-                        r2_json = await r2.json()
-                        if 'tree' in r2_json:
-                            collected_depots = []
-                            for i in r2_json['tree']:
-                                result = await get_manifest(sha, i['path'], steam_path, selected_repo, session)
-                                collected_depots.extend(result)
-                            if collected_depots:
-                                if isSteamTools:
-                                    migrate(st_use=True)
-                                    await stool_add(collected_depots, app_id)
-                                    log.info(' ✅ 找到SteamTools，已添加解锁文件')
-                                if isGreenLuma:
-                                    migrate(st_use=False)
-                                    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' ✅ 清单最后更新时间：{date}')
-                                log.info(f' ✅ 入库成功: {app_id}')
-                                os.system('pause')
-                                return True
-        log.error(f' ⚠ 清单下载或生成失败: {app_id}')
-        os.system('pause')
-        return False

common/manifest_down.py

@@ -1,37 +0,0 @@
-from aiohttp import ClientError
-from tqdm.asyncio import tqdm_asyncio
-
-from .log import log
-
-
-async def get(sha, path, repo, session):
-    url_list = [
-        f'https://cdn.jsdmirror.com/gh/{repo}@{sha}/{path}',
-        f'https://jsd.onmicrosoft.cn/gh/{repo}@{sha}/{path}',
-        f'https://mirror.ghproxy.com/https://raw.githubusercontent.com/{repo}/{sha}/{path}',
-        f'https://raw.githubusercontent.com/{repo}/{sha}/{path}',
-    ]
-    retry = 3
-    while retry:
-        for url in url_list:
-            try:
-                async with session.get(url, ssl=False) as r:
-                    if r.status == 200:
-                        total_size = int(r.headers.get('Content-Length', 0))
-                        chunk_size = 1024
-                        content = bytearray()
-
-                        with tqdm_asyncio(total=total_size, unit='B', unit_scale=True, desc=f'下载 {path}') as pbar:
-                            async for chunk in r.content.iter_chunked(chunk_size):
-                                content.extend(chunk)
-                                pbar.update(len(chunk))
-                        
-                        return content
-                    else:
-                        log.error(f' 🔄 获取失败: {path} - 状态码: {r.status}')
-            except ClientError:
-                log.error(f' 🔄 获取失败: {path} - 连接错误')
-        retry -= 1
-        log.warning(f' 🔄 重试剩余次数: {retry} - {path}')
-    log.error(f' 🔄 超过最大重试次数: {path}')
-    raise Exception(f' 🔄 无法下载: {path}')

common/migration.py

@@ -1,44 +0,0 @@
-import os
-import subprocess
-import requests
-
-from pathlib import Path
-from .log import log
-from .get_steam_path import steam_path
-
-directory = Path(steam_path / "config" / "stplug-in")
-
-def migrate(st_use):
-    if st_use == True:
-        log.info('检测到你正在使用SteamTools，尝试迁移旧文件')
-        if os.path.exists(directory):
-            for filename in os.listdir(directory):
-                if filename.startswith("Onekey_unlock_"):
-                    new_filename = filename[len("Onekey_unlock_"):]
-    
-                    old_file = os.path.join(directory, filename)
-                    new_file = os.path.join(directory, new_filename)
-
-                    try:
-                        os.replace(old_file, new_file)
-                        log.info(f'Renamed: {filename} -> {new_filename}')
-                    except Exception as e:
-                        log.error(f'Failed to rename {filename} -> {new_filename}: {e}')
-        else:
-            log.error('故障，正在重新安装SteamTools')
-            temp_path = './temp'
-            if not os.path.exists(temp_path):
-                os.mkdir(temp_path)
-            down_url = 'https://steamtools.net/res/SteamtoolsSetup.exe'
-            out_path = './temp/SteamtoolsSetup.exe'
-            with requests.get(down_url, stream=True) as r:
-                if r.status_code == 200:
-                    with open(out_path, 'wb') as f:
-                        for chunk in r.iter_content(chunk_size=8192):
-                            f.write(chunk)
-                else:
-                    log.error('网络错误')
-            subprocess.run(str(out_path))
-            os.rmdir(temp_path)
-    else:
-        log.info('未使用SteamTools，停止迁移')

common/st_unlock.py

@@ -1,25 +0,0 @@
-import os
-import asyncio
-import subprocess
-import aiofiles
-
-from .log import log
-from .get_steam_path import steam_path
-
-lock = asyncio.Lock()
-
-async def stool_add(depot_data, app_id):
-    lua_filename = f"{app_id}.lua"
-    lua_filepath = steam_path / "config" / "stplug-in" / lua_filename
-
-    async with lock:
-        log.info(f' ✅ SteamTools解锁文件生成: {lua_filepath}')
-        async with aiofiles.open(lua_filepath, mode="w", encoding="utf-8") as lua_file:
-            await lua_file.write(f'addappid({app_id}, 1, "None")\n')
-            for depot_id, depot_key in depot_data:
-                await lua_file.write(f'addappid({depot_id}, 1, "{depot_key}")\n')
-
-    luapacka_path = steam_path / "config" / "stplug-in" / "luapacka.exe"
-    subprocess.run([str(luapacka_path), str(lua_filepath)])
-    os.remove(lua_filepath)
-    return True

common/stack_error.py

@@ -1,5 +0,0 @@
-import traceback
-
-def stack_error(exception):
-    stack_trace = traceback.format_exception(type(exception), exception, exception.__traceback__)
-    return ''.join(stack_trace)

main.py

@@ -1,30 +1,494 @@
+import sys
 import os
+import traceback
 import time
+import logging
+import subprocess
 import asyncio
-
-from common.log import log
-from common.stack_error import stack_error
-from common.init_text import init
-from common.main_func import main
-
-lock = asyncio.Lock()
+import aiofiles
+import colorlog
+import httpx
+import winreg
+import ujson as json
+import vdf
+from typing import Any
+from pathlib import Path
+from colorama import init, Fore, Back, Style
 
 init()
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
 
-repos = [
-         'ikun0014/ManifestHub',
-         'Auiowu/ManifestAutoUpdate',
-         'tymolu233/ManifestAutoUpdate'
-        ]
-if __name__ == '__main__':
+lock = asyncio.Lock()
+client = httpx.AsyncClient(verify=False)
+
+
+DEFAULT_CONFIG = {
+    "Github_Personal_Token": "",
+    "Custom_Steam_Path": "",
+    "QA1": "温馨提示: Github_Personal_Token可在Github设置的最底下开发者选项找到, 详情看教程",
+    "教程": "https://ikunshare.com/Onekey_tutorial"
+}
+
+LOG_FORMAT = '%(log_color)s%(message)s'
+LOG_COLORS = {
+    'INFO': 'cyan',
+    'WARNING': 'yellow',
+    'ERROR': 'red',
+    'CRITICAL': 'purple',
+}
+
+
+def init_log(level=logging.DEBUG) -> logging.Logger:
+    """ 初始化日志模块 """
+    logger = logging.getLogger('Onekey')
+    logger.setLevel(level)
+
+    stream_handler = logging.StreamHandler()
+    stream_handler.setLevel(level)
+
+    fmt = colorlog.ColoredFormatter(LOG_FORMAT, log_colors=LOG_COLORS)
+    stream_handler.setFormatter(fmt)
+
+    if not logger.handlers:
+        logger.addHandler(stream_handler)
+
+    return logger
+
+
+log = init_log()
+
+
+def init():
+    """ 输出初始化信息 """
+    banner_lines = [
+        "  _____   __   _   _____   _   _    _____  __    __ ",
+        " /  _  \\ |  \\ | | | ____| | | / /  | ____| \\ \\  / /",
+        " | | | | |   \\| | | |__   | |/ /   | |__    \\ \\/ /",
+        " | | | | | |\\   | |  __|  | |\\ \\   |  __|    \\  / ",
+        " | |_| | | | \\  | | |___  | | \\ \\  | |___    / /",
+        " \\_____/ |_|  \\_| |_____| |_|  \\_\\ |_____|  /_/",
+    ]
+    for line in banner_lines:
+        log.info(line)
+
+    log.info('作者: ikun0014')
+    log.warning('本项目采用GNU General Public License v3开源许可证, 请勿用于商业用途')
+    log.info('版本: 1.3.6')
+    log.info(
+        '项目Github仓库: https://github.com/ikunshare/Onekey \n Gitee: https://gitee.com/ikun0014/Onekey'
+    )
+    log.info('官网: ikunshare.com')
+    log.warning(
+        '本项目完全开源免费, 如果你在淘宝, QQ群内通过购买方式获得, 赶紧回去骂商家死全家\n   交流群组:\n    https://t.me/ikunshare_qun'
+    )
+    log.info('App ID可以在SteamDB, SteamUI或Steam商店链接页面查看')
+
+
+def stack_error(exception: Exception) -> str:
+    """ 处理错误堆栈 """
+    stack_trace = traceback.format_exception(
+        type(exception), exception, exception.__traceback__)
+    return ''.join(stack_trace)
+
+
+async def gen_config_file():
+    """ 生成配置文件 """
     try:
-        while True:
-            log.info('App ID可以在SteamDB或Steam商店链接页面查看')
-            app_id = input("请输入游戏AppID：").strip()
-            asyncio.run(main(app_id, repos))
+        async with aiofiles.open("./config.json", mode="w", encoding="utf-8") as f:
+            await f.write(json.dumps(DEFAULT_CONFIG, indent=2, ensure_ascii=False, escape_forward_slashes=False))
+
+        log.info('程序可能为第一次启动或配置重置,请填写配置文件后重新启动程序')
     except KeyboardInterrupt:
-        exit()
+        log.info("程序已退出")
     except Exception as e:
-        log.error(f' ⚠ 发生错误: {stack_error(e)}，将在5秒后退出')
-        time.sleep(5)
-        os.system('pause')
+        log.error(f'配置文件生成失败,{stack_error(e)}')
+
+
+async def load_config() -> dict:
+    """ 加载配置文件 """
+    if not os.path.exists('./config.json'):
+        await gen_config_file()
+        os.system('pause')
+        sys.exit()
+
+    try:
+        async with aiofiles.open("./config.json", mode="r", encoding="utf-8") as f:
+            config = json.loads(await f.read())
+            return config
+    except KeyboardInterrupt:
+        log.info("程序已退出")
+    except Exception as e:
+        log.error(f"配置文件加载失败，原因: {stack_error(e)},重置配置文件中...")
+        os.remove("./config.json")
+        await gen_config_file()
+        os.system('pause')
+        sys.exit()
+
+config = asyncio.run(load_config())
+
+
+async def check_github_api_rate_limit(headers):
+    """ 检查Github请求数 """
+
+    if headers != None:
+        log.info(f"您已配置Github Token")
+
+    url = 'https://api.github.com/rate_limit'
+    try:
+        r = await client.get(url, headers=headers)
+        r_json = r.json()
+        if r.status_code == 200:
+            rate_limit = r_json.get('rate', {})
+            remaining_requests = rate_limit.get('remaining', 0)
+            reset_time = rate_limit.get('reset', 0)
+            reset_time_formatted = time.strftime(
+                '%Y-%m-%d %H:%M:%S', time.localtime(reset_time))
+            log.info(f'剩余请求次数: {remaining_requests}')
+            if remaining_requests == 0:
+                log.warning(f'GitHub API 请求数已用尽, 将在 {reset_time_formatted} 重置,建议生成一个填在配置文件里')
+        else:
+            log.error('Github请求数检查失败, 网络错误')
+    except KeyboardInterrupt:
+        log.info("程序已退出")
+    except httpx.ConnectError as e:
+        log.error(f'检查Github API 请求数失败, {stack_error(e)}')
+    except httpx.ConnectTimeout as e:
+        log.error(f'检查Github API 请求数超时: {stack_error(e)}')
+    except Exception as e:
+        log.error(f'发生错误: {stack_error(e)}')
+
+
+async def checkcn() -> bool:
+    try:
+        req = await client.get('https://mips.kugou.com/check/iscn?&format=json')
+        body = req.json()
+        scn = bool(body['flag'])
+        if (not scn):
+            log.info(
+                f"您在非中国大陆地区({body['country']})上使用了项目, 已自动切换回Github官方下载CDN")
+            os.environ['IS_CN'] = 'no'
+            return False
+        else:
+            os.environ['IS_CN'] = 'yes'
+            return True
+    except KeyboardInterrupt:
+        log.info("程序已退出")
+    except httpx.ConnectError as e:
+        os.environ['IS_CN'] = 'yes'
+        log.warning('检查服务器位置失败，已忽略，自动认为你在中国大陆')
+        log.warning(stack_error(e))
+        return False
+
+
+async def depotkey_merge(config_path: Path, depots_config: dict) -> bool:
+    if not config_path.exists():
+        async with lock:
+            log.error('Steam默认配置不存在, 可能是没有登录账号')
+        return False
+
+    try:
+        async with aiofiles.open(config_path, encoding='utf-8') as f:
+            content = await f.read()
+
+        config = vdf.loads(content)
+        steam = config.get('InstallConfigStore', {}).get('Software', {}).get('Valve') or config.get('InstallConfigStore', {}).get('Software', {}).get('valve')
+
+        if steam is None:
+            log.error('找不到Steam配置, 请检查配置文件')
+            return False
+
+        depots = steam.setdefault('depots', {})
+        depots.update(depots_config.get('depots', {}))
+
+        async with aiofiles.open(config_path, mode='w', encoding='utf-8') as f:
+            new_context = vdf.dumps(config, pretty=True)
+            await f.write(new_context)
+
+        log.info('成功合并')
+        return True
+    except KeyboardInterrupt:
+        log.info("程序已退出")
+    except Exception as e:
+        async with lock:
+            log.error(f'合并失败, 原因: {e}')
+        return False
+
+
+async def get(sha: str, path: str, repo: str):
+    if os.environ.get('IS_CN') == 'yes':
+        url_list = [
+            f'https://jsdelivr.pai233.top/gh/{repo}@{sha}/{path}',
+            f'https://cdn.jsdmirror.com/gh/{repo}@{sha}/{path}',
+            f'https://raw.gitmirror.com/{repo}/{sha}/{path}',
+            f'https://raw.dgithub.xyz/{repo}/{sha}/{path}',
+            f'https://gh.akass.cn/{repo}/{sha}/{path}'
+        ]
+    else:
+        url_list = [
+            f'https://raw.githubusercontent.com/{repo}/{sha}/{path}'
+        ]
+    retry = 3
+    while retry > 0:
+        for url in url_list:
+            try:
+                r = await client.get(url, timeout=30)
+                if r.status_code == 200:
+                    return r.read()
+                else:
+                    log.error(f'获取失败: {path} - 状态码: {r.status_code}')
+            except KeyboardInterrupt:
+                log.info("程序已退出")
+            except httpx.ConnectError as e:
+                log.error(f'获取失败: {path} - 连接错误: {str(e)}')
+            except httpx.ConnectTimeout as e:
+                log.error(f'连接超时: {url} - 错误: {str(e)}')
+
+        retry -= 1
+        log.warning(f'重试剩余次数: {retry} - {path}')
+
+    log.error(f'超过最大重试次数: {path}')
+    raise Exception(f'无法下载: {path}')
+
+
+async def get_manifest(sha: str, path: str, steam_path: Path, repo: str) -> list:
+    collected_depots = []
+    depot_cache_path = steam_path / 'depotcache'
+    try:
+        depot_cache_path.mkdir(exist_ok=True)
+        if path.endswith('.manifest'):
+            save_path = depot_cache_path / path
+            if save_path.exists():
+                log.warning(f'已存在清单: {save_path}')
+                return collected_depots
+            content = await get(sha, path, repo)
+            log.info(f'清单下载成功: {path}')
+            async with aiofiles.open(save_path, 'wb') as f:
+                await f.write(content)
+        elif path == 'Key.vdf':
+            content = await get(sha, path, repo)
+            log.info(f'密钥下载成功: {path}')
+            depots_config = vdf.loads(content.decode('utf-8'))
+            depots = dict(depots_config.get('depots'))
+            collected_depots = [
+                (depot_id, depot_info['DecryptionKey'])
+                for depot_id, depot_info in depots.items()
+            ]
+    except KeyboardInterrupt:
+        log.info("程序已退出")
+    except Exception as e:
+        log.error(f'处理失败: {path} - {stack_error(e)}')
+        raise
+    return collected_depots
+
+
+def get_steam_path() -> Path:
+    try:
+        key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, r'Software\Valve\Steam')
+        steam_path = Path(winreg.QueryValueEx(key, 'SteamPath')[0])
+
+        custom_steam_path = config.get("Custom_Steam_Path", "").strip()
+        return Path(custom_steam_path) if custom_steam_path else steam_path
+    except KeyboardInterrupt:
+        log.info("程序已退出")
+    except Exception as e:
+        log.error(f'Steam路径获取失败, {stack_error(e)}, 请检查是否正确安装Steam')
+        os.system('pause')
+        return Path()
+
+
+SP = get_steam_path()
+GL = any((SP / dll).exists() for dll in ['GreenLuma_2024_x86.dll', 'GreenLuma_2024_x64.dll', 'User32.dll'])
+ST = (SP / 'config' / 'stUI').is_dir()
+ST_PT = Path(SP) / "config" / "stplug-in"
+TP = Path('./temp')
+ST_STP_URL = 'https://steamtools.net/res/SteamtoolsSetup.exe'
+STP_FILE = TP / 'SteamtoolsSetup.exe'
+
+
+async def download_setup_file() -> None:
+    log.info('开始下载 SteamTools 安装程序...')
+    try:
+        r = await client.get(ST_STP_URL, timeout=30)
+        if r.status_code == 200:
+            async with aiofiles.open(STP_FILE, mode='wb') as f:
+                await f.write(r.read())
+            log.info('安装程序下载完成')
+        else:
+            log.error(f'网络错误，无法下载安装程序，状态码: {r.status_code}')
+    except KeyboardInterrupt:
+        log.info("程序已退出")
+    except httpx.ConnectTimeout:
+        log.error('下载时超时')
+    except Exception as e:
+        log.error(f'下载失败: {e}')
+
+
+async def migrate() -> None:
+    try:
+        log.info('检测到你正在使用 SteamTools,尝试迁移旧文件')
+        if SP.exists():
+            for file in SP.iterdir():
+                if file.is_file() and file.name.startswith("Onekey_unlock_"):
+                    new_filename = file.name[len("Onekey_unlock_"):]
+                    try:
+                        file.rename(SP / new_filename)
+                        log.info(f'Renamed: {file.name} -> {new_filename}')
+                    except Exception as e:
+                        log.error(
+                            f'重命名失败 {file.name} -> {new_filename}: {e}')
+        else:
+            log.error('故障,正在重新安装 SteamTools')
+            TP.mkdir(parents=True, exist_ok=True)
+            await download_setup_file(client)
+            subprocess.run(str(STP_FILE), check=True)
+            for file in TP.iterdir():
+                file.unlink()
+            TP.rmdir()
+    except KeyboardInterrupt:
+        log.info("程序已退出")
+
+
+async def stool_add(depot_data: list, app_id: str) -> bool:
+    lua_filename = f"{app_id}.lua"
+    lua_filepath = SP / "config" / "stplug-in" / lua_filename
+    async with lock:
+        log.info(f'SteamTools 解锁文件生成: {lua_filepath}')
+        try:
+            async with aiofiles.open(lua_filepath, mode="w", encoding="utf-8") as lua_file:
+                await lua_file.write(f'addappid({app_id}, 1, "None")\n')
+                for depot_id, depot_key in depot_data:
+                    await lua_file.write(f'addappid({depot_id}, 1, "{depot_key}")\n')
+            luapacka_path = SP / "config" / "stplug-in" / "luapacka.exe"
+            log.info(f'正在处理文件: {lua_filepath}')
+            result = subprocess.run(
+                [str(luapacka_path), str(lua_filepath)],
+                capture_output=True
+            )
+            if result.returncode != 0:
+                log.error(f'调用失败: {result.stderr.decode()}')
+                return False
+            log.info('处理完成')
+        except KeyboardInterrupt:
+            log.info("程序已退出")
+        except Exception as e:
+            log.error(f'处理过程出现错误: {e}')
+            return False
+        finally:
+            if lua_filepath.exists():
+                os.remove(lua_filepath)
+                log.info(f'删除临时文件: {lua_filepath}')
+    return True
+
+
+async def greenluma_add(depot_id_list: list) -> bool:
+    app_list_path = SP / 'AppList'
+    try:
+        app_list_path.mkdir(parents=True, exist_ok=True)
+        for file in app_list_path.glob('*.txt'):
+            file.unlink(missing_ok=True)
+        depot_dict = {
+            int(i.stem): int(i.read_text(encoding='utf-8').strip())
+            for i in app_list_path.iterdir() if i.is_file() and i.stem.isdecimal() and i.suffix == '.txt'
+        }
+        for depot_id in map(int, depot_id_list):
+            if depot_id not in depot_dict.values():
+                index = max(depot_dict.keys(), default=-1) + 1
+                while index in depot_dict:
+                    index += 1
+                (app_list_path /
+                 f'{index}.txt').write_text(str(depot_id), encoding='utf-8')
+                depot_dict[index] = depot_id
+        return True
+    except Exception as e:
+        print(f'处理时出错: {e}')
+        return False
+
+
+async def fetch_branch_info(url, headers) -> str | None:
+    try:
+        r = await client.get(url, headers=headers)
+        return r.json()
+    except KeyboardInterrupt:
+        log.info("程序已退出")
+    except Exception as e:
+        log.error(f'获取信息失败: {stack_error(e)}')
+        return None
+    except httpx.ConnectTimeout as e:
+        log.error(f'获取信息时超时: {stack_error(e)}')
+        return None
+
+
+async def get_latest_repo_info(repos: list, app_id: str, headers) -> Any | None:
+    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(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]
+    github_token = config.get("Github_Personal_Token", "")
+    headers = {'Authorization': f'Bearer {github_token}'} if github_token else None
+    await checkcn()
+    await check_github_api_rate_limit(headers)
+    selected_repo, latest_date = await get_latest_repo_info(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(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(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'], SP, selected_repo)
+                    collected_depots.extend(result)
+                if collected_depots:
+                    if ST:
+                        await migrate()
+                        await stool_add(collected_depots, app_id)
+                        log.info('找到SteamTools, 已添加解锁文件')
+                    elif GL:
+                        await greenluma_add([app_id])
+                        depot_config = {'depots': {depot_id: {'DecryptionKey': depot_key} for depot_id, depot_key in collected_depots}}
+                        await depotkey_merge(SP / '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}')
+                    await client.aclose()
+                    os.system('pause')
+                    return True
+    log.error(f'清单下载或生成失败: {app_id}')
+    await client.aclose()
+    os.system('pause')
+    return False
+
+
+if __name__ == '__main__':
+    init()
+    try:
+        repos = [
+            'ikun0014/ManifestHub',
+            'Auiowu/ManifestAutoUpdate',
+        ]
+        app_id = input(f"{Fore.CYAN}{Back.BLACK}{Style.BRIGHT}请输入游戏AppID: {Style.RESET_ALL}").strip()
+        asyncio.run(main(app_id, repos))
+    except KeyboardInterrupt:
+        log.info("程序已退出")
+    except SystemExit:
+        sys.exit()

package.json

@@ -1,6 +1,6 @@
 {
   "name": "onekey",
-  "version": "1.2.1",
+  "version": "1.3.6",
   "description": "一个Steam仓库清单下载器",
   "main": "index.js",
   "scripts": {

requirements.txt

@@ -1,8 +1,7 @@
-aiofiles
-colorlog
-ujson
-vdf
-nuitka
-requests
-imageio
-aiohttp
+aiofiles==24.1.0
+colorama==0.4.6
+colorlog==6.8.2
+httpx==0.27.2
+rich==13.9.4
+ujson==5.10.0
+vdf==3.4