github_sync.py

import os
import requests
import base64
import logging
from urllib.parse import urlparse

# --- 配置日志 ---
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

# --- 从环境变量读取配置 (保留原项目的环境变量名) ---
GIT_TOKEN = os.getenv("GIT_TOKEN")
GIT_REPO_URL = os.getenv("GIT_REPO_URL")
FEED_FILE_PATH = "feed.xml" # 相对于仓库根目录的文件路径

def parse_repo_url(url):
    """从 GitHub URL 解析 owner 和 repo 名称"""
    if not url:
        return None, None
    try:
        parsed = urlparse(url)
        path_parts = parsed.path.strip('/').split('/')
        if len(path_parts) >= 2 and parsed.netloc == "github.com":
            owner = path_parts[0]
            repo = path_parts[1].replace('.git', '') # 移除可能的 .git 后缀
            return owner, repo
        else:
            logging.error(f"无法从 URL 解析 owner 和 repo: {url}")
            return None, None
    except Exception as e:
        logging.error(f"解析 URL 时出错 {url}: {e}")
        return None, None

OWNER, REPO = parse_repo_url(GIT_REPO_URL) # 使用 GIT_REPO_URL

def get_github_api_headers(token):
    """构造 GitHub API 请求头"""
    if not token:
        # 根据原 github_sync.py 的风格，这里可以抛出 ValueError 或者仅记录错误并返回 None
        # github_sync_new.py 抛出 ValueError，这里保持一致
        raise ValueError("GIT_TOKEN 环境变量未设置")
    return {
        "Authorization": f"token {token}",
        "Accept": "application/vnd.github.v3+json",
    }

def get_remote_feed():
    """从 GitHub 仓库获取 feed.xml 的内容和 SHA"""
    if not OWNER or not REPO:
        logging.error("无法确定 GitHub owner 或 repo。请检查 GIT_REPO_URL。")
        return None, None

    api_url = f"https://api.github.com/repos/{OWNER}/{REPO}/contents/{FEED_FILE_PATH}"
    try:
        headers = get_github_api_headers(GIT_TOKEN)
    except ValueError as e:
        logging.error(f"获取 API 请求头失败: {e}")
        return None, None

    try:
        response = requests.get(api_url, headers=headers)
        logging.info(f"GitHub API /contents 响应状态码: {response.status_code} for URL: {api_url}")

        if response.status_code == 200:
            data = response.json()
            logging.debug(f"GitHub API /contents 完整响应: {data}") # 改为 debug 级别，避免日志过多

            sha = data.get("sha")
            content_base64 = data.get("content")
            download_url = data.get("download_url")

            if sha is None:
                logging.error("GitHub API /contents 响应中缺少 'sha' 字段。")
                return None, None # SHA 是必须的

            if content_base64:
                # 文件较小，直接从 content 字段解码
                logging.info(f"文件 '{FEED_FILE_PATH}' 小于1MB，直接从 content 字段解码。")
                content_bytes = base64.b64decode(content_base64)
                content = content_bytes.decode('utf-8')
                logging.info(f"成功从 GitHub 获取 '{FEED_FILE_PATH}' (SHA: {sha})")
                return content, sha
            elif download_url:
                # 文件较大 (content 为 null)，通过 download_url 获取
                logging.info(f"文件 '{FEED_FILE_PATH}' 大于1MB 或 content 为空，尝试从 download_url 获取: {download_url}")
                try:
                    # 注意：访问 download_url 通常不需要额外的认证头，因为它通常是预签名的 S3 URL
                    # 但如果遇到权限问题，可以尝试也加上 headers
                    download_response = requests.get(download_url) # 可以考虑添加 headers=headers 如果需要
                    download_response.raise_for_status() # 如果下载失败则抛出异常
                    content = download_response.text # 或者 .content.decode('utf-8') 如果编码有问题
                    logging.info(f"成功通过 download_url 获取 '{FEED_FILE_PATH}' 内容 (SHA: {sha})")
                    return content, sha
                except requests.exceptions.RequestException as e_download:
                    logging.error(f"通过 download_url 获取文件内容时出错: {e_download}")
                    return None, None
            else:
                logging.error(f"GitHub API /contents 响应缺少 'content' 和 'download_url'。无法获取文件内容。SHA: {sha}")
                return None, None

        elif response.status_code == 404:
            logging.info(f"远程仓库中未找到 '{FEED_FILE_PATH}'。将创建新文件。")
            return None, None # 文件不存在，返回 None SHA
        else:
            logging.error(f"获取远程 feed 元数据失败: {response.status_code} - {response.text}")
            return None, None
    except requests.exceptions.RequestException as e_api:
        logging.error(f"请求 GitHub API (/contents) 时出错: {e_api}")
        return None, None
    except Exception as e_general:
        logging.error(f"处理 GitHub API (/contents) 响应时发生意外错误: {e_general}")
        return None, None


