diff --git a/assets/images/placeholders/MusicPlaceholder.svg b/assets/images/placeholders/MusicPlaceholder.svg new file mode 100644 index 0000000..4a3917b --- /dev/null +++ b/assets/images/placeholders/MusicPlaceholder.svg @@ -0,0 +1 @@ + \ No newline at end of file diff --git a/migrations/2024-05-18-231505_add_user_admin/down.sql b/migrations/2024-05-18-231505_add_user_admin/down.sql new file mode 100644 index 0000000..8c2de0f --- /dev/null +++ b/migrations/2024-05-18-231505_add_user_admin/down.sql @@ -0,0 +1 @@ +ALTER TABLE users DROP COLUMN admin; diff --git a/migrations/2024-05-18-231505_add_user_admin/up.sql b/migrations/2024-05-18-231505_add_user_admin/up.sql new file mode 100644 index 0000000..58a6e49 --- /dev/null +++ b/migrations/2024-05-18-231505_add_user_admin/up.sql @@ -0,0 +1 @@ +ALTER TABLE users ADD COLUMN admin BOOLEAN DEFAULT FALSE NOT NULL; diff --git a/src/auth.rs b/src/auth.rs index 58ba6f4..37f861f 100644 --- a/src/auth.rs +++ b/src/auth.rs @@ -21,9 +21,10 @@ use crate::users::UserCredentials; pub async fn signup(new_user: User) -> Result<(), ServerFnError> { use crate::users::create_user; - // Ensure the user has no id + // Ensure the user has no id, and is not a self-proclaimed admin let new_user = User { id: None, + admin: false, ..new_user }; @@ -143,3 +144,37 @@ pub async fn get_user() -> Result { auth_session.user.ok_or(ServerFnError::::ServerError("User not logged in".to_string())) } + +/// Check if a user is an admin +/// Returns a Result with a boolean indicating if the user is logged in and an admin +#[server(endpoint = "check_admin")] +pub async fn check_admin() -> Result { + let auth_session = extract::>().await + .map_err(|e| ServerFnError::::ServerError(format!("Error getting auth session: {}", e)))?; + + Ok(auth_session.user.as_ref().map(|u| u.admin).unwrap_or(false)) +} + +/// Require that a user is logged in and an admin +/// Returns a Result with the error message if the user is not logged in or is not an admin +/// Intended to be used at the start of a protected route, to ensure the user is logged in and an admin: +/// ```rust +/// use leptos::*; +/// use libretunes::auth::require_admin; +/// #[server(endpoint = "protected_admin_route")] +/// pub async fn protected_admin_route() -> Result<(), ServerFnError> { +/// require_admin().await?; +/// // Continue with protected route +/// Ok(()) +/// } +/// ``` +#[cfg(feature = "ssr")] +pub async fn require_admin() -> Result<(), ServerFnError> { + check_admin().await.and_then(|is_admin| { + if is_admin { + Ok(()) + } else { + Err(ServerFnError::::ServerError(format!("Unauthorized"))) + } + }) +} diff --git a/src/models.rs b/src/models.rs index f0665ba..83b716d 100644 --- a/src/models.rs +++ b/src/models.rs @@ -41,6 +41,8 @@ pub struct User { /// The time the user was created #[cfg_attr(feature = "ssr", diesel(deserialize_as = SystemTime))] pub created_at: Option, + /// Whether the user is an admin + pub admin: bool, } impl User { @@ -304,6 +306,26 @@ impl Artist { Ok(my_songs) } + + /// Display a list of artists as a string. + /// + /// For one artist, displays [artist1]. For two artists, displays [artist1] & [artist2]. + /// For three or more artists, displays [artist1], [artist2], & [artist3]. + pub fn display_list(artists: &Vec) -> String { + let mut artist_list = String::new(); + + for (i, artist) in artists.iter().enumerate() { + if i == 0 { + artist_list.push_str(&artist.name); + } else if i == artists.len() - 1 { + artist_list.push_str(&format!(" & {}", artist.name)); + } else { + artist_list.push_str(&format!(", {}", artist.name)); + } + } + + artist_list + } } /// Model for an album @@ -433,6 +455,32 @@ impl Song { Ok(my_artists) } + + /// Get the album for this song from the database + /// + /// # Arguments + /// + /// * `conn` - A mutable reference to a database connection + /// + /// # Returns + /// + /// * `Result, Box>` - A result indicating success with an album, or None if + /// the song does not have an album, or an error + /// + #[cfg(feature = "ssr")] + pub fn get_album(self: &Self, conn: &mut PgPooledConn) -> Result, Box> { + use crate::schema::albums::dsl::*; + + if let Some(album_id) = self.album_id { + let my_album = albums + .filter(id.eq(album_id)) + .first::(conn)?; + + Ok(Some(my_album)) + } else { + Ok(None) + } + } } /// Model for a history entry diff --git a/src/pages/signup.rs b/src/pages/signup.rs index 69f68c7..f02dfab 100644 --- a/src/pages/signup.rs +++ b/src/pages/signup.rs @@ -25,6 +25,7 @@ pub fn Signup() -> impl IntoView { email: email.get(), password: Some(password.get()), created_at: None, + admin: false, }; log!("new user: {:?}", new_user); diff --git a/src/playbar.rs b/src/playbar.rs index 7bfc1b0..7ff2458 100644 --- a/src/playbar.rs +++ b/src/playbar.rs @@ -1,3 +1,4 @@ +use crate::models::Artist; use crate::playstatus::PlayStatus; use leptos::ev::MouseEvent; use leptos::html::{Audio, Div}; @@ -243,26 +244,27 @@ fn PlayDuration(elapsed_secs: MaybeSignal, total_secs: MaybeSignal) -> fn MediaInfo(status: RwSignal) -> impl IntoView { let name = Signal::derive(move || { status.with(|status| { - status.queue.front().map_or("No media playing".into(), |song| song.name.clone()) + status.queue.front().map_or("No media playing".into(), |song| song.title.clone()) }) }); let artist = Signal::derive(move || { status.with(|status| { - status.queue.front().map_or("".into(), |song| song.artist.clone()) + status.queue.front().map_or("".into(), |song| format!("{}", Artist::display_list(&song.artists))) }) }); let album = Signal::derive(move || { status.with(|status| { - status.queue.front().map_or("".into(), |song| song.album.clone()) + status.queue.front().map_or("".into(), |song| + song.album.as_ref().map_or("".into(), |album| album.title.clone())) }) }); let image = Signal::derive(move || { status.with(|status| { - // TODO Use some default / unknown image? - status.queue.front().map_or("".into(), |song| song.image_path.clone()) + status.queue.front().map_or("/images/placeholders/MusicPlaceholder.svg".into(), + |song| song.image_path.clone()) }) }); @@ -400,7 +402,7 @@ pub fn PlayBar(status: RwSignal) -> impl IntoView { status.with_untracked(|status| { // Start playing the first song in the queue, if available if let Some(song) = status.queue.front() { - log!("Starting playing with song: {}", song.name); + log!("Starting playing with song: {}", song.title); // Don't use the set_play_src / set_playing helper function // here because we already have access to the audio element @@ -453,7 +455,7 @@ pub fn PlayBar(status: RwSignal) -> impl IntoView { let prev_song = status.queue.pop_front(); if let Some(prev_song) = prev_song { - log!("Adding song to history: {}", prev_song.name); + log!("Adding song to history: {}", prev_song.title); status.history.push_back(prev_song); } else { log!("Queue empty, no previous song to add to history"); diff --git a/src/queue.rs b/src/queue.rs index 39f0ee0..8a819b9 100644 --- a/src/queue.rs +++ b/src/queue.rs @@ -1,3 +1,4 @@ +use crate::models::Artist; use crate::playstatus::PlayStatus; use crate::song::Song; use leptos::ev::MouseEvent; @@ -98,7 +99,7 @@ pub fn Queue(status: RwSignal) -> impl IntoView { on:dragenter=move |e: DragEvent| on_drag_enter(e, index) on:dragover=on_drag_over > - + Varchar, password -> Varchar, created_at -> Timestamp, + admin -> Bool, } } diff --git a/src/songdata.rs b/src/songdata.rs index 7e0dd0c..23ce415 100644 --- a/src/songdata.rs +++ b/src/songdata.rs @@ -1,12 +1,25 @@ +use crate::models::{Album, Artist, Song}; + +use time::Date; + /// Holds information about a song -#[derive(Debug, Clone)] +/// +/// Intended to be used in the front-end, as it includes artist and album objects, rather than just their ids. pub struct SongData { + /// Song id + pub id: i32, /// Song name - pub name: String, - /// Song artist - pub artist: String, + pub title: String, + /// Song artists + pub artists: Vec, /// Song album - pub album: String, + pub album: Option, + /// The track number of the song on the album + pub track: Option, + /// The duration of the song in seconds + pub duration: i32, + /// The song's release date + pub release_date: Option, /// Path to song file, relative to the root of the web server. /// For example, `"/assets/audio/Song.mp3"` pub song_path: String, @@ -14,3 +27,71 @@ pub struct SongData { /// For example, `"/assets/images/Song.jpg"` pub image_path: String, } + +#[cfg(feature = "ssr")] +impl TryInto for Song { + type Error = Box; + + /// Convert a Song object into a SongData object + /// + /// This conversion is expensive, as it requires database queries to get the artist and album objects. + /// The SongData/Song conversions are also not truly reversible, + /// due to the way the image_path, album, and artist data is handled. + fn try_into(self) -> Result { + use crate::database; + let mut db_con = database::get_db_conn(); + + let album = self.get_album(&mut db_con)?; + + // Use the song's image path if it exists, otherwise use the album's image path, or fallback to the placeholder + let image_path = self.image_path.clone().unwrap_or_else(|| { + album + .as_ref() + .and_then(|album| album.image_path.clone()) + .unwrap_or_else(|| "/assets/images/placeholder.jpg".to_string()) + }); + + Ok(SongData { + id: self.id.ok_or("Song id must be present (Some) to convert to SongData")?, + title: self.title.clone(), + artists: self.get_artists(&mut db_con)?, + album: album, + track: self.track, + duration: self.duration, + release_date: self.release_date, + // TODO https://gitlab.mregirouard.com/libretunes/libretunes/-/issues/35 + song_path: self.storage_path, + image_path: image_path, + }) + } +} + +impl TryInto for SongData { + type Error = Box; + + /// Convert a SongData object into a Song object + /// + /// The SongData/Song conversions are also not truly reversible, + /// due to the way the image_path, album, and and artist data is handled. + fn try_into(self) -> Result { + Ok(Song { + id: Some(self.id), + title: self.title, + album_id: self.album.map(|album| + album.id.ok_or("Album id must be present (Some) to convert to Song")).transpose()?, + track: self.track, + duration: self.duration, + release_date: self.release_date, + // TODO https://gitlab.mregirouard.com/libretunes/libretunes/-/issues/35 + storage_path: self.song_path, + + // Note that if the source of the image_path was the album, the image_path + // will be set to the album's image_path instead of None + image_path: if self.image_path == "/assets/images/placeholder.jpg" { + None + } else { + Some(self.image_path) + }, + }) + } +}