class Account < ApplicationRecord include Syncable, Monetizable, Issuable validates :name, :balance, :currency, presence: true belongs_to :family belongs_to :import, optional: true belongs_to :plaid_account, optional: true has_many :import_mappings, as: :mappable, dependent: :destroy, class_name: "Import::Mapping" has_many :entries, dependent: :destroy, class_name: "Account::Entry" has_many :transactions, through: :entries, source: :entryable, source_type: "Account::Transaction" has_many :valuations, through: :entries, source: :entryable, source_type: "Account::Valuation" has_many :trades, through: :entries, source: :entryable, source_type: "Account::Trade" has_many :holdings, dependent: :destroy, class_name: "Account::Holding" has_many :balances, dependent: :destroy has_many :issues, as: :issuable, dependent: :destroy monetize :balance, :cash_balance enum :classification, { asset: "asset", liability: "liability" }, validate: { allow_nil: true } scope :active, -> { where(is_active: true, scheduled_for_deletion: false) } scope :assets, -> { where(classification: "asset") } scope :liabilities, -> { where(classification: "liability") } scope :alphabetically, -> { order(:name) } scope :manual, -> { where(plaid_account_id: nil) } has_one_attached :logo delegated_type :accountable, types: Accountable::TYPES, dependent: :destroy accepts_nested_attributes_for :accountable, update_only: true class << self def by_group(period: Period.all, currency: Money.default_currency.iso_code) grouped_accounts = { assets: ValueGroup.new("Assets", currency), liabilities: ValueGroup.new("Liabilities", currency) } Accountable.by_classification.each do |classification, types| types.each do |type| accounts = self.where(accountable_type: type) if accounts.any? group = grouped_accounts[classification.to_sym].add_child_group(type, currency) accounts.each do |account| group.add_value_node( account, account.balance_money.exchange_to(currency, fallback_rate: 0), account.series(period: period, currency: currency) ) end end end end grouped_accounts end def create_and_sync(attributes) attributes[:accountable_attributes] ||= {} # Ensure accountable is created, even if empty account = new(attributes.merge(cash_balance: attributes[:balance])) transaction do # Create 2 valuations for new accounts to establish a value history for users to see account.entries.build( name: "Current Balance", date: Date.current, amount: account.balance, currency: account.currency, entryable: Account::Valuation.new ) account.entries.build( name: "Initial Balance", date: 1.day.ago.to_date, amount: 0, currency: account.currency, entryable: Account::Valuation.new ) account.save! end account.sync_later account end end def destroy_later update!(scheduled_for_deletion: true) DestroyJob.perform_later(self) end def sync_data(start_date: nil) update!(last_synced_at: Time.current) Syncer.new(self, start_date: start_date).run end def post_sync broadcast_remove_to(family, target: "syncing-notice") resolve_stale_issues accountable.post_sync end def series(period: Period.last_30_days, currency: nil) balance_series = balances.in_period(period).where(currency: currency || self.currency) if balance_series.empty? && period.date_range.end == Date.current TimeSeries.new([ { date: Date.current, value: balance_money.exchange_to(currency || self.currency) } ]) else TimeSeries.from_collection(balance_series, :balance_money, favorable_direction: asset? ? "up" : "down") end rescue Money::ConversionError TimeSeries.new([]) end def original_balance balance_amount = balances.chronological.first&.balance || balance Money.new(balance_amount, currency) end def current_holdings holdings.where(currency: currency, date: holdings.maximum(:date)).order(amount: :desc) end def favorable_direction classification == "asset" ? "up" : "down" end def enrich_data DataEnricher.new(self).run end def enrich_data_later EnrichDataJob.perform_later(self) end def update_with_sync!(attributes) transaction do update!(attributes) update_balance!(attributes[:balance]) if attributes[:balance] end sync_later end def update_balance!(balance) valuation = entries.account_valuations.find_by(date: Date.current) if valuation valuation.update! amount: balance else entries.create! \ date: Date.current, name: "Balance update", amount: balance, currency: currency, entryable: Account::Valuation.new end end end