feat: add check about to expiry command

rust-lang · Apr 5, 2024 · 0a1214c · 0a1214c
1 parent c1964bf
commit 0a1214c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/src/admin/enqueue_job.rs b/src/admin/enqueue_job.rs
@@ -38,6 +38,7 @@ pub enum Command {
         #[arg(long)]
         force: bool,
     },
+    CheckAboutToExpireToken,
 }
 
 pub fn run(command: Command) -> Result<()> {
@@ -118,6 +119,9 @@ pub fn run(command: Command) -> Result<()> {
 
             jobs::CheckTyposquat::new(&name).enqueue(conn)?;
         }
+        Command::CheckAboutToExpireToken => {
+            jobs::CheckAboutToExpireToken.enqueue(conn)?;
+        }
     };
 
     Ok(())

diff --git a/src/worker/jobs/mod.rs b/src/worker/jobs/mod.rs
@@ -19,6 +19,7 @@ pub use self::downloads::{
     CleanProcessedLogFiles, ProcessCdnLog, ProcessCdnLogQueue, UpdateDownloads,
 };
 pub use self::dump_db::DumpDb;
+pub use self::expiry_notification::CheckAboutToExpireToken;
 pub use self::git::{NormalizeIndex, SquashIndex, SyncToGitIndex, SyncToSparseIndex};
 pub use self::readmes::RenderAndUploadReadme;
 pub use self::sync_admins::SyncAdmins;

diff --git a/src/worker/mod.rs b/src/worker/mod.rs
@@ -32,5 +32,6 @@ impl RunnerExt for Runner<Arc<Environment>> {
             .register_job_type::<jobs::SyncToGitIndex>()
             .register_job_type::<jobs::SyncToSparseIndex>()
             .register_job_type::<jobs::UpdateDownloads>()
+            .register_job_type::<jobs::CheckAboutToExpireToken>()
     }
 }