Replace wag with wrag
This commit is contained in:
Henrik Lissner
parent
cc14bfe1a0
commit
d08e0ac93a
66
bin/wag
66
bin/wag
|
|
@ -1,66 +0,0 @@
|
||||||
#!/usr/bin/env ruby
|
|
||||||
|
|
||||||
# A hacky script that allows you to search with ag
|
|
||||||
# <https://github.com/ggreer/the_silver_searcher> and edit the results in
|
|
||||||
# $EDITOR. The changes are propagated to all listed files, if any.
|
|
||||||
#
|
|
||||||
# Usage: wag [pattern]
|
|
||||||
#
|
|
||||||
# TODO Also rename files whose names have been changed
|
|
||||||
|
|
||||||
if ARGV.empty?
|
|
||||||
puts "You didn't enter in anything!"
|
|
||||||
exit 1
|
|
||||||
end
|
|
||||||
|
|
||||||
require 'tempfile'
|
|
||||||
require 'shellwords'
|
|
||||||
|
|
||||||
t = Tempfile.new('.ag-results')
|
|
||||||
out = system("ag #{ARGV.join(' ')} --nogroup --nocolor > #{t.path}")
|
|
||||||
if [false, nil].include? out
|
|
||||||
puts "No matches" if out === false
|
|
||||||
|
|
||||||
t.unlink
|
|
||||||
exit 1
|
|
||||||
end
|
|
||||||
|
|
||||||
old_lines = File.open(t.path).readlines
|
|
||||||
system("#{ENV['EDITOR']} #{t.path}")
|
|
||||||
lines = File.open(t.path).readlines
|
|
||||||
t.unlink
|
|
||||||
|
|
||||||
if old_lines.length != lines.length
|
|
||||||
puts "There was a line-number mismatch. Abort! Abort! Abort!"
|
|
||||||
exit 2
|
|
||||||
end
|
|
||||||
|
|
||||||
i = nil
|
|
||||||
lines.each_with_index do |line, i|
|
|
||||||
next if line.empty? || line.eql?(old_lines[i])
|
|
||||||
|
|
||||||
l = line.split(":", 3)
|
|
||||||
unless l.length == 3
|
|
||||||
puts "Line #{i} is malformed as was skipped"
|
|
||||||
next
|
|
||||||
end
|
|
||||||
|
|
||||||
filename = Shellwords.escape(l[0])
|
|
||||||
lineno = Shellwords.escape(l[1])
|
|
||||||
content = l[2].gsub("'", "\\'")
|
|
||||||
|
|
||||||
if system("sed -i .bak '#{lineno}c\\
|
|
||||||
#{content}' #{filename}")
|
|
||||||
puts "Replaced line #{lineno} in #{filename}"
|
|
||||||
else
|
|
||||||
puts "Failed to replace line #{lineno} in #{filename}"
|
|
||||||
end
|
|
||||||
|
|
||||||
system("rm -f #{filename}.bak") if File.exists?("#{filename}.bak")
|
|
||||||
end
|
|
||||||
|
|
||||||
if i === nil
|
|
||||||
puts "Didn't do nuthin'"
|
|
||||||
else
|
|
||||||
puts "Done!"
|
|
||||||
end
|
|
||||||
107
bin/wrag
Executable file
107
bin/wrag
Executable file
|
|
@ -0,0 +1,107 @@
|
||||||
|
#!/usr/bin/env ruby
|
||||||
|
|
||||||
|
# A hacky script that allows you to search with ag
|
||||||
|
# <https://github.com/ggreer/the_silver_searcher> and edit the results in
|
||||||
|
# $EDITOR. The changes are propagated to all listed files, if any.
|
||||||
|
#
|
||||||
|
# Usage: wrag [pattern]
|
||||||
|
#
|
||||||
|
# TODO Also rename changed filenames
|
||||||
|
# TODO Assemble sed string to make this atomic
|
||||||
|
|
||||||
|
require 'optparse'
|
||||||
|
require 'tempfile'
|
||||||
|
require 'shellwords'
|
||||||
|
|
||||||
|
def usage
|
||||||
|
puts "Usage: #{File.basename($0)} [-f|--rename-files] [-b|--backup] search_string"
|
||||||
|
puts "\nAny of ag's flags will work with wrag"
|
||||||
|
exit
|
||||||
|
end
|
||||||
|
|
||||||
|
usage if ARGV.empty?
|
||||||
|
|
||||||
|
options = {:rename => false,
|
||||||
|
:backup => false}
|
||||||
|
|
||||||
|
opt = OptionParser.new
|
||||||
|
opt.on('-f', '--rename-files') { options[:rename] = true }
|
||||||
|
opt.on('-b', '--backup') { options[:backup] = true }
|
||||||
|
|
||||||
|
t = Tempfile.new('.ag-results')
|
||||||
|
out = system("ag #{ARGV.join(' ')} --group --nocolor > #{t.path}")
|
||||||
|
if [false, nil].include? out
|
||||||
|
puts "No matches" if out === false
|
||||||
|
|
||||||
|
t.unlink
|
||||||
|
exit 1
|
||||||
|
end
|
||||||
|
|
||||||
|
old_lines = File.open(t.path).readlines
|
||||||
|
system("#{ENV['EDITOR']} #{t.path}")
|
||||||
|
lines = File.open(t.path).readlines
|
||||||
|
t.unlink
|
||||||
|
|
||||||
|
if old_lines.length != lines.length
|
||||||
|
puts "There was a line-number mismatch. Abort! Abort! Abort!"
|
||||||
|
exit 2
|
||||||
|
end
|
||||||
|
|
||||||
|
i = nil
|
||||||
|
current_filename = nil
|
||||||
|
lines.each_with_index do |line, i|
|
||||||
|
line.strip!
|
||||||
|
|
||||||
|
next if line.empty?
|
||||||
|
|
||||||
|
if line =~ /^([0-9]+):(.+)/ # make sure it isn't actually a file
|
||||||
|
filename = current_filename
|
||||||
|
lineno = $1
|
||||||
|
content = $2
|
||||||
|
elsif line =~ /^(.+):([0-9]+):(.+)/
|
||||||
|
if current_file
|
||||||
|
puts "Mixing and matching groups and non-groups! Aborting."
|
||||||
|
exit 3
|
||||||
|
end
|
||||||
|
|
||||||
|
filename = $1
|
||||||
|
lineno = $2
|
||||||
|
content = $3
|
||||||
|
elsif File.exists?(line)
|
||||||
|
if File.readable?(line)
|
||||||
|
current_filename = line
|
||||||
|
else
|
||||||
|
puts "#{line} isn't readable. Skipping."
|
||||||
|
end
|
||||||
|
next
|
||||||
|
else
|
||||||
|
puts "Malformed line: #{line}"
|
||||||
|
exit 4
|
||||||
|
end
|
||||||
|
|
||||||
|
next if line.eql?(old_lines[i].strip)
|
||||||
|
|
||||||
|
filename = Shellwords.escape(filename)
|
||||||
|
lineno = Shellwords.escape(lineno)
|
||||||
|
content = content.gsub("'", "\\'")
|
||||||
|
|
||||||
|
puts "#{filename}:#{lineno}:#{content}"
|
||||||
|
|
||||||
|
if system("sed -i .bak '#{lineno}c\\
|
||||||
|
#{content}
|
||||||
|
' #{filename}")
|
||||||
|
puts "Replaced: line #{lineno} in #{filename}"
|
||||||
|
else
|
||||||
|
puts "Failed to replace line #{lineno} in #{filename}"
|
||||||
|
end
|
||||||
|
|
||||||
|
unless options[:backup]
|
||||||
|
system("rm -f #{filename}.bak") if File.exists?("#{filename}.bak")
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
if i === nil
|
||||||
|
puts "Didn't do nuthin'"
|
||||||
|
else
|
||||||
|
puts "Done!"
|
||||||
|
end
|
||||||
Loading…
Reference in a new issue