feat: add announced checkbox to track rows

Add a persistent "announced" checkbox after each track's Announce button.
The state is stored in a new `announced` column on `show_tracks` and is
auto-set when the Announce button is pressed. The checkbox is also freely
togglable, and announced tracks have their Announce button disabled.

Also fixes .env leakage in test_config.py (pass _env_file=None) and adds
tests for the new DB method, API endpoint, and announce side-effect.

Made-with: Cursor
This commit is contained in:
cottongin
2026-04-01 22:56:43 -04:00
parent a5f77187b3
commit 11f13c86b5
7 changed files with 218 additions and 17 deletions

1
.gitignore vendored
View File

@@ -27,3 +27,4 @@ build/
# AI session artifacts # AI session artifacts
chat-summaries/ chat-summaries/
.superpowers/

View File

@@ -44,6 +44,12 @@ class AnnounceRequest(BaseModel):
position: int position: int
class AnnouncedRequest(BaseModel):
show_id: int
position: int
announced: bool
class PingRequest(BaseModel): class PingRequest(BaseModel):
target: str target: str
message: str message: str
@@ -126,8 +132,24 @@ def create_dashboard_router(
f"{track['title']} by {track['artist']} - {track['permalink_url']}" f"{track['title']} by {track['artist']} - {track['permalink_url']}"
) )
await manager.broadcast({"type": "announce", "message": message}) await manager.broadcast({"type": "announce", "message": message})
db.set_track_announced(body.show_id, body.position, True)
return {"status": "announced", "message": message} return {"status": "announced", "message": message}
@router.post("/admin/announced")
async def set_announced(body: AnnouncedRequest, request: Request):
user = _get_session_user(request)
if user is None:
auth_header = request.headers.get("authorization", "")
if not auth_header.startswith("Bearer ") or auth_header.removeprefix("Bearer ") != admin_token:
raise HTTPException(status_code=401, detail="Unauthorized")
track = db.get_show_track_by_position(body.show_id, body.position)
if track is None:
raise HTTPException(status_code=404, detail=f"No track at position {body.position}")
db.set_track_announced(body.show_id, body.position, body.announced)
return {"status": "ok"}
@router.post("/admin/ping") @router.post("/admin/ping")
async def ping(body: PingRequest, request: Request): async def ping(body: PingRequest, request: Request):
user = _get_session_user(request) user = _get_session_user(request)

View File

@@ -55,6 +55,11 @@ class Database:
conn.commit() conn.commit()
except sqlite3.OperationalError: except sqlite3.OperationalError:
pass pass
try:
conn.execute("ALTER TABLE show_tracks ADD COLUMN announced INTEGER NOT NULL DEFAULT 0")
conn.commit()
except sqlite3.OperationalError:
pass
conn.close() conn.close()
def upsert_track(self, track: Track) -> None: def upsert_track(self, track: Track) -> None:
@@ -154,9 +159,9 @@ class Database:
conn = self._connect() conn = self._connect()
rows = conn.execute( rows = conn.execute(
""" """
SELECT st.show_id, st.track_id, st.position, t.title, t.artist, SELECT st.show_id, st.track_id, st.position, st.announced,
t.permalink_url, t.artwork_url, t.duration_ms, t.license, t.title, t.artist, t.permalink_url, t.artwork_url,
t.liked_at, t.raw_json t.duration_ms, t.license, t.liked_at, t.raw_json
FROM show_tracks st FROM show_tracks st
JOIN tracks t ON st.track_id = t.id JOIN tracks t ON st.track_id = t.id
WHERE st.show_id = ? WHERE st.show_id = ?
@@ -173,9 +178,9 @@ class Database:
conn = self._connect() conn = self._connect()
row = conn.execute( row = conn.execute(
""" """
SELECT st.show_id, st.track_id, st.position, t.title, t.artist, SELECT st.show_id, st.track_id, st.position, st.announced,
t.permalink_url, t.artwork_url, t.duration_ms, t.license, t.title, t.artist, t.permalink_url, t.artwork_url,
t.liked_at, t.raw_json t.duration_ms, t.license, t.liked_at, t.raw_json
FROM show_tracks st FROM show_tracks st
JOIN tracks t ON st.track_id = t.id JOIN tracks t ON st.track_id = t.id
WHERE st.show_id = ? AND st.position = ? WHERE st.show_id = ? AND st.position = ?
@@ -185,6 +190,17 @@ class Database:
conn.close() conn.close()
return dict(row) if row else None return dict(row) if row else None
def set_track_announced(
self, show_id: int, position: int, announced: bool
) -> None:
conn = self._connect()
conn.execute(
"UPDATE show_tracks SET announced = ? WHERE show_id = ? AND position = ?",
(int(announced), show_id, position),
)
conn.commit()
conn.close()
def set_show_tracks(self, show_id: int, track_ids: list[int]) -> None: def set_show_tracks(self, show_id: int, track_ids: list[int]) -> None:
conn = self._connect() conn = self._connect()
if track_ids: if track_ids:

View File

@@ -66,7 +66,8 @@
border-bottom-color: #4caf50; border-bottom-color: #4caf50;
font-weight: 600; font-weight: 600;
} }
.btn-group { display: flex; gap: 4px; } .btn-group { display: flex; gap: 8px; align-items: center; }
.announced-check { accent-color: #4caf50; cursor: pointer; width: 28px; height: 28px; margin: 0; }
.ping-section { .ping-section {
margin-top: 2rem; margin-top: 2rem;
padding-top: 1.5rem; padding-top: 1.5rem;
@@ -149,7 +150,10 @@
html += '<th class="track-num">#</th><th>Title</th><th>Artist</th><th></th>'; html += '<th class="track-num">#</th><th>Title</th><th>Artist</th><th></th>';
html += '</tr></thead><tbody>'; html += '</tr></thead><tbody>';
for (const t of tracks) { for (const t of tracks) {
const disabled = subscriberCount === 0 ? 'disabled title="No bots connected"' : ""; const noBot = subscriberCount === 0;
const announced = !!t.announced;
const annDisabled = (noBot || announced) ? 'disabled' : '';
const annTitle = noBot ? 'title="No bots connected"' : announced ? 'title="Already announced"' : '';
const copyText = `${t.title} by ${t.artist} - ${t.permalink_url}`; const copyText = `${t.title} by ${t.artist} - ${t.permalink_url}`;
html += `<tr> html += `<tr>
<td class="track-num">${t.position}</td> <td class="track-num">${t.position}</td>
@@ -159,8 +163,11 @@
<div class="btn-group"> <div class="btn-group">
<button class="btn-sm copy-btn outline" <button class="btn-sm copy-btn outline"
onclick="copyTrack(this, '${esc(copyText).replace(/'/g, "\\'")}')">Copy</button> onclick="copyTrack(this, '${esc(copyText).replace(/'/g, "\\'")}')">Copy</button>
<button class="btn-sm announce-btn" ${disabled} <button class="btn-sm announce-btn" ${annDisabled} ${annTitle}
onclick="announce(${showId}, ${t.position}, this)">Announce</button> onclick="announce(${showId}, ${t.position}, this)">Announce</button>
<input type="checkbox" class="announced-check"
${announced ? 'checked' : ''}
onchange="toggleAnnounced(${showId}, ${t.position}, this)">
</div> </div>
</td> </td>
</tr>`; </tr>`;
@@ -275,10 +282,18 @@
} }
btn.textContent = "\u2713"; btn.textContent = "\u2713";
btn.classList.add("success"); btn.classList.add("success");
const cb = btn.closest("tr").querySelector(".announced-check");
if (cb) cb.checked = true;
const cached = showCache[showId];
if (cached) {
const track = cached.tracks.find(t => t.position === position);
if (track) track.announced = 1;
}
setTimeout(() => { setTimeout(() => {
btn.textContent = "Announce"; btn.textContent = "Announce";
btn.classList.remove("success"); btn.classList.remove("success");
btn.disabled = false; btn.disabled = true;
btn.title = "Already announced";
}, 2000); }, 2000);
} catch (e) { } catch (e) {
showToast(e.message, true); showToast(e.message, true);
@@ -287,6 +302,39 @@
} }
} }
async function toggleAnnounced(showId, position, cb) {
const newVal = cb.checked;
try {
const resp = await fetch("/admin/announced", {
method: "POST",
headers: {"Content-Type": "application/json"},
body: JSON.stringify({show_id: showId, position: position, announced: newVal}),
});
if (!resp.ok) {
const data = await resp.json().catch(() => ({}));
throw new Error(data.detail || "Failed to update");
}
const cached = showCache[showId];
if (cached) {
const track = cached.tracks.find(t => t.position === position);
if (track) track.announced = newVal ? 1 : 0;
}
const annBtn = cb.closest("tr").querySelector(".announce-btn");
if (annBtn) {
if (newVal) {
annBtn.disabled = true;
annBtn.title = "Already announced";
} else if (subscriberCount > 0) {
annBtn.disabled = false;
annBtn.title = "";
}
}
} catch (e) {
cb.checked = !newVal;
showToast(e.message, true);
}
}
async function sendPing(btn) { async function sendPing(btn) {
const target = document.getElementById("ping-target").value.trim(); const target = document.getElementById("ping-target").value.trim();
const message = document.getElementById("ping-message").value.trim(); const message = document.getElementById("ping-message").value.trim();
@@ -344,9 +392,12 @@
detail.style.display = "none"; detail.style.display = "none";
} }
document.querySelectorAll(".announce-btn").forEach(btn => { document.querySelectorAll(".announce-btn").forEach(btn => {
if (count === 0) { const row = btn.closest("tr");
const cb = row ? row.querySelector(".announced-check") : null;
const isAnnounced = cb && cb.checked;
if (count === 0 || isAnnounced) {
btn.disabled = true; btn.disabled = true;
btn.title = "No bots connected"; btn.title = isAnnounced ? "Already announced" : "No bots connected";
} else { } else {
btn.disabled = false; btn.disabled = false;
btn.title = ""; btn.title = "";

View File

@@ -2,7 +2,7 @@ from ntr_fetcher.config import Settings
def test_settings_defaults(): def test_settings_defaults():
settings = Settings(admin_token="test-secret") settings = Settings(admin_token="test-secret", _env_file=None)
assert settings.port == 8000 assert settings.port == 8000
assert settings.host == "127.0.0.1" assert settings.host == "127.0.0.1"
assert settings.db_path == "./ntr_fetcher.db" assert settings.db_path == "./ntr_fetcher.db"
@@ -17,7 +17,7 @@ def test_settings_from_env(monkeypatch):
monkeypatch.setenv("NTR_HOST", "0.0.0.0") monkeypatch.setenv("NTR_HOST", "0.0.0.0")
monkeypatch.setenv("NTR_ADMIN_TOKEN", "my-secret") monkeypatch.setenv("NTR_ADMIN_TOKEN", "my-secret")
monkeypatch.setenv("NTR_SOUNDCLOUD_USER", "someoneelse") monkeypatch.setenv("NTR_SOUNDCLOUD_USER", "someoneelse")
settings = Settings() settings = Settings(_env_file=None)
assert settings.port == 9090 assert settings.port == 9090
assert settings.host == "0.0.0.0" assert settings.host == "0.0.0.0"
assert settings.admin_token == "my-secret" assert settings.admin_token == "my-secret"
@@ -27,7 +27,7 @@ def test_settings_from_env(monkeypatch):
def test_settings_admin_token_required(): def test_settings_admin_token_required():
import pytest import pytest
with pytest.raises(Exception): with pytest.raises(Exception):
Settings() Settings(_env_file=None)
def test_dashboard_config_absent(monkeypatch): def test_dashboard_config_absent(monkeypatch):
@@ -35,7 +35,7 @@ def test_dashboard_config_absent(monkeypatch):
monkeypatch.delenv("NTR_WEB_USER", raising=False) monkeypatch.delenv("NTR_WEB_USER", raising=False)
monkeypatch.delenv("NTR_WEB_PASSWORD", raising=False) monkeypatch.delenv("NTR_WEB_PASSWORD", raising=False)
monkeypatch.delenv("NTR_SECRET_KEY", raising=False) monkeypatch.delenv("NTR_SECRET_KEY", raising=False)
s = Settings() s = Settings(_env_file=None)
assert s.web_user is None assert s.web_user is None
assert s.web_password is None assert s.web_password is None
assert s.secret_key is None assert s.secret_key is None
@@ -47,7 +47,7 @@ def test_dashboard_config_present(monkeypatch):
monkeypatch.setenv("NTR_WEB_USER", "nick") monkeypatch.setenv("NTR_WEB_USER", "nick")
monkeypatch.setenv("NTR_WEB_PASSWORD", "secret") monkeypatch.setenv("NTR_WEB_PASSWORD", "secret")
monkeypatch.setenv("NTR_SECRET_KEY", "signme") monkeypatch.setenv("NTR_SECRET_KEY", "signme")
s = Settings() s = Settings(_env_file=None)
assert s.web_user == "nick" assert s.web_user == "nick"
assert s.web_password == "secret" assert s.web_password == "secret"
assert s.secret_key == "signme" assert s.secret_key == "signme"

View File

@@ -187,3 +187,66 @@ def test_ws_subscribe_with_invalid_token(app):
ws.send_json({"type": "subscribe", "token": "wrong"}) ws.send_json({"type": "subscribe", "token": "wrong"})
with pytest.raises(Exception): with pytest.raises(Exception):
ws.receive_json() ws.receive_json()
# --- Announced endpoint tests ---
def test_announce_sets_announced_flag(client, db):
show = _seed_show(db)
resp = client.post(
"/admin/announce",
json={"show_id": show.id, "position": 1},
headers={"Authorization": "Bearer test-token"},
)
assert resp.status_code == 200
tracks = db.get_show_tracks(show.id)
assert tracks[0]["announced"] == 1
def test_set_announced_with_bearer(client, db):
show = _seed_show(db)
resp = client.post(
"/admin/announced",
json={"show_id": show.id, "position": 1, "announced": True},
headers={"Authorization": "Bearer test-token"},
)
assert resp.status_code == 200
assert resp.json()["status"] == "ok"
tracks = db.get_show_tracks(show.id)
assert tracks[0]["announced"] == 1
def test_set_announced_toggle_off(client, db):
show = _seed_show(db)
client.post(
"/admin/announced",
json={"show_id": show.id, "position": 1, "announced": True},
headers={"Authorization": "Bearer test-token"},
)
resp = client.post(
"/admin/announced",
json={"show_id": show.id, "position": 1, "announced": False},
headers={"Authorization": "Bearer test-token"},
)
assert resp.status_code == 200
tracks = db.get_show_tracks(show.id)
assert tracks[0]["announced"] == 0
def test_set_announced_without_auth(client, db):
_seed_show(db)
resp = client.post(
"/admin/announced",
json={"show_id": 1, "position": 1, "announced": True},
)
assert resp.status_code == 401
def test_set_announced_invalid_position(client, db):
_seed_show(db)
resp = client.post(
"/admin/announced",
json={"show_id": 1, "position": 99, "announced": True},
headers={"Authorization": "Bearer test-token"},
)
assert resp.status_code == 404

View File

@@ -295,3 +295,51 @@ def test_has_track_in_show(db):
db.set_show_tracks(show.id, [t1.id]) db.set_show_tracks(show.id, [t1.id])
assert db.has_track_in_show(show.id, 1) is True assert db.has_track_in_show(show.id, 1) is True
assert db.has_track_in_show(show.id, 999) is False assert db.has_track_in_show(show.id, 999) is False
def test_announced_defaults_to_zero(db):
week_start = datetime(2026, 3, 13, 2, 0, 0, tzinfo=timezone.utc)
week_end = datetime(2026, 3, 20, 2, 0, 0, tzinfo=timezone.utc)
show = db.get_or_create_show(week_start, week_end)
t1 = _make_track(1, "2026-03-14T01:00:00+00:00")
db.upsert_track(t1)
db.set_show_tracks(show.id, [t1.id])
tracks = db.get_show_tracks(show.id)
assert tracks[0]["announced"] == 0
def test_set_track_announced(db):
week_start = datetime(2026, 3, 13, 2, 0, 0, tzinfo=timezone.utc)
week_end = datetime(2026, 3, 20, 2, 0, 0, tzinfo=timezone.utc)
show = db.get_or_create_show(week_start, week_end)
t1 = _make_track(1, "2026-03-14T01:00:00+00:00")
db.upsert_track(t1)
db.set_show_tracks(show.id, [t1.id])
db.set_track_announced(show.id, 1, True)
tracks = db.get_show_tracks(show.id)
assert tracks[0]["announced"] == 1
def test_set_track_announced_toggle_off(db):
week_start = datetime(2026, 3, 13, 2, 0, 0, tzinfo=timezone.utc)
week_end = datetime(2026, 3, 20, 2, 0, 0, tzinfo=timezone.utc)
show = db.get_or_create_show(week_start, week_end)
t1 = _make_track(1, "2026-03-14T01:00:00+00:00")
db.upsert_track(t1)
db.set_show_tracks(show.id, [t1.id])
db.set_track_announced(show.id, 1, True)
db.set_track_announced(show.id, 1, False)
tracks = db.get_show_tracks(show.id)
assert tracks[0]["announced"] == 0
def test_get_show_track_by_position_includes_announced(db):
week_start = datetime(2026, 3, 13, 2, 0, 0, tzinfo=timezone.utc)
week_end = datetime(2026, 3, 20, 2, 0, 0, tzinfo=timezone.utc)
show = db.get_or_create_show(week_start, week_end)
t1 = _make_track(1, "2026-03-14T01:00:00+00:00")
db.upsert_track(t1)
db.set_show_tracks(show.id, [t1.id])
db.set_track_announced(show.id, 1, True)
result = db.get_show_track_by_position(show.id, 1)
assert result["announced"] == 1