Merge 7ac3053673 into 549ab089ee

app/models/account.rb

@@ -501,6 +501,19 @@ class Account < ApplicationRecord
     save!
   end
 
+  def private_key
+    if account_secret
+      account_secret.private_key
+    else
+      super
+    end
+  end
+
+  def private_key=(value)
+    self.account_secret ||= AccountSecret.new
+    account_secret.private_key = value
+  end
+
   private
 
   def prepare_contents

app/models/account_secret.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+# == Schema Information
+#
+# Table name: account_secrets
+#
+#  id          :bigint(8)        not null, primary key
+#  private_key :text
+#  account_id  :bigint(8)        not null
+#  created_at  :datetime         not null
+#  updated_at  :datetime         not null
+#
+class AccountSecret < ApplicationRecord
+  belongs_to :account
+
+  encrypts :private_key
+end

app/models/concerns/account/associations.rb

@@ -77,5 +77,8 @@ module Account::Associations
 
     # Imports
     has_many :bulk_imports, inverse_of: :account, dependent: :delete_all
+
+    # Secrets
+    has_one :account_secret, inverse_of: :account, dependent: :destroy, autosave: true
   end
 end

db/migrate/20240726143215_create_account_secrets.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+class CreateAccountSecrets < ActiveRecord::Migration[7.1]
+  def change
+    create_table :account_secrets do |t|
+      t.text :private_key
+      t.references :account, null: false, foreign_key: { on_delete: :cascade }
+
+      t.timestamps
+    end
+  end
+end

db/schema.rb

@@ -10,7 +10,7 @@
 #
 # It's strongly recommended that you check this file into your version control system.
 
-ActiveRecord::Schema[7.1].define(version: 2024_07_24_181224) do
+ActiveRecord::Schema[7.1].define(version: 2024_07_26_143215) do
   # These are extensions that must be enabled in order to support this database
   enable_extension "plpgsql"
 
@@ -103,6 +103,14 @@ ActiveRecord::Schema[7.1].define(version: 2024_07_24_181224) do
     t.index ["relationship_severance_event_id"], name: "idx_on_relationship_severance_event_id_403f53e707"
   end
 
+  create_table "account_secrets", force: :cascade do |t|
+    t.text "private_key"
+    t.bigint "account_id", null: false
+    t.datetime "created_at", null: false
+    t.datetime "updated_at", null: false
+    t.index ["account_id"], name: "index_account_secrets_on_account_id"
+  end
+
   create_table "account_stats", force: :cascade do |t|
     t.bigint "account_id", null: false
     t.bigint "statuses_count", default: 0, null: false
@@ -1274,6 +1282,7 @@ ActiveRecord::Schema[7.1].define(version: 2024_07_24_181224) do
   add_foreign_key "account_pins", "accounts", on_delete: :cascade
   add_foreign_key "account_relationship_severance_events", "accounts", on_delete: :cascade
   add_foreign_key "account_relationship_severance_events", "relationship_severance_events", on_delete: :cascade
+  add_foreign_key "account_secrets", "accounts", on_delete: :cascade
   add_foreign_key "account_stats", "accounts", on_delete: :cascade
   add_foreign_key "account_statuses_cleanup_policies", "accounts", on_delete: :cascade
   add_foreign_key "account_warnings", "accounts", column: "target_account_id", on_delete: :cascade

spec/models/account_spec.rb

@@ -1067,4 +1067,15 @@ RSpec.describe Account do
       expect(subject.reload.followers_count).to eq 15
     end
   end
+
+  describe 'private key' do
+    it 'encrypts and decrypts the key' do
+      account = Fabricate(:account)
+
+      account.private_key = 'secret'
+      account.save!
+
+      expect(account.reload.private_key).to eq 'secret'
+    end
+  end
 end

spec/models/concerns/account/counters_spec.rb

@@ -46,16 +46,14 @@ describe Account::Counters do
     end
 
     it 'preserves last_status_at when decrementing statuses_count' do
-      account_stat = Fabricate(
-        :account_stat,
-        account: account,
-        last_status_at: 3.days.ago,
-        statuses_count: 10
-      )
+      account.statuses_count = 10
+      account.save!
+      account.account_stat.last_status_at = 3.days.ago
+      account.account_stat.save!
 
       expect { account.decrement_count!(:statuses_count) }
-        .to change(account_stat.reload, :statuses_count).by(-1)
-        .and not_change(account_stat.reload, :last_status_at)
+        .to change(account.account_stat.reload, :statuses_count).by(-1)
+        .and not_change(account.account_stat.reload, :last_status_at)
     end
   end
 end

spec/requests/api/v1/directories_spec.rb

@@ -17,7 +17,7 @@ describe 'Directories API' do
           user: Fabricate(:user, confirmed_at: nil, approved: true),
           username: 'local_unconfirmed'
         )
-        local_unconfirmed_account.create_account_stat!
+        local_unconfirmed_account.account_stat.save!
 
         local_unapproved_account = Fabricate(
           :account,
@@ -25,7 +25,7 @@ describe 'Directories API' do
           user: Fabricate(:user, confirmed_at: 10.days.ago),
           username: 'local_unapproved'
         )
-        local_unapproved_account.create_account_stat!
+        local_unapproved_account.account_stat.save!
         local_unapproved_account.user.update(approved: false)
 
         local_undiscoverable_account = Fabricate(
@@ -35,7 +35,7 @@ describe 'Directories API' do
           discoverable: false,
           username: 'local_undiscoverable'
         )
-        local_undiscoverable_account.create_account_stat!
+        local_undiscoverable_account.account_stat.save!
 
         excluded_from_timeline_account = Fabricate(
           :account,
@@ -43,7 +43,7 @@ describe 'Directories API' do
           discoverable: true,
           username: 'remote_excluded_from_timeline'
         )
-        excluded_from_timeline_account.create_account_stat!
+        excluded_from_timeline_account.account_stat.save!
         Fabricate(:block, account: user.account, target_account: excluded_from_timeline_account)
 
         domain_blocked_account = Fabricate(
@@ -52,11 +52,11 @@ describe 'Directories API' do
           discoverable: true,
           username: 'remote_domain_blocked'
         )
-        domain_blocked_account.create_account_stat!
+        domain_blocked_account.account_stat.save!
         Fabricate(:account_domain_block, account: user.account, domain: 'test.example')
 
-        local_discoverable_account.create_account_stat!
-        eligible_remote_account.create_account_stat!
+        local_discoverable_account.account_stat.save!
+        eligible_remote_account.account_stat.save!
       end
 
       let(:local_discoverable_account) do
@@ -93,8 +93,8 @@ describe 'Directories API' do
       let(:remote_account) { Fabricate(:account, domain: 'host.example') }
 
       before do
-        local_account.create_account_stat!
-        remote_account.create_account_stat!
+        local_account.account_stat.save!
+        remote_account.account_stat.save!
       end
 
       it 'returns only the local accounts' do