Separate token into own class files

changelogs/unreleased/refactoring-entities-file-25.yml

+---
+title: Separate token entities into own class files
+merge_request: 24974
+author: Rajendra Kadam
+type: added

lib/api/entities.rb

@@ -169,36 +169,6 @@ module API
       expose :last_pipeline, using: Entities::PipelineBasic
       expose :variables, using: Entities::Variable
     end
-
-    class ImpersonationToken < PersonalAccessToken
-      expose :impersonation
-    end
-
-    class ImpersonationTokenWithToken < PersonalAccessTokenWithToken
-      expose :impersonation
-    end
-
-    class FeatureGate < Grape::Entity
-      expose :key
-      expose :value
-    end
-
-    class Feature < Grape::Entity
-      expose :name
-      expose :state
-      expose :gates, using: FeatureGate do |model|
-        model.gates.map do |gate|
-          value = model.gate_values[gate.key]
-
-          # By default all gate values are populated. Only show relevant ones.
-          if (value.is_a?(Integer) && value.zero?) || (value.is_a?(Set) && value.empty?)
-            next
-          end
-
-          { key: gate.key, value: value }
-        end.compact
-      end
-    end
   end
 end
 

lib/api/entities/feature.rb

+# frozen_string_literal: true
+
+module API
+  module Entities
+    class Feature < Grape::Entity
+      expose :name
+      expose :state
+      expose :gates, using: Entities::FeatureGate do |model|
+        model.gates.map do |gate|
+          value = model.gate_values[gate.key]
+
+          # By default all gate values are populated. Only show relevant ones.
+          if (value.is_a?(Integer) && value.zero?) || (value.is_a?(Set) && value.empty?)
+            next
+          end
+
+          { key: gate.key, value: value }
+        end.compact
+      end
+    end
+  end
+end

lib/api/entities/feature_gate.rb

+# frozen_string_literal: true
+
+module API
+  module Entities
+    class FeatureGate < Grape::Entity
+      expose :key
+      expose :value
+    end
+  end
+end

lib/api/entities/impersonation_token.rb

+# frozen_string_literal: true
+
+module API
+  module Entities
+    class ImpersonationToken < Entities::PersonalAccessToken
+      expose :impersonation
+    end
+  end
+end

lib/api/entities/impersonation_token_with_token.rb

+# frozen_string_literal: true
+
+module API
+  module Entities
+    class ImpersonationTokenWithToken < Entities::PersonalAccessTokenWithToken
+      expose :impersonation
+    end
+  end
+end