feat: update cell outputs status

lib/tasks/migration/update_output_cells_status.rake

@@ -0,0 +1,76 @@
+class UpdateCellOutputsStatus
+  include Rake::DSL
+
+  def initialize
+    @address_ids = []
+
+    namespace :migration do
+      desc "Usage: RAILS_ENV=production bundle exec rake migration:update_output_cells_status"
+      task update_output_cells_status: :environment do
+        ApplicationRecord.transaction do
+          outputs = CellOutput.pending.includes(:ckb_transaction).where(ckb_transaction: { tx_status: "committed" })
+          outputs.each do |output|
+            output.live!
+            update_udt_account(output)
+            @address_ids << output.address_id
+          end
+        end
+
+        update_addresses_live_cells_count
+      end
+    end
+  end
+
+  private
+
+  def update_addresses_live_cells_count
+    Address.where(id: @address_ids.uniq).find_each do |address|
+      address.live_cells_count = address.cell_outputs.live.count
+      address.cal_balance!
+      address.save!
+    end
+  end
+
+  def update_udt_account(udt_output)
+    return unless udt_output.cell_type.in?(%w(udt m_nft_token nrc_721_token))
+
+    address = Address.find(udt_output.address_id)
+    udt_type = udt_type(udt_output.cell_type)
+    udt_account = address.udt_accounts.where(type_hash: udt_output.type_hash, udt_type: udt_type).select(:id,
+                                                                                                         :created_at).first
+    amount = udt_account_amount(udt_type, udt_output.type_hash, address)
+
+    if udt_account.present?
+      udt_account.update(amount: amount)
+    else
+      udt = Udt.where(type_hash: udt_output.type_hash, udt_type: udt_type).select(:id, :udt_type, :full_name,
+                                                                                  :symbol, :decimal, :published, :code_hash, :type_hash, :created_at).take!
+      nft_token_id =
+        udt_type == "nrc_721_token" ? CkbUtils.parse_nrc_721_args(udt_output.type_script.args).token_id : nil
+      new_udt_accounts_attribute = {
+        address_id: udt_output.address_id, udt_type: udt.udt_type, full_name: udt.full_name, symbol: udt.symbol, decimal: udt.decimal,
+        published: udt.published, code_hash: udt.code_hash, type_hash: udt.type_hash, amount: amount, udt_id: udt.id, nft_token_id: nft_token_id,
+        created_at: Time.current, updated_at: Time.current
+      }
+
+      UdtAccount.insert!(new_udt_accounts_attribute)
+    end
+  end
+
+  def udt_type(cell_type)
+    cell_type == "udt" ? "sudt" : cell_type
+  end
+
+  def udt_account_amount(udt_type, type_hash, address)
+    case udt_type
+    when "sudt"
+      address.cell_outputs.live.udt.where(type_hash: type_hash).sum(:udt_amount)
+    when "m_nft_token"
+      address.cell_outputs.live.m_nft_token.where(type_hash: type_hash).sum(:udt_amount)
+    else
+      0
+    end
+  end
+end
+
+UpdateCellOutputsStatus.new