music(chord): 拆两个 tab + 抓两种 (letters/functional)
deploy music / build-and-deploy (push) Successful in 1m54s

- yopu 切 /song?title=&artist= 搜索(避免歌手词被搜糊)
- 抓的版本按搜索结果 nier-snippet svg <text> 数区分:
  >0 = 字母谱 (G/Em7/C 弹唱谱);==0 = 功能谱 (1/4/5/6m 数字级数)
- sidecar fetch/status/state/image 都走 (id, mode) 维度,文件落 /data/chord-fetch/{id}-{mode}.png
- backend chord_fetch / chord_status 接 ?mode=letters|functional,import 时 role 分别为 chord_letters / chord_functional
- 前端 chord tab 拆「吉他谱」+「功能谱」,state/error/poll 各自独立;旧 role='chord' 显示在「吉他谱」兼容历史 import
- verified 标记探测:匿名访问 yopu HTML 里 0 hits(要登录可见),暂时只能按 svg_text 区分
This commit is contained in:
Fam Zheng
2026-05-10 15:10:03 +01:00
parent f836c8dab7
commit fd80116168
5 changed files with 291 additions and 160 deletions
+50 -37
View File
@@ -36,36 +36,38 @@ OUT_DIR.mkdir(parents=True, exist_ok=True)
app = FastAPI()
VALID_MODES = ('letters', 'functional')
# in-memory job state. piece_id -> {status, error, query}
state: dict[int, dict] = {}
# in-memory job state. (piece_id, mode) -> {status, error, title, artist}
state: dict[tuple[int, str], dict] = {}
state_lock = threading.Lock()
job_q: queue.Queue = queue.Queue()
def out_path(piece_id: int) -> Path:
return OUT_DIR / f"{piece_id}.png"
def out_path(piece_id: int, mode: str) -> Path:
return OUT_DIR / f"{piece_id}-{mode}.png"
def worker():
while True:
piece_id, query = job_q.get()
piece_id, mode, title, artist = job_q.get()
key = (piece_id, mode)
with state_lock:
state[piece_id] = {'status': 'processing', 'error': '', 'query': query}
logger.info("[piece=%d] start fetch query=%r", piece_id, query)
state[key] = {'status': 'processing', 'error': '', 'title': title, 'artist': artist}
logger.info("[piece=%d mode=%s] start: title=%r artist=%r", piece_id, mode, title, artist)
try:
ok, msg = yopu.fetch_chord_chart(query, str(out_path(piece_id)))
ok, msg = yopu.fetch_chord_chart(title, artist, str(out_path(piece_id, mode)), mode=mode)
with state_lock:
if ok:
state[piece_id] = {'status': 'completed', 'error': '', 'query': query}
logger.info("[piece=%d] completed: %s", piece_id, msg)
state[key] = {'status': 'completed', 'error': '', 'title': title, 'artist': artist}
logger.info("[piece=%d mode=%s] completed: %s", piece_id, mode, msg)
else:
state[piece_id] = {'status': 'failed', 'error': msg, 'query': query}
logger.warning("[piece=%d] failed: %s", piece_id, msg)
state[key] = {'status': 'failed', 'error': msg, 'title': title, 'artist': artist}
logger.warning("[piece=%d mode=%s] failed: %s", piece_id, mode, msg)
except Exception as e:
logger.exception("[piece=%d] worker crash", piece_id)
logger.exception("[piece=%d mode=%s] worker crash", piece_id, mode)
with state_lock:
state[piece_id] = {'status': 'failed', 'error': str(e), 'query': query}
state[key] = {'status': 'failed', 'error': str(e), 'title': title, 'artist': artist}
finally:
job_q.task_done()
@@ -79,58 +81,69 @@ def healthz():
@app.post('/fetch')
def fetch(piece_id: int, query: str):
"""加入 fetch 队列。query 一般是 '<artist> <title>'
def fetch(piece_id: int, title: str, artist: str = '', mode: str = 'functional'):
"""加入 fetch 队列。
mode='letters' = 弹唱谱字母版;mode='functional' = 数字级数版。
幂等:已 completed 且文件还在,直接返回 completed。"""
if piece_id <= 0 or not query.strip():
raise HTTPException(400, 'piece_id / query required')
if piece_id <= 0 or not title.strip():
raise HTTPException(400, 'piece_id / title required')
if mode not in VALID_MODES:
raise HTTPException(400, f'mode must be one of {VALID_MODES}')
key = (piece_id, mode)
with state_lock:
cur = state.get(piece_id, {})
if cur.get('status') == 'completed' and out_path(piece_id).exists():
cur = state.get(key, {})
if cur.get('status') == 'completed' and out_path(piece_id, mode).exists():
return {'status': 'completed'}
if cur.get('status') in ('pending', 'processing'):
return {'status': cur['status']}
state[piece_id] = {'status': 'pending', 'error': '', 'query': query}
state[key] = {'status': 'pending', 'error': '', 'title': title, 'artist': artist}
job_q.put((piece_id, query))
job_q.put((piece_id, mode, title.strip(), artist.strip()))
return {'status': 'pending'}
@app.get('/status/{piece_id}')
def status(piece_id: int):
@app.get('/status/{piece_id}/{mode}')
def status(piece_id: int, mode: str):
if mode not in VALID_MODES:
raise HTTPException(400, f'mode must be one of {VALID_MODES}')
key = (piece_id, mode)
with state_lock:
cur = state.get(piece_id, {})
file_exists = out_path(piece_id).exists()
cur = state.get(key, {})
file_exists = out_path(piece_id, mode).exists()
if cur.get('status') == 'completed' and not file_exists:
return {'status': 'failed', 'error': 'png 文件丢了'}
return {'status': 'failed', 'error': 'png 文件丢了', 'mode': mode}
if not cur and file_exists:
return {'status': 'completed'}
return {'status': 'completed', 'mode': mode, 'file_exists': True}
return {
'status': cur.get('status', 'none'),
'error': cur.get('error', ''),
'query': cur.get('query', ''),
'mode': mode,
'file_exists': file_exists,
}
@app.get('/image/{piece_id}')
def image(piece_id: int):
p = out_path(piece_id)
@app.get('/image/{piece_id}/{mode}')
def image(piece_id: int, mode: str):
if mode not in VALID_MODES:
raise HTTPException(400, f'mode must be one of {VALID_MODES}')
p = out_path(piece_id, mode)
if not p.exists():
raise HTTPException(404, 'not found')
return FileResponse(p, media_type='image/png')
@app.delete('/state/{piece_id}')
def reset(piece_id: int):
@app.delete('/state/{piece_id}/{mode}')
def reset(piece_id: int, mode: str):
"""music backend import 完后清状态 + 删 png(防 PVC 越积越多)。"""
if mode not in VALID_MODES:
raise HTTPException(400, f'mode must be one of {VALID_MODES}')
with state_lock:
state.pop(piece_id, None)
p = out_path(piece_id)
state.pop((piece_id, mode), None)
p = out_path(piece_id, mode)
if p.exists():
try:
p.unlink()
except Exception as e:
logger.warning("[piece=%d] cleanup unlink: %s", piece_id, e)
logger.warning("[piece=%d mode=%s] cleanup unlink: %s", piece_id, mode, e)
return {'ok': True}
+76 -28
View File
@@ -54,19 +54,58 @@ def setup_driver(window="1920,5000"):
return webdriver.Chrome(service=service, options=o)
def find_first_chord_chart(driver, search_url):
"""搜索页找最佳的「功能谱」结果
def find_chart(driver, title: str, artist: str, prefer: str = 'functional'):
""" /song?title=&artist= 找最佳候选 view
yopu 现在搜索结果里同一首歌有多个版本
- 字母谱(chord chart):nier-snippet 里有 SVG <text> 渲染的 chord 字母(G/Em7/C
- 功能谱(数字 / 级数):nier-snippet 里没 SVG <text>(用 HTML/CSS 显示数字 1/4/5
yopu 同一首歌一般有多个版本,按搜索结果里 nier-snippet 内的
SVG <text> 数量区分:
- svg_text > 0 → chord 字母版(G/Em7/C),民间叫弹唱谱
- svg_text == 0 → 功能谱 / 数字级数版
我们优先取第一个**功能谱**svgTextCount === 0),fallback 到第一个字母谱
`prefer` ∈ {'letters', 'functional'},按需求挑第一个匹配的
实在没匹配就 fallback 到第一个非空候选。
"""
logger.info("loading search: %s", search_url)
from urllib.parse import urlencode
base = 'https://yopu.co/song'
# /song 用 hash 传参(跟 yopu 前端约定一致)
search_url = f"{base}#title={quote(title)}&artist={quote(artist)}"
logger.info("loading /song: %s", search_url)
driver.get(search_url)
time.sleep(3)
hits = driver.execute_script("""
var out = [];
var posts = document.querySelectorAll('a.post-main');
for (var i = 0; i < posts.length; i++) {
var p = posts[i];
var titleEl = p.querySelector('.title-line .title, .title');
var subEl = p.querySelector('.title-line .subtitle, .subtitle');
var info = p.querySelector('.one-line-info');
var snippet = p.querySelector('.nier-snippet');
var svgTextCount = snippet ? snippet.querySelectorAll('svg text').length : 0;
// 任何子元素 class 含 'verified' 都算(svelte 加了 hash class
var isVerified = p.querySelectorAll('[class*="verified"]').length > 0;
out.push({
href: p.href,
title: titleEl ? (titleEl.textContent || '').trim() : '',
subtitle: subEl ? (subEl.textContent || '').trim() : '',
info: info ? (info.textContent || '').trim() : '',
svgTextCount: svgTextCount,
isLetters: svgTextCount > 0,
isFunctional: svgTextCount === 0,
isVerified: isVerified,
});
}
return out;
""")
if not hits:
logger.warning("no a.post-main found at /song — fallback to /explore")
# fallback: yopu /song 偶尔没结果,回退到 /explore
from urllib.parse import quote as _q
q = (artist + ' ' + title).strip()
driver.get(f"https://yopu.co/explore#q={_q(q)}")
time.sleep(3)
hits = driver.execute_script("""
var out = [];
var posts = document.querySelectorAll('a.post-main');
@@ -82,35 +121,44 @@ def find_first_chord_chart(driver, search_url):
title: titleEl ? (titleEl.textContent || '').trim() : '',
subtitle: subEl ? (subEl.textContent || '').trim() : '',
info: info ? (info.textContent || '').trim() : '',
isFunctional: svgTextCount === 0,
svgTextCount: svgTextCount,
isLetters: svgTextCount > 0,
isFunctional: svgTextCount === 0,
isVerified: false,
});
}
return out;
""")
if not hits:
logger.warning("no a.post-main found — yopu DOM changed?")
return None
# 优先功能谱
functional = [h for h in hits if h['isFunctional']]
if functional:
chosen = functional[0]
kind = 'functional'
else:
chosen = hits[0]
kind = 'letter-chord (no functional version found)'
# 优先匹配 prefer;同时优先 verified(虽然匿名访问大概率全是 false)
def _key(h):
match_pref = (prefer == 'letters' and h['isLetters']) or \
(prefer == 'functional' and h['isFunctional'])
# 数值越小越优先:first match_pref+verified, then match_pref, then verified, then all
return (0 if (match_pref and h['isVerified']) else
1 if match_pref else
2 if h['isVerified'] else 3)
sorted_hits = sorted(hits, key=_key)
chosen = sorted_hits[0]
matched = (prefer == 'letters' and chosen['isLetters']) or \
(prefer == 'functional' and chosen['isFunctional'])
kind = prefer if matched else f"{prefer}-fallback"
href = chosen['href']
if href.startswith('/'):
p = urlparse(search_url)
p = urlparse(driver.current_url)
href = f"{p.scheme}://{p.netloc}{href}"
elif not href.startswith('http'):
href = urljoin(search_url, href)
logger.info("[%s] %s%s [%s] (%d total: %d functional, %d letter)",
href = urljoin(driver.current_url, href)
logger.info("[%s] %s%s [%s] verified=%s (total %d, letters=%d, functional=%d, verified=%d)",
kind, chosen['title'], chosen['subtitle'], chosen['info'],
len(hits), len(functional), len(hits) - len(functional))
chosen['isVerified'], len(hits),
sum(1 for h in hits if h['isLetters']),
sum(1 for h in hits if h['isFunctional']),
sum(1 for h in hits if h['isVerified']))
return {
'url': href,
'title': chosen.get('title') or '',
@@ -261,24 +309,24 @@ def _save_debug(driver, tag: str):
logger.warning("debug snapshot failed: %s", e)
def fetch_chord_chart(query: str, output_path: str, *,
def fetch_chord_chart(title: str, artist: str, output_path: str, *,
mode: str = 'functional',
sheet_style: str = '功能谱',
chord_style: str = '级数名',
verbose: bool = False) -> tuple[bool, str]:
"""
搜 yopu.co、进 view 页、按 row 选样式、截图
返回 (ok, msg)。msg 在失败时是错误说明。
"""搜 yopu /song、按 mode 挑候选 view、截图。
mode='functional' → 数字级数版;mode='letters' → 字母版(弹唱谱)
返回 (ok, msg)。
"""
if verbose:
logging.basicConfig(level=logging.DEBUG, format='%(asctime)s %(levelname)s %(message)s')
else:
logging.basicConfig(level=logging.INFO, format='%(asctime)s %(levelname)s %(message)s')
search_url = f"https://yopu.co/explore#q={quote(query)}"
driver = None
try:
driver = setup_driver()
result = find_first_chord_chart(driver, search_url)
result = find_chart(driver, title, artist, prefer=mode)
if not result:
_save_debug(driver, 'no-search-hit')
return False, '未找到和弦谱'
+4 -4
View File
@@ -62,12 +62,12 @@ export function attachmentUrl(id) {
return `/api/attachments/${id}`
}
export function chordFetch(pieceId) {
return fetch(`/api/pieces/${pieceId}/chord/fetch`, { method: 'POST' }).then(jsonOrThrow)
export function chordFetch(pieceId, mode = 'functional') {
return fetch(`/api/pieces/${pieceId}/chord/fetch?mode=${encodeURIComponent(mode)}`, { method: 'POST' }).then(jsonOrThrow)
}
export function chordStatus(pieceId) {
return fetch(`/api/pieces/${pieceId}/chord/status`).then(jsonOrThrow)
export function chordStatus(pieceId, mode = 'functional') {
return fetch(`/api/pieces/${pieceId}/chord/status?mode=${encodeURIComponent(mode)}`).then(jsonOrThrow)
}
// ---- chat ----
+92 -61
View File
@@ -130,8 +130,43 @@
>{{ line.text }}</div>
</div>
<!-- 谱面 -->
<div v-show="['chord', 'numbered', 'staff'].includes(activeTab)" class="sheet-box">
<!-- 谱面弹唱谱 / 功能谱 共用渲染 -->
<div v-show="['chord', 'functional'].includes(activeTab)" class="sheet-box">
<img
v-for="att in chordTabAttachments(activeTab)"
:key="att.id"
:src="attachmentUrl(att.id)"
:alt="att.filename"
class="sheet-img"
@click="fullscreenSrc = attachmentUrl(att.id)"
/>
<div
v-if="chordTabAttachments(activeTab).length === 0"
class="auto-fetch"
>
<p v-if="chordStateOf(activeTab) === 'idle'" class="hint-line">
<span v-if="activeTab === 'chord'"> yopu.co <b>弹唱谱字母 G/Em/C</b></span>
<span v-else> yopu.co <b>功能谱数字 1/4/5/6m</b></span>
</p>
<p v-else-if="['pending','processing'].includes(chordStateOf(activeTab))" class="hint-line">
正在抓取 30-60s
</p>
<p v-else-if="chordStateOf(activeTab) === 'failed'" class="hint-line err">
抓取失败{{ chordErrors[modeForTab(activeTab)] }}
</p>
<button
class="btn-fetch"
:disabled="['pending','processing'].includes(chordStateOf(activeTab))"
@click="startChordFetch(modeForTab(activeTab))"
>
<span v-if="['pending','processing'].includes(chordStateOf(activeTab))" class="spin"></span>
<span v-else>🎸 自动抓取{{ activeTab === 'chord' ? '弹唱谱' : '功能谱' }}</span>
</button>
</div>
</div>
<!-- 简谱 / 五线谱手动上传的图 -->
<div v-show="['numbered', 'staff'].includes(activeTab)" class="sheet-box">
<img
v-for="att in roleAttachments(activeTab)"
:key="att.id"
@@ -140,28 +175,6 @@
class="sheet-img"
@click="fullscreenSrc = attachmentUrl(att.id)"
/>
<div
v-if="activeTab === 'chord' && roleAttachments('chord').length === 0"
class="auto-fetch"
>
<p v-if="chordState === 'idle'" class="hint-line">
yopu.co <b>功能谱 + 级数名</b>
</p>
<p v-else-if="chordState === 'pending' || chordState === 'processing'" class="hint-line">
正在抓取浏览器后台跑 chromium 截图 30-60s
</p>
<p v-else-if="chordState === 'failed'" class="hint-line err">
抓取失败{{ chordError }}
</p>
<button
class="btn-fetch"
:disabled="chordState === 'pending' || chordState === 'processing'"
@click="startChordFetch"
>
<span v-if="chordState === 'pending' || chordState === 'processing'" class="spin"></span>
<span v-else>🎸 自动抓取吉他谱</span>
</button>
</div>
</div>
<!-- PDF -->
@@ -339,11 +352,14 @@ let lastReportedId = null
// fullscreen 乐谱
const fullscreenSrc = ref(null)
// chord
const chordState = ref('idle')
const chordError = ref('')
let chordPollTimer = null
let chordPollStarted = 0
// chord —— 两个 mode 各自独立 state
const chordStates = ref({ letters: 'idle', functional: 'idle' })
const chordErrors = ref({ letters: '', functional: '' })
const chordPollTimers = { letters: null, functional: null }
const chordPollStarted = { letters: 0, functional: 0 }
function chordStateOf(tab) {
return chordStates.value[modeForTab(tab)] || 'idle'
}
// chat
const chatBodyEl = ref(null)
@@ -380,11 +396,26 @@ function roleAttachments(role) {
)
}
// chord 兼容:吉他谱 tab 显示历史 role='chord' + 新 role='chord_letters';功能谱 tab 显示 role='chord_functional'
function chordTabAttachments(tab) {
const set = tab === 'chord'
? new Set(['chord', 'chord_letters'])
: new Set(['chord_functional'])
return (selected.value?.attachments || []).filter(
a => a.kind === 'image' && set.has(a.role),
)
}
function modeForTab(tab) {
return tab === 'functional' ? 'functional' : 'letters'
}
const tabs = computed(() => {
if (!selected.value) return []
const list = []
if (selected.value.lyrics) list.push({ key: 'lyrics', label: '歌词', count: 0 })
list.push({ key: 'chord', label: '吉他谱', count: roleAttachments('chord').length })
list.push({ key: 'chord', label: '吉他谱', count: chordTabAttachments('chord').length })
list.push({ key: 'functional', label: '功能谱', count: chordTabAttachments('functional').length })
const num = roleAttachments('numbered').length
if (num) list.push({ key: 'numbered', label: '简谱', count: num })
const staff = roleAttachments('staff').length
@@ -504,9 +535,9 @@ async function promptNewPlaylist() {
async function loadPiece(id) {
selected.value = null
notesDraft.value = ''
stopChordPoll()
chordState.value = 'idle'
chordError.value = ''
stopChordPoll('letters'); stopChordPoll('functional')
chordStates.value = { letters: 'idle', functional: 'idle' }
chordErrors.value = { letters: '', functional: '' }
abortChat()
chatMessages.value = []
chatStreamText.value = ''
@@ -660,54 +691,54 @@ function onNotesInput() {
}
// chord
async function startChordFetch() {
async function startChordFetch(mode) {
if (!selectedId.value) return
chordState.value = 'pending'
chordError.value = ''
chordStates.value = { ...chordStates.value, [mode]: 'pending' }
chordErrors.value = { ...chordErrors.value, [mode]: '' }
try {
const r = await chordFetch(selectedId.value)
const r = await chordFetch(selectedId.value, mode)
if (r.status === 'completed') {
await reloadPiece()
chordState.value = 'completed'
chordStates.value = { ...chordStates.value, [mode]: 'completed' }
return
}
chordState.value = r.status || 'pending'
chordPollStarted = Date.now()
if (chordPollTimer) clearInterval(chordPollTimer)
chordPollTimer = setInterval(pollChord, 3000)
chordStates.value = { ...chordStates.value, [mode]: r.status || 'pending' }
chordPollStarted[mode] = Date.now()
if (chordPollTimers[mode]) clearInterval(chordPollTimers[mode])
chordPollTimers[mode] = setInterval(() => pollChord(mode), 3000)
} catch (e) {
chordState.value = 'failed'
chordError.value = e.message || String(e)
chordStates.value = { ...chordStates.value, [mode]: 'failed' }
chordErrors.value = { ...chordErrors.value, [mode]: e.message || String(e) }
}
}
async function pollChord() {
if (!selectedId.value) { stopChordPoll(); return }
if (Date.now() - chordPollStarted > 90_000) {
stopChordPoll()
chordState.value = 'failed'
chordError.value = '抓取超时'
async function pollChord(mode) {
if (!selectedId.value) { stopChordPoll(mode); return }
if (Date.now() - chordPollStarted[mode] > 90_000) {
stopChordPoll(mode)
chordStates.value = { ...chordStates.value, [mode]: 'failed' }
chordErrors.value = { ...chordErrors.value, [mode]: '抓取超时' }
return
}
try {
const r = await chordStatus(selectedId.value)
chordState.value = r.status || 'pending'
chordError.value = r.error || ''
const r = await chordStatus(selectedId.value, mode)
chordStates.value = { ...chordStates.value, [mode]: r.status || 'pending' }
chordErrors.value = { ...chordErrors.value, [mode]: r.error || '' }
if (r.status === 'completed') {
stopChordPoll()
stopChordPoll(mode)
await reloadPiece()
} else if (r.status === 'failed') {
stopChordPoll()
stopChordPoll(mode)
}
} catch (e) {
chordError.value = e.message || String(e)
chordErrors.value = { ...chordErrors.value, [mode]: e.message || String(e) }
}
}
function stopChordPoll() {
if (chordPollTimer) {
clearInterval(chordPollTimer)
chordPollTimer = null
function stopChordPoll(mode) {
if (chordPollTimers[mode]) {
clearInterval(chordPollTimers[mode])
chordPollTimers[mode] = null
}
}
@@ -843,7 +874,7 @@ onMounted(async () => {
onBeforeUnmount(() => {
document.removeEventListener('keydown', onKeyDown)
if (notesTimer) clearTimeout(notesTimer)
stopChordPoll()
stopChordPoll('letters'); stopChordPoll('functional')
abortChat()
})
</script>
+70 -31
View File
@@ -1082,25 +1082,52 @@ async fn playlist_remove_piece(
// ---------- handlers: chord auto-fetch ----------
/// `POST /api/pieces/:id/chord/fetch` — 触发 sidecar 抓取 yopu 和弦谱。
/// 已经有 chord attachment 的曲目直接返回 completed。
#[derive(Deserialize)]
struct ChordModeQuery {
/// 'letters' = 弹唱谱字母版;'functional' = 数字级数版。默认 functional 兼容旧调用。
mode: Option<String>,
}
fn chord_mode_to_role(mode: &str) -> &'static str {
match mode {
"letters" => "chord_letters",
_ => "chord_functional",
}
}
fn parse_mode(q: &ChordModeQuery) -> Result<&'static str, AppError> {
let m = q.mode.as_deref().unwrap_or("functional");
match m {
"letters" => Ok("letters"),
"functional" => Ok("functional"),
other => Err(AppError::bad_request(format!(
"mode must be 'letters' or 'functional', got '{other}'"
))),
}
}
/// `POST /api/pieces/:id/chord/fetch?mode=letters|functional` — 触发 sidecar 抓 yopu 谱。
/// 已有该 mode 的 attachment 直接 completed。
async fn chord_fetch(
State(s): State<AppState>,
Path(piece_id): Path<i64>,
Query(q): Query<ChordModeQuery>,
) -> Result<JsonResp<Value>, AppError> {
let (title, artist, has_chord) = chord_piece_meta(&s, piece_id)?;
if has_chord {
return Ok(JsonResp(json!({ "status": "completed", "reason": "已有吉他谱" })));
let mode = parse_mode(&q)?;
let (title, artist, has) = chord_piece_meta(&s, piece_id, mode)?;
if has {
return Ok(JsonResp(json!({ "status": "completed", "mode": mode, "reason": "already imported" })));
}
let query = match artist.as_deref() {
Some(a) if !a.is_empty() => format!("{a} {title}"),
_ => title,
};
let url = format!("{}/fetch", s.chord_url);
let resp = s
.http
.post(&url)
.query(&[("piece_id", piece_id.to_string()), ("query", query)])
.query(&[
("piece_id", piece_id.to_string()),
("title", title),
("artist", artist.unwrap_or_default()),
("mode", mode.to_string()),
])
.timeout(std::time::Duration::from_secs(15))
.send()
.await
@@ -1117,17 +1144,19 @@ async fn chord_fetch(
Ok(JsonResp(body))
}
/// `GET /api/pieces/:id/chord/status` — 查询抓取状态。完成时把 png import 成 attachment。
/// `GET /api/pieces/:id/chord/status?mode=letters|functional`
async fn chord_status(
State(s): State<AppState>,
Path(piece_id): Path<i64>,
Query(q): Query<ChordModeQuery>,
) -> Result<JsonResp<Value>, AppError> {
let (_title, _artist, has_chord) = chord_piece_meta(&s, piece_id)?;
if has_chord {
return Ok(JsonResp(json!({ "status": "completed", "imported": true })));
let mode = parse_mode(&q)?;
let (_title, _artist, has) = chord_piece_meta(&s, piece_id, mode)?;
if has {
return Ok(JsonResp(json!({ "status": "completed", "mode": mode, "imported": true })));
}
let url = format!("{}/status/{}", s.chord_url, piece_id);
let url = format!("{}/status/{}/{}", s.chord_url, piece_id, mode);
let resp = s
.http
.get(&url)
@@ -1150,16 +1179,16 @@ async fn chord_status(
.unwrap_or(false);
if st == "completed" && file_exists {
let attachment_id = import_chord_png(&s, piece_id).await?;
// 通知 sidecar 清掉 state + 文件,避免重复 import
let attachment_id = import_chord_png(&s, piece_id, mode).await?;
let _ = s
.http
.delete(format!("{}/state/{}", s.chord_url, piece_id))
.delete(format!("{}/state/{}/{}", s.chord_url, piece_id, mode))
.timeout(std::time::Duration::from_secs(5))
.send()
.await;
return Ok(JsonResp(json!({
"status": "completed",
"mode": mode,
"imported": true,
"attachment_id": attachment_id,
})));
@@ -1171,6 +1200,7 @@ async fn chord_status(
fn chord_piece_meta(
s: &AppState,
piece_id: i64,
mode: &str,
) -> Result<(String, Option<String>, bool), AppError> {
let conn = s.db.lock().unwrap();
let row: Option<(String, Option<String>)> = conn
@@ -1181,38 +1211,40 @@ fn chord_piece_meta(
)
.optional()?;
let (title, artist) = row.ok_or(AppError::NotFound)?;
let has_chord: bool = conn
let role = chord_mode_to_role(mode);
let has: bool = conn
.query_row(
"SELECT 1 FROM attachments
WHERE piece_id = ?1 AND kind = 'image' AND role = 'chord' LIMIT 1",
params![piece_id],
WHERE piece_id = ?1 AND kind = 'image' AND role = ?2 LIMIT 1",
params![piece_id, role],
|_| Ok(true),
)
.optional()?
.unwrap_or(false);
Ok((title, artist, has_chord))
Ok((title, artist, has))
}
async fn import_chord_png(s: &AppState, piece_id: i64) -> Result<i64, AppError> {
let src = std::path::PathBuf::from(format!("/data/chord-fetch/{piece_id}.png"));
let bytes = tokio::fs::metadata(&src).await.map_err(AppError::Io)?;
let size = bytes.len() as i64;
async fn import_chord_png(s: &AppState, piece_id: i64, mode: &str) -> Result<i64, AppError> {
let src = std::path::PathBuf::from(format!("/data/chord-fetch/{piece_id}-{mode}.png"));
let meta = tokio::fs::metadata(&src).await.map_err(AppError::Io)?;
let size = meta.len() as i64;
let role = chord_mode_to_role(mode);
let filename = format!("{role}.png");
let attachment_id = {
let conn = s.db.lock().unwrap();
conn.execute(
"INSERT INTO attachments
(piece_id, kind, role, mime, filename, size_bytes, sort_order)
VALUES (?1, 'image', 'chord', 'image/png', 'chord.png', ?2,
VALUES (?1, 'image', ?2, 'image/png', ?3, ?4,
COALESCE((SELECT MAX(sort_order) FROM attachments WHERE piece_id = ?1), 0) + 1)",
params![piece_id, size],
params![piece_id, role, filename, size],
)?;
conn.last_insert_rowid()
};
let dst = s.blobs_dir.join(attachment_id.to_string());
if let Err(e) = tokio::fs::copy(&src, &dst).await {
// 失败回滚 db 行
let conn = s.db.lock().unwrap();
let _ = conn.execute("DELETE FROM attachments WHERE id = ?1", params![attachment_id]);
return Err(AppError::Io(e));
@@ -1232,10 +1264,17 @@ async fn upload_attachments(
) -> Result<JsonResp<Value>, AppError> {
let role = match q.role.as_deref().map(str::trim).filter(|s| !s.is_empty()) {
None => None,
Some(r) if matches!(r, "chord" | "numbered" | "staff") => Some(r.to_string()),
Some(r)
if matches!(
r,
"chord" | "chord_letters" | "chord_functional" | "numbered" | "staff"
) =>
{
Some(r.to_string())
}
Some(other) => {
return Err(AppError::bad_request(format!(
"unsupported role '{other}', expect one of: chord / numbered / staff"
"unsupported role '{other}', expect one of: chord / chord_letters / chord_functional / numbered / staff"
)));
}
};