view: optimize view for different message type

Author: ramsayleung

app/controllers/trained_messages_controller.rb

@@ -118,9 +118,10 @@ def update
   # DELETE /trained_messages/1 or /trained_messages/1.json
   def destroy
     @trained_message.destroy!
+    return_url = params[:return_url] || trained_messages_path
 
     respond_to do |format|
-      format.html { redirect_to trained_messages_path, notice: "Trained message was successfully destroyed.", status: :see_other }
+      format.html { redirect_to return_url, notice: "Trained message was successfully destroyed.", status: :see_other }
       format.json { head :no_content }
     end
   end

app/helpers/trained_messages_helper.rb

@@ -1,2 +1,18 @@
 module TrainedMessagesHelper
+  def message_type_classes(message_type)
+    case message_type
+    when "spam"
+      "bg-red-100 text-red-800"
+    when "ham"
+      "bg-green-100 text-green-800"
+    when "maybe_spam"
+      "bg-orange-100 text-orange-800"
+    when "maybe_ham"
+      "bg-lime-100 text-lime-800"
+    when "untrained"
+      "bg-gray-100 text-gray-800"
+    else
+      "bg-gray-100 text-gray-800"
+    end
+  end
 end

app/views/trained_messages/index.html.erb

@@ -178,8 +178,7 @@
                       </span>
                     <% end %>
                   </td>
-                  <td class="px-6 py-4 whitespace-nowrap text-sm text-gray-500
-                             <%= trained_message.message_type == 'spam' ? 'bg-red-100 text-red-800' : trained_message.message_type == 'ham' ? 'bg-green-100 text-green-800' :  'bg-yellow-100 text-green-800'%>">
+                  <td class="px-6 py-4 whitespace-nowrap text-sm text-gray-500 <%= message_type_classes(trained_message.message_type) %>">
                     <%= trained_message.message_type&.humanize || 'Unknown' %>
                   </td>
                   <td class="px-6 py-4 whitespace-nowrap text-sm text-gray-500">

app/views/trained_messages/show.html.erb

@@ -11,5 +11,8 @@
 
   <%= link_to "Edit this trained message", edit_trained_message_path(@trained_message), class: "w-full sm:w-auto text-center rounded-md px-3.5 py-2.5 bg-gray-100 hover:bg-gray-50 inline-block font-medium" %>
   <%= link_to "Back to trained messages", trained_messages_path(index_params), class: "w-full sm:w-auto text-center mt-2 sm:mt-0 sm:ml-2 rounded-md px-3.5 py-2.5 bg-gray-100 hover:bg-gray-50 inline-block font-medium" %>
-  <%= button_to "Destroy this trained message", @trained_message, method: :delete, form_class: "sm:inline-block mt-2 sm:mt-0 sm:ml-2", class: "w-full rounded-md px-3.5 py-2.5 text-white bg-red-600 hover:bg-red-500 font-medium cursor-pointer", data: { turbo_confirm: "Are you sure?" } %>
+  <%
+  return_url = trained_messages_path(index_params)
+  %>
+  <%= button_to "Destroy this trained message", @trained_message, method: :delete, form_class: "sm:inline-block mt-2 sm:mt-0 sm:ml-2", class: "w-full rounded-md px-3.5 py-2.5 text-white bg-red-600 hover:bg-red-500 font-medium cursor-pointer", data: { turbo_confirm: "Are you sure?" }, params: { return_url: return_url } %>
 </div>