magic_frozen_string_literal 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 5271e012fd545661378c77a655a973dcf5784742
-  data.tar.gz: 37ed692a99df681f6d044b9d5042ece055552105
+  metadata.gz: 9f4a7f937c5bdae40737a09dc33c88db763840be
+  data.tar.gz: 722fb36293bf02e654b3fcf81114c8219c433d7b
 SHA512:
-  metadata.gz: 7381678e8e283f22e3648c4a9a80fc6828b703c5d04cd0bbb5578cbc651e288a1f88d5c4497eaed6c0b96a8f2d861bc3d43cc23f64f1591fd29ec72d66d35015
-  data.tar.gz: b50bac58839c7995482ede8a117666f627212506d25a459572c06d47855285a56e0678c56187354f6b5728cb4fe6359c8d6f164e1b9404c9a78545797901dbf2
+  metadata.gz: b6e58b53690c83f77b75bd8e0f3fb44df88770e85e9cdb0f2effb59bc8a47e10a1d65d2fc427feec31f84fe9ed5ff914c53864e1de9b2c46832865d192c2c8f4
+  data.tar.gz: 539915a742a2ec149f110fd7126614d656840df65f1eeca3125e8d8121c4a33ce31c2d2fb275b4d461cadb2c8dedde19689cba8f964770a3f71a83e06772947a

data/lib/add_magic_comment.rb

@@ -1,45 +1,37 @@
 # frozen_string_literal: true
 
-# A simple library to prepend magic comments to multiple ".rb" files
+# A simple library to prepend magic comments to all Ruby files in a given folder
 
 module AddMagicComment
   MAGIC_COMMENT_PREFIX  = "frozen_string_literal"
-  MAGIC_COMMENT_PATTERN = /^-?# *#{MAGIC_COMMENT_PREFIX}/
+  MAGIC_COMMENT_PATTERN = /^(-|(<%))?#\s*#{MAGIC_COMMENT_PREFIX}\s*(%>)?/
   MAGIC_COMMENT         = "#{MAGIC_COMMENT_PREFIX}: true"
 
-  EXTENSIONS = {
-    'rb'   => '# {text}',
-    'rake' => '# {text}',
-    'haml' => '-# {text}',
+  EXTENSION_COMMENTS = {
+    "rb"   => "# #{MAGIC_COMMENT}",
+    "rake" => "# #{MAGIC_COMMENT}",
+    "haml" => "-# #{MAGIC_COMMENT}",
+    "erb"  => "<%# #{MAGIC_COMMENT} %>"
   }
 
-  # Options :
-  # 1 : Encoding
-  # 2 : Path
-  # TODO : check that the encoding specified is a valid encoding
-  # TODO : allow use of only one option, so the encoding would be guessed (maybe using `file --mime`?)
-  def self.process(options)
-
-    directory = options[0] || Dir.pwd
-
-
-    # TODO : add options for recursivity (and application of the script to a single file)
+  def self.process(argv)
+    directory = argv.first || Dir.pwd
 
     count = 0
-    EXTENSIONS.each do |ext, comment_style|
-      rbfiles = File.join(directory, "**", "*.#{ext}")
-      Dir.glob(rbfiles).each do |filename|
+    EXTENSION_COMMENTS.each do |ext, comment|
+      filename_pattern = File.join(directory, "**", "*.#{ext}")
+      Dir.glob(filename_pattern).each do |filename|
         next unless File.size(filename) > 0
         File.open(filename, "r+") do |file|
           lines = file.readlines
 
           # remove current encoding comment(s)
-          while lines[0].match(MAGIC_COMMENT_PATTERN)
+          while lines.first && lines.first.match(MAGIC_COMMENT_PATTERN)
             lines.shift
           end
 
           # set current encoding
-          lines.insert(0, comment_style.sub('{text}', MAGIC_COMMENT + "\n"))
+          lines.insert(0, comment + "\n")
           count += 1
 
           file.pos = 0

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: magic_frozen_string_literal
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Colin Kelley after Jared Roesch after Manuel Ryan
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-08-21 00:00:00.000000000 Z
+date: 2016-02-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -68,12 +68,12 @@ required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 1.3.6
+      version: '0'
 requirements: []
 rubyforge_project: 
 rubygems_version: 2.2.2
 signing_key: 
 specification_version: 4
-summary: 'Easily add magic comments ''# frozen_string_literal: true'' on multiple
-  ruby source files'
+summary: 'Easily add magic comments ''# frozen_string_literal: true'' to multiple
+  Ruby source files'
 test_files: []