toil(front): add feature flag for pipeline rebuilds in the UI (#419)

## 📝 Description
Add feature flag to control if we want to display rebuild pipeline
button in the UI

## ✅ Checklist
- [x] I have tested this change
- [ ] This change requires documentation update

Author: skipi

front/lib/front_web/controllers/pipeline_controller.ex

@@ -5,7 +5,14 @@ defmodule FrontWeb.PipelineController do
   alias Front.Models.Pipeline
   alias Front.Models.Switch
   alias Front.WorkflowPage.PipelineStatus
-  alias FrontWeb.Plugs.{FetchPermissions, PageAccess, PublicPageAccess, PutProjectAssigns}
+
+  alias FrontWeb.Plugs.{
+    FeatureEnabled,
+    FetchPermissions,
+    PageAccess,
+    PublicPageAccess,
+    PutProjectAssigns
+  }
 
   require Logger
 
@@ -24,6 +31,11 @@ defmodule FrontWeb.PipelineController do
   plug(:assign_pipeline_without_blocks when action in [:status, :switch, :stop, :rebuild])
   plug(:preload_switch when action in [:show, :poll, :switch])
 
+  plug(
+    FeatureEnabled,
+    [:ui_partial_ppl_rebuild] when action in [:rebuild]
+  )
+
   def path(conn, params) do
     organization_id = conn.assigns.organization_id
     workflow_id = conn.assigns.workflow.id

front/lib/front_web/templates/workflow/status/_interactive_pipeline.html.eex

@@ -21,7 +21,7 @@
       <span class="gray mh2">&middot;</span>
       <%= link "Stop Pipeline", to: pipeline_stop_path(@conn, :stop, @workflow.id, @pipeline.id), class: "btn btn-secondary btn-tiny", pipeline_stop_button: "true" %>
     <% end %>
-    <%= if @conn.assigns.permissions["project.job.rerun"] && FrontWeb.PipelineView.pipeline_rebuildable?(@pipeline) && !FrontWeb.PipelineView.anonymous?(@conn) do %>
+    <%= if FeatureProvider.feature_enabled?(:ui_partial_ppl_rebuild, param: @conn.assigns[:organization_id]) && @conn.assigns.permissions["project.job.rerun"] && FrontWeb.PipelineView.pipeline_rebuildable?(@pipeline) && !FrontWeb.PipelineView.anonymous?(@conn) do %>
       <span class="gray mh2">&middot;</span>
       <%= link "Rebuild Pipeline", to: pipeline_rebuild_path(@conn, :rebuild, @workflow.id, @pipeline.id), class: "btn btn-secondary btn-tiny", pipeline_rebuild_button: "true", title: "Rerun only failed jobs in this pipeline" %>
     <% end %>

front/test/front_web/controllers/pipeline_controller_test.exs

@@ -308,4 +308,38 @@ defmodule FrontWeb.PipelineControllerTest do
       assert conn.status == 404
     end
   end
+
+  describe "rebuild => with ui_partial_ppl_rebuild feature flag" do
+    test "returns 404 when feature flag is disabled", %{
+      conn: conn,
+      workflow_id: workflow_id,
+      pipeline_id: pipeline_id
+    } do
+      org = Support.Stubs.DB.first(:organizations)
+      Support.Stubs.Feature.disable_feature(org.id, :ui_partial_ppl_rebuild)
+
+      conn =
+        conn
+        |> post("/workflows/#{workflow_id}/pipelines/#{pipeline_id}/rebuild")
+
+      assert conn.status == 404
+    end
+
+    test "returns 200 when feature flag is enabled", %{
+      conn: conn,
+      workflow_id: workflow_id,
+      pipeline_id: pipeline_id
+    } do
+      org = Support.Stubs.DB.first(:organizations)
+      Support.Stubs.Feature.enable_feature(org.id, :ui_partial_ppl_rebuild)
+
+      conn =
+        conn
+        |> post("/workflows/#{workflow_id}/pipelines/#{pipeline_id}/rebuild")
+
+      assert conn.status == 200
+      assert json_response(conn, 200)["message"] == "Pipeline rebuild initiated successfully."
+      assert json_response(conn, 200)["pipeline_id"] != nil
+    end
+  end
 end

front/test/support/stubs/feature.ex

@@ -137,7 +137,8 @@ defmodule Support.Stubs.Feature do
       {"new_project_onboarding", state: :ENABLED, quantity: 1},
       {"open_id_connect_filter", state: :ENABLED, quantity: 1},
       {"wf_editor_via_jobs", state: :HIDDEN, quantity: 0},
-      {"ui_reports", state: :ENABLED, quantity: 1}
+      {"ui_reports", state: :ENABLED, quantity: 1},
+      {"ui_partial_ppl_rebuild", state: :ENABLED, quantity: 1}
     ]
   end