From c53d9d3c600af6a710b38f7bf6282d2e3d339250 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?R=C3=A2u=20Cao?= <raucao@kosmos.org>
Date: Sat, 17 May 2025 16:44:28 +0400
Subject: [PATCH] WIP Sync Mastodon IDs/profiles to local accounts

---
 app/services/mastodon_manager/fetch_user.rb   | 12 ++++
 app/services/mastodon_manager/find_user.rb    | 14 +++++
 .../mastodon_manager/sync_account_profiles.rb | 57 +++++++++++++++++++
 .../user_manager/import_remote_avatar.rb      | 12 ++++
 4 files changed, 95 insertions(+)
 create mode 100644 app/services/mastodon_manager/fetch_user.rb
 create mode 100644 app/services/mastodon_manager/find_user.rb
 create mode 100644 app/services/mastodon_manager/sync_account_profiles.rb
 create mode 100644 app/services/user_manager/import_remote_avatar.rb

diff --git a/app/services/mastodon_manager/fetch_user.rb b/app/services/mastodon_manager/fetch_user.rb
new file mode 100644
index 0000000..3d7d3d0
--- /dev/null
+++ b/app/services/mastodon_manager/fetch_user.rb
@@ -0,0 +1,12 @@
+module MastodonManager
+  class FetchUser < MastodonManagerService
+    def initialize(mastodon_id:)
+      @mastodon_id = mastodon_id
+    end
+
+    def call
+      user = get "v1/admin/accounts/#{@mastodon_id}"
+      user.with_indifferent_access
+    end
+  end
+end
diff --git a/app/services/mastodon_manager/find_user.rb b/app/services/mastodon_manager/find_user.rb
new file mode 100644
index 0000000..d196c6b
--- /dev/null
+++ b/app/services/mastodon_manager/find_user.rb
@@ -0,0 +1,14 @@
+module MastodonManager
+  class FindUser < MastodonManagerService
+    def initialize(username:)
+      @username = username
+    end
+
+    def call
+      users = get "v2/admin/accounts?username=#{@username}&origin=local"
+      users = users.map { |u| u.with_indifferent_access }
+      # Results may contain partial matches
+      users.find { |u| u.dig(:username) == @username }
+    end
+  end
+end
diff --git a/app/services/mastodon_manager/sync_account_profiles.rb b/app/services/mastodon_manager/sync_account_profiles.rb
new file mode 100644
index 0000000..71b08cf
--- /dev/null
+++ b/app/services/mastodon_manager/sync_account_profiles.rb
@@ -0,0 +1,57 @@
+module MastodonManager
+  class SyncAccountProfiles < MastodonManagerService
+    def initialize(direction: "down", overwrite: false)
+      @direction = direction
+      @overwrite = overwrite
+
+      if @direction != "down"
+        raise NotImplementedError
+      end
+    end
+
+    def call
+      Rails.logger.debug { "Syncing account profiles (direction: #{@direction}, overwrite: #{@overwrite})"}
+
+      User.find_each do |user|
+        if user.mastodon_id.blank?
+          mastodon_user = MastodonManager::FindUser.call username: user.cn
+          if mastodon_user
+            Rails.logger.debug { "Setting mastodon_id for user #{user.cn}" }
+            user.update! mastodon_id: @user[:id]
+          else
+            Rails.logger.debug { "No Mastodon user found for username #{user.cn}" }
+            next
+          end
+        end
+
+        next if user.avatar.attached? && user.display_name.present?
+
+        unless mastodon_user
+          Rails.logger.debug { "Fetching Mastodon account with ID #{user.mastodon_id} for #{user.cn}" }
+          mastodon_user = MastodonManager::FetchUser.call mastodon_id: user.mastodon_id
+        end
+
+        if user.display_name.blank?
+          if mastodon_display_name = mastodon_user.dig(:account, :display_name)
+            Rails.logger.debug { "Setting display name for user #{user.cn} from Mastodon" }
+            LdapManager::UpdateDisplayName.call(
+              dn: user.dn, display_name: mastodon_display_name
+            )
+          end
+        end
+
+        if !user.avatar.attached?
+          if avatar_url = mastodon_user.dig(:account, :avatar_static)
+            Rails.logger.debug { "Importing Mastodon avatar for user #{user.cn}" }
+            UserManager::ImportRemoteAvatar.call(
+              user: user, avatar_url: avatar_url
+            )
+          end
+        end
+      rescue => e
+        Sentry.capture_exception(e) if Setting.sentry_enabled?
+        Rails.logger.error e
+      end
+    end
+  end
+end
diff --git a/app/services/user_manager/import_remote_avatar.rb b/app/services/user_manager/import_remote_avatar.rb
new file mode 100644
index 0000000..26ff98b
--- /dev/null
+++ b/app/services/user_manager/import_remote_avatar.rb
@@ -0,0 +1,12 @@
+module UserManager
+  class ImportRemoteAvatar < UserManagerService
+    def initialize(user:, avatar_url:)
+      @user = user
+      @avatar_url = avatar_url
+    end
+
+    def call
+
+    end
+  end
+end