import io import zipfile from datetime import datetime, timezone from urllib.parse import quote from fastapi import APIRouter, Depends, File, HTTPException, UploadFile from fastapi.responses import Response from pydantic import BaseModel from sqlalchemy.orm import Session as DBSession from app.core.notes import import_files, next_position, reorder, rename_section, set_section from app.db.database import get_db from app.db.models import NotePage router = APIRouter(prefix="/api/notes", tags=["notes"]) class NotePageResponse(BaseModel): id: int title: str content: str position: int section: str | None updated_at: datetime class Config: from_attributes = True class NotePageCreate(BaseModel): title: str = "新しいページ" content: str = "" section: str | None = None class NotePageUpdate(BaseModel): title: str | None = None content: str | None = None section: str | None = None class SectionRenameRequest(BaseModel): old_name: str new_name: str | None = None class ReorderRequest(BaseModel): page_id: int section: str | None = None order: list[int] # ファイル名に使えない文字（Windows の禁止文字。制御文字は _safe_filename 側で弾く） _UNSAFE_FILENAME_CHARS = '\\/:*?"<>|' def _safe_filename(name: str, fallback: str) -> str: """タイトルやセクション名をファイル名に使える形へ落とす。空になったら fallback を使う。""" cleaned = "".join("_" if c in _UNSAFE_FILENAME_CHARS or ord(c) < 32 else c for c in name) cleaned = cleaned.strip().strip(".") return cleaned[:80] or fallback def _attachment_headers(filename: str) -> dict[str, str]: """日本語のファイル名は Content-Disposition にそのまま置けないため RFC 5987 形式を併記する。""" ascii_fallback = filename.encode("ascii", "replace").decode("ascii").replace("?", "_") return { "Content-Disposition": ( f'attachment; filename="{ascii_fallback}"; ' f"filename*=UTF-8''{quote(filename, safe='')}" ) } @router.get("", response_model=list[NotePageResponse]) def list_pages(db: DBSession = Depends(get_db)): return db.query(NotePage).order_by(NotePage.position, NotePage.id).all() @router.get("/export") def export_all_pages(db: DBSession = Depends(get_db)): """全ページを 1ページ1ファイルの Markdown にして ZIP でダウンロードさせる。 本文はそのまま書き出す（1つのファイルに連結すると、ページ内の見出しの階層が ページ間の見出しと混ざってしまうため）。タブの並び順を残すためファイル名には 通し番号を付け、セクションはフォルダにする。 """ pages = db.query(NotePage).order_by(NotePage.position, NotePage.id).all() buffer = io.BytesIO() with zipfile.ZipFile(buffer, "w", zipfile.ZIP_DEFLATED) as archive: for i, page in enumerate(pages, start=1): entry = f"{i:02d}_{_safe_filename(page.title, f'page_{page.id}')}.md" if page.section: entry = f"{_safe_filename(page.section, 'section')}/{entry}" archive.writestr(entry, page.content) filename = f"notes_{datetime.now(timezone.utc).strftime('%Y%m%d')}.zip" return Response( content=buffer.getvalue(), media_type="application/zip", headers=_attachment_headers(filename), ) @router.get("/{page_id}/export") def export_page(page_id: int, db: DBSession = Depends(get_db)): """1ページを Markdown ファイルとしてダウンロードさせる。本文はそのまま、ファイル名がタイトルになる。""" page = db.get(NotePage, page_id) if not page: raise HTTPException(status_code=404, detail="Note page not found") filename = f"{_safe_filename(page.title, f'page_{page.id}')}.md" return Response( content=page.content, media_type="text/markdown; charset=utf-8", headers=_attachment_headers(filename), ) @router.post("", response_model=NotePageResponse, status_code=201) def create_page(req: NotePageCreate, db: DBSession = Depends(get_db)): title = req.title.strip() or "新しいページ" page = NotePage(title=title, content=req.content, position=next_position(db)) db.add(page) db.commit() db.refresh(page) if req.section is not None and req.section.strip(): page = set_section(db, page.id, req.section) return page @router.post("/import", response_model=list[NotePageResponse], status_code=201) async def import_pages(files: list[UploadFile] = File(...), db: DBSession = Depends(get_db)): """アップロードされたファイルをページとして取り込む。 .md / .txt はそのまま1ページ、.zip は中のテキストファイルを1つずつページにする （ZIP内のフォルダ名がセクションになるので、エクスポートしたZIPをそのまま戻せる）。 対象外の拡張子や大きすぎるファイルは黙って飛ばすので、1つも取り込めなければ 400。 """ payload = [(f.filename or "無題", await f.read()) for f in files] created = import_files(db, payload) if not created: raise HTTPException( status_code=400, detail="取り込めるファイルがありませんでした（.md / .txt / .zip のみ）", ) return created @router.post("/sections/rename") def rename_section_endpoint(req: SectionRenameRequest, db: DBSession = Depends(get_db)): """old_name セクションに属する全ページを new_name セクションへ一括で付け替える。new_name が空ならグループ解除。""" count = rename_section(db, req.old_name, req.new_name) return {"renamed": count} @router.post("/reorder") def reorder_pages(req: ReorderRequest, db: DBSession = Depends(get_db)): """タブのドラッグ&ドロップによる並べ替え。order は全ページIDを新しい順序で並べたもの。""" ok = reorder(db, req.page_id, req.section, req.order) if not ok: raise HTTPException(status_code=400, detail="order must contain exactly the current page ids") return {"reordered": len(req.order)} @router.patch("/{page_id}", response_model=NotePageResponse) def update_page(page_id: int, req: NotePageUpdate, db: DBSession = Depends(get_db)): page = db.get(NotePage, page_id) if not page: raise HTTPException(status_code=404, detail="Note page not found") if req.title is not None: title = req.title.strip() if not title: raise HTTPException(status_code=400, detail="title is empty") page.title = title if req.content is not None: page.content = req.content db.commit() db.refresh(page) # section はページの並び順にも影響するため、変更があれば専用ロジックで反映する if "section" in req.model_fields_set: page = set_section(db, page_id, req.section) return page @router.delete("/{page_id}", status_code=204) def delete_page(page_id: int, db: DBSession = Depends(get_db)): page = db.get(NotePage, page_id) if not page: raise HTTPException(status_code=404, detail="Note page not found") db.delete(page) db.commit()