135 lines
6.3 KiB
Python
135 lines
6.3 KiB
Python
import os, sys, json, subprocess, time as _time
|
|
from fastapi import APIRouter, HTTPException, UploadFile, File, Form
|
|
from pydantic import BaseModel
|
|
from typing import Optional
|
|
|
|
router = APIRouter()
|
|
|
|
# ── Native "Save As" dialog (fix bug: export MIDI/WAV không hỏi nơi lưu) ──
|
|
# WebView2 standalone chặn anchor-download → không thấy save dialog. Giải pháp:
|
|
# frontend gọi /save-dialog → engine mở NATIVE SaveFileDialog (Windows Explorer)
|
|
# → trả đường dẫn user chọn → frontend gửi bytes qua /write → ghi file thật.
|
|
SAVE_DIALOG_TIMEOUT = 120 # fallback native (PowerShell) — user có thể để dialog mở lâu
|
|
# Bridge timeout phải LỚN HƠN Rust wait (300s trong lib.rs) — nếu không, user
|
|
# chọn path sau 300s thì engine đã bỏ cuộc → save_file.response bị bỏ lại, file
|
|
# không được ghi dù user đã chọn nơi lưu (bug "hỏi nơi lưu nhưng không có file").
|
|
SAVE_BRIDGE_TIMEOUT = 360
|
|
|
|
|
|
def _save_file_via_tauri_bridge(default_name: str) -> Optional[str]:
|
|
"""Tauri shell (Rust watcher trong lib.rs) mở NATIVE SaveFileDialog qua
|
|
file IPC: engine ghi save_file.request (chứa tên mặc định) → Rust mở dialog
|
|
→ ghi save_file.response. Trả None nếu bridge không tồn tại (standalone) →
|
|
caller fallback native. Trả "" nếu bridge ĐÃ dùng nhưng user hủy / hết hạn →
|
|
KHÔNG fallback (tránh 2 dialog cùng lúc + tránh ghi đè path user đã chọn)."""
|
|
root = os.environ.get("APPDATA") or os.path.expanduser("~")
|
|
ipc = os.path.join(root, "SonicForgeDAW", "ipc")
|
|
if not os.path.isdir(ipc):
|
|
return None
|
|
# Bridge chi dung khi engine do Tauri spawn (co SF_PARENT_PID). Marker co
|
|
# the con sot tu lan Tauri chay truoc → chay standalone se cho response
|
|
# khong bao gio den (treo 120s). Check nay loai bo truong hop do.
|
|
if not os.environ.get("SF_PARENT_PID"):
|
|
return None
|
|
if not os.path.exists(os.path.join(ipc, "tauri_bridge_ready")):
|
|
return None
|
|
req = os.path.join(ipc, "save_file.request")
|
|
resp = os.path.join(ipc, "save_file.response")
|
|
try:
|
|
for f in (req, resp):
|
|
if os.path.exists(f):
|
|
os.remove(f)
|
|
with open(req, "w", encoding="utf-8") as fh:
|
|
fh.write(default_name or "untitled")
|
|
deadline = _time.time() + SAVE_BRIDGE_TIMEOUT
|
|
while _time.time() < deadline:
|
|
if os.path.exists(resp):
|
|
try:
|
|
with open(resp, "r", encoding="utf-8") as fh:
|
|
val = fh.read().strip()
|
|
finally:
|
|
os.remove(resp)
|
|
return val # "" khi user hủy — KHÔNG fallback PowerShell (tránh 2 dialog)
|
|
_time.sleep(0.1)
|
|
except Exception:
|
|
return ""
|
|
return "" # bridge đã dùng nhưng hết hạn — KHÔNG fallback (dialog Rust vẫn có thể đang mở)
|
|
|
|
|
|
def _save_file_native_engine(default_name: str, filter_name: str, filter_ext: str) -> Optional[str]:
|
|
"""Fallback khi không có Tauri bridge: PowerShell SaveFileDialog (Windows),
|
|
osascript (macOS), zenity/kdialog (Linux)."""
|
|
if os.name == "nt":
|
|
# FileName mặc định + Filter — single-quoted trong PS, filename không
|
|
# chứa nháy đơn (SonicForge tự sinh tên).
|
|
ps = (
|
|
"Add-Type -AssemblyName System.Windows.Forms; "
|
|
f"$f = New-Object System.Windows.Forms.SaveFileDialog; "
|
|
f"$f.FileName = '{default_name}'; "
|
|
f"$f.Filter = '{filter_name} (*.{filter_ext})|*.{filter_ext}|Tất cả các tệp (*.*)|*.*'; "
|
|
"$f.Title = 'Chọn nơi lưu tệp xuất'; "
|
|
"if ($f.ShowDialog() -eq [System.Windows.Forms.DialogResult]::OK) { Write-Output $f.FileName }"
|
|
)
|
|
try:
|
|
r = subprocess.run(
|
|
["powershell", "-NoProfile", "-STA", "-Command", ps],
|
|
capture_output=True, text=True, timeout=SAVE_DIALOG_TIMEOUT,
|
|
)
|
|
return r.stdout.strip() or None
|
|
except Exception:
|
|
return None
|
|
if sys.platform == "darwin":
|
|
try:
|
|
r = subprocess.run(
|
|
["osascript", "-e",
|
|
'set f to choose file name with prompt "Chọn nơi lưu tệp xuất" default name "' + (default_name or "untitled") + '"'],
|
|
capture_output=True, text=True, timeout=SAVE_DIALOG_TIMEOUT,
|
|
)
|
|
return r.stdout.strip() or None
|
|
except Exception:
|
|
return None
|
|
for cmd in (["zenity", "--file-selection", "--save", "--filename=" + (default_name or "untitled")],
|
|
["kdialog", "--getsavefilename", os.path.join(os.path.expanduser("~"), default_name or "untitled")]):
|
|
try:
|
|
r = subprocess.run(cmd, capture_output=True, text=True, timeout=SAVE_DIALOG_TIMEOUT)
|
|
if r.returncode == 0:
|
|
p = r.stdout.strip()
|
|
if p:
|
|
return p
|
|
except Exception:
|
|
continue
|
|
return None
|
|
|
|
|
|
class SaveDialogRequest(BaseModel):
|
|
default_name: str = "untitled"
|
|
filter_name: str = "File"
|
|
filter_ext: str = "bin"
|
|
|
|
|
|
@router.post("/save-dialog")
|
|
def save_file_dialog(req: SaveDialogRequest):
|
|
"""Mở NATIVE Save As dialog. Không cần auth (desktop local, chỉ mở dialog).
|
|
Trả {"path": "<đường dẫn user chọn>"} hoặc {"path": None} (hủy/không có
|
|
dialog). def (sync) → FastAPI chạy threadpool — không block loop."""
|
|
path = _save_file_via_tauri_bridge(req.default_name)
|
|
if path is None: # không có bridge (standalone thuần) → fallback native
|
|
path = _save_file_native_engine(req.default_name, req.filter_name, req.filter_ext)
|
|
return {"path": path or None}
|
|
|
|
|
|
@router.post("/write")
|
|
async def write_export_file(path: str = Form(...), file: UploadFile = File(...)):
|
|
"""Ghi bytes đã export vào đường dẫn tuyệt đối user chọn qua save dialog.
|
|
Trả {"path": ..., "bytes": N} — frontend toast kèm đường dẫn thật."""
|
|
try:
|
|
data = await file.read()
|
|
dirname = os.path.dirname(path)
|
|
if dirname:
|
|
os.makedirs(dirname, exist_ok=True)
|
|
with open(path, "wb") as f:
|
|
f.write(data)
|
|
return {"path": path, "bytes": len(data)}
|
|
except Exception as e:
|
|
raise HTTPException(status_code=400, detail=f"Không thể ghi file: {str(e)}")
|