xenum 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6bbf9d98e7579cf718f304fdde5a13459d5c3b225383d48d20fba38bc25d6f59
-  data.tar.gz: 5d448b96d97c4438fbb491ebfe89e8aee222977c0df42edb568972ef41cb1ccb
+  metadata.gz: a22147054180684998dfbea79223da8a588abf7e0f923e3a1189b76a52d4335e
+  data.tar.gz: bd0b21403fa4d114cbb5139f1c514226648f8de71406da76b6bfb086375346fb
 SHA512:
-  metadata.gz: be579d94a4ce9c11aac9e2be34c2542e92e6c6fef2d7624e662cbfcc5f8f277262d58094bb5b245d44d6e6949a85811faf081f1ffed2592dd3bdc8e62d907565
-  data.tar.gz: 6bdabc0c8e78e6dd35ee33d7af734cc15b86bf66e6d40e26b14804900d211dae1ddbd53c8cc86904cd2c90e723bafb55a3d36083551c84154f5feb4f379a497e
+  metadata.gz: 94efb08dd1112f2bd67b3380b2e104d48392ff0467f39b28e9830b686de9778d5956a43db1f353b31b371e77585ea836a260b9951c89d5ba91d705ad0617f6e4
+  data.tar.gz: fdbaafde2073b72fbb450dcab5a134d7002cdefab5a89428a3977caaee0aac7eaaf1ef5dd9e6cf4ae82417032ddcc723931579f7a271507ab326634e88fb7329

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    xenum (0.1.0)
+    xenum (0.1.1)
 
 GEM
   remote: https://rubygems.org/

data/README.md

@@ -30,6 +30,18 @@ works like `Array#flatten`, but for all `Enumerable`, and it is lazy
 # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]
 ```
 
+### lazy_insert
+
+works like `Array#insert`, but not limited to `Array`
+
+```ruby
+10.times.lazy_insert(3, 'a', 'b').to_a
+# [0, 1, 2, "a", "b", 3, 4, 5, 6, 7, 8, 9]
+
+10.times.lazy_insert(-3, 'a', 'b').to_a
+# [0, 1, 2, 3, 4, 5, 6, 7, "a", "b", 8, 9]
+```
+
 ### lazy_product
 
 works like `Array#product`, but not limited to `Array`

data/lib/xenum/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Xenum
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

data/lib/xenum.rb

@@ -26,6 +26,16 @@ module Enumerable
     end
   end
 
+  def lazy_insert(index, *objs)
+    return self if objs.empty?
+
+    if index >= 0
+      lazy_insert_pos(index, objs)
+    else
+      lazy_insert_neg(index, objs)
+    end
+  end
+
   def lazy_product(*enums)
     if enums.empty?
       return Enumerator.new do |yielder|
@@ -92,4 +102,48 @@ module Enumerable
       end
     end
   end
+
+  private
+
+  def lazy_insert_neg(index, objs)
+    these = Enumerator === self ? self : self.to_enum
+    queue = []
+
+    (index.abs - 1).times do
+      begin
+        queue << these.next
+      rescue StopIteration
+        raise IndexError, "index #{index} too small; minimum: #{-(queue.size + 1)}"
+      end
+    end
+
+    Enumerator.new do |yielder|
+      loop do
+        begin
+          queue << these.next
+          yielder << queue.shift
+        rescue StopIteration
+          break
+        end
+      end
+      objs.each{ |obj| yielder << obj }
+      queue.each{ |this| yielder << this }
+    end
+  end
+
+  def lazy_insert_pos(index, objs)
+    these = self
+    Enumerator.new do |yielder|
+      current_index = -1
+      these.each do |this|
+        current_index += 1
+        objs.each{ |obj| yielder << obj } if current_index == index
+        yielder << this
+      end
+      if current_index < index
+        ((current_index + 1)...index).each{ yielder << nil }
+        objs.each{ |obj| yielder << obj }
+      end
+    end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: xenum
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - ken
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-03-15 00:00:00.000000000 Z
+date: 2024-03-18 00:00:00.000000000 Z
 dependencies: []
 description:
 email: