Update app.py

app.py

@@ -1,8 +1,8 @@
-from fastapi import FastAPI, BackgroundTasks, UploadFile, File, Form
-from fastapi.responses import HTMLResponse, JSONResponse, Response
+from fastapi import FastAPI, BackgroundTasks, UploadFile, File, Form, Request, Query
+from fastapi.responses import HTMLResponse, JSONResponse, Response, RedirectResponse
 from fastapi.staticfiles import StaticFiles
-import pathlib, os, uvicorn, base64, json, shutil
-from typing import Dict, List, Any
+import pathlib, os, uvicorn, base64, json, shutil, uuid, time, urllib.parse
+from typing import Dict, List, Any, Optional
 import asyncio
 import logging
 import threading
@@ -31,6 +31,12 @@ CACHE_DIR = BASE / "cache"
 if not CACHE_DIR.exists():
     CACHE_DIR.mkdir(parents=True)
 
+# PDF 메타데이터 디렉토리 및 파일 설정
+METADATA_DIR = pathlib.Path("/data/metadata") if os.path.exists("/data") else BASE / "metadata"
+if not METADATA_DIR.exists():
+    METADATA_DIR.mkdir(parents=True)
+PDF_METADATA_FILE = METADATA_DIR / "pdf_metadata.json"
+
 # 관리자 비밀번호
 ADMIN_PASSWORD = os.getenv("PASSWORD", "admin")  # 환경 변수에서 가져오기, 기본값은 테스트용
 
@@ -38,6 +44,42 @@ ADMIN_PASSWORD = os.getenv("PASSWORD", "admin")  # 환경 변수에서 가져오
 pdf_cache: Dict[str, Dict[str, Any]] = {}
 # 캐싱 락
 cache_locks = {}
+# PDF 메타데이터 (ID to 경로 매핑)
+pdf_metadata: Dict[str, str] = {}
+
+# PDF 메타데이터 로드
+def load_pdf_metadata():
+    global pdf_metadata
+    if PDF_METADATA_FILE.exists():
+        try:
+            with open(PDF_METADATA_FILE, "r") as f:
+                pdf_metadata = json.load(f)
+            logger.info(f"PDF 메타데이터 로드 완료: {len(pdf_metadata)} 항목")
+        except Exception as e:
+            logger.error(f"메타데이터 로드 오류: {e}")
+            pdf_metadata = {}
+    else:
+        pdf_metadata = {}
+
+# PDF 메타데이터 저장
+def save_pdf_metadata():
+    try:
+        with open(PDF_METADATA_FILE, "w") as f:
+            json.dump(pdf_metadata, f)
+    except Exception as e:
+        logger.error(f"메타데이터 저장 오류: {e}")
+
+# PDF ID 생성 (파일명 + 타임스탬프 기반)
+def generate_pdf_id(filename: str) -> str:
+    # 파일명에서 확장자 제거
+    base_name = os.path.splitext(filename)[0]
+    # URL 안전 문자열로 변환
+    safe_name = urllib.parse.quote(base_name, safe='')
+    # 타임스탬프 추가로 고유성 보장
+    timestamp = int(time.time())
+    # 짧은 임의 문자열 추가
+    random_suffix = uuid.uuid4().hex[:6]
+    return f"{safe_name}_{timestamp}_{random_suffix}"
 
 # PDF 파일 목록 가져오기 (메인 디렉토리용)
 def get_pdf_files():
@@ -53,7 +95,6 @@ def get_permanent_pdf_files():
         pdf_files = [f for f in PERMANENT_PDF_DIR.glob("*.pdf")]
     return pdf_files
 
-# PDF 썸네일 생성 및 프로젝트 데이터 준비
 # PDF 썸네일 생성 및 프로젝트 데이터 준비
 def generate_pdf_projects():
     projects_data = []
@@ -75,15 +116,28 @@ def generate_pdf_projects():
     
     # 중복 제거된 파일들로 프로젝트 데이터 생성
     for pdf_file in unique_files.values():
