execute_order.rb

# frozen_string_literal: true

class ExecuteOrder
  delegate :portfolio, :shares, :stock, to: :order, private: true
  delegate :price_cents, to: :stock, prefix: true, private: true

  def initialize(order)
    @order = order
  end

  def self.execute(...)
    new(...).execute
  end

  def execute
    return unless order.pending?

    unless valid_order?
      cancel_order!
      return
    end

    ActiveRecord::Base.transaction do
      create_portfolio_transaction
      create_portfolio_stock
      update_order_status
    end
  end

  private

  attr_accessor :order, :portfolio_stock, :portfolio_transaction

  def create_portfolio_transaction
    @portfolio_transaction = if order.buy?
                               portfolio
                                 .portfolio_transactions
                                 .debit
                                 .create!(amount_cents: purchase_cost)
                             else
                               portfolio
                                 .portfolio_transactions
                                 .credit
                                 .create!(amount_cents: purchase_cost)
                             end
  end

  def create_portfolio_stock
    share_amount = order.sell? ? -shares : shares
    @portfolio_stock = portfolio
      .portfolio_stocks
      .create!(stock:, shares: share_amount, purchase_price: stock.current_price)
  end

  def update_order_status
    order.update!(status: :completed, portfolio_stock:, portfolio_transaction:)
  end

  def purchase_cost
    order.purchase_cost
  end

  def valid_order?
    return order.sufficient_funds? if order.buy?
    return order.sufficient_shares? if order.sell?

    true
  end

  def cancel_order!
    order.cancel!
  end
end