Make it possible to choose an avatar

plume-models/src/users.rs

@@ -491,6 +491,13 @@ impl User {
             }
         })
     }
+
+    pub fn set_avatar(&self, conn: &PgConnection, id: i32) {
+        diesel::update(self)
+            .set(users::avatar_id.eq(id))
+            .execute(conn)
+            .expect("Couldn't update user avatar");
+    }
 }
 
 impl<'a, 'r> FromRequest<'a, 'r> for User {

src/main.rs

@@ -68,6 +68,7 @@ fn main() {
             routes::medias::upload,
             routes::medias::details,
             routes::medias::delete,
+            routes::medias::set_avatar,
             routes::medias::static_files,
 
             routes::notifications::paginated_notifications,

src/routes/medias.rs

@@ -104,8 +104,14 @@ fn delete(id: i32, _user: User, conn: DbConn) -> Redirect {
     Redirect::to(uri!(list))
 }
 
+#[get("/medias/<id>/avatar")]
+fn set_avatar(id: i32, user: User, conn: DbConn) -> Redirect {
+    let media = Media::get(&*conn, id).expect("Media to delete not found");
+    user.set_avatar(&*conn, media.id);
+    Redirect::to(uri!(details: id = id))
+}
+
 #[get("/static/media/<file..>", rank = 1)]
 fn static_files(file: PathBuf) -> Option<NamedFile> {
     NamedFile::open(Path::new("media/").join(file)).ok()
 }
-