+        # 해당 파일의 PDF ID 찾기
+        pdf_id = None
+        for pid, path in pdf_metadata.items():
+            if os.path.basename(path) == pdf_file.name:
+                pdf_id = pid
+                break
+        
+        # ID가 없으면 새로 생성하고 메타데이터에 추가
+        if not pdf_id:
+            pdf_id = generate_pdf_id(pdf_file.name)
+            pdf_metadata[pdf_id] = str(pdf_file)
+            save_pdf_metadata()
+        
         projects_data.append({
             "path": str(pdf_file),
             "name": pdf_file.stem,
+            "id": pdf_id,
             "cached": pdf_file.stem in pdf_cache and pdf_cache[pdf_file.stem].get("status") == "completed"
         })
     
     return projects_data
 
-
 # 캐시 파일 경로 생성
 def get_cache_path(pdf_name: str):
     return CACHE_DIR / f"{pdf_name}_cache.json"
@@ -242,10 +296,40 @@ async def cache_pdf(pdf_path: str):
             pdf_cache[pdf_name]["status"] = "error"
             pdf_cache[pdf_name]["error"] = str(e)
 
+# PDF ID로 PDF 경로 찾기
+def get_pdf_path_by_id(pdf_id: str) -> str:
+    if pdf_id in pdf_metadata:
+        path = pdf_metadata[pdf_id]
+        # 파일 존재 확인
+        if os.path.exists(path):
+            return path
+        
+        # 영구 저장소에서 파일명으로 찾기
+        filename = os.path.basename(path)
+        perm_path = PERMANENT_PDF_DIR / filename
+        if perm_path.exists():
+            # 메타데이터 업데이트
+            pdf_metadata[pdf_id] = str(perm_path)
+            save_pdf_metadata()
+            return str(perm_path)
+            
+        # 메인 디렉토리에서 파일명으로 찾기
+        main_path = PDF_DIR / filename
+        if main_path.exists():
+            # 메타데이터 업데이트
+            pdf_metadata[pdf_id] = str(main_path)
+            save_pdf_metadata()
+            return str(main_path)
+    
+    return None
+
 # 시작 시 모든 PDF 파일 캐싱
 async def init_cache_all_pdfs():
     logger.info("PDF 캐싱 작업 시작")
     
+    # PDF 메타데이터 로드
+    load_pdf_metadata()
+    
     # 메인 및 영구 디렉토리에서 PDF 파일 모두 가져오기
     pdf_files = get_pdf_files() + get_permanent_pdf_files()
     
@@ -253,6 +337,25 @@ async def init_cache_all_pdfs():
     unique_pdf_paths = set(str(p) for p in pdf_files)
     pdf_files = [pathlib.Path(p) for p in unique_pdf_paths]
     
+    # 파일 기반 메타데이터 업데이트
+    for pdf_file in pdf_files:
+        # ID가 없는 파일에 대해 ID 생성
+        found = False
+        for pid, path in pdf_metadata.items():
+            if os.path.basename(path) == pdf_file.name:
+                found = True
+                # 경로 업데이트 필요한 경우
+                if not os.path.exists(path):
+                    pdf_metadata[pid] = str(pdf_file)
+                break
+        
+        if not found:
+            pdf_id = generate_pdf_id(pdf_file.name)
+            pdf_metadata[pdf_id] = str(pdf_file)
+    
+    # 메타데이터 저장
+    save_pdf_metadata()
+    
     # 이미 캐시된 PDF 파일 로드 (빠른 시작을 위해 먼저 수행)
     for cache_file in CACHE_DIR.glob("*_cache.json"):
         try:
@@ -290,14 +393,43 @@ async def get_permanent_pdf_projects_api():
     projects_data = []
     
     for pdf_file in pdf_files:
+        # PDF ID 찾기
+        pdf_id = None
+        for pid, path in pdf_metadata.items():
+            if os.path.basename(path) == pdf_file.name:
+                pdf_id = pid
+                break
+        
+        # ID가 없으면 생성
+        if not pdf_id:
+            pdf_id = generate_pdf_id(pdf_file.name)
+            pdf_metadata[pdf_id] = str(pdf_file)
+            save_pdf_metadata()
+            
         projects_data.append({
             "path": str(pdf_file),
             "name": pdf_file.stem,
+            "id": pdf_id,
             "cached": pdf_file.stem in pdf_cache and pdf_cache[pdf_file.stem].get("status") == "completed"
         })
     
     return projects_data
 
