robotag 2.0.0 → 3.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: dec985055f4c9d4c0433884d83032aa9da032146a59a834775e48a9887d10784
-  data.tar.gz: 3f24cd962472659a3cbc99a6b4780e3df748eac943055cbd8b66288387cd65b1
+  metadata.gz: ab44b297b1fd2c5f003e7825deb734e12d46dc62470b02abd3da9feed0b9bdbc
+  data.tar.gz: ab1d7b1ddae29e276e481a7c72d957c1526c5cac46414ba604e30132a5983d7d
 SHA512:
-  metadata.gz: 3f47d741e49ff2be4722de01b33eb5b410e459a5766ed4c543c25e828a1903447bcc438c4d087ab5f39a00841608ee85315cc7950cee75a829944524e418d4d6
-  data.tar.gz: 1a72c50d795145710e075bde7e5e08c215ddcf148c3f3ea6de5a38ef993a1b3bc6414e8d686d691adb0ffc30c74ed5121b54214e07e9f768981f8dfe109fef10
+  metadata.gz: 2ea9fed578bce9a570ce0a526d0e5ec2c83d466a50eac068b5d95cdda7cdfb5cf3e741efe7ac011b6c9fc89b25c20496fd462fd02a2d431ac047ac69659d4919
+  data.tar.gz: f22e93f4f1dfc97ee5851795214a3f909cf285466f306cd32bd0fc59f05bf192c71ebe17b81df86fbdcfc4ca14e8fe6a85d8038df6a382784b186c369e042cfc

data/lib/robotag.rb

@@ -36,35 +36,44 @@ class Robotag
   def tag_all_with(tag)
     warn('This script will potentially rewrite all of your feature files. Please be patient and remember to tip your source control system.') # rubocop:disable Metrics/LineLength
 
-    # Analysis and output
-    self.chain_state.map { |query_result| query_result[:self] }.each do |test|
-      if (test.is_a?(CukeModeler::Scenario) || test.is_a?(CukeModeler::Outline))
-        add_tag(test, tag)
-      else
-        raise("Unknown test type: #{test.class}")
-      end
-    end
+    analysis_and_output(tag, :add)
+    self
+  end
+
+  def remove_all(tag)
+    warn('This script will potentially rewrite all of your feature files. Please be patient and remember to tip your source control system.') # rubocop:disable Metrics/LineLength
+
+    analysis_and_output(tag, :remove)
     self
   end
 
-  def add_tag(test, tag)
-    return if has_tag?(test, tag)
+  def process_tag(test, tag, action)
+    if action == :add
+      return if has_tag?(test, tag)
+    elsif action == :remove
+      return unless has_tag?(test, tag)
+    end
 
-    tag_test(test, tag)
+    modify_test_tag(test, tag, action)
   end
 
-  def tag_test(test, tag)
+  def modify_test_tag(test, tag, action)
     feature_file = test.get_ancestor(:feature_file)
     file_path = feature_file.path
 
     tag_index = (test.source_line - 2)
 
     file_lines = if self.write_directives[file_path].nil?
-                  File.readlines(file_path)
-                else
-                  self.write_directives[file_path].split("\n").map {|line| line + "\n"}
-                end
-    file_lines[tag_index] = "#{file_lines[tag_index].chomp} #{tag}\n"
+                   File.readlines(file_path)
+                 else
+                   self.write_directives[file_path].split("\n").map { |line| line + "\n" }
+                 end
+
+    if action == :add
+      file_lines[tag_index] = "#{file_lines[tag_index].chomp} #{tag}\n"
+    elsif action == :remove
+      file_lines[tag_index].delete!(tag)
+    end
 
     self.write_directives[file_path] = file_lines.join
 
@@ -100,4 +109,16 @@ class Robotag
       self.chain_state
     end
   end
+
+  private
+
+  def analysis_and_output(tag, action)
+    self.chain_state.map { |query_result| query_result[:self] }.each do |test|
+      if (test.is_a?(CukeModeler::Scenario) || test.is_a?(CukeModeler::Outline))
+        process_tag(test, tag, action)
+      else
+        raise("Unknown test type: #{test.class}")
+      end
+    end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: robotag
 version: !ruby/object:Gem::Version
-  version: 2.0.0
+  version: 3.0.0
 platform: ruby
 authors:
 - David West
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-05-26 00:00:00.000000000 Z
+date: 2023-05-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec