mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-21 09:28:14 +00:00
6bf34e051b
脚本从 scripts/ 移到 deploy/scripts/ 后目录深度 1→2，而两层 dirname
是写死的，于是资产目录解析成 deploy/dist/release，上传直接
FileNotFoundError（v1.0.1 首次上传即因此失败）。
这与 v0.7.2 的 2c5f9ff 把 package/ 移到 deploy/packaging/ 打断
build.sh 的 PROJECT_ROOT 是同一个坑：目录搬家没更新相对路径。改成
向上找 go.mod，以后脚本放哪都不会错。
顺带把两个静默失败改为显式报错：目录不存在、目录下无可识别产物
（原先前者抛裸 FileNotFoundError，后者会打出 ALL OK 却一个都没传）。
130 lines
4.7 KiB
Python
Executable File
130 lines
4.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
"""上传 release 资产到 gitcode（两步：取签名 URL → PUT 到 OBS）。
|
||
|
||
用法: upload_assets.py <tag> <token> [file...]
|
||
不传 file 时上传 dist/release/ 下全部发布产物。
|
||
|
||
环境变量：
|
||
GITCODE_REPO 目标仓库，默认 JianFeeeee/HomeAgent（SDK 仓传 JianFeeeee/homeagent-sdk）
|
||
ASSET_DIR 资产目录，默认 <repo>/dist/release
|
||
|
||
为何两步：gitcode 的 release 附件不走 API 直传，而是先向
|
||
`releases/<tag>/upload_url` 要一个 OBS 预签名 URL（带 x-obs-* 回调头），
|
||
再把文件 PUT 到那个 URL。回调头必须原样透传，否则 OBS 收下了文件但
|
||
gitcode 侧不会登记为 release 附件。
|
||
"""
|
||
import json
|
||
import os
|
||
import sys
|
||
import urllib.error
|
||
import urllib.parse
|
||
import urllib.request
|
||
|
||
REPO = os.environ.get("GITCODE_REPO", "JianFeeeee/HomeAgent")
|
||
API = "https://gitcode.com/api/v5/repos"
|
||
|
||
# 发布产物后缀。注意 Windows 安装器是 HomeAgent_v*_win64.exe，
|
||
# 与 bin/ 里的裸 .exe 靠 _win64.exe 后缀区分。
|
||
ARTIFACT_SUFFIXES = (
|
||
".tar.gz",
|
||
".zip",
|
||
".deb",
|
||
".rpm",
|
||
".pkg",
|
||
"_win64.exe",
|
||
)
|
||
|
||
|
||
def is_artifact(name: str) -> bool:
|
||
return name == "SHA256SUMS" or name.endswith(ARTIFACT_SUFFIXES)
|
||
|
||
|
||
def get_upload_url(tag: str, token: str, filename: str) -> tuple[str, dict]:
|
||
q = urllib.parse.urlencode({"file_name": filename})
|
||
url = f"{API}/{REPO}/releases/{tag}/upload_url?{q}"
|
||
req = urllib.request.Request(url, headers={"private-token": token})
|
||
with urllib.request.urlopen(req, timeout=30) as r:
|
||
data = json.loads(r.read())
|
||
return data["url"], data.get("headers", {})
|
||
|
||
|
||
def put_file(url: str, headers: dict, path: str) -> tuple[int, str]:
|
||
size = os.path.getsize(path)
|
||
with open(path, "rb") as f:
|
||
body = f.read()
|
||
req = urllib.request.Request(url, data=body, method="PUT")
|
||
for k, v in headers.items():
|
||
req.add_header(k, v)
|
||
req.add_header("Content-Length", str(size))
|
||
try:
|
||
# 大文件（Full 变体安装包近 100MB）给足超时。
|
||
with urllib.request.urlopen(req, timeout=900) as r:
|
||
return r.status, r.read().decode("utf-8", "replace")[:300]
|
||
except urllib.error.HTTPError as e:
|
||
return e.code, e.read().decode("utf-8", "replace")[:300]
|
||
except Exception as e: # noqa: BLE001
|
||
return 0, f"{type(e).__name__}: {e}"
|
||
|
||
|
||
def project_root() -> str:
|
||
"""向上找带 go.mod 的目录作为仓库根。
|
||
|
||
为何不数 dirname：本脚本初版在 scripts/（深度 1），移到 deploy/scripts/
|
||
（深度 2）后写死的两层 dirname 就指向了 deploy/dist/release，上传直接
|
||
FileNotFoundError。这正是 v0.7.2 那次 package/ → deploy/packaging/ 打断
|
||
PROJECT_ROOT 的同一个坑，改成按标记文件定位以后怎么挑位置都不会错。
|
||
"""
|
||
d = os.path.dirname(os.path.abspath(__file__))
|
||
while d != os.path.dirname(d):
|
||
if os.path.exists(os.path.join(d, "go.mod")):
|
||
return d
|
||
d = os.path.dirname(d)
|
||
# 实在找不到（脚本被单独拷出仓库）就回退到 cwd，给 ASSET_DIR 一个机会
|
||
return os.getcwd()
|
||
|
||
|
||
def main() -> int:
|
||
if len(sys.argv) < 3:
|
||
print(__doc__)
|
||
return 2
|
||
tag, token = sys.argv[1], sys.argv[2]
|
||
outdir = os.environ.get("ASSET_DIR") or os.path.join(
|
||
project_root(), "dist", "release"
|
||
)
|
||
if not os.path.isdir(outdir):
|
||
print(f"error: 资产目录不存在: {outdir}")
|
||
print(" 用 ASSET_DIR=<目录> 显式指定，或先跑构建生成 dist/release/")
|
||
return 2
|
||
files = sys.argv[3:] or sorted(
|
||
f for f in os.listdir(outdir) if is_artifact(f)
|
||
)
|
||
if not files:
|
||
print(f"error: {outdir} 下没有可识别的发布产物")
|
||
return 2
|
||
print(f"repo={REPO} tag={tag} dir={outdir}", flush=True)
|
||
failed = []
|
||
for name in files:
|
||
path = os.path.join(outdir, name)
|
||
if not os.path.isfile(path):
|
||
print(f"skip (missing): {name}", flush=True)
|
||
continue
|
||
mib = os.path.getsize(path) / 1048576
|
||
print(f"==> {name} ({mib:.1f} MiB)", flush=True)
|
||
try:
|
||
url, headers = get_upload_url(tag, token, name)
|
||
except Exception as e: # noqa: BLE001
|
||
print(f" upload_url FAILED: {e}", flush=True)
|
||
failed.append(name)
|
||
continue
|
||
status, body = put_file(url, headers, path)
|
||
ok = 200 <= status < 300
|
||
print(f" PUT -> {status} {'OK' if ok else body}", flush=True)
|
||
if not ok:
|
||
failed.append(name)
|
||
print(f"\n{'ALL OK' if not failed else f'{len(failed)} FAILED: ' + ', '.join(failed)}")
|
||
return 1 if failed else 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|