+# API 엔드포인트: PDF ID로 정보 가져오기
+@app.get("/api/pdf-info-by-id/{pdf_id}")
+async def get_pdf_info_by_id(pdf_id: str):
+    pdf_path = get_pdf_path_by_id(pdf_id)
+    if pdf_path:
+        pdf_file = pathlib.Path(pdf_path)
+        return {
+            "path": pdf_path,
+            "name": pdf_file.stem,
+            "id": pdf_id,
+            "exists": True,
+            "cached": pdf_file.stem in pdf_cache and pdf_cache[pdf_file.stem].get("status") == "completed"
+        }
+    return {"exists": False, "error": "PDF를 찾을 수 없습니다"}
+
 # API 엔드포인트: PDF 썸네일 제공 (최적화)
 @app.get("/api/pdf-thumbnail")
 async def get_pdf_thumbnail(path: str):
@@ -422,7 +554,6 @@ async def get_pdf_content(path: str, background_tasks: BackgroundTasks):
         logger.error(f"PDF 콘텐츠 로드 오류: {str(e)}\n{error_details}")
         return JSONResponse(content={"error": str(e)}, status_code=500)
 
-
 # PDF 업로드 엔드포인트 - 영구 저장소에 저장 및 메인 화면에 자동 표시
 @app.post("/api/upload-pdf")
 async def upload_pdf(file: UploadFile = File(...)):
@@ -446,11 +577,22 @@ async def upload_pdf(file: UploadFile = File(...)):
         with open(PDF_DIR / file.filename, "wb") as buffer:
             buffer.write(content)
         
+        # PDF ID 생성 및 메타데이터 저장
+        pdf_id = generate_pdf_id(file.filename)
+        pdf_metadata[pdf_id] = str(file_path)
+        save_pdf_metadata()
+        
         # 백그라운드에서 캐싱 시작
         asyncio.create_task(cache_pdf(str(file_path)))
         
         return JSONResponse(
-            content={"success": True, "path": str(file_path), "name": file_path.stem},
+            content={
+                "success": True, 
+                "path": str(file_path), 
+                "name": file_path.stem,
+                "id": pdf_id,
+                "viewUrl": f"/view/{pdf_id}"
+            },
             status_code=200
         )
     except Exception as e:
@@ -462,7 +604,6 @@ async def upload_pdf(file: UploadFile = File(...)):
             status_code=500
         )
 
-
 # 관리자 인증 엔드포인트
 @app.post("/api/admin-login")
 async def admin_login(password: str = Form(...)):
@@ -491,6 +632,17 @@ async def delete_pdf(path: str):
         if pdf_name in pdf_cache:
             del pdf_cache[pdf_name]
             
+        # 메타데이터에서 해당 파일 ID 제거
+        to_remove = []
+        for pid, fpath in pdf_metadata.items():
+            if os.path.basename(fpath) == pdf_file.name:
+                to_remove.append(pid)
+        
+        for pid in to_remove:
+            del pdf_metadata[pid]
+        
+        save_pdf_metadata()
+            
         return {"success": True}
     except Exception as e:
         logger.error(f"PDF 삭제 오류: {str(e)}")
@@ -528,13 +680,67 @@ async def unfeature_pdf(path: str):
         logger.error(f"PDF 표시 해제 오류: {str(e)}")
         return {"success": False, "message": str(e)}
 
+# 직접 PDF 뷰어 URL 접근용 라우트
+@app.get("/view/{pdf_id}")
+async def view_pdf_by_id(pdf_id: str):
+    # PDF ID 유효한지 확인
+    pdf_path = get_pdf_path_by_id(pdf_id)
+    if not pdf_path:
+        return HTMLResponse(
+            content=f"<html><body><h1>PDF를 찾을 수 없습니다</h1><p>ID: {pdf_id}</p><a href='/'>홈으로 돌아가기</a></body></html>",
+            status_code=404
+        )
+    
+    # 메인 페이지로 리다이렉트하되, PDF ID 파라미터 추가
+    return get_html_content(pdf_id=pdf_id)
+
 # HTML 파일 읽기 함수
