mastodon/app/models/list_account.rb

# frozen_string_literal: true

# == Schema Information
#
# Table name: list_accounts
#
#  id                :bigint(8)        not null, primary key
#  list_id           :bigint(8)        not null
#  account_id        :bigint(8)        not null
#  follow_id         :bigint(8)
#  follow_request_id :bigint(8)
#

class ListAccount < ApplicationRecord
  belongs_to :list
  belongs_to :account
  belongs_to :follow, optional: true
  belongs_to :follow_request, optional: true

  validates :account_id, uniqueness: { scope: :list_id }
  validate :validate_relationship

  before_validation :set_follow, unless: :list_owner_account_is_account?

  private

  def set_follow
    self.follow = Follow.find_by(account_id: list.account_id, target_account_id: account.id)
    self.follow_request = FollowRequest.find_by(account_id: list.account_id, target_account_id: account.id) if follow.nil?
  end

  def validate_relationship
    return if list_owner_account_is_account?

    errors.add(:account_id, :must_be_following) if follow_id.nil? && follow_request_id.nil?
    errors.add(:follow, :invalid) if follow_id.present? && follow.target_account_id != account_id
    errors.add(:follow_request, :invalid) if follow_request_id.present? && follow_request.target_account_id != account_id
  end

  def list_owner_account_is_account?
    list.account_id == account_id
  end
end