rake_script 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8cc103a638ff21e24588737201910ec53ed682abb285567f0cd6e21bd5e6b66e
-  data.tar.gz: b11cbc1465ce68e498321533470f423764e171e4d5e3addcad1f9f25c3790ae7
+  metadata.gz: bb8f2fb9d883a772838e6a3ea624031319db197ed8a7909b881b3f5abb5ae390
+  data.tar.gz: 1dcac5081bb2f0e7dc07beb2d1a5646700eb21510a3ccb17102e9f7940a494d7
 SHA512:
-  metadata.gz: c0ea2e5bc75e86bf9b7445873493e9613bc32249260eaec5714f2a66077c0e5dbbf421dac0036aacc8e020b6aa1ac01c6cfeed0e336e120e31d6bc51e208328b
-  data.tar.gz: 3e1325f937068bf7afa760c79bc41d8ee1b305f07973dcaaad8c7aa79c061b0146506f6430548d49d7c39c2f190e99108627afd1783ac532a0a48a6f374f955b
+  metadata.gz: a64af8382fa822d6cc9f01bea71a8f41dcf2e4a805759304027fe3767a750e36620c174e924edd360365aeb1c62612efa4dae5c6c878da866fbffebf94506230
+  data.tar.gz: a70b64a07a0b68373f64f055d211bc7db3e78b05f91f091212b3af0cbeab25623e2d677408adf291f36f644c3c70f38ecfd394df7d53d09a3c19d504c0530a74

data/lib/rake_script/file_system.rb

@@ -54,9 +54,13 @@ module RakeScript
       Dir.chdir(path) { yield }
     end
 
-    def append_file(filepath, string, verbose: false, before: nil, after: nil)
+    def append_file(filepath, string, verbose: false, before: nil, after: nil, safe: false)
       raise ArgumentError, "can't use :before and :after in same time" if before && after
 
+      unless File.exist?(filepath)
+        return failure!(safe, verbose) { "can't find file at #{filepath}" }
+      end
+
       after_index = nil
       if after || before
         content = File.read(filepath)
@@ -69,7 +73,56 @@ module RakeScript
         f.seek(after_index, IO::SEEK_SET) if after_index
         f.write(string)
       end
-      puts("Append #{string.inspect} to #{filepath}") if verbose
+      puts_verbose("Append #{string.inspect} to #{filepath}") if verbose
+    end
+
+    # @param safe [Boolean]
+    # @param verbose [Boolean]
+    # @raise [ArgumentError]
+    # @return [nil]
+    # @example
+    #   unless File.exist?(filepath)
+    #     return failure!(true, true) { "can't find file at #{filepath}" }
+    #   end
+    def failure!(safe = false, verbose = false, &block)
+      raise ArgumentError, block.call unless safe
+      puts_verbose("[Error] #{block.call}") if verbose
+      nil
+    end
+
+    def puts_verbose(msg, color: :red, style: :normal)
+      if respond_to?(:puts_colored)
+        puts_colored(msg, color: color, style: style)
+      else
+        puts(msg)
+      end
+    end
+
+    def replace_file(filepath, new_string, old:, verbose: false, safe: false)
+      unless File.exist?(filepath)
+        return failure!(safe, verbose) { "can't find file at #{filepath}" }
+      end
+
+      content = File.read(filepath)
+      index_start = content.index(old)
+      if index_start.nil?
+        return failure!(safe, verbose) { "can't find #{old.inspect} in #{filepath}" }
+      end
+
+      if old.is_a?(Regexp)
+        old_string = old.match(content[index_start..-1]).to_a.first
+      else
+        old_string = old
+      end
+      index_end = index_start + old_string.size
+      new_content = content[0..(index_start - 1)] + new_string + content[index_end..-1]
+      # File.write(filepath, new_content, mode: 'w')
+      File.open(filepath, 'w') do |f|
+        f.sync = true
+        f.write(new_content)
+      end
+
+      puts_verbose("Replace #{old.inspect} with #{new_string.inspect} in #{filepath}") if verbose
     end
   end
 end

data/lib/rake_script/version.rb

@@ -1,3 +1,3 @@
 module RakeScript
-  VERSION = '0.3.0'.freeze
+  VERSION = '0.4.0'.freeze
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rake_script
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Denis Talakevich