115 lines
3.8 KiB
Python
115 lines
3.8 KiB
Python
import os
|
|
import re
|
|
import requests
|
|
import yaml
|
|
import logging
|
|
from pathlib import Path
|
|
|
|
# 设置日志
|
|
logging.basicConfig(level=logging.INFO,
|
|
format='%(asctime)s - %(levelname)s - %(message)s',
|
|
handlers=[
|
|
logging.FileHandler("/tmp/github-release-mirror-log.txt"),
|
|
logging.StreamHandler()
|
|
])
|
|
|
|
# 获取当前脚本所在目录的绝对路径
|
|
script_dir = os.path.dirname(os.path.abspath(__file__))
|
|
config_path = os.path.join(script_dir, "config.yaml")
|
|
|
|
# 读取配置文件
|
|
with open(config_path, 'r') as file:
|
|
config = yaml.safe_load(file)
|
|
|
|
# 配置代理
|
|
proxies = {}
|
|
if 'proxy' in config and config['proxy'].get('enabled', False):
|
|
if 'http' in config['proxy']:
|
|
proxies['http'] = config['proxy']['http']
|
|
if 'https' in config['proxy']:
|
|
proxies['https'] = config['proxy']['https']
|
|
|
|
if proxies:
|
|
logging.info(f"已启用代理: {proxies}")
|
|
|
|
def download_file(url, filename):
|
|
"""下载文件到指定路径"""
|
|
request_kwargs = {'stream': True}
|
|
if proxies:
|
|
request_kwargs['proxies'] = proxies
|
|
|
|
with requests.get(url, **request_kwargs) as r:
|
|
with open(filename, 'wb') as f:
|
|
for chunk in r.iter_content(chunk_size=8192):
|
|
f.write(chunk)
|
|
logging.info(f"文件下载完成：{filename}")
|
|
|
|
def change_ownership(path, uid, gid):
|
|
"""更改文件或目录的所有者"""
|
|
os.chown(path, uid, gid)
|
|
for root, dirs, files in os.walk(path):
|
|
for d in dirs:
|
|
os.chown(os.path.join(root, d), uid, gid)
|
|
for f in files:
|
|
os.chown(os.path.join(root, f), uid, gid)
|
|
logging.info(f"已更改所有权：{path}")
|
|
|
|
def process_repo(repo_config, base_path, uid, gid):
|
|
repo_info = repo_config['repo-info']
|
|
folder_name = repo_config.get('folder-name', repo_info.split('/')[-1])
|
|
get_changes_info = repo_config.get('get-changes-info', True)
|
|
file_type = repo_config.get('file-type', [])
|
|
regex_pattern = repo_config.get('regex', None)
|
|
|
|
logging.info(f"\n正在处理项目：{folder_name}")
|
|
logging.info(f"仓库名称：{repo_info}")
|
|
|
|
API_URL = f"https://api.github.com/repos/{repo_info}/releases/latest"
|
|
|
|
request_kwargs = {}
|
|
if proxies:
|
|
request_kwargs['proxies'] = proxies
|
|
|
|
response = requests.get(API_URL, **request_kwargs)
|
|
latest_release = response.json()
|
|
latest_version = latest_release['tag_name']
|
|
release_notes = latest_release['body']
|
|
assets = latest_release['assets']
|
|
|
|
version_dir = base_path / folder_name / latest_version
|
|
if not version_dir.exists():
|
|
logging.info(f"发现新版本：{latest_version}，正在下载...")
|
|
version_dir.mkdir(parents=True)
|
|
|
|
if get_changes_info:
|
|
with open(version_dir / "CHANGELOG.md", "w") as f:
|
|
f.write(release_notes)
|
|
logging.info("已保存变更日志。")
|
|
|
|
for asset in assets:
|
|
if file_type and not any(asset['name'].endswith(ext) for ext in file_type):
|
|
continue
|
|
if regex_pattern and not re.match(regex_pattern, asset['name']):
|
|
continue
|
|
|
|
download_url = asset['browser_download_url']
|
|
filename = version_dir / asset['name']
|
|
download_file(download_url, filename)
|
|
|
|
if config['run-as-root'] and config['fix-chown']:
|
|
change_ownership(version_dir, uid, gid)
|
|
logging.info(f"版本 {latest_version} 下载完成。")
|
|
else:
|
|
logging.info(f"最新版本 {latest_version} 已存在。")
|
|
|
|
def main():
|
|
saved_path = Path(config['saved-path'])
|
|
uid = config['apply-uid']
|
|
gid = config['apply-gid']
|
|
|
|
for track_item, track_config in config['track-list'].items():
|
|
process_repo(track_config, saved_path, uid, gid)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
|