def push_feed_to_github(local_file_path, commit_message, remote_sha):
    """将本地 feed.xml 推送到 GitHub 仓库"""
    if not OWNER or not REPO:
        logging.error("无法确定 GitHub owner 或 repo。请检查 GIT_REPO_URL。")
        return False

    if not os.path.exists(local_file_path):
        logging.error(f"本地文件未找到: {local_file_path}")
        return False

    url = f"https://api.github.com/repos/{OWNER}/{REPO}/contents/{FEED_FILE_PATH}"
    try:
        headers = get_github_api_headers(GIT_TOKEN) # 使用 GIT_TOKEN
    except ValueError as e:
        logging.error(f"获取 API 请求头失败: {e}")
        return False

    try:
        with open(local_file_path, 'rb') as f:
            content_bytes = f.read()

        content_base64 = base64.b64encode(content_bytes).decode('utf-8')

        data = {
            "message": commit_message,
            "content": content_base64,
            "branch": "main" # 或者你的默认分支名
        }

        # 如果 remote_sha 存在，说明是更新现有文件，需要提供 SHA
        if remote_sha:
            data["sha"] = remote_sha
        # 如果 remote_sha 为 None，说明是创建新文件，不需要 SHA

        response = requests.put(url, headers=headers, json=data)

        if response.status_code == 200:
            logging.info(f"成功更新远程 '{FEED_FILE_PATH}'")
            return True
        elif response.status_code == 201:
            logging.info(f"成功创建远程 '{FEED_FILE_PATH}'")
            return True
        else:
            logging.error(f"推送 feed 到 GitHub 失败: {response.status_code} - {response.text}")
            try:
                error_data = response.json()
                logging.error(f"GitHub API 错误详情: {error_data.get('message', '无详细信息')}")
            except Exception:
                pass
            return False

    except requests.exceptions.RequestException as e:
        logging.error(f"请求 GitHub API 时出错: {e}")
        return False
    except Exception as e:
        logging.error(f"读取本地文件或处理推送时发生意外错误: {e}")
        return False

def sync_feed_to_github():
    """执行核心的 feed 文件同步逻辑"""
    if not GIT_TOKEN or not GIT_REPO_URL:
        logging.error("错误：请设置 GIT_TOKEN 和 GIT_REPO_URL 环境变量。")
        return False # 或者可以抛出异常
    elif not OWNER or not REPO:
        logging.error("错误：无法从 GIT_REPO_URL 解析仓库信息。请检查其格式。")
        return False

    logging.info(f"配置: Owner={OWNER}, Repo={REPO}")

    # 1. 尝试获取远程文件
    logging.info(f"--- 正在尝试从 GitHub 获取 {FEED_FILE_PATH} ---")
    remote_content, current_sha = get_remote_feed()

    if remote_content is not None:
        logging.info(f"成功获取远程内容 (SHA: {current_sha})。")
    elif current_sha is None and remote_content is None:
        logging.warning(f"远程文件 {FEED_FILE_PATH} 不存在或获取失败。")
    # else: # file not found case
        # logging.info(f"远程文件 {FEED_FILE_PATH} 不存在。")

    # 2. 准备一个本地文件用于测试推送
    if os.path.exists(FEED_FILE_PATH):
        logging.info(f"--- 正在尝试将本地 {FEED_FILE_PATH} 推送到 GitHub ---")
        commit_msg = f"Update {FEED_FILE_PATH} via script"
        success = push_feed_to_github(FEED_FILE_PATH, commit_msg, current_sha)
        if success:
            logging.info("推送成功！")
            return True
        else:
            logging.error("推送失败。")
            return False
    else:
        logging.info(f"跳过推送，因为本地文件 {FEED_FILE_PATH} 不存在。")
        return False # 或者根据需求返回其他状态

# --- 主执行逻辑 (用于直接运行脚本) ---
if __name__ == "__main__":
    sync_feed_to_github() # 调用新的主函数