about summary refs log tree commit diff
path: root/lib/enumerable.rb
diff options
context:
space:
mode:
authorThibG <thib@sitedethib.com>2021-01-11 13:44:57 +0100
committerGitHub <noreply@github.com>2021-01-11 13:44:57 +0100
commita7a47834e50645746355e2b9a88244c1804c73cd (patch)
tree3b81b1d2a4b2c2d2e4b29948c72a7995a21fa449 /lib/enumerable.rb
parentd42e7e01dcd464f80637682d4eee6e5a7f36f26e (diff)
parent31e68bf3d35b80e794c1a44b3d60dabcf10f2a3d (diff)
Merge pull request #1483 from ThibG/glitch-soc/merge-upstream
Merge upstream changes
Diffstat (limited to 'lib/enumerable.rb')
-rw-r--r--lib/enumerable.rb26
1 files changed, 26 insertions, 0 deletions
diff --git a/lib/enumerable.rb b/lib/enumerable.rb
new file mode 100644
index 000000000..66918f65e
--- /dev/null
+++ b/lib/enumerable.rb
@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+module Enumerable
+  # TODO: Remove this once stop to support Ruby 2.6
+  if RUBY_VERSION < '2.7.0'
+    def filter_map
+      if block_given?
+        result = []
+        each do |element|
+          res = yield element
+          result << res if res
+        end
+        result
+      else
+        Enumerator.new do |yielder|
+          result = []
+          each do |element|
+            res = yielder.yield element
+            result << res if res
+          end
+          result
+        end
+      end
+    end
+  end
+end