mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
144 lines
4.2 KiB
Python
144 lines
4.2 KiB
Python
import requests
|
|
from fastapi import APIRouter, HTTPException, Request
|
|
from fastapi.responses import JSONResponse
|
|
|
|
from openhands.server.shared import openhands_config
|
|
from openhands.utils.async_utils import call_sync_from_async
|
|
|
|
app = APIRouter(prefix='/api/github')
|
|
|
|
|
|
@app.get('/repositories')
|
|
async def get_github_repositories(
|
|
request: Request,
|
|
page: int = 1,
|
|
per_page: int = 10,
|
|
sort: str = 'pushed',
|
|
installation_id: int | None = None,
|
|
):
|
|
openhands_config.verify_github_repo_list(installation_id)
|
|
|
|
# Add query parameters
|
|
params: dict[str, str] = {
|
|
'page': str(page),
|
|
'per_page': str(per_page),
|
|
}
|
|
# Construct the GitHub API URL
|
|
if installation_id:
|
|
github_api_url = (
|
|
f'https://api.github.com/user/installations/{installation_id}/repositories'
|
|
)
|
|
else:
|
|
github_api_url = 'https://api.github.com/user/repos'
|
|
params['sort'] = sort
|
|
|
|
# Set the authorization header with the GitHub token
|
|
headers = generate_github_headers(request.state.github_token)
|
|
|
|
# Fetch repositories from GitHub
|
|
try:
|
|
response = await call_sync_from_async(
|
|
requests.get, github_api_url, headers=headers, params=params
|
|
)
|
|
response.raise_for_status() # Raise an error for HTTP codes >= 400
|
|
except requests.exceptions.RequestException as e:
|
|
raise HTTPException(
|
|
status_code=response.status_code if response else 500,
|
|
detail=f'Error fetching repositories: {str(e)}',
|
|
)
|
|
|
|
# Create response with the JSON content
|
|
json_response = JSONResponse(content=response.json())
|
|
response.close()
|
|
|
|
# Forward the Link header if it exists
|
|
if 'Link' in response.headers:
|
|
json_response.headers['Link'] = response.headers['Link']
|
|
|
|
return json_response
|
|
|
|
|
|
@app.get('/user')
|
|
async def get_github_user(request: Request):
|
|
headers = generate_github_headers(request.state.github_token)
|
|
try:
|
|
response = await call_sync_from_async(
|
|
requests.get, 'https://api.github.com/user', headers=headers
|
|
)
|
|
response.raise_for_status()
|
|
except requests.exceptions.RequestException as e:
|
|
raise HTTPException(
|
|
status_code=response.status_code if response else 500,
|
|
detail=f'Error fetching user: {str(e)}',
|
|
)
|
|
|
|
json_response = JSONResponse(content=response.json())
|
|
response.close()
|
|
|
|
return json_response
|
|
|
|
|
|
@app.get('/installations')
|
|
async def get_github_installation_ids(request: Request):
|
|
headers = generate_github_headers(request.state.github_token)
|
|
try:
|
|
response = await call_sync_from_async(
|
|
requests.get, 'https://api.github.com/user/installations', headers=headers
|
|
)
|
|
response.raise_for_status()
|
|
except requests.exceptions.RequestException as e:
|
|
raise HTTPException(
|
|
status_code=response.status_code if response else 500,
|
|
detail=f'Error fetching installations: {str(e)}',
|
|
)
|
|
|
|
data = response.json()
|
|
ids = [installation['id'] for installation in data['installations']]
|
|
json_response = JSONResponse(content=ids)
|
|
response.close()
|
|
|
|
return json_response
|
|
|
|
|
|
@app.get('/search/repositories')
|
|
async def search_github_repositories(
|
|
request: Request,
|
|
query: str,
|
|
per_page: int = 5,
|
|
sort: str = 'stars',
|
|
order: str = 'desc',
|
|
):
|
|
headers = generate_github_headers(request.state.github_token)
|
|
params = {
|
|
'q': query,
|
|
'per_page': per_page,
|
|
'sort': sort,
|
|
'order': order,
|
|
}
|
|
|
|
try:
|
|
response = await call_sync_from_async(
|
|
requests.get,
|
|
'https://api.github.com/search/repositories',
|
|
headers=headers,
|
|
params=params,
|
|
)
|
|
response.raise_for_status()
|
|
except requests.exceptions.RequestException as e:
|
|
raise HTTPException(
|
|
status_code=response.status_code if response else 500,
|
|
detail=f'Error searching repositories: {str(e)}',
|
|
)
|
|
|
|
json_response = JSONResponse(content=response.json())
|
|
response.close()
|
|
|
|
return json_response
|
|
|
|
|
|
def generate_github_headers(token: str) -> dict[str, str]:
|
|
return {
|
|
'Authorization': f'Bearer {token}',
|
|
'Accept': 'application/vnd.github.v3+json',
|
|
}
|