app.py

# -*- coding: utf-8 -*-
import os
import re
import time
import json
import tempfile
from typing import Optional, Tuple, List, Any

import matplotlib.pyplot as plt
import gradio as gr

# --- 引入 Google API 相關套件 ---
from googleapiclient.discovery import build, Resource
from googleapiclient.http import MediaIoBaseUpload
from googleapiclient.errors import HttpError
from google.oauth2.service_account import Credentials

# ====== CONFIGURATION ======
DEFAULT_EXTENT = (119.0, 123.5, 21.5, 25.5)  # 預設地圖範圍 (台灣)
SCOPES = ("https://www.googleapis.com/auth/drive.file",)  # Google Drive API 權限範圍，僅限於本應用程式建立的檔案
DEFAULT_FOLDER_URL = "https://drive.google.com/drive/folders/1B7D4Z6vV_G30nIrBr8qFsgNA4l2G-dRo"  # 預設上傳的資料夾 (可以是你的共用雲端硬碟資料夾)
MIMETYPE_PNG = "image/png"
MIMETYPE_FOLDER = "application/vnd.google-apps.folder"

# ====== UTILITY FUNCTIONS ======
def clean_folder_id(folder_input: str) -> Optional[str]:
    """
    從 Google Drive 資料夾的 URL 或純 ID 中，提取出乾淨的 folder ID。
    能自動處理常見的複製貼上錯誤，如多餘的標點符號。
    """
    if not folder_input:
        return None
    raw = folder_input.strip()

    # 優先從 URL 格式中擷取 ID (例如 .../folders/<ID>)
    match = re.search(r"/folders/([A-Za-z0-9_-]+)", raw)
    if match:
        raw = match.group(1)

    # 移除 URL 的 query string (例如 ?usp=sharing)
    raw = raw.split("?")[0]
    # 移除頭尾的引號
    raw = raw.strip().strip('"\'')
    # 移除使用者可能不小心複製到的結尾標點符號
    raw = raw.rstrip(" .。；;，,！!？?、/\\』」）》）】]}")

    # 最終驗證 ID 格式是否合法 (Google Drive ID 通常是長度超過 20 的英數混合字串)
    return raw if re.fullmatch(r"[A-Za-z0-9_-]{20,}", raw) else None


def parse_points(csv_text: str) -> List[Tuple[float, float]]:
    """從文字輸入中解析出經緯度座標點列表。"""
    points: List[Tuple[float, float]] = []
    if not csv_text:
        return points
    for line in csv_text.strip().splitlines():
        line = line.strip()
        # 跳過空行或註解行
        if not line or line.startswith("#"):
            continue
        # 將 tab 轉換為逗號，並以逗號分割
        parts = [p.strip() for p in line.replace("\t", ",").split(",")]
        if len(parts) < 2:
            continue
        try:
            # 嘗試轉換為浮點數
            lon, lat = float(parts[0]), float(parts[1])
            points.append((lon, lat))
        except ValueError:
            # 轉換失敗則忽略此行
            continue
    return points


def make_figure(points: List[Tuple[float, float]], title: str,
                extent: Tuple[float, float, float, float] = DEFAULT_EXTENT,
                dpi: int = 150) -> str:
    """使用 Matplotlib 繪製地圖並存成暫存圖檔。"""
    min_lon, max_lon, min_lat, max_lat = extent
    fig, ax = plt.subplots(figsize=(6.5, 6.5), dpi=dpi)
    ax.set_title(title or "Map Plot", pad=10)
    ax.set_xlabel("Longitude"); ax.set_ylabel("Latitude")
    ax.set_xlim(min_lon, max_lon); ax.set_ylim(min_lat, max_lat)
    ax.grid(True, linestyle=":", linewidth=0.8)

    # 繪製座標點
    if points:
        xs = [p[0] for p in points]
        ys = [p[1] for p in points]
        ax.scatter(xs, ys, s=60, marker="*", label="Input points", zorder=10)
        ax.legend(loc="upper right")

    # 儲存圖片到系統暫存目錄
    out_path = os.path.join(tempfile.gettempdir(), f"map_{int(time.time())}.png")
    fig.tight_layout()
    fig.savefig(out_path)
    plt.close(fig)  # 關閉圖表以釋放記憶體，在 Web App 中很重要
    return out_path


def drive_service_from_secret() -> Tuple[Optional[Resource], Optional[str], Optional[str]]:
    """
    從環境變數讀取服務帳號 JSON，並建立 Google Drive API 服務物件。
    """
    raw_secret = os.environ.get("SERVICE_ACCOUNT_JSON", "").strip()
    if not raw_secret:
        return None, None, "在 Secrets 中找不到 `SERVICE_ACCOUNT_JSON`。"
    try:
        info = json.loads(raw_secret)
    except json.JSONDecodeError as e:
        return None, None, f"`SERVICE_ACCOUNT_JSON` 格式錯誤，不是有效的 JSON: {e}"

    client_email = info.get("client_email")
    if not client_email:
        return None, None, "`SERVICE_ACCOUNT_JSON` 中缺少 `client_email` 欄位。"

    try:
        creds = Credentials.from_service_account_info(info, scopes=SCOPES)
        # cache_discovery=False 在無伺服器或容器環境中是個好習慣，可避免快取問題
        service = build("drive", "v3", credentials=creds, cache_discovery=False)
        return service, client_email, None
    except Exception as e:
        return None, None, f"建立 Drive 服務時發生錯誤: {e}"


