admin/delete_crate: Add --deleted-by CLI option

Author: Turbo87

src/bin/crates-admin/delete_crate.rs

@@ -2,7 +2,7 @@ use crate::dialoguer;
 use anyhow::Context;
 use chrono::{NaiveDateTime, Utc};
 use colored::Colorize;
-use crates_io::models::NewDeletedCrate;
+use crates_io::models::{NewDeletedCrate, User};
 use crates_io::schema::{crate_downloads, deleted_crates};
 use crates_io::worker::jobs;
 use crates_io::{db, schema::crates};
@@ -29,6 +29,10 @@ pub struct Opts {
     /// Don't ask for confirmation: yes, we are sure. Best for scripting.
     #[arg(short, long)]
     yes: bool,
+
+    /// Your GitHub username.
+    #[arg(long)]
+    deleted_by: String,
 }
 
 pub async fn run(opts: Opts) -> anyhow::Result<()> {
@@ -47,6 +51,10 @@ pub async fn run(opts: Opts) -> anyhow::Result<()> {
         .await
         .context("Failed to look up crate name from the database")?;
 
+    let deleted_by = User::async_find_by_login(&mut conn, &opts.deleted_by)
+        .await
+        .context("Failed to look up `--deleted-by` user from the database")?;
+
     println!("Deleting the following crates:");
     println!();
     for name in &crate_names {
@@ -73,6 +81,7 @@ pub async fn run(opts: Opts) -> anyhow::Result<()> {
             let deleted_crate = NewDeletedCrate::builder(name)
                 .created_at(&created_at)
                 .deleted_at(&now)
+                .deleted_by(deleted_by.id)
                 .available_at(&now)
                 .build();
 

src/models/user.rs

@@ -51,6 +51,20 @@ impl User {
             .first(conn)
     }
 
+    pub async fn async_find_by_login(
+        conn: &mut AsyncPgConnection,
+        login: &str,
+    ) -> QueryResult<User> {
+        use diesel_async::RunQueryDsl;
+
+        users::table
+            .filter(lower(users::gh_login).eq(login.to_lowercase()))
+            .filter(users::gh_id.ne(-1))
+            .order(users::gh_id.desc())
+            .first(conn)
+            .await
+    }
+
     pub fn owning(krate: &Crate, conn: &mut impl Conn) -> QueryResult<Vec<Owner>> {
         use diesel::RunQueryDsl;