string_literal_extractor 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f7c72d3ff93f4a3ca4dd93f3d74bc949d1681060
-  data.tar.gz: 16552e6fbfc788ff2f2bb37d75ef18bdf4c44d1a
+  metadata.gz: 543cc852e84b1bd0ef01ab0f63b6e27fac160462
+  data.tar.gz: 5cb35717eb6f5228d29f0be84508929712490ccf
 SHA512:
-  metadata.gz: 1c43345413e6792721fe79fe7a2ffec837d8d57c1654ab5452f296a0f861b717805e6f0b43946d1bedfaead30339388c3250cca1fdbe9fe32baefd6fb9706f64
-  data.tar.gz: 02e7b32e76ee2f038d176761cd78db7c2c602e67ef175db52e3bfef67ec865f22c19108dafdd71eb5f87b7a2434564b7f6900991a4bca499f279589d286e9c24
+  metadata.gz: 1eabe26d0a05c6479ce79f90bdd95e12af6d766b53a6bc694bcdf2a41b3946b6067014be05d472de0ad8af5b75ecfd9336cd486a15dbb025d5927e2278b5cb02
+  data.tar.gz: ee1607412100eed4348aeb92680d21031f6291ef8a185684662107894800de027dda78569848005906041824b6173df69e1e254685daf2f9d60ea8f98e1d5c03

data/string_literal_extractor.rb

@@ -0,0 +1,75 @@
+require 'ripper'
+
+class StringLiteralExtractor < Ripper
+  include Enumerable
+
+  %i(heredoc_beg tstring_beg).each do |event|
+    module_eval(<<-CODE, __FILE__, __LINE__ + 1)
+    def on_#{event}(tok)
+      @buf ||= []
+      @buf << [[lineno, column]]
+    end
+    CODE
+  end
+
+  def on_tstring_end(tok)
+    @buf.last << [lineno, column]
+  end
+
+  def on_heredoc_end(tok)
+    @buf.last << [lineno, column + tok.size-1]
+  end
+
+  def on_CHAR(literal)
+    @string_literal_each.call(literal, [[lineno, column], [lineno, column + 1]])
+  end
+
+  def on_string_literal(*args)
+    pos = @buf.pop
+    lineno_pos = pos.map(&:first)
+    column_pos = pos.map(&:last)
+    lines = @src.lines[lineno_pos.first-1...lineno_pos.last]
+    if lineno_pos.first == lineno_pos.last
+      lines[0] = lines[0][column_pos.first..column_pos.last]
+    else
+      lines[0] = lines[0].dup.tap { |l| l[0...column_pos.first] = '' }
+      lines[-1] = lines[-1].dup.tap { |l| l[column_pos.last+1..-1] = '' }
+    end
+    @string_literal_each.call(lines.join, pos)
+
+    args.unshift(:string_literal)
+    args
+  end
+
+  def initialize(src, filename="(ripper)", lineno=1)
+    @src = src
+    super
+  end
+
+  def each
+    @string_literal_each = -> (literal, pos) { yield literal, pos }
+    parse
+  end
+end
+
+return unless $0 == __FILE__
+
+StringLiteralExtractor.new(%q{"#{'hi' + 'hoi'}"}).each do |literal, pos|
+  p [literal, pos]
+end
+
+StringLiteralExtractor.new(%q{"a" "b"}).each do |literal, pos|
+  p [literal, pos]
+end
+
+StringLiteralExtractor.new("<<EOS\n  foo  bar\n\nEOS").each do |literal, pos|
+  p [literal, pos]
+end
+
+StringLiteralExtractor.new('"#{hi}"').each do |literal, pos|
+  p [literal, pos]
+end
+
+StringLiteralExtractor.new('?a').each do |literal, pos|
+  p [literal, pos]
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: string_literal_extractor
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Seiei Miyagi
@@ -37,7 +37,8 @@ email: hanachin@gmail.com
 executables: []
 extensions: []
 extra_rdoc_files: []
-files: []
+files:
+- string_literal_extractor.rb
 homepage: https://github.com/hanachin/string_literal_extractor
 licenses:
 - MIT