檔案插入指定字串(Find 與檔案操作)

#usage: inser_ga_code.rb  GA_ID  dir
require 'Find'

gaID = ARGV[0]
gaCode = %Q{
<script src="http://www.google-analytics.com/urchin.js" type="text/javascript">
</script>
<script type="text/javascript">
_uacct = "#{gaID}";
urchinTracker();
</script>
}

dir = ARGV[1]
Find.find(dir) do |filepath|
#%w[c:/index.htm c:/index2.html].each do |filepath|  #for test
    next if FileTest.directory? filepath
    next if File.extname(filepath).scan(/\.html?$/i).empty?
    puts filepath

    File.open(filepath, 'rb+') do |fh|
        content = fh.read
        if Regexp.new(%Q(_uacct = "#{gaID}")) =~ content then
            puts 'ok'
        else
            i = content.rindex /<\/body>/i
            next unless i
            fh.pos = i
            fh.print gaCode, content[i..content.length]
            #fh.rewind
            #newContent = content[0...i] + gaCode + content[i..content.length]
            #File.open('index.htm.new', 'w') {|nfh| nfh << newContent}
            puts 'add'
        end
    end
end