Feature: Add the ability to "revert" a CSV import (#1814)

* Allow reverting imports

* Fix tests

* Add currency column to all imports

* Don't auto-enrich demo account

Author: zachgoll

app/controllers/imports_controller.rb

@@ -1,5 +1,5 @@
 class ImportsController < ApplicationController
-  before_action :set_import, only: %i[show publish destroy]
+  before_action :set_import, only: %i[show publish destroy revert]
 
   def publish
     @import.publish_later
@@ -31,6 +31,11 @@ def show
     end
   end
 
+  def revert
+    @import.revert_later
+    redirect_to imports_path, notice: "Import is reverting in the background."
+  end
+
   def destroy
     @import.destroy
 

app/jobs/revert_import_job.rb

@@ -0,0 +1,7 @@
+class RevertImportJob < ApplicationJob
+  queue_as :latency_low
+
+  def perform(import)
+    import.revert
+  end
+end

app/models/demo/generator.rb

@@ -24,7 +24,7 @@ def reset_data!(family_names)
     puts "Data cleared"
 
     family_names.each_with_index do |family_name, index|
-      create_family_and_user!(family_name, "user#{index == 0 ? "" : index + 1}@maybe.local", data_enrichment_enabled: index == 0)
+      create_family_and_user!(family_name, "user#{index == 0 ? "" : index + 1}@maybe.local")
     end
 
     puts "Users reset"

app/models/import.rb

@@ -6,7 +6,14 @@ class Import < ApplicationRecord
 
   scope :ordered, -> { order(created_at: :desc) }
 
-  enum :status, { pending: "pending", complete: "complete", importing: "importing", failed: "failed" }, validate: true
+  enum :status, {
+    pending: "pending",
+    complete: "complete",
+    importing: "importing",
+    reverting: "reverting",
+    revert_failed: "revert_failed",
+    failed: "failed"
+  }, validate: true, default: "pending"
 
   validates :type, inclusion: { in: TYPES }
   validates :col_sep, inclusion: { in: [ ",", ";" ] }
@@ -35,6 +42,27 @@ def publish
     update! status: :failed, error: error.message
   end
 
+  def revert_later
+    raise "Import is not revertable" unless revertable?
+
+    update! status: :reverting
+
+    RevertImportJob.perform_later(self)
+  end
+
+  def revert
+    Import.transaction do
+      accounts.destroy_all
+      entries.destroy_all
+    end
+
+    family.sync
+
+    update! status: :pending
+  rescue => error
+    update! status: :revert_failed, error: error.message
+  end
+
   def csv_rows
     @csv_rows ||= parsed_csv
   end
@@ -113,6 +141,10 @@ def publishable?
     cleaned? && mappings.all?(&:valid?)
   end
 
+  def revertable?
+    complete? || revert_failed?
+  end
+
   def has_unassigned_account?
     mappings.accounts.where(key: "").any?
   end

app/views/import/configurations/_account_import.html.erb

@@ -4,6 +4,7 @@
   <%= form.select :entity_type_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Entity Type" } %>
   <%= form.select :name_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Name (optional)" } %>
   <%= form.select :amount_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Balance" } %>
+  <%= form.select :currency_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Currency" } %>
 
   <%= form.submit "Apply configuration", class: "w-full btn btn--primary", disabled: import.complete? %>
 <% end %>

app/views/import/configurations/_mint_import.html.erb

@@ -16,6 +16,8 @@
     <%= form.select :signage_convention, [["Incomes are negative", "inflows_negative"], ["Incomes are positive", "inflows_positive"]], { label: true }, disabled: import.complete? %>
   </div>
 
+  <%= form.select :currency_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Currency" }, disabled: import.complete? %>
+
   <%= form.select :account_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Account (optional)" }, disabled: import.complete? %>
   <%= form.select :name_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Name (optional)" }, disabled: import.complete? %>
   <%= form.select :category_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Category (optional)" }, disabled: import.complete? %>

app/views/import/configurations/_trade_import.html.erb

@@ -11,6 +11,8 @@
     <%= form.select :signage_convention, [["Buys are positive qty", "inflows_positive"], ["Buys are negative qty", "inflows_negative"]], label: true %>
   </div>
 
+  <%= form.select :currency_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Currency" } %>
+
   <%= form.select :ticker_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Ticker" } %>
   <%= form.select :price_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Price" } %>
   <%= form.select :account_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Account (optional)" } %>

app/views/import/configurations/_transaction_import.html.erb

@@ -11,6 +11,8 @@
     <%= form.select :signage_convention, [["Incomes are positive", "inflows_positive"], ["Incomes are negative", "inflows_negative"]], label: true %>
   </div>
 
+  <%= form.select :currency_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Currency" } %>
+
   <%= form.select :account_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Account (optional)" } %>
   <%= form.select :name_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Name (optional)" } %>
   <%= form.select :category_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Category (optional)" } %>

app/views/imports/_import.html.erb

@@ -17,6 +17,14 @@
       <span class="px-1 py text-xs rounded-full bg-red-500/5 text-red-500 border border-alpha-black-50">
         <%= t(".failed") %>
       </span>
+    <% elsif import.reverting? %>
+      <span class="px-1 py text-xs rounded-full bg-orange-500/5 text-orange-500 border border-alpha-black-50">
+        <%= t(".reverting") %>
+      </span>
+    <% elsif import.revert_failed? %>
+      <span class="px-1 py text-xs rounded-full bg-red-500/5 text-red-500 border border-alpha-black-50">
+        <%= t(".revert_failed") %>
+      </span>
     <% elsif import.complete? %>
       <span class="px-1 py text-xs rounded-full bg-green-500/5 text-green-500 border border-alpha-black-50">
         <%= t(".complete") %>
@@ -33,7 +41,16 @@
         <span><%= t(".view") %></span>
       <% end %>
 
-      <% unless import.complete? %>
+      <% if import.complete? || import.revert_failed? %>
+        <%= button_to revert_import_path(import), 
+                      method: :put,
+                      class: "block w-full py-2 px-3 space-x-2 text-orange-600 hover:bg-orange-50 flex items-center rounded-lg",
+                      data: { turbo_confirm: true } do %>
+          <%= lucide_icon "rotate-ccw", class: "w-5 h-5" %>
+
+          <span>Revert</span>
+        <% end %>
+      <% else %>
         <%= button_to import_path(import),
                       method: :delete,
                       class: "block w-full py-2 px-3 space-x-2 text-red-600 hover:bg-red-50 flex items-center rounded-lg",

app/views/imports/_revert_failure.html.erb

@@ -0,0 +1,18 @@
+<%# locals: (import:) %>
+
+<div class="h-full flex flex-col justify-center items-center">
+  <div class="space-y-6 max-w-sm">
+    <div class="mx-auto bg-red-500/5 h-8 w-8 rounded-full flex items-center justify-center">
+      <%= lucide_icon "alert-octagon", class: "w-5 h-5 text-red-500" %>
+    </div>
+
+    <div class="text-center space-y-2">
+      <h1 class="font-medium text-gray-900 text-center text-3xl">Reverting import failed</h1>
+      <p class="text-sm text-gray-500">Please try again or contact support.</p>
+    </div>
+
+    <div>
+      <%= button_to "Try again", revert_import_path(import), class: "btn btn--primary text-center w-full" %>
+    </div>
+  </div>
+</div>