Compare commits
1 Commits
main
...
fix/auto-2
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
37d171d198 |
@ -3,9 +3,13 @@ from sqlmodel.ext.asyncio.session import AsyncSession
|
||||
from sqlalchemy.ext.asyncio import create_async_engine
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
from sqlalchemy import text
|
||||
from sqlalchemy.exc import DBAPIError, OperationalError
|
||||
import os
|
||||
import logging
|
||||
from dotenv import load_dotenv
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
load_dotenv()
|
||||
|
||||
DB_USER = os.getenv("DB_USER")
|
||||
@ -21,7 +25,10 @@ engine = create_async_engine(
|
||||
echo=True,
|
||||
future=True,
|
||||
pool_pre_ping=True,
|
||||
pool_recycle=300,
|
||||
pool_recycle=180,
|
||||
pool_size=5,
|
||||
max_overflow=10,
|
||||
pool_timeout=30,
|
||||
)
|
||||
|
||||
async def init_db():
|
||||
@ -55,9 +62,21 @@ async def init_db():
|
||||
except Exception:
|
||||
pass # Already applied
|
||||
|
||||
async def get_session() -> AsyncSession:
|
||||
async_session = sessionmaker(
|
||||
_async_session_factory = sessionmaker(
|
||||
engine, class_=AsyncSession, expire_on_commit=False
|
||||
)
|
||||
async with async_session() as session:
|
||||
)
|
||||
|
||||
_MAX_RETRIES = 2
|
||||
|
||||
async def get_session() -> AsyncSession:
|
||||
for attempt in range(_MAX_RETRIES):
|
||||
try:
|
||||
async with _async_session_factory() as session:
|
||||
yield session
|
||||
return
|
||||
except (DBAPIError, OperationalError, ConnectionResetError, OSError) as e:
|
||||
if attempt < _MAX_RETRIES - 1:
|
||||
logger.warning("Database connection error (attempt %d/%d): %s", attempt + 1, _MAX_RETRIES, e)
|
||||
await engine.dispose()
|
||||
continue
|
||||
raise
|
||||
|
||||
@ -1,10 +1,9 @@
|
||||
"""
|
||||
Gitea API 客户端
|
||||
"""
|
||||
import re
|
||||
import httpx
|
||||
import os
|
||||
from typing import Tuple, Optional
|
||||
from typing import Tuple
|
||||
|
||||
|
||||
class GiteaClient:
|
||||
@ -13,7 +12,7 @@ class GiteaClient:
|
||||
def __init__(self, gitea_url: str = None, token: str = None):
|
||||
self.gitea_url = (gitea_url or os.getenv("GITEA_URL", "")).rstrip("/")
|
||||
self.token = token or os.getenv("GITEA_TOKEN", "")
|
||||
self.base_api_url = f"{self.gitea_url}/api/v1" if self.gitea_url else ""
|
||||
self.base_api_url = f"{self.gitea_url}/api/v1"
|
||||
self.client = httpx.Client(timeout=30)
|
||||
|
||||
def _headers(self) -> dict:
|
||||
@ -23,31 +22,21 @@ class GiteaClient:
|
||||
"Content-Type": "application/json",
|
||||
}
|
||||
|
||||
def _parse_pr_url(self, pr_url: str) -> Tuple[str, str, str, int]:
|
||||
def _extract_owner_repo_from_pr_url(self, pr_url: str) -> Tuple[str, str, int]:
|
||||
"""
|
||||
从 PR URL 提取 base_url, owner, repo, pr_number
|
||||
从 PR URL 提取 owner, repo, pr_number
|
||||
例如: https://gitea.airlabs.art/owner/repo/pulls/45
|
||||
返回: ("https://gitea.airlabs.art", "owner", "repo", 45)
|
||||
"""
|
||||
match = re.search(r'(https?://[^/]+)/([^/]+)/([^/]+)/pulls/(\d+)', pr_url)
|
||||
import re
|
||||
match = re.search(r'([^/]+)/([^/]+)/pulls/(\d+)', pr_url)
|
||||
if not match:
|
||||
raise ValueError(f"无法解析 PR URL: {pr_url}")
|
||||
|
||||
base_url, owner, repo, pr_number = match.groups()
|
||||
return base_url, owner, repo, int(pr_number)
|
||||
|
||||
def _get_api_url(self, pr_url: Optional[str] = None) -> str:
|
||||
"""获取 API base URL,优先从 pr_url 解析,否则用 self.base_api_url"""
|
||||
if pr_url:
|
||||
match = re.search(r'(https?://[^/]+)/', pr_url)
|
||||
if match:
|
||||
return f"{match.group(1)}/api/v1"
|
||||
if self.base_api_url:
|
||||
return self.base_api_url
|
||||
raise ValueError("无法确定 Gitea API 地址:未配置 GITEA_URL 且 PR URL 中无法提取")
|
||||
owner, repo, pr_number = match.groups()
|
||||
return owner, repo, int(pr_number)
|
||||
|
||||
def merge_pr(
|
||||
self, owner: str, repo: str, pr_number: int, api_base_url: str = None
|
||||
self, owner: str, repo: str, pr_number: int
|
||||
) -> Tuple[bool, str]:
|
||||
"""
|
||||
合并 PR
|
||||
@ -56,13 +45,11 @@ class GiteaClient:
|
||||
owner: 仓库所有者
|
||||
repo: 仓库名称
|
||||
pr_number: PR 编号
|
||||
api_base_url: 可选,API base URL(从 pr_url 解析得到)
|
||||
|
||||
Returns:
|
||||
(是否成功, 消息)
|
||||
"""
|
||||
base = api_base_url or self.base_api_url
|
||||
url = f"{base}/repos/{owner}/{repo}/pulls/{pr_number}/merge"
|
||||
url = f"{self.base_api_url}/repos/{owner}/{repo}/pulls/{pr_number}/merge"
|
||||
payload = {
|
||||
"Do": "merge", # merge / squash / rebase
|
||||
"MergeMessageField": f"Merge PR #{pr_number} (approved via Log Center)",
|
||||
@ -88,7 +75,7 @@ class GiteaClient:
|
||||
return False, str(e)
|
||||
|
||||
def close_pr(
|
||||
self, owner: str, repo: str, pr_number: int, reason: str = "", api_base_url: str = None
|
||||
self, owner: str, repo: str, pr_number: int, reason: str = ""
|
||||
) -> Tuple[bool, str]:
|
||||
"""
|
||||
关闭 PR(可选添加评论说明原因)
|
||||
@ -98,24 +85,20 @@ class GiteaClient:
|
||||
repo: 仓库名称
|
||||
pr_number: PR 编号
|
||||
reason: 关闭原因(将作为评论添加)
|
||||
api_base_url: 可选,API base URL(从 pr_url 解析得到)
|
||||
|
||||
Returns:
|
||||
(是否成功, 消息)
|
||||
"""
|
||||
base = api_base_url or self.base_api_url
|
||||
|
||||
# Step 1: 如果提供了原因,先添加评论
|
||||
if reason:
|
||||
comment_success, comment_msg = self.add_pr_comment(
|
||||
owner, repo, pr_number, f"## ❌ 修复被拒绝\n\n**原因:**\n{reason}",
|
||||
api_base_url=base,
|
||||
owner, repo, pr_number, f"## ❌ 修复被拒绝\n\n**原因:**\n{reason}"
|
||||
)
|
||||
if not comment_success:
|
||||
return False, f"添加评论失败: {comment_msg}"
|
||||
|
||||
# Step 2: 关闭 PR
|
||||
url = f"{base}/repos/{owner}/{repo}/pulls/{pr_number}"
|
||||
url = f"{self.base_api_url}/repos/{owner}/{repo}/pulls/{pr_number}"
|
||||
payload = {"state": "closed"}
|
||||
|
||||
try:
|
||||
@ -128,7 +111,7 @@ class GiteaClient:
|
||||
return False, str(e)
|
||||
|
||||
def add_pr_comment(
|
||||
self, owner: str, repo: str, pr_number: int, comment: str, api_base_url: str = None
|
||||
self, owner: str, repo: str, pr_number: int, comment: str
|
||||
) -> Tuple[bool, str]:
|
||||
"""
|
||||
添加 PR 评论
|
||||
@ -138,14 +121,12 @@ class GiteaClient:
|
||||
repo: 仓库名称
|
||||
pr_number: PR 编号
|
||||
comment: 评论内容
|
||||
api_base_url: 可选,API base URL
|
||||
|
||||
Returns:
|
||||
(是否成功, 消息)
|
||||
"""
|
||||
base = api_base_url or self.base_api_url
|
||||
# 注意: Gitea 中 PR 和 Issue 共用评论 API
|
||||
url = f"{base}/repos/{owner}/{repo}/issues/{pr_number}/comments"
|
||||
url = f"{self.base_api_url}/repos/{owner}/{repo}/issues/{pr_number}/comments"
|
||||
payload = {"body": comment}
|
||||
|
||||
try:
|
||||
@ -156,19 +137,17 @@ class GiteaClient:
|
||||
return False, str(e)
|
||||
|
||||
def merge_pr_by_url(self, pr_url: str) -> Tuple[bool, str]:
|
||||
"""通过 PR URL 直接合并(从 URL 自动解析 Gitea 地址)"""
|
||||
"""通过 PR URL 直接合并"""
|
||||
try:
|
||||
base_url, owner, repo, pr_number = self._parse_pr_url(pr_url)
|
||||
api_base_url = f"{base_url}/api/v1"
|
||||
return self.merge_pr(owner, repo, pr_number, api_base_url=api_base_url)
|
||||
owner, repo, pr_number = self._extract_owner_repo_from_pr_url(pr_url)
|
||||
return self.merge_pr(owner, repo, pr_number)
|
||||
except Exception as e:
|
||||
return False, str(e)
|
||||
|
||||
def close_pr_by_url(self, pr_url: str, reason: str = "") -> Tuple[bool, str]:
|
||||
"""通过 PR URL 直接关闭(从 URL 自动解析 Gitea 地址)"""
|
||||
"""通过 PR URL 直接关闭"""
|
||||
try:
|
||||
base_url, owner, repo, pr_number = self._parse_pr_url(pr_url)
|
||||
api_base_url = f"{base_url}/api/v1"
|
||||
return self.close_pr(owner, repo, pr_number, reason, api_base_url=api_base_url)
|
||||
owner, repo, pr_number = self._extract_owner_repo_from_pr_url(pr_url)
|
||||
return self.close_pr(owner, repo, pr_number, reason)
|
||||
except Exception as e:
|
||||
return False, str(e)
|
||||
|
||||
39
app/main.py
39
app/main.py
@ -85,8 +85,7 @@ def generate_fingerprint(log: ErrorLogCreate) -> str:
|
||||
|
||||
if source == "cicd":
|
||||
ctx = log.context or {}
|
||||
# 加入 error_message 避免同一 job 不同错误被去重
|
||||
raw = f"{log.project_id}|cicd|{log.error.get('type')}|{ctx.get('job_name', 'unknown')}|{ctx.get('step_name', 'unknown')}|{log.error.get('message', '')}"
|
||||
raw = f"{log.project_id}|cicd|{log.error.get('type')}|{ctx.get('job_name', 'unknown')}|{ctx.get('step_name', 'unknown')}"
|
||||
elif source == "deployment":
|
||||
ctx = log.context or {}
|
||||
raw = f"{log.project_id}|deployment|{log.error.get('type')}|{ctx.get('namespace', 'default')}|{ctx.get('deployment_name', 'unknown')}"
|
||||
@ -106,28 +105,13 @@ async def report_log(log_data: ErrorLogCreate, session: AsyncSession = Depends(g
|
||||
existing_log = results.first()
|
||||
|
||||
if existing_log:
|
||||
# If exists and not resolved, update error content but keep status
|
||||
# If exists and not resolved, just ignore or update count (implied)
|
||||
if existing_log.status not in [LogStatus.DEPLOYED, LogStatus.FIXED, LogStatus.VERIFIED]:
|
||||
# 更新错误内容,确保始终反映最新的错误信息
|
||||
existing_log.error_message = log_data.error.get("message", existing_log.error_message)
|
||||
existing_log.stack_trace = log_data.error.get("stack_trace", existing_log.stack_trace)
|
||||
existing_log.context = log_data.context or existing_log.context
|
||||
existing_log.timestamp = log_data.timestamp or datetime.utcnow()
|
||||
if log_data.commit_hash:
|
||||
existing_log.commit_hash = log_data.commit_hash
|
||||
session.add(existing_log)
|
||||
await session.commit()
|
||||
await session.refresh(existing_log)
|
||||
return {"message": "Log deduplicated (content updated)", "id": existing_log.id, "status": existing_log.status}
|
||||
# If it was resolved but happened again -> Regression! Reset to NEW
|
||||
return {"message": "Log deduplicated", "id": existing_log.id, "status": existing_log.status}
|
||||
# If it was resolved but happened again -> Regression! Reset to NEW?
|
||||
existing_log.status = LogStatus.NEW
|
||||
existing_log.error_message = log_data.error.get("message", existing_log.error_message)
|
||||
existing_log.stack_trace = log_data.error.get("stack_trace", existing_log.stack_trace)
|
||||
existing_log.context = log_data.context or existing_log.context
|
||||
existing_log.timestamp = log_data.timestamp or datetime.utcnow()
|
||||
existing_log.retry_count = 0
|
||||
if log_data.commit_hash:
|
||||
existing_log.commit_hash = log_data.commit_hash
|
||||
existing_log.retry_count = 0 # Reset retries for new occurrence
|
||||
session.add(existing_log)
|
||||
await session.commit()
|
||||
await session.refresh(existing_log)
|
||||
@ -604,19 +588,6 @@ async def update_project(project_id: str, data: ProjectUpdate, session: AsyncSes
|
||||
await session.refresh(project)
|
||||
return project
|
||||
|
||||
@app.delete("/api/v1/projects/{project_id}", tags=["Projects"])
|
||||
async def delete_project(project_id: str, session: AsyncSession = Depends(get_session)):
|
||||
"""Delete a project by project_id"""
|
||||
statement = select(Project).where(Project.project_id == project_id)
|
||||
results = await session.exec(statement)
|
||||
project = results.first()
|
||||
if not project:
|
||||
raise HTTPException(status_code=404, detail="Project not found")
|
||||
|
||||
await session.delete(project)
|
||||
await session.commit()
|
||||
return {"message": "Project deleted"}
|
||||
|
||||
@app.get("/", tags=["Health"])
|
||||
async def health_check():
|
||||
return {"status": "ok"}
|
||||
|
||||
@ -32,9 +32,6 @@ spec:
|
||||
value: "log_center"
|
||||
- name: DB_PASSWORD
|
||||
value: "JogNQdtrd3WY8CBCAiYfYEGx"
|
||||
# Gitea Token(URL 从 PR URL 自动解析)
|
||||
- name: GITEA_TOKEN
|
||||
value: "443f7f2f556b4832f90e46df9af3e21ccb06b8a3"
|
||||
resources:
|
||||
requests:
|
||||
memory: "128Mi"
|
||||
|
||||
@ -38,7 +38,6 @@ class ClaudeService:
|
||||
cmd = [
|
||||
self.cli_path,
|
||||
"-p", prompt,
|
||||
"--model", settings.claude_model,
|
||||
"--dangerously-skip-permissions",
|
||||
]
|
||||
|
||||
|
||||
@ -19,7 +19,6 @@ class Settings(BaseSettings):
|
||||
# Claude CLI
|
||||
claude_cli_path: str = Field(default="claude", description="Claude CLI 路径")
|
||||
claude_timeout: int = Field(default=1000, description="Claude CLI 超时时间(秒)")
|
||||
claude_model: str = Field(default="claude-opus-4-6", description="Claude 模型")
|
||||
|
||||
# Git
|
||||
git_user_name: str = Field(default="repair-agent", description="Git 用户名")
|
||||
|
||||
@ -160,9 +160,6 @@ export const getProjectDetail = (projectId: string) =>
|
||||
export const updateProject = (projectId: string, data: Partial<Project>) =>
|
||||
api.put<Project>(`/api/v1/projects/${projectId}`, data);
|
||||
|
||||
export const deleteProject = (projectId: string) =>
|
||||
api.delete(`/api/v1/projects/${projectId}`);
|
||||
|
||||
export const updateTaskStatus = (taskId: number, status: string, message?: string) =>
|
||||
api.put(`/api/v1/tasks/${taskId}/status`, { status, message });
|
||||
|
||||
|
||||
@ -1,6 +1,6 @@
|
||||
import { useState, useEffect } from 'react';
|
||||
import { getProjects, updateProject, deleteProject, type Project } from '../api';
|
||||
import { Save, X, Pencil, Trash2 } from 'lucide-react';
|
||||
import { getProjects, updateProject, type Project } from '../api';
|
||||
import { Save, X, Pencil } from 'lucide-react';
|
||||
|
||||
export default function ProjectList() {
|
||||
const [projects, setProjects] = useState<Project[]>([]);
|
||||
@ -9,11 +9,6 @@ export default function ProjectList() {
|
||||
const [editForm, setEditForm] = useState({ name: '', repo_url: '', local_path: '', description: '' });
|
||||
const [saving, setSaving] = useState(false);
|
||||
|
||||
// Delete confirmation state
|
||||
const [deleteTarget, setDeleteTarget] = useState<Project | null>(null);
|
||||
const [deleteConfirmName, setDeleteConfirmName] = useState('');
|
||||
const [deleting, setDeleting] = useState(false);
|
||||
|
||||
const fetchProjects = async () => {
|
||||
setLoading(true);
|
||||
try {
|
||||
@ -55,32 +50,6 @@ export default function ProjectList() {
|
||||
}
|
||||
};
|
||||
|
||||
const openDeleteConfirm = (p: Project) => {
|
||||
setDeleteTarget(p);
|
||||
setDeleteConfirmName('');
|
||||
};
|
||||
|
||||
const closeDeleteConfirm = () => {
|
||||
setDeleteTarget(null);
|
||||
setDeleteConfirmName('');
|
||||
};
|
||||
|
||||
const confirmDelete = async () => {
|
||||
if (!deleteTarget) return;
|
||||
setDeleting(true);
|
||||
try {
|
||||
await deleteProject(deleteTarget.project_id);
|
||||
closeDeleteConfirm();
|
||||
await fetchProjects();
|
||||
} catch (err) {
|
||||
console.error('删除失败:', err);
|
||||
} finally {
|
||||
setDeleting(false);
|
||||
}
|
||||
};
|
||||
|
||||
const isDeleteConfirmed = deleteTarget && deleteConfirmName === deleteTarget.project_id;
|
||||
|
||||
const ConfigBadge = ({ value }: { value: string | null }) => (
|
||||
value ? (
|
||||
<span style={{ fontSize: '12px', color: 'var(--text-secondary)', wordBreak: 'break-all' }}>{value}</span>
|
||||
@ -195,18 +164,9 @@ export default function ProjectList() {
|
||||
{new Date(p.updated_at).toLocaleString()}
|
||||
</td>
|
||||
<td>
|
||||
<div style={{ display: 'flex', gap: '6px' }}>
|
||||
<button className="btn-link" onClick={() => startEdit(p)}>
|
||||
<Pencil size={14} /> 编辑
|
||||
</button>
|
||||
<button
|
||||
className="btn-link"
|
||||
onClick={() => openDeleteConfirm(p)}
|
||||
style={{ color: 'var(--error)' }}
|
||||
>
|
||||
<Trash2 size={14} /> 删除
|
||||
</button>
|
||||
</div>
|
||||
</td>
|
||||
</>
|
||||
)}
|
||||
@ -273,18 +233,9 @@ export default function ProjectList() {
|
||||
<>
|
||||
<div className="mobile-card-top">
|
||||
<strong>{p.project_id}</strong>
|
||||
<div style={{ display: 'flex', gap: '8px' }}>
|
||||
<button className="btn-link" onClick={() => startEdit(p)}>
|
||||
<Pencil size={14} />
|
||||
</button>
|
||||
<button
|
||||
className="btn-link"
|
||||
onClick={() => openDeleteConfirm(p)}
|
||||
style={{ color: 'var(--error)' }}
|
||||
>
|
||||
<Trash2 size={14} />
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
{p.name && <div style={{ fontSize: '13px' }}>{p.name}</div>}
|
||||
<div className="mobile-card-meta" style={{ flexDirection: 'column', alignItems: 'flex-start', gap: '4px', marginTop: '6px' }}>
|
||||
@ -302,76 +253,6 @@ export default function ProjectList() {
|
||||
</div>
|
||||
</>
|
||||
)}
|
||||
|
||||
{/* Delete confirmation modal */}
|
||||
{deleteTarget && (
|
||||
<div
|
||||
style={{
|
||||
position: 'fixed',
|
||||
top: 0,
|
||||
left: 0,
|
||||
right: 0,
|
||||
bottom: 0,
|
||||
background: 'rgba(0,0,0,0.5)',
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'center',
|
||||
zIndex: 1000,
|
||||
}}
|
||||
onClick={closeDeleteConfirm}
|
||||
>
|
||||
<div
|
||||
style={{
|
||||
background: 'var(--bg-secondary)',
|
||||
borderRadius: 'var(--radius-md)',
|
||||
padding: '24px',
|
||||
maxWidth: '440px',
|
||||
width: '90%',
|
||||
boxShadow: '0 4px 12px rgba(0,0,0,0.15)',
|
||||
border: '1px solid var(--border)',
|
||||
}}
|
||||
onClick={(e) => e.stopPropagation()}
|
||||
>
|
||||
<h3 style={{ margin: '0 0 8px 0', fontSize: '16px' }}>删除项目</h3>
|
||||
<p style={{ margin: '0 0 16px 0', fontSize: '13px', color: 'var(--text-secondary)' }}>
|
||||
此操作不可撤销。请输入项目 ID <strong style={{ color: 'var(--error)' }}>{deleteTarget.project_id}</strong> 以确认删除。
|
||||
</p>
|
||||
<input
|
||||
type="text"
|
||||
value={deleteConfirmName}
|
||||
onChange={(e) => setDeleteConfirmName(e.target.value)}
|
||||
placeholder={deleteTarget.project_id}
|
||||
className="edit-input"
|
||||
style={{ width: '100%', marginBottom: '16px', boxSizing: 'border-box' }}
|
||||
autoFocus
|
||||
/>
|
||||
<div style={{ display: 'flex', justifyContent: 'flex-end', gap: '8px' }}>
|
||||
<button
|
||||
className="btn-link"
|
||||
onClick={closeDeleteConfirm}
|
||||
style={{ color: 'var(--text-tertiary)', padding: '6px 12px' }}
|
||||
>
|
||||
取消
|
||||
</button>
|
||||
<button
|
||||
onClick={confirmDelete}
|
||||
disabled={!isDeleteConfirmed || deleting}
|
||||
style={{
|
||||
background: isDeleteConfirmed ? 'var(--error)' : 'var(--bg-tertiary)',
|
||||
color: isDeleteConfirmed ? '#fff' : 'var(--text-tertiary)',
|
||||
border: 'none',
|
||||
borderRadius: 'var(--radius-sm, 4px)',
|
||||
padding: '6px 16px',
|
||||
fontSize: '13px',
|
||||
cursor: isDeleteConfirmed ? 'pointer' : 'not-allowed',
|
||||
}}
|
||||
>
|
||||
{deleting ? '删除中...' : '确认删除'}
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user