lib/graphql-preview/extensions/enabled_previews.rb in graphql-preview-0.1.0 vs lib/graphql-preview/extensions/enabled_previews.rb in graphql-preview-0.1.1

@@ -1,9 +1,15 @@
 module GraphQLPreview
   module Extensions
     module EnabledPreviews
-      def enabled_previews
-        metadata[:enabled_previews]
+      def enabled_previews(environment = nil)
+        metadata[:enabled_previews].select do |preview|
+          if environment.nil?
+            true
+          else
+            preview.environments.nil? || preview.environments.include?(environment)
+          end
+        end
       end
     end
   end
 end