From 5d58ef40a90df22922c568ef5ad8be12f710842b Mon Sep 17 00:00:00 2001
From: alemi <me@alemi.dev>
Date: Thu, 2 May 2024 02:59:26 +0200
Subject: [PATCH] feat: added users update task

---
 src/cli/mod.rs    |  3 +++
 src/cli/update.rs | 38 ++++++++++++++++++++++++++++++++++++++
 src/main.rs       | 14 ++++++++++++--
 3 files changed, 53 insertions(+), 2 deletions(-)
 create mode 100644 src/cli/update.rs

diff --git a/src/cli/mod.rs b/src/cli/mod.rs
index ef23ff0e..e107431e 100644
--- a/src/cli/mod.rs
+++ b/src/cli/mod.rs
@@ -12,3 +12,6 @@ pub use relay::*;
 
 mod register;
 pub use register::*;
+
+mod update;
+pub use update::*;
diff --git a/src/cli/update.rs b/src/cli/update.rs
new file mode 100644
index 00000000..05afda65
--- /dev/null
+++ b/src/cli/update.rs
@@ -0,0 +1,38 @@
+use futures::TryStreamExt;
+use sea_orm::{ColumnTrait, EntityTrait, IntoActiveModel, QueryFilter};
+
+use crate::server::{fetcher::Fetcher, Context};
+
+pub async fn update_users(db: sea_orm::DatabaseConnection, domain: String, days: i64) -> crate::Result<()> {
+	let ctx = Context::new(db, domain).await?;
+	let mut count = 0;
+	let mut insertions = Vec::new();
+
+	{
+		let mut stream = crate::model::user::Entity::find()
+			.filter(crate::model::user::Column::Updated.lt(chrono::Utc::now() - chrono::Duration::days(days)))
+			.stream(ctx.db())
+			.await?;
+
+
+		while let Some(user) = stream.try_next().await? {
+			match ctx.pull_user(&user.id).await {
+				Err(e) => tracing::warn!("could not update user {}: {e}", user.id),
+				Ok(u) => {
+					insertions.push(u);
+					count += 1;
+				},
+			}
+		}
+	}
+
+	for u in insertions {
+		tracing::info!("updating user {}", u.id);
+		crate::model::user::Entity::delete_by_id(&u.id).exec(ctx.db()).await?;
+		crate::model::user::Entity::insert(u.into_active_model()).exec(ctx.db()).await?;
+	}
+
+	tracing::info!("updated {count} users");
+
+	Ok(())
+}
diff --git a/src/main.rs b/src/main.rs
index d26b318a..9f80fe68 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -88,6 +88,13 @@ enum CliCommand {
 		#[arg(long, default_value_t = false)]
 		/// fix replies counts for posts
 		replies: bool,
+	},
+
+	/// update remote users
+	Update {
+		#[arg(long, short, default_value_t = 7)]
+		/// number of days after which users should get updated
+		days: i64,
 	}
 }
 
@@ -131,8 +138,11 @@ async fn main() {
 
 		CliCommand::Fix { likes, shares, replies } =>
 			cli::fix(db, likes, shares, replies)
-				.await
-				.expect("failed running fix task"),
+				.await.expect("failed running fix task"),
+
+		CliCommand::Update { days } =>
+			cli::update_users(db, args.domain, days)
+				.await.expect("error updating users"),
 
 		CliCommand::Serve => {
 			let ctx = server::Context::new(db, args.domain)