1 files changed, 27 insertions, 7 deletions
diff --git a/app/controllers/backends/terraform_aws_controller.rb b/app/controllers/backends/terraform_aws_controller.rb
index 72c7e3a..5669eb1 100644
--- a/app/controllers/backends/terraform_aws_controller.rb
+++ b/app/controllers/backends/terraform_aws_controller.rb
@@ -4,17 +4,22 @@ class Backends::TerraformAwsController < ApplicationController
   end
 
   def create
-    backend = Backends::TerraformAws.create(backend_params)
+    backend = Backends::TerraformAws.create(create_params)
 
-    flash[:success] = "#{backend.label} created"
+    flash[:success] = "Backend #{backend.label} created"
 
-    redirect_to setup_path
+    redirect_to terraform_aws_backend_path(backend)
   end
 
   def update
-    backend = Backends::TerraformAws.update(params[:id], backend_params)
+    @backend = Backends::TerraformAws.update(
+      params[:id],
+      update_params
+    )
+
+    flash[:success] = "Backend #{@backend.label} updated"
 
-    flash[:success] = "#{backend.label} updated"
+    render :show
   end
 
   def show
@@ -24,18 +29,33 @@ class Backends::TerraformAwsController < ApplicationController
   def destroy
     backend = Backends::TerraformAws.find(params[:id])
 
+    flash[:success] = "Backend #{backend.label} deleted"
     backend.delete
+
+    redirect_to setup_path
   end
 
   private
 
-  def backend_params
+  def create_params
     params
+      .require(:backends_terraform_aws)
       .permit(
         :label,
+        :domain,
         :aws_region,
         :aws_access_key_id,
-        :aws_secret_access_key,
+        :aws_secret_access_key
+      )
+  end
+
+  def update_params
+    params
+      .require(:backends_terraform_aws)
+      .permit(
+        :label,
+        :aws_access_key_id,
+        :aws_secret_access_key
       )
   end
 end