#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""从 GitHub 星标时间生成仓库内的浅色、深色 SVG 和按天聚合数据。
使用 GITHUB_TOKEN 或 GH_TOKEN；仅保存日期和数量，不保存用户资料。
数据来自当前仍保留的星标，无法重建已取消星标的完整历史。
"""
import json
import os
import sys
import urllib.error
import urllib.request
from datetime import datetime, timezone
REPO = os.environ.get("STAR_REPO", "DSH-APP/DSHA")
OUT_DIR = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "docs")
W, H = 820, 420
PAD_L, PAD_R, PAD_T, PAD_B = 62, 24, 44, 52
def api(url, token, accept="application/vnd.github+json"):
req = urllib.request.Request(url, headers={
"Authorization": "token " + token,
"Accept": accept,
"User-Agent": "DSHA-star-history",
})
with urllib.request.urlopen(req, timeout=60) as r:
return json.load(r), r.headers.get("Link", "")
def next_link(link_header):
"""从 Link 头里取 rel="next" 的 URL；没有就返回 None。"""
for part in link_header.split(","):
if 'rel="next"' in part:
return part.split(";")[0].strip().strip("<>")
return None
def fetch_stars(token):
"""拉全部 stargazers 的 starred_at（升序）。"""
url = ("https://api.github.com/repos/%s/stargazers?per_page=100" % REPO)
stamps = []
while url:
try:
data, link = api(url, token, "application/vnd.github.star+json")
except urllib.error.HTTPError as e:
body = e.read().decode("utf-8", "replace")[:300]
print("拉 stargazers 失败 HTTP %s：%s" % (e.code, body), file=sys.stderr)
if e.code in (401, 403, 404):
print("请检查 token 是否具有本仓库读取权限及 API 限额。", file=sys.stderr)
raise
for it in data:
at = it.get("starred_at")
if at:
stamps.append(at)
url = next_link(link)
stamps.sort()
return stamps
def daily_series(stamps):
"""按天聚合成累计曲线：[(date, 累计数), …]。点少、曲线平滑、文件也小。"""
out, total = [], 0
for at in stamps:
day = at[:10]
total += 1
if out and out[-1][0] == day:
out[-1][1] = total
else:
out.append([day, total])
return [(d, n) for d, n in out]
def nice_max(v):
"""把 y 轴上限取整到好看的刻度。"""
if v <= 10:
return 10
for step in (10, 20, 25, 50, 100, 200, 250, 500, 1000, 2000, 5000, 10000):
if v <= step * 5:
return ((v + step - 1) // step) * step
return v
def days_between(a, b):
fmt = "%Y-%m-%d"
return (datetime.strptime(b, fmt) - datetime.strptime(a, fmt)).days
def esc(s):
return (s.replace("&", "&").replace("<", "<").replace(">", ">"))
def render(series, dark):
"""画 SVG。刻意不用任何绘图库 —— 这脚本要在干净的 runner 上零依赖跑起来。"""
if not series:
return ('' % W)
bg = "#0d1117" if dark else "#ffffff"
fg = "#e6edf3" if dark else "#24292f"
sub = "#8b949e" if dark else "#57606a"
grid = "#30363d" if dark else "#e1e4e8"
line = "#58a6ff" if dark else "#0969da"
fill = "rgba(88,166,255,0.16)" if dark else "rgba(9,105,218,0.10)"
first_day, last_day = series[0][0], series[-1][0]
today = datetime.now(timezone.utc).strftime("%Y-%m-%d")
if days_between(last_day, today) > 0:
series = series + [(today, series[-1][1])] # 曲线画到今天，末尾持平
last_day = today
span = max(days_between(first_day, last_day), 1)
ymax = nice_max(series[-1][1])
plot_w = W - PAD_L - PAD_R
plot_h = H - PAD_T - PAD_B
def px(day):
return PAD_L + plot_w * days_between(first_day, day) / span
def py(n):
return PAD_T + plot_h * (1 - n / ymax)
parts = ['")
return "".join(parts)
def main():
token = os.environ.get("GITHUB_TOKEN") or os.environ.get("GH_TOKEN")
if not token:
print("需要 GITHUB_TOKEN（能读本仓库即可）", file=sys.stderr)
return 1
stamps = fetch_stars(token)
series = daily_series(stamps)
if not os.path.isdir(OUT_DIR):
os.makedirs(OUT_DIR)
with open(os.path.join(OUT_DIR, "star-history.svg"), "w", encoding="utf-8") as f:
f.write(render(series, dark=False))
with open(os.path.join(OUT_DIR, "star-history-dark.svg"), "w", encoding="utf-8") as f:
f.write(render(series, dark=True))
with open(os.path.join(OUT_DIR, "star-history.json"), "w", encoding="utf-8") as f:
# 刻意**不写**「生成时间」这类每次都变的字段：写了的话，即使一颗新 star 都没有，
# 每次跑都会产出一次「文件变了」的提交 —— 首次上线就撞到了，CI 那趟自动提交
# 改的就只有这一行。只留与数据本身有关的东西，数据没变就完全没有 diff。
json.dump({"repo": REPO, "total": series[-1][1] if series else 0,
"latest": series[-1][0] if series else None,
"points": [{"date": d, "stars": n} for d, n in series]},
f, ensure_ascii=False, indent=1)
print("STAR_HISTORY_OK %s：%d stars，%d 个数据点（%s → %s）"
% (REPO, series[-1][1] if series else 0, len(series), series[0][0] if series else "-", series[-1][0] if series else "-"))
return 0
if __name__ == "__main__":
sys.exit(main())