main.py

from fastapi import FastAPI, Request
from fastapi.responses import JSONResponse, FileResponse
from pydantic import BaseModel
import httpx
import tempfile
from prometheus_fastapi_instrumentator import Instrumentator

from src.READMECreater.GithubFetcher import DownloadRepoFiles
from src.READMECreater.READMEGenerator import GenerateREADME
from src.TagCreater.Models import ExtractJson, ModelThreading
from src.Upload.Upload2DB import (
    GetVersion,
    SaveGitData,
    GetNextReadmeID,
    SavingCareerDB,
)
from src.Upload.Upload2Storage import UploadREADME
from src.Utils.GetJWT import GetDataFromToken, GetTokenFromHeader
from src.Utils.GetImage import GetImageInGithub
from src.Utils.DBClient import ReadGithubFromUserID, ReadREADMEDB, ReadImageFromUserID

app = FastAPI(title="GitHub AI API")
instrumentator = Instrumentator().instrument(app).expose(app)


class RepoRequest(BaseModel):
    git_url: str


# README 생성 API
@app.post("/api/github-service/readme")
async def GenerateReadme(request: RepoRequest, fast_api: Request):
    try:
        # README 쓰기
        repoURL = request.git_url
        repoFiles = DownloadRepoFiles(repoURL)
        readmeContent = GenerateREADME(repoURL, repoFiles)
        metaData = readmeContent[:1000]

        accessToken = GetTokenFromHeader(fast_api)
        userID = GetDataFromToken(accessToken, "user_id")

        version = GetVersion(repoURL)
        readmeID = GetNextReadmeID()

        # README Storage & DB에 저장
        downloadURL = UploadREADME(readmeContent, userID, repoURL, version)
        SaveGitData(version, repoURL, readmeID, userID, downloadURL, metaData)

        return {
            "status": 200,
            "message": "요청에 성공하였습니다.",
            "data": downloadURL,  # 스토리지 내에 데이터를 user에게 streaming 하는 API Call
        }

    except Exception:
        return JSONResponse(
            status_code=400,
            content={"status": 400, "message": "README 생성 실패", "data": None},
        )


# README 다운로드 요청 처리 API
@app.get("/api/github-service/download")
async def DownloadFile(downloadURL: str):
    async with httpx.AsyncClient() as client:
        try:
            resp = await client.get(downloadURL)
            resp.raise_for_status()

            # 다운로드를 위한 임시 파일 생성
            with tempfile.NamedTemporaryFile(delete=False, suffix=".md") as tmp:
                tmp.write(resp.content)
                tmpPath = tmp.name

            return FileResponse(
                path=tmpPath,
                filename="README.md",
                media_type="application/octet-stream",
                headers={"Content-Disposition": 'attachment; filename="README.md"'},
            )

        except Exception:
            return JSONResponse(
                status_code=400,
                content={
                    "status": 400,
                    "message": "README 다운로드 실패",
                    "data": None,
                },
            )


# Tag 생성 API
@app.post("/api/github-service/tag")
async def GenerateTag(request: RepoRequest, fast_api: Request):
    try:
        githubURL = request.git_url
        response = ModelThreading(githubURL)
        tags = ExtractJson(response.text)

        accessToken = GetTokenFromHeader(fast_api)
        userID = GetDataFromToken(accessToken, "user_id")

        imageURL = GetImageInGithub(githubURL)
        SavingCareerDB(tags, userID, githubURL, imageURL)

        return {
            "status": 200,
            "message": "요청에 성공하였습니다.",
            "data": tags,
        }

    except Exception as e:
        print("[ERROR] Exception in GenerateTag:", e)
        import traceback

        traceback.print_exc()
        return JSONResponse(
            status_code=400,
            content={"status": 400, "message": f"Tag 생성 실패: {e}", "data": None},
        )


# 유저가 올린 모든 github 정보 읽기
@app.get("/api/github-service/db/user")
async def ReadUserGithub(request: Request):
    try:
        accessToken = GetTokenFromHeader(request)
        userID = GetDataFromToken(accessToken, "user_id")

        data = ReadGithubFromUserID(userID)

        return {
            "status": 200,
            "message": "요청에 성공하였습니다.",
            "data": data,
        }
    except Exception:
        return JSONResponse(
            status_code=400,
            content={"status": 400, "message": "DB read 실패", "data": None},
        )


# 유저가 올린 github url의 README 데이터 읽기
@app.get("/api/github-service/db/readme")
async def ReadREADME(userID: int, gitURL: str):
    try:
        # README 데이터 읽기
        data = ReadREADMEDB(userID, gitURL)

        if not data:
            return JSONResponse(
                status_code=404,
                content={
                    "status": 404,
                    "message": "해당 README 데이터를 찾을 수 없습니다.",
                    "data": None,
                },
            )

        firstData = data[0]

        # 이미지 URL 가져오기
        imgResult = ReadImageFromUserID(userID, gitURL)

        imgURL = None
        if isinstance(imgResult, dict) and "result" in imgResult:
            imgURL = imgResult["result"].get("img_url")
        elif isinstance(imgResult, str):  # img_url이 직접 반환될 수도 있음
            imgURL = imgResult

        print(f"Image URL: {imgURL}")

        # 결과 데이터에 img_url 추가
        firstData["img_url"] = imgURL if imgURL else ""

        return {
            "status": 200,
            "message": "요청에 성공하였습니다.",
            "data": firstData,
        }

    except Exception as e:
        print(f"Error: {e}")
        return JSONResponse(
            status_code=400,
            content={"status": 400, "message": "DB read 실패", "data": None},
        )


# 헬스 체크
@app.get("/api/github-service/health-check")
async def HealthCheck():
    return {"status": 200, "message": "서버 상태 확인", "data": "Working"}