Add plugin hooks for init and scheduled tasks (fixes #6488, fixes #6489) (#6493)

* Add plugin hooks for init and scheduled tasks (fixes #6488, fixes #6489)

* fix test, add init

* clippy

* 1 min hook

Author: Nutomic

crates/api/api_utils/src/plugins.rs

@@ -187,7 +187,7 @@ pub fn plugin_metadata() -> Vec<PluginMetadata> {
 }
 
 #[derive(Clone)]
-struct LemmyPlugins {
+pub struct LemmyPlugins {
   plugins: Vec<LemmyPlugin>,
   captcha_plugin: Option<LemmyPlugin>,
 }
@@ -240,6 +240,11 @@ impl LemmyPlugin {
       .insert("lemmy_version".to_string(), VERSION.to_string());
     let builder = move || PluginBuilder::new(manifest.clone()).with_wasi(true).build();
     let pool = Pool::new(builder);
+    if let Some(mut p) = pool.get(GET_PLUGIN_TIMEOUT)?
+      && p.function_exists("init")
+    {
+      p.call::<_, ()>("init", ())?;
+    }
     Ok(LemmyPlugin {
       pool,
       filename: filename.unwrap_or(settings.file),
@@ -263,7 +268,7 @@ impl LemmyPlugin {
 
 impl LemmyPlugins {
   /// Load and initialize all plugins
-  fn get_or_init() -> Self {
+  pub fn get_or_init() -> Self {
     static PLUGINS: LazyLock<LemmyPlugins> = LazyLock::new(|| {
       let mut plugins: Vec<_> = SETTINGS
         .plugins

crates/routes/src/utils/scheduled_tasks.rs

@@ -18,6 +18,7 @@ use diesel_async::{AsyncPgConnection, RunQueryDsl};
 use diesel_uplete::uplete;
 use lemmy_api_utils::{
   context::LemmyContext,
+  plugins::plugin_hook_after,
   send_activity::{ActivityChannel, SendActivityData},
   utils::send_webmention,
 };
@@ -64,6 +65,11 @@ pub async fn setup(context: Data<LemmyContext>) -> LemmyResult<()> {
   // https://github.com/mdsherry/clokwerk/issues/38
   let mut scheduler = AsyncScheduler::with_tz(Utc);
 
+  // Every 1 minute run plugin hooks
+  scheduler.every(CTimeUnits::minutes(1)).run(async move || {
+    plugin_hook_after("scheduled_task_1_min", &());
+  });
+
   let context_1 = context.clone();
   // Every 10 minutes update hot ranks, delete expired captchas and publish scheduled posts
   scheduler.every(CTimeUnits::minutes(10)).run(move || {
@@ -78,6 +84,7 @@ pub async fn setup(context: Data<LemmyContext>) -> LemmyResult<()> {
         .await
         .inspect_err(|e| warn!("Failed to publish scheduled posts: {e}"))
         .ok();
+      plugin_hook_after("scheduled_task_10_mins", &());
     }
   });
 
@@ -102,6 +109,7 @@ pub async fn setup(context: Data<LemmyContext>) -> LemmyResult<()> {
         .await
         .inspect_err(|e| warn!("Failed to delete expired instance bans: {e}"))
         .ok();
+      plugin_hook_after("scheduled_task_1_hour", &());
     }
   });
 
@@ -141,6 +149,7 @@ pub async fn setup(context: Data<LemmyContext>) -> LemmyResult<()> {
         .await
         .inspect_err(|e| warn!("Failed to clear old activities: {e}"))
         .ok();
+      plugin_hook_after("scheduled_task_daily", &());
     }
   });
 

crates/server/src/lib.rs

@@ -11,6 +11,7 @@ use clap::{Parser, Subcommand};
 use lemmy_api::sitemap::get_sitemap;
 use lemmy_api_utils::{
   context::LemmyContext,
+  plugins::LemmyPlugins,
   request::client_builder,
   send_activity::ActivityChannel,
   utils::local_site_rate_limit_to_rate_limit_config,
@@ -175,6 +176,9 @@ pub async fn start_lemmy_server(args: CmdArgs) -> LemmyResult<()> {
     startup_server_handle = Some(create_startup_server()?);
   }
 
+  // Initialize plugins in background
+  tokio::task::spawn_blocking(LemmyPlugins::get_or_init);
+
   // Set up the connection pool
   let pool = build_db_pool()?;