-def get_html_content():
+def get_html_content(pdf_id: str = None):
     html_path = BASE / "flipbook_template.html"
+    content = ""
     if html_path.exists():
         with open(html_path, "r", encoding="utf-8") as f:
-            return f.read()
-    return HTML  # 기본 HTML 사용
+            content = f.read()
+    else:
+        content = HTML  # 기본 HTML 사용
+    
+    # PDF ID가 제공된 경우, 자동 로드 스크립트 추가
+    if pdf_id:
+        auto_load_script = f"""
+        <script>
+            // 페이지 로드 시 자동으로 해당 PDF 열기
+            document.addEventListener('DOMContentLoaded', async function() {{
+                try {{
+                    // PDF 정보 가져오기
+                    const response = await fetch('/api/pdf-info-by-id/{pdf_id}');
+                    const pdfInfo = await response.json();
+                    
+                    if (pdfInfo.exists && pdfInfo.path) {{
+                        // 약간의 지연 후 PDF 뷰어 열기 (UI가 준비된 후)
+                        setTimeout(() => {{
+                            openPdfById('{pdf_id}', pdfInfo.path, pdfInfo.cached);
+                        }}, 500);
+                    }} else {{
+                        showError("요청한 PDF를 찾을 수 없습니다.");
+                    }}
+                }} catch (e) {{
+                    console.error("자동 PDF 로드 오류:", e);
+                }}
+            }});
+        </script>
+        """
+        
+        # body 종료 태그 전에 스크립트 삽입
+        content = content.replace("</body>", auto_load_script + "</body>")
+    
+    return HTMLResponse(content=content)
+
+@app.get("/", response_class=HTMLResponse)
+async def root(request: Request, pdf_id: Optional[str] = Query(None)):
+    # PDF ID가 쿼리 파라미터로 제공된 경우 /view/{pdf_id}로 리다이렉트
+    if pdf_id:
+        return RedirectResponse(url=f"/view/{pdf_id}")
+    return get_html_content()
 
 # HTML 문자열 (UI 수정 버전)
 HTML = """
@@ -1298,6 +1504,9 @@ HTML = """
     // 현재 페이지 로딩 상태
     let currentLoadingPdfPath = null;
     let pageLoadingInterval = null;
+    
+    // 현재 열린 PDF의 ID
+    let currentPdfId = null;
 
     /* 🔊 오디오 unlock – 내장 Audio 와 같은 MP3 경로 사용 */
     ['click','touchstart'].forEach(evt=>{
@@ -1371,7 +1580,7 @@ HTML = """
           await loadServerPDFs();
           
           // 성공 메시지
-          showMessage("PDF가 성공적으로 업로드되었습니다!");
+          showMessage("PDF가 성공적으로 업로드되었습니다! 공유 URL: " + result.viewUrl);
         } else {
           hideLoading();
           showError("업로드 실패: " + (result.message || "알 수 없는 오류"));
@@ -1383,26 +1592,31 @@ HTML = """
       }
     }
     
