warg 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 424f245fa2ac9ae05aff9e1f4c66e9c5a7a9b5660436e0d5d91a5653fc062d44
-  data.tar.gz: a8bc6efa101af3b880fb2a5e012f55e8a8b3510180fe33248df0cffa3f54c724
+  metadata.gz: 1dae787424c000c9f5503d44a1a0f4653c5f2ae9fcc50e0afe24a030de34716b
+  data.tar.gz: 925e761cbb6409606f42791bd5c18c8cfc2c7648f996ace762d01201eca527c7
 SHA512:
-  metadata.gz: 97c01e3dfdf6df8786a91ba9532917bf3a9e270256eea18bf0cb33e2754fad957403ab9139072bb69ff1a0d6cfbcd082bbdf8cd017916531eb6846bd1fadb986
-  data.tar.gz: fd3d3247c2635dded91bfe9fb954a3d18fe55a6b789656e50b593579b069621421da7e6ddb339bb87ea9585604e27556bb4ca97dd890d6069b66e01f716882b0
+  metadata.gz: 3604a78c642404593ca9ee445e6cf8e861b688ac6df2aeadb700f21dafa20ad283adf4a05d0d375cecc854d3a93d20966656fcdede08c2b49b5a74574ad914ec
+  data.tar.gz: 0f8557c8b1ab338728813b6d4b4da8ccaa8629f47ee381d0585c4efe011741943edfdc0fdd84cf6539ab0ab46429388783a06e6b81d7fc067edc73861fc75c44

data/lib/warg.rb

@@ -1159,22 +1159,8 @@ module Warg
       @hosts.length
     end
 
-    def create_file_from(content, path:, mode: 0644)
-      each do |host|
-        host.create_file_from(content, path: path, mode: mode)
-      end
-    end
-
-    def upload(file, to:)
-      each do |host|
-        host.upload(file, to: to)
-      end
-    end
-
-    def download(path)
-      map do |host|
-        host.download(path)
-      end
+    def lazy
+      LazilyFilteredHostCollection.new(self)
     end
 
     def each
@@ -1187,24 +1173,46 @@ module Warg
       self
     end
 
-    def run_script(script, order: :parallel, &setup)
-      run(order: order) do |host, result|
-        result.update host.run_script(script, &setup)
+    module Interaction
+      def create_file_from(content, path:, mode: 0644)
+        each do |host|
+          host.create_file_from(content, path: path, mode: mode)
+        end
       end
-    end
 
-    def run_command(command, order: :parallel, &setup)
-      run(order: order) do |host, result|
-        result.update host.run_command(command, &setup)
+      def upload(file, to:)
+        each do |host|
+          host.upload(file, to: to)
+        end
+      end
+
+      def download(path)
+        map do |host|
+          host.download(path)
+        end
+      end
+
+      def run_script(script, order: :parallel, &setup)
+        run(order: order) do |host, result|
+          result.update host.run_script(script, &setup)
+        end
       end
-    end
 
-    def run(order:, &block)
-      strategy = Executor.for(order)
-      executor = strategy.new(self)
-      executor.run(&block)
+      def run_command(command, order: :parallel, &setup)
+        run(order: order) do |host, result|
+          result.update host.run_command(command, &setup)
+        end
+      end
+
+      def run(order:, &block)
+        strategy = Executor.for(order)
+        executor = strategy.new(self)
+        executor.run(&block)
+      end
     end
 
+    include Interaction
+
     def to_a
       @hosts.dup
     end
@@ -1214,6 +1222,78 @@ module Warg
     attr_reader :hosts
   end
 
+  class LazilyFilteredHostCollection
+    include Enumerable
+    include HostCollection::Interaction
+
+    attr_reader :filters
+
+    def self.from(value)
+      new HostCollection.from(value)
+    end
+
+    def initialize(hosts)
+      @hosts = hosts
+      @filters = {}
+    end
+
+    def one
+      host = to_a.sample
+
+      if host.nil?
+        raise "no hosts matched `#{inspect}'"
+      else
+        HostCollection.from(host)
+      end
+    end
+
+    def add(host_data)
+      host = Host.from(host_data)
+
+      if host.matches?(**@filters)
+        @hosts.add(host)
+        self
+      else
+        false
+      end
+    end
+
+    def with(**filters)
+      combined = @filters.merge(filters)
+
+      HostCollection.new(@hosts.select { |host| host.matches?(combined) })
+    end
+
+    def ==(other)
+      case other
+      when HostCollection, LazilyFilteredHostCollection
+        to_a == other.to_a
+      when Enumerable
+        to_a == other
+      else
+        super
+      end
+    end
+
+    alias eql? ==
+
+    def length
+      count
+    end
+
+    def each
+      if block_given?
+        @hosts.each do |host|
+          if host.matches?(**@filters)
+            yield host
+          end
+        end
+      else
+        enum_for(:each)
+      end
+    end
+  end
+
   class Config
     attr_accessor :default_user
     attr_reader :hosts
@@ -1357,7 +1437,6 @@ module Warg
       @argv = argv
       @playlist = playlist
       @parser = OptionParser.new
-      @filters = {}
 
       @parser.on("-t", "--target HOSTS", Array, "hosts to use") do |hosts_data|
         hosts_data.each { |host_data| hosts.add(host_data) }
@@ -1366,15 +1445,16 @@ module Warg
       @parser.on("-f", "--filter key=value", String, "filters to apply to hosts") do |filter|
         key, value = filter.split("=", 2)
 
-        @filters[key] = value
+        @hosts.filters[key] = value
       end
 
       super()
+
+      @hosts = @hosts.lazy
     end
 
     def parse_options!
       @parser.parse(@argv)
-      @hosts = @hosts.with(**@filters)
     end
 
     def copy(config)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: warg
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - Eduardo Gutierrez
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-11-11 00:00:00.000000000 Z
+date: 2023-11-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bcrypt_pbkdf