feat: add AtomGit code and release sync workflows

2026-05-06 05:38:50 +00:00 · 2026-02-16 13:36:45 +08:00
parent f9e379bdcc
commit d9e05e594a
3 changed files with 156 additions and 0 deletions
@@ -0,0 +1,123 @@
+import os
+import time
+import requests
+from github import Github
+
+ATOMGIT_API = "https://api.atomgit.com/api/v5"
+ATOMGIT_OWNER = "naiba"
+ATOMGIT_REPO = "nezha-dashboard"
+GITHUB_REPO = "nezhahq/nezha"
+
+
+def get_github_latest_release():
+    g = Github()
+    repo = g.get_repo(GITHUB_REPO)
+    release = repo.get_latest_release()
+    if not release:
+        print("No releases found.")
+        return
+
+    print(f"Latest release tag is: {release.tag_name}")
+    print(f"Latest release info is: {release.body}")
+    files = []
+    for asset in release.get_assets():
+        url = asset.browser_download_url
+        name = asset.name
+
+        response = requests.get(url)
+        if response.status_code == 200:
+            with open(name, "wb") as f:
+                f.write(response.content)
+            print(f"Downloaded {name}")
+        else:
+            print(f"Failed to download {name}")
+        files.append(get_abs_path(name))
+    sync_to_atomgit(release.tag_name, release.body, files)
+
+
+def sync_to_atomgit(tag, body, files):
+    access_token = os.environ["ATOMGIT_PAT"]
+    release_api_uri = f"{ATOMGIT_API}/repos/{ATOMGIT_OWNER}/{ATOMGIT_REPO}/releases"
+
+    release_data = {
+        "access_token": access_token,
+        "tag_name": tag,
+        "name": tag,
+        "body": body,
+        "prerelease": False,
+        "target_commitish": "master",
+    }
+
+    release_resp = None
+    for attempt in range(3):
+        try:
+            release_resp = requests.post(release_api_uri, json=release_data, timeout=30)
+            release_resp.raise_for_status()
+            break
+        except requests.exceptions.Timeout:
+            print(
+                f"Create release timed out, retrying in 30s... (attempt {attempt + 1})"
+            )
+            time.sleep(30)
+        except requests.exceptions.RequestException as err:
+            print(f"Create release failed: {err}")
+            if release_resp is not None:
+                print(f"Response: {release_resp.text}")
+            break
+
+    if release_resp is None or release_resp.status_code not in (200, 201):
+        print("Failed to create release on AtomGit, aborting.")
+        return
+
+    print(f"Created release {tag} on AtomGit")
+
+    for file_path in files:
+        upload_asset(access_token, tag, file_path)
+
+    print("Sync is completed!")
+
+
+def upload_asset(access_token, tag, file_path):
+    file_name = os.path.basename(file_path)
+    upload_url_api = (
+        f"{ATOMGIT_API}/repos/{ATOMGIT_OWNER}/{ATOMGIT_REPO}"
+        f"/releases/{tag}/upload_url?file_name={file_name}"
+    )
+
+    for attempt in range(3):
+        try:
+            resp = requests.get(
+                upload_url_api,
+                headers={"Authorization": f"Bearer {access_token}"},
+                timeout=30,
+            )
+            resp.raise_for_status()
+            upload_info = resp.json()
+
+            obs_url = upload_info["url"]
+            obs_headers = upload_info["headers"]
+
+            with open(file_path, "rb") as f:
+                put_resp = requests.put(
+                    obs_url, headers=obs_headers, data=f, timeout=120
+                )
+
+            if put_resp.text.strip() == "success" or put_resp.status_code in (200, 201):
+                print(f"Uploaded {file_name}")
+                return
+            else:
+                print(
+                    f"Upload {file_name} failed: {put_resp.status_code} {put_resp.text}"
+                )
+        except requests.exceptions.RequestException as err:
+            print(f"Upload {file_name} attempt {attempt + 1} failed: {err}")
+            time.sleep(10)
+
+    print(f"Failed to upload {file_name} after 3 attempts")
+
+
+def get_abs_path(path):
+    return os.path.join(os.getcwd(), path)
+
+
+get_github_latest_release()