113 lines
4.9 KiB
Python
113 lines
4.9 KiB
Python
"""
|
|
Gitea REST API клиент — fallback когда Gitea MCP недоступен.
|
|
Используется напрямую из агента если uvx/gitea-mcp не установлен.
|
|
|
|
REST-клиент для Gitea — альтернатива когда uvx gitea-mcp не установлен. Умеет:
|
|
ensure_repo(repo) — создать репозиторий если не существует
|
|
upload_file(file_path, content, commit_message) — залить файл, при обновлении сам находит SHA
|
|
get_file(file_path) — прочитать файл из репозитория
|
|
list_files(repo, path) — список файлов в папке
|
|
"""
|
|
|
|
import base64
|
|
import os
|
|
from typing import Optional
|
|
|
|
import requests
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
|
|
class GiteaRestClient:
|
|
def __init__(self):
|
|
self.base_url = os.getenv("GITEA_URL", "https://git.brojs.ru").rstrip("/")
|
|
self.token = os.getenv("GITEA_TOKEN")
|
|
self.username = os.getenv("GITEA_USERNAME")
|
|
self.repo = os.getenv("GITEA_REPO", "dz")
|
|
self.api = f"{self.base_url}/api/v1"
|
|
self.session = requests.Session()
|
|
self.session.headers.update({
|
|
"Authorization": f"token {self.token}",
|
|
"Content-Type": "application/json",
|
|
})
|
|
|
|
def ensure_repo(self, repo: str | None = None) -> bool:
|
|
"""Создаёт репозиторий если не существует. Возвращает True при успехе."""
|
|
target = repo or self.repo
|
|
resp = self.session.get(f"{self.api}/repos/{self.username}/{target}")
|
|
if resp.status_code == 200:
|
|
return True
|
|
resp = self.session.post(
|
|
f"{self.api}/user/repos",
|
|
json={
|
|
"name": target,
|
|
"description": "Решения домашних заданий",
|
|
"private": False,
|
|
"auto_init": True,
|
|
"default_branch": "main",
|
|
},
|
|
)
|
|
if resp.ok:
|
|
print(f"[GITEA] 🆕 Создан репозиторий: {target}")
|
|
return True
|
|
print(f"[GITEA] ❌ Ошибка создания репо: {resp.status_code} {resp.text[:200]}")
|
|
return False
|
|
|
|
def upload_file(self, file_path: str, content: str, commit_message: str, repo: str | None = None) -> str:
|
|
"""
|
|
Загружает файл в репозиторий.
|
|
Возвращает URL файла на Gitea.
|
|
"""
|
|
target = repo or self.repo
|
|
self.ensure_repo(target)
|
|
|
|
url = f"{self.api}/repos/{self.username}/{target}/contents/{file_path}"
|
|
content_b64 = base64.b64encode(content.encode("utf-8")).decode("utf-8")
|
|
|
|
# Проверяем sha (для обновления)
|
|
existing = self.session.get(url, params={"ref": "main"})
|
|
payload = {
|
|
"message": commit_message,
|
|
"content": content_b64,
|
|
"branch": "main",
|
|
}
|
|
if existing.status_code == 200:
|
|
payload["sha"] = existing.json().get("sha")
|
|
|
|
resp = self.session.put(url, json=payload)
|
|
resp.raise_for_status()
|
|
|
|
file_url = f"{self.base_url}/{self.username}/{target}/src/branch/main/{file_path}"
|
|
print(f"[GITEA] ✅ Загружен: {file_url}")
|
|
return file_url
|
|
|
|
def get_file(self, file_path: str, repo: str | None = None) -> str | None:
|
|
"""Читает содержимое файла из репозитория. Возвращает None если файл не найден."""
|
|
target = repo or self.repo
|
|
url = f"{self.api}/repos/{self.username}/{target}/contents/{file_path}"
|
|
resp = self.session.get(url, params={"ref": "main"})
|
|
if resp.status_code != 200:
|
|
return None
|
|
content_b64 = resp.json().get("content", "")
|
|
return base64.b64decode(content_b64).decode("utf-8")
|
|
|
|
def list_files(self, repo: str | None = None, path: str = "") -> list[str]:
|
|
"""Возвращает список имён файлов в корне репозитория."""
|
|
target = repo or self.repo
|
|
url = f"{self.api}/repos/{self.username}/{target}/contents/{path}"
|
|
resp = self.session.get(url, params={"ref": "main"})
|
|
if resp.status_code != 200:
|
|
return []
|
|
items = resp.json()
|
|
if not isinstance(items, list):
|
|
return []
|
|
return [item["name"] for item in items if item.get("type") == "file"]
|
|
|
|
def upload_solution(self, assignment_id: str, title: str, code: str, language: str = "py") -> str:
|
|
"""Удобный метод: залить решение и получить ссылку."""
|
|
safe = "".join(c if c.isalnum() or c in "-_" else "_" for c in title)[:40]
|
|
file_path = f"solutions/{assignment_id}_{safe}/solution.{language}"
|
|
commit_msg = f"Решение задания: {title}"
|
|
return self.upload_file(file_path, code, commit_msg)
|