-    function addCard(i, thumb, title, isCached = false) {
+    function addCard(i, thumb, title, isCached = false, pdfId = null) {
       const d = document.createElement('div');
       d.className = 'card fade-in';
       d.onclick = () => open(i);
       
+      // PDF ID가 있으면 데이터 속성으로 저장
+      if (pdfId) {
+          d.dataset.pdfId = pdfId;
+      }
+      
       // 제목 처리
       const displayTitle = title ? 
-        (title.length > 15 ? title.substring(0, 15) + '...' : title) : 
-        '프로젝트 ' + (i+1);
+          (title.length > 15 ? title.substring(0, 15) + '...' : title) : 
+          '프로젝트 ' + (i+1);
       
       // 캐시 상태 뱃지 추가
       const cachedBadge = isCached ? 
-        '<div class="cached-status">캐시됨</div>' : '';
+          '<div class="cached-status">캐시됨</div>' : '';
       
       d.innerHTML = `
-        <div class="card-inner">
-          ${cachedBadge}
-          <img src="${thumb}" alt="${displayTitle}" loading="lazy">
-          <p title="${title || '프로젝트 ' + (i+1)}">${displayTitle}</p>
-        </div>
+          <div class="card-inner">
+              ${cachedBadge}
+              <img src="${thumb}" alt="${displayTitle}" loading="lazy">
+              <p title="${title || '프로젝트 ' + (i+1)}">${displayTitle}</p>
+          </div>
       `;
       grid.appendChild(d);
       
@@ -1411,9 +1625,9 @@ HTML = """
     }
 
     /* ── 프로젝트 저장 ── */
-    function save(pages, title, isCached = false){
-      const id=projects.push(pages)-1;
-      addCard(id, pages[0].thumb, title, isCached);
+    function save(pages, title, isCached = false, pdfId = null) {
+      const id = projects.push(pages) - 1;
+      addCard(id, pages[0].thumb, title, isCached, pdfId);
     }
 
     /* ── 서버 PDF 로드 및 캐시 상태 확인 ── */
@@ -1465,7 +1679,8 @@ HTML = """
               return {
                 pages,
                 name: project.name,
-                isCached
+                isCached,
+                id: project.id
               };
             }
           } catch (err) {
@@ -1480,7 +1695,7 @@ HTML = """
         
         // 성공적으로 가져온 결과만 표시
         results.filter(result => result !== null).forEach(result => {
-          save(result.pages, result.name, result.isCached);
+          save(result.pages, result.name, result.isCached, result.id);
         });
         
         hideLoading();
@@ -1565,11 +1780,120 @@ HTML = """
       }
     }
 
+    /* ── PDF ID로 PDF 열기 ── */
+    async function openPdfById(pdfId, pdfPath, isCached = false) {
+      try {
+        // 먼저 홈 화면에서 카드를 찾아서 클릭하는 방법 시도
+        let foundCard = false;
+        const cards = document.querySelectorAll('.card');
+        
+        for (let i = 0; i < cards.length; i++) {
+          if (cards[i].dataset.pdfId === pdfId) {
+            cards[i].click();
+            foundCard = true;
+            break;
+          }
+        }
+        
+        // 카드를 찾지 못한 경우 직접 오픈
+        if (!foundCard) {
+          toggle(false);
+          showLoading("PDF 준비 중...");
+          
+          let pages = [];
+          
+          // 이미 캐시된 경우 캐시된 데이터 사용
+          if (isCached) {
+            try {
+              const response = await fetch(`/api/cached-pdf?path=${encodeURIComponent(pdfPath)}`);
+              const cachedData = await response.json();
+              
+              if (cachedData.status === "completed" && cachedData.pages) {
+                hideLoading();
+                createFlipBook(cachedData.pages);
+                // 현재 열린 PDF의 ID 저장
+                currentPdfId = pdfId;
+                return;
+              }
+            } catch (error) {
+              console.error("캐시 데이터 로드 실패:", error);
+            }
+          }
+          
+          // 썸네일 가져오기
+          try {
+            const thumbResponse = await fetch(`/api/pdf-thumbnail?path=${encodeURIComponent(pdfPath)}`);
+            const thumbData = await thumbResponse.json();
+            
+            if (thumbData.thumbnail) {
+              pages = [{
+                src: thumbData.thumbnail,
+                thumb: thumbData.thumbnail,
+                path: pdfPath,
+                cached: isCached
+              }];
+            }
+          } catch (error) {
+            console.error("썸네일 로드 실패:", error);
+          }
+          
+          // 일단 기본 페이지 추가
+          if (pages.length === 0) {
+            pages = [{
+              path: pdfPath,
+              cached: isCached
+            }];
+          }
+          
+          // 프로젝트에 추가하고 뷰어 실행
+          const projectId = projects.push(pages) - 1;
+          hideLoading();
+          open(projectId);
+          
+          // 현재 열린 PDF의 ID 저장
+          currentPdfId = pdfId;
+        }
+      } catch (error) {
+        console.error("PDF ID로 열기 실패:", error);
+        hideLoading();
+        showError("PDF를 열 수 없습니다. 다시 시도해주세요.");
+      }
+    }
+
+    /* ── 현재 PDF의 고유 URL 생성 및 복사 ── */
+    function copyPdfShareUrl() {
+      if (!currentPdfId) {
+        showError("공유할 PDF가 없습니다.");
+        return;
+      }
+      
+      // 현재 도메인 기반 전체 URL 생성
+      const shareUrl = `${window.location.origin}/view/${currentPdfId}`;
+      
+      // 클립보드에 복사
+      navigator.clipboard.writeText(shareUrl)
+        .then(() => {
+          showMessage("PDF 링크가 복사되었습니다!");
+        })
+        .catch(err => {
+          console.error("클립보드 복사 실패:", err);
+          showError("링크 복사에 실패했습니다.");
+        });
+    }
+
     /* ── 카드 → FlipBook ── */
     async function open(i) {
       toggle(false);
       const pages = projects[i];
       
+      // PDF ID 찾기 및 저장
+      const card = document.querySelectorAll('.card')[i];
+      if (card && card.dataset.pdfId) {
+        currentPdfId = card.dataset.pdfId;
+      } else {
+        currentPdfId = null;
+      }
+      
       // 기존 FlipBook 정리
       if(fb) {
         fb.destroy();
@@ -1833,7 +2157,7 @@ HTML = """
             enableDownload: false,
             enablePrint: false,
             enableZoom: true,
-            enableShare: false,
+            enableShare: true, // 공유 버튼 활성화
             enableSearch: true,
             enableAutoPlay: true,
             enableAnnotation: false,
@@ -1850,7 +2174,8 @@ HTML = """
             pageTextureSize: 1024, // 페이지 텍스처 크기
             thumbnails: true, // 섬네일 활성화
             autoHideControls: false, // 자동 숨김 비활성화
-            controlsTimeout: 8000 // 컨트롤 표시 시간 연장
+            controlsTimeout: 8000, // 컨트롤 표시 시간 연장
+            shareHandler: copyPdfShareUrl // 공유 핸들러 설정
           }
         });
         
@@ -1905,6 +2230,7 @@ HTML = """
       }
       $id('loadingPages').style.display = 'none';
       currentLoadingPdfPath = null;
+      currentPdfId = null;
     };
 
     function toggle(showHome){
@@ -2019,12 +2345,18 @@ HTML = """
               const card = document.createElement('div');
               card.className = 'admin-card card fade-in';
               
+              // 고유 URL 생성
+              const viewUrl = `${window.location.origin}/view/${pdf.id}`;
+              
               // 썸네일 및 정보
               card.innerHTML = `
                 <div class="card-inner">
                   ${pdf.cached ? '<div class="cached-status">캐시됨</div>' : ''}
                   <img src="${thumbData.thumbnail || ''}" alt="${pdf.name}" loading="lazy">
                   <p title="${pdf.name}">${pdf.name.length > 15 ? pdf.name.substring(0, 15) + '...' : pdf.name}</p>
+                  <div style="position: absolute; bottom: 130px; left: 50%; transform: translateX(-50%); z-index:10;">
+                    <a href="${viewUrl}" target="_blank" style="color:#4a6ee0; font-size:12px;">바로가기 링크</a>
+                  </div>
                   ${isMainDisplayed ? 
                     `<button class="unfeature-btn" data-path="${pdf.path}">메인에서 제거</button>` : 
                     `<button class="feature-btn" data-path="${pdf.path}">메인에 표시</button>`}
@@ -2279,9 +2611,5 @@ HTML = """
 </html>
 """
 
-@app.get("/", response_class=HTMLResponse)
-async def root():
-    return get_html_content()
-
 if __name__ == "__main__":
     uvicorn.run("app:app", host="0.0.0.0", port=int(os.getenv("PORT", 7860)))