Add EnvironmentSerializer to EnvironmentsController

1 files changed, 3 insertions, 16 deletions

diff --git a/app/controllers/projects/environments_controller.rb b/app/controllers/projects/environments_controller.rb
index 42f882b55ab..6bd4cb3f2f5 100644
--- a/app/controllers/projects/environments_controller.rb
+++ b/app/controllers/projects/environments_controller.rb
@@ -13,7 +13,9 @@ class Projects::EnvironmentsController < Projects::ApplicationController
     respond_to do |format|
       format.html
       format.json do
-        render json: serialize_as_json(@environments)
+        render json: EnvironmentSerializer
+          .new(project: @project)
+          .represent(@environments)
       end
     end
   end
@@ -63,19 +65,4 @@ class Projects::EnvironmentsController < Projects::ApplicationController
   def environment
     @environment ||= project.environments.find(params[:id])
   end
-  
-  def serialize_as_json(resource)
-    resource.as_json(
-      include: {
-        last_deployment: {
-          include: {
-            user: { only: [:id, :name, :username], methods: [:avatar_url] },
-            deployable: { only: [:id, :name, :ref, :tag] }
-          },
-          methods: [:short_sha, :commit_title, :commit]
-        },
-        project: { methods: [:namespace] }
-      }
-    )
-  end
 end