1 files changed, 33 insertions, 14 deletions
diff --git a/app/controllers/backends/terraform_libvirt_controller.rb b/app/controllers/backends/terraform_libvirt_controller.rb
index f854bc2..6dcba17 100644
--- a/app/controllers/backends/terraform_libvirt_controller.rb
+++ b/app/controllers/backends/terraform_libvirt_controller.rb
@@ -1,30 +1,26 @@
 class Backends::TerraformLibvirtController < ApplicationController
   def new
     @backend = Backends::TerraformLibvirt.new
-    @backend.uri = "qemu:///system"
+    @backend.uri = 'qemu:///system'
   end
 
   def create
-    backend = Backends::TerraformLibvirt.create(
-      params
-        .require(:backends_terraform_libvirt)
-        .permit(:label, :uri)
-    )
+    backend = Backends::TerraformLibvirt.create(create_params)
 
-    flash[:success] = "#{backend.label} created"
+    flash[:success] = "Backend #{backend.label} created"
 
-    redirect_to setup_path
+    redirect_to terraform_libvirt_backend_path(backend)
   end
 
   def update
-    backend = Backends::TerraformLibvirt.update(
-      params[:id],
-      params
-        .require(:backends_terraform_libvirt)
-        .permit(:label, :uri)
+    @backend = Backends::TerraformLibvirt.update(
+      params['id'],
+      update_params
     )
 
-    flash[:success] = "#{backend.label} updated"
+    flash[:success] = "Backend #{@backend.label} updated"
+
+    render :show
   end
 
   def show
@@ -34,6 +30,29 @@ class Backends::TerraformLibvirtController < ApplicationController
   def destroy
     backend = Backends::TerraformLibvirt.find(params['id'])
 
+    flash[:success] = "Backend #{backend.label} deleted"
     backend.delete
+
+    redirect_to setup_path
+  end
+
+  private
+
+  def create_params
+    params
+      .require(:backends_terraform_libvirt)
+      .permit(
+        :label,
+        :domain,
+        :uri
+      )
+  end
+
+  def update_params
+    params
+      .require(:backends_terraform_libvirt)
+      .permit(
+        :label
+      )
   end
 end