improve webui

This commit is contained in:
Mark 2024-10-25 22:50:18 +02:00
parent c03c35f1c9
commit eedd348c5c

View File

@ -4,7 +4,7 @@ use std::sync::{mpsc, Arc, Mutex};
use musicdb_lib::data::album::Album; use musicdb_lib::data::album::Album;
use musicdb_lib::data::artist::Artist; use musicdb_lib::data::artist::Artist;
use musicdb_lib::data::database::Database; use musicdb_lib::data::database::Database;
use musicdb_lib::data::queue::{QueueContent, QueueFolder}; use musicdb_lib::data::queue::{Queue, QueueContent, QueueFolder};
use musicdb_lib::data::song::Song; use musicdb_lib::data::song::Song;
use musicdb_lib::data::SongId; use musicdb_lib::data::SongId;
use musicdb_lib::server::Command; use musicdb_lib::server::Command;
@ -34,7 +34,7 @@ use rocket::{get, routes, Config, State};
*/ */
const HTML_START: &'static str = const HTML_START: &'static str =
"<!DOCTYPE html><html><head><meta name=\"color-scheme\" content=\"light dark\">"; "<!DOCTYPE html><html><head><meta charset=\"UTF-8\"><meta name=\"color-scheme\" content=\"light dark\">";
const HTML_SEP: &'static str = "</head><body>"; const HTML_SEP: &'static str = "</head><body>";
const HTML_END: &'static str = "</body></html>"; const HTML_END: &'static str = "</body></html>";
@ -44,8 +44,10 @@ struct Data {
} }
#[get("/")] #[get("/")]
async fn index(data: &State<Data>) -> RawHtml<String> { fn index(data: &State<Data>) -> RawHtml<String> {
dbg!(());
let script = r#"<script> let script = r#"<script>
const sleep = ms => new Promise(r => setTimeout(r, ms));
async function performSearch() { async function performSearch() {
var searchResultDiv = document.getElementById("searchResultDiv"); var searchResultDiv = document.getElementById("searchResultDiv");
searchResultDiv.innerHTML = "Loading..."; searchResultDiv.innerHTML = "Loading...";
@ -68,7 +70,7 @@ async function performSearch() {
} }
query += "title=" + encodeURIComponent(sfTitle); query += "title=" + encodeURIComponent(sfTitle);
} }
if (query || confirm("You didn't search for anything specific. If you continue, the whole library will be loaded, which can take a while and use a lot of bandwidth!")) { if (query || confirm("You didn't search for anything specific. If you continue, the whole library will be loaded, which can take a while, use a lot of bandwidth, and may crash your browser!")) {
console.log("Performing search with query '" + query + "'."); console.log("Performing search with query '" + query + "'.");
var r1 = await fetch("/search?" + query); var r1 = await fetch("/search?" + query);
var r2 = await r1.text(); var r2 = await r1.text();
@ -81,39 +83,206 @@ async function addSong(id) {
await fetch("/add-song/" + id); await fetch("/add-song/" + id);
} }
</script>"#; </script>"#;
let buttons = "<button onclick=\"fetch('/play').then(() => location.reload())\">play</button><button onclick=\"fetch('/pause').then(() => location.reload())\">pause</button><button onclick=\"fetch('/skip').then(() => location.reload())\">skip</button><button onclick=\"fetch('/clear-queue').then(() => location.reload())\">clear queue</button><button onclick=\"location.reload()\">reload</button>"; let script2 = r#"<script>
const searchDiv = document.getElementById("searchDiv");
searchDiv.style.display = "";
document.getElementById("warnLag").innerText = "connecting...";
const nowPlayingDiv = document.getElementById("nowPlayingDiv");
const queueDiv = document.getElementById("queueDiv");
var didFinish = false;
var averageLoopTimeMs = 250;
async function runLoop() {
while (true) {
await sleep(1000);
didFinish = false;
var startTime = new Date();
sleep(averageLoopTimeMs*2).then(async function() {
while (!didFinish) {
var elapsed = new Date() - startTime;
document.getElementById("warnLag").innerText = "Warning: slow connection, server may be busy. be patient. (" + Math.round(averageLoopTimeMs) + "ms exceeded by " + Math.round((elapsed-averageLoopTimeMs)/averageLoopTimeMs) + "x)";
await sleep(100);
}
});
nowPlayingDiv.innerHTML = await (await fetch("/now-playing-html")).text();
queueDiv.innerHTML = await (await fetch("/queue-html")).text();
var elapsedTime = new Date() - startTime;
didFinish = true;
averageLoopTimeMs = ((averageLoopTimeMs * 4) + elapsedTime) / 5;
document.getElementById("warnLag").innerText = "Average update time: " + Math.round(averageLoopTimeMs) + "ms";
}
}
runLoop();</script>"#;
let buttons = "<button onclick=\"fetch('/play')\">play</button><button onclick=\"fetch('/pause')\">pause</button><button onclick=\"fetch('/skip')\">skip</button><button onclick=\"fetch('/clear-queue')\">clear queue</button>";
let search = "<input id=\"searchFieldArtist\" placeholder=\"artist\"><input id=\"searchFieldAlbum\" placeholder=\"album\"><input id=\"searchFieldTitle\" placeholder=\"title\"> let search = "<input id=\"searchFieldArtist\" placeholder=\"artist\"><input id=\"searchFieldAlbum\" placeholder=\"album\"><input id=\"searchFieldTitle\" placeholder=\"title\">
<button onclick=\"performSearch()\">search</button><div id=\"searchResultDiv\"></div>"; <button onclick=\"performSearch()\">search</button><div id=\"searchResultDiv\"></div>";
let db = data.db.lock().unwrap(); let db = data.db.lock().unwrap();
let now_playing = let now_playing = gen_now_playing(&db);
if let Some(current_song) = db.queue.get_current_song().and_then(|id| db.get_song(id)) { let mut queue = String::new();
format!( gen_queue_html(&db.queue, &mut queue, &db);
"<h1>Now Playing</h1><h4>{}</h4>", dbg!(&queue);
html_escape::encode_safe(&current_song.title),
)
} else {
format!("<h1>Now Playing</h1><p>nothing</p>",)
};
drop(db); drop(db);
RawHtml(format!( RawHtml(format!(
"{HTML_START}<title>MusicDb</title>{script}{HTML_SEP}{now_playing}<div>{buttons}</div><div>{search}</div>{HTML_END}", "{HTML_START}<title>MusicDb</title>{script}{HTML_SEP}<div id=\"warnLag\">no javascript? reload to see updated information.</div><div id=\"nowPlayingDiv\">{now_playing}</div><div>{buttons}</div><div id=\"searchDiv\" style=\"display:none;\">{search}</div><div id=\"queueDiv\">{queue}</div>{script2}{HTML_END}",
)) ))
} }
#[get("/now-playing-html")]
fn now_playing_html(data: &State<Data>) -> RawHtml<String> {
RawHtml(gen_now_playing(&*data.db.lock().unwrap()))
}
#[get("/queue-html")]
fn queue_html(data: &State<Data>) -> RawHtml<String> {
let mut str = String::new();
let db = data.db.lock().unwrap();
gen_queue_html(&db.queue, &mut str, &db);
RawHtml(str)
}
fn gen_now_playing(db: &Database) -> String {
if let Some(current_song) = db.queue.get_current_song().and_then(|id| db.get_song(id)) {
format!(
"<h1>Now Playing</h1><h4>{}</h4>",
html_escape::encode_safe(&current_song.title),
)
} else {
format!("<h1>Now Playing</h1><p>nothing</p>",)
}
}
fn gen_queue_html(queue: &Queue, str: &mut String, db: &Database) {
gen_queue_html_impl(queue, str, db, true, &mut "".to_owned());
}
fn gen_queue_html_impl(
queue: &Queue,
str: &mut String,
db: &Database,
active_highlight: bool,
path: &mut String,
) {
match queue.content() {
QueueContent::Song(id) => {
if let Some(song) = db.songs().get(id) {
str.push_str("<div>");
if active_highlight {
str.push_str("<b>");
}
str.push_str(&format!("<button onclick=\"fetch('/queue-goto/{path}')\">"));
str.push_str(&html_escape::encode_text(&song.title));
str.push_str("</button>");
if active_highlight {
str.push_str("</b>");
}
str.push_str("<small>");
if let Some(artist) = db.artists().get(&song.artist) {
str.push_str(" by ");
str.push_str(&html_escape::encode_text(&artist.name));
}
if let Some(album) = song.album.as_ref().and_then(|id| db.albums().get(id)) {
str.push_str(" on ");
str.push_str(&html_escape::encode_text(&album.name));
}
str.push_str(&format!(
"<button onclick=\"fetch('/queue-remove/{path}')\">rm</button>"
));
str.push_str("</small></div>");
} else {
str.push_str("<div><small>unknown song</small></div>");
}
}
QueueContent::Folder(f) => {
let html_shuf: &'static str = " <small><small>shuffled</small></small>";
if f.content.is_empty() {
str.push_str("[0/0] ");
if active_highlight {
str.push_str("<b>");
}
str.push_str(&html_escape::encode_text(&f.name));
if active_highlight {
str.push_str("</b>");
}
if f.order.is_some() {
str.push_str(html_shuf);
}
} else {
str.push_str(&format!("[{}/{}] ", f.index + 1, f.content.len(),));
if active_highlight {
str.push_str("<b>");
}
str.push_str(&html_escape::encode_text(&f.name));
if active_highlight {
str.push_str("</b>");
}
if f.order.is_some() {
str.push_str(html_shuf);
}
str.push_str("<ol>");
for (i, v) in f.iter().enumerate() {
str.push_str("<li>");
if !path.is_empty() {
path.push('_');
}
path.push_str(&format!("{i}"));
gen_queue_html_impl(v, str, db, active_highlight && i == f.index, path);
while !(path.is_empty() || path.ends_with('_')) {
path.pop();
}
path.pop();
str.push_str("</li>");
}
str.push_str("</ol>");
}
}
QueueContent::Loop(d, t, i) => {
if active_highlight {
str.push_str("<b>");
}
if *t == 0 {
str.push_str(&format!("<small>[{}/&infin;]</small>", d + 1));
} else {
str.push_str(&format!("<small>[{}/{}]</small>", d + 1, t));
}
if active_highlight {
str.push_str("</b>");
}
if !path.is_empty() {
path.push('_');
}
path.push('0');
gen_queue_html_impl(i, str, db, active_highlight, path);
while !(path.is_empty() || path.ends_with('_')) {
path.pop();
}
path.pop();
}
}
}
#[get("/queue-remove/<path>")]
fn queue_remove(data: &State<Data>, path: &str) {
if let Some(path) = path.split('_').map(|v| v.parse().ok()).collect() {
data.command_sender
.send(Command::QueueRemove(path))
.unwrap();
}
}
#[get("/queue-goto/<path>")]
fn queue_goto(data: &State<Data>, path: &str) {
if let Some(path) = path.split('_').map(|v| v.parse().ok()).collect() {
data.command_sender.send(Command::QueueGoto(path)).unwrap();
}
}
#[get("/play")] #[get("/play")]
async fn play(data: &State<Data>) { fn play(data: &State<Data>) {
data.command_sender.send(Command::Resume).unwrap(); data.command_sender.send(Command::Resume).unwrap();
} }
#[get("/pause")] #[get("/pause")]
async fn pause(data: &State<Data>) { fn pause(data: &State<Data>) {
data.command_sender.send(Command::Pause).unwrap(); data.command_sender.send(Command::Pause).unwrap();
} }
#[get("/skip")] #[get("/skip")]
async fn skip(data: &State<Data>) { fn skip(data: &State<Data>) {
data.command_sender.send(Command::NextSong).unwrap(); data.command_sender.send(Command::NextSong).unwrap();
} }
#[get("/clear-queue")] #[get("/clear-queue")]
async fn clear_queue(data: &State<Data>) { fn clear_queue(data: &State<Data>) {
data.command_sender data.command_sender
.send(Command::QueueUpdate( .send(Command::QueueUpdate(
vec![], vec![],
@ -129,7 +298,7 @@ async fn clear_queue(data: &State<Data>) {
} }
#[get("/add-song/<id>")] #[get("/add-song/<id>")]
async fn add_song(data: &State<Data>, id: SongId) { fn add_song(data: &State<Data>, id: SongId) {
data.command_sender data.command_sender
.send(Command::QueueAdd( .send(Command::QueueAdd(
vec![], vec![],
@ -139,7 +308,7 @@ async fn add_song(data: &State<Data>, id: SongId) {
} }
#[get("/search?<artist>&<album>&<title>&<artist_tags>&<album_tags>&<song_tags>")] #[get("/search?<artist>&<album>&<title>&<artist_tags>&<album_tags>&<song_tags>")]
async fn search( fn search(
data: &State<Data>, data: &State<Data>,
artist: Option<&str>, artist: Option<&str>,
album: Option<&str>, album: Option<&str>,
@ -371,7 +540,19 @@ pub async fn main(
.manage(Data { db, command_sender }) .manage(Data { db, command_sender })
.mount( .mount(
"/", "/",
routes![index, play, pause, skip, clear_queue, add_song, search], routes![
index,
play,
pause,
skip,
clear_queue,
queue_goto,
queue_remove,
add_song,
search,
now_playing_html,
queue_html
],
) )
.launch() .launch()
.await .await