From 5b02150e12b00f6e1a99b2c848884965a58e9f86 Mon Sep 17 00:00:00 2001
From: Ariel Kirkwood <ariel.kirkwood@me.com>
Date: Sun, 5 May 2024 18:35:36 -0400
Subject: [PATCH] Display fund's type and sector

---
 app/models/fund.rb                        |  2 +-
 app/views/funds/_fund.html.erb            |  9 +++++++--
 lib/holdings/extraction_strategies/tcw.rb | 11 +++++++++--
 3 files changed, 17 insertions(+), 5 deletions(-)

diff --git a/app/models/fund.rb b/app/models/fund.rb
index 4c550e8..aaad4e4 100644
--- a/app/models/fund.rb
+++ b/app/models/fund.rb
@@ -14,7 +14,7 @@ class Fund < ApplicationRecord
   has_many :assets, through: :holdings
 
   delegate :extract_holdings, to: :holdings_extractor
-  delegate :name, to: :underlying_asset
+  delegate :name, :type, :sector, to: :underlying_asset
 
   validates :underlying_asset_id, uniqueness: true
 
diff --git a/app/views/funds/_fund.html.erb b/app/views/funds/_fund.html.erb
index 241d80a..cb5a2a4 100644
--- a/app/views/funds/_fund.html.erb
+++ b/app/views/funds/_fund.html.erb
@@ -1,6 +1,13 @@
 <div class="h3"><%= render @ticker %></div>
 <%= tag.h1 fund.name %>
 
+<div class="h6 mb-3">
+  <span class="badge text-bg-primary"><%= tag.strong fund.type.underscore.humanize %></span>
+  <span class="badge text-bg-primary"><%= tag.strong fund.sector %></span>
+</div>
+
+<hr>
+
 <%= tag.blockquote class: 'blockquote' do %>
   <figcaption class="blockquote-footer mt-0 mb-3">
     <cite><strong>Betterment</strong></cite> says about this fund:
@@ -8,6 +15,4 @@
   <%= fund.betterment_detail %>
 <% end %>
 
-<hr>
-
 <%= render fund.latest_portfolio %>
diff --git a/lib/holdings/extraction_strategies/tcw.rb b/lib/holdings/extraction_strategies/tcw.rb
index e2f80c9..b4a5353 100644
--- a/lib/holdings/extraction_strategies/tcw.rb
+++ b/lib/holdings/extraction_strategies/tcw.rb
@@ -25,9 +25,16 @@ def date
         @date ||= Date.parse(worksheet[1][1].value)
       end
 
-      def extract_holdings # rubocop:disable Metrics/AbcSize, Metrics/CyclomaticComplexity, Metrics/MethodLength
+      def extract_holdings # rubocop:disable Metrics/AbcSize, Metrics/CyclomaticComplexity, Metrics/MethodLength, Metrics/PerceivedComplexity
         working_sheet.map do |row|
-          asset = Asset.find_or_create_by!(name: row[:description], asset_class: asset_type(row[:security_type]))
+          asset = if (ticker = Assets::Ticker.where(ticker: row[:ticker]).first)
+                    ticker.asset
+                  elsif (cusip = Assets::CUSIP.where(cusip: row[:cusip]).first)
+                    cusip.asset
+                  elsif (sedol = Assets::SEDOL.where(sedol: row[:sedol]).first)
+                    sedol.asset
+                  end
+          asset = Asset.find_or_create_by!(name: row[:description], asset_class: asset_type(row[:security_type])) if asset.blank?
 
           Assets::Ticker.find_or_create_by(asset:, ticker: row[:ticker]) if row[:ticker] != 'n/a'
           Assets::Currency.find_or_create_by(asset:, currency: row[:asset_currency]) if asset_type(row[:security_type]) == 'CashEquivalent' && row[:asset_currency].present?