def upload_png_to_drive(local_png_path: str, service: Resource, folder_id: Optional[str], make_public: bool) -> Tuple[str, str]:
    """上傳本地圖片檔案到指定的 Google Drive 資料夾。"""
    file_metadata = {"name": os.path.basename(local_png_path)}
    if folder_id:
        file_metadata["parents"] = [folder_id]

    # 準備上傳的媒體內容
    with open(local_png_path, "rb") as f:
        media = MediaIoBaseUpload(f, mimetype=MIMETYPE_PNG)
        # 執行上傳請求
        # supportsAllDrives=True 是關鍵，讓服務帳號可以寫入共用雲端硬碟
        created_file = (
            service.files()
            .create(body=file_metadata, media_body=media,
                    fields="id, webViewLink",
                    supportsAllDrives=True)
            .execute()
        )

    file_id = created_file.get("id", "")
    web_view_link = created_file.get("webViewLink", "")

    # 如果使用者勾選了「公開」，則設定檔案權限
    if make_public and file_id:
        try:
            service.permissions().create(
                fileId=file_id,
                body={"role": "reader", "type": "anyone"},
                supportsAllDrives=True,
            ).execute()
        except HttpError as e:
            # 即使設定權限失敗，也不中斷流程，僅在連結後附加錯誤訊息
            web_view_link += f"\n(但設定公開連結失敗: {e})"

    return file_id, web_view_link


# ====== GRADIO HANDLER FUNCTIONS ======
def generate_and_upload(points_text, title, min_lon, max_lon, min_lat, max_lat,
                        folder_input, make_public):
    """Gradio 按鈕點擊後的主要處理函式。"""
    # 1. 產生本地圖片檔
    try:
        pts = parse_points(points_text)
        png_path = make_figure(pts, title, extent=(min_lon, max_lon, min_lat, max_lat))
    except Exception as e:
        return None, f"❌ 繪圖失敗: {e}"

    # 2. 建立 Google Drive 服務
    service, client_email, err_msg = drive_service_from_secret()
    if service is None:
        return png_path, (f"⚠️ **Google Drive 未設定或設定錯誤**\n"
                          f"請確認您已在 Hugging Face Space 的 Settings → Secrets 中新增了 `SERVICE_ACCOUNT_JSON`。\n"
                          f"詳細錯誤: {err_msg}")

    # 3. 決定要使用的 Folder ID (優先順序: UI 輸入 > 環境變數 > 程式碼預設值)
    env_folder = os.environ.get("DRIVE_FOLDER_ID", "").strip()
    raw_folder_input = folder_input or env_folder or DEFAULT_FOLDER_URL
    folder_id = clean_folder_id(raw_folder_input)

    if not folder_id:
        return png_path, (f"❌ **無效的 Google Drive 資料夾網址或 ID**\n"
                          f"請貼上整個資料夾網址，或是正確的 ID。\n"
                          f"收到的內容: `{raw_folder_input}`")

    # 4. 上傳檔案
    try:
        file_id, view_link = upload_png_to_drive(png_path, service, folder_id, make_public)
        msg = [
            "✅ **上傳成功！**",
            f"• **服務帳號**: `{client_email}`",
            f"• **目標資料夾 ID**: `{folder_id}`",
            f"• **檔案 ID**: `{file_id}`",
            f"• **檔案連結**: {view_link}"
        ]
        if not make_public:
            msg.append("\n> _提示: 未勾選公開，檔案將繼承資料夾的權限設定。_")
        return png_path, "\n".join(msg)
    except HttpError as e:
        return png_path, f"❌ **上傳失敗 (HttpError)**: {e}"
    except Exception as e:
        return png_path, f"❌ **上傳失敗 (未知錯誤)**: {e}"


def check_config():
    """檢查環境變數與 API 連線狀態。"""
    service, client_email, err_msg = drive_service_from_secret()
    if service is None:
        return f"❌ **服務建立失敗**: {err_msg}"

    env_folder = os.environ.get("DRIVE_FOLDER_ID", "").strip()
    folder_id = clean_folder_id(env_folder or DEFAULT_FOLDER_URL)

    output = [
        f"✅ **服務帳號已載入**: `{client_email}`",
        f"➡️ **將使用的資料夾 ID**: `{folder_id}` (來自: {'Secret' if env_folder else '預設值'})"
    ]

    if folder_id:
        try:
            # 嘗試取得資料夾名稱來驗證 ID 是否有效以及權限是否足夠
            meta = service.files().get(
                fileId=folder_id, fields="id, name", supportsAllDrives=True
            ).execute()
            output.append(f"✅ **目標資料夾名稱**: **{meta.get('name')}**")
        except HttpError as e:
            output.append(f"⚠️ **無法讀取資料夾資訊**: {e}\n> _請檢查服務帳號是否有此資料夾的「檢視者」或更高權限。_")
    return "\n".join(output)


def list_folder(folder_input):
    """列出指定資料夾內的檔案。"""
    service, client_email, err_msg = drive_service_from_secret()
    if service is None:
        return f"❌ **服務建立失敗**: {err_msg}"

    env_folder = os.environ.get("DRIVE_FOLDER_ID", "").strip()
    raw_folder_input = folder_input or env_folder or DEFAULT_FOLDER_URL
    folder_id = clean_folder_id(raw_folder_input)
    if not folder_id:
        return "❌ **無效的 Google Drive 資料夾網址或 ID。**"

    try:
        # 驗證目標是否為資料夾
        meta = service.files().get(
            fileId=folder_id, fields="name, mimeType", supportsAllDrives=True
        ).execute()
        if meta.get("mimeType") != MIMETYPE_FOLDER:
            return f"⚠️ **目標不是資料夾**: `{meta.get('name')}` (類型: {meta.get('mimeType')})"
    except HttpError as e:
        return f"❌ **取得資料夾資訊失敗**: {e}\n> _請檢查 ID 是否正確，以及服務帳號是否有權限存取。_"

    try:
        # 列出資料夾內容
        results = service.files().list(
            q=f"'{folder_id}' in parents and trashed=false",
            fields="files(id, name, mimeType, webViewLink)",
            pageSize=50,
            supportsAllDrives=True,
            corpora="allDrives" # corpora='allDrives' 可同時搜尋個人、共用雲端硬碟
        ).execute()
        files = results.get("files", [])
    except HttpError as e:
        return f"❌ **列出檔案失敗**: {e}"

    lines = [
        f"✅ **服務帳號**: `{client_email}`",
        f"✅ **資料夾**: **{meta.get('name')}** (`{folder_id}`)",
        "\n--- 內容 (最多 50 筆) ---"
    ]
    if not files:
        lines.append("（此資料夾為空，或服務帳號無權限看見任何檔案）")
    else:
        for f in files:
            lines.append(f"- **{f['name']}** (`{f['mimeType']}`) - [開啟連結]({f['webViewLink']})")
    return "\n".join(lines)

# ====== GRADIO UI DEFINITION ======
with gr.Blocks(title="地圖產生器 → Google Drive", theme=gr.themes.Soft()) as demo:
    gr.Markdown(f"""
    # 🗺️ 地圖產生器 → Google Drive 上傳工具
    這個工具會根據輸入的經緯度座標點產生一張地圖圖片，並將其上傳到指定的 Google Drive 資料夾。
    - **預設上傳資料夾**: `{DEFAULT_FOLDER_URL}`
    - 可在下方欄位輸入新的資料夾 URL/ID 來覆蓋預設值。
    - 也可在 Space 的 **Settings → Secrets** 中設定 `DRIVE_FOLDER_ID` 來改變預設值。
    """)

    with gr.Row():
        with gr.Column(scale=2):
            points_in = gr.Textbox(
                label="座標點 (每行一組 經度,緯度)",
                value="121.5,25.0\n120.9,24.5\n121.0,23.9\n# 這是註解，會被忽略",
                lines=8,
            )
            title_in = gr.Textbox(label="地圖標題", value="台灣座標點範例")

            with gr.Accordion("進階設定", open=False):
                folder_in = gr.Textbox(
                    label="（可選）覆蓋預設資料夾",
                    placeholder="貼上新的 Google Drive 資料夾 URL 或 純 ID",
                )
                make_public_in = gr.Checkbox(
                    label="建立公開分享連結 (權限: 擁有連結的任何人皆可檢視)", value=False
                )
                with gr.Row():
                    min_lon_in = gr.Number(label="最小經度", value=DEFAULT_EXTENT[0])
                    max_lon_in = gr.Number(label="最大經度", value=DEFAULT_EXTENT[1])
                    min_lat_in = gr.Number(label="最小緯度", value=DEFAULT_EXTENT[2])
                    max_lat_in = gr.Number(label="最大緯度", value=DEFAULT_EXTENT[3])

            with gr.Row():
                run_btn = gr.Button("🚀 產生圖片並上傳", variant="primary")
                list_btn = gr.Button("📄 列出資料夾內容")
                cfg_btn = gr.Button("🧪 檢查設定")

        with gr.Column(scale=3):
            out_image = gr.Image(label="預覽圖", type="filepath")
            out_msg = gr.Markdown(label="執行結果")
            with gr.Accordion("除錯與列表資訊", open=False):
                cfg_out = gr.Markdown(label="設定檢查結果")
                list_out = gr.Markdown(label="資料夾內容列表")

    # --- 按鈕事件綁定 ---
    run_btn.click(
        fn=generate_and_upload,
        inputs=[points_in, title_in, min_lon_in, max_lon_in, min_lat_in, max_lat_in,
                folder_in, make_public_in],
        outputs=[out_image, out_msg],
    )
    cfg_btn.click(fn=check_config, inputs=None, outputs=cfg_out)
    list_btn.click(fn=list_folder, inputs=folder_in, outputs=list_out)

if __name__ == "__main__":
    demo.launch()