from flask import Flask, request, render_template, jsonify, send_from_directory
from pathlib import Path
import io, shutil, zipfile
from werkzeug.exceptions import HTTPException

app = Flask(__name__)

# 전역 초기화
UPLOAD_DIR = Path("./uploads").resolve() # ./uploads를 절대경로화 -> /app/uploads
UPLOAD_DIR.mkdir(parents=True, exist_ok=True)
count = 1 # 파일명 카운터

# GET / : 업로드 페이지
@app.route("/", methods=["GET"])
def main():   
    return render_template('upload.html')

# POST /upload : 업로드 처리
@app.route("/upload", methods=["POST"])
def upload():
    global count
    # multipart 폼 데이터에서 file 필드가 없으면 400 JSON 반환.
    if "file" not in request.files:
        return jsonify(ok=False, error="no file"), 400

    # multipart 폼 데이터에서 파일명이 비어있으면 400 JSON 반환.
    f = request.files["file"]
    if f.filename == "":
        return jsonify(ok=False, error="empty filename"), 400

    # ZIP으로 판단되면 “압축 해제 저장”
    # 확장자가 .zip이거나 Content-Type 문자열에 zip이 포함되면 ZIP으로 취급
    if (f.filename or "").lower().endswith(".zip") or "zip" in (f.content_type or "").lower():
        data = f.read() # 파일을 통째로 메모리에 읽음
        zf = zipfile.ZipFile(io.BytesIO(data)) # zipfile.ZipFile로 파싱
        names = []
        # infolist()로 ZIP 내부 엔트리를 순회
        for info in zf.infolist():
            target = UPLOAD_DIR / info.filename
            # 디렉터리면 mkdir
            if info.is_dir():
                target.mkdir(parents=True, exist_ok=True)
            # 파일이면 zf.open(info)로 꺼내서 target에 복사 저장
            else:
                target.parent.mkdir(parents=True, exist_ok=True)
                with zf.open(info) as src, open(target, "wb") as dst:
                    shutil.copyfileobj(src, dst)
            # 저장한 엔트리 이름 목록을 saved로 JSON 응답
            names.append(info.filename)
        return jsonify(ok=True, saved=names)
    
    # ZIP이 아니면 “단일 파일 저장”
    ext = Path(f.filename).suffix.lower()
    # 원래 파일명은 버리고, 카운터 기반 파일명으로 저장, 확장자만 유지
    file_name = f"{count}{ext}" if ext else f"{count}"
    count += 1
    out = UPLOAD_DIR / file_name
    out.parent.mkdir(parents=True, exist_ok=True)
    f.save(out)
    return jsonify(ok=True, saved=str(file_name))    

# GET /test.html : 템플릿 렌더링
@app.route("/test.html", methods=["GET"])
def test_page():
    return render_template("test.html")

# GET /uploads/<path:filename> : 업로드 파일 서빙
"""
1. 
URL: /uploads/<path:filename>
<path:filename> -> filename에 슬래시 포함 허용, 폴더까지 포함해서 파일 요청 가능

2.
send_from_directory(directory, path ...)
directory와 path를 결합해서 실제 파일 경로 만들기
그 경로의 파일을 찾아서 존재하면 파일 내용과 적절한 헤더를 붙여 응답
... 같은 경로로 directory 밖으로 빠지는 시도는 일반적으로 차단

3.
as_attachment=False 브라우저가 파일을 인라인으로 렌더링
"""
@app.route("/uploads/<path:filename>")
def serve_uploads(filename):
    return send_from_directory(UPLOAD_DIR, filename, as_attachment=False)
    
app.config.update(PROPAGATE_EXCEPTIONS=False)

# 모든 에러 400 처리
@app.errorhandler(Exception)
def _all_errors_to_400(e):
    return jsonify(ok=False, error="bad request"), 400

@app.errorhandler(HTTPException)
def _http_errors_to_400(e):
    return jsonify(ok=False, error="bad request"), 400

# flask 실행
if __name__ == "__main__":
    app.run(host="0.0.0.0", port=5000, debug=True)

target = UPLOAD_DIR / info.filename