#!/usr/bin/env ruby
#
# Adds a new string to res/values/strings.xml for Android projects
#
# Usage: add_string <key> <word>+
#

require 'fileutils'

def print_help
  STDERR.puts 'Usage: ' + File.basename($0) + ' <options>* <key> <word>+'
  STDERR.puts 'where options include'
  STDERR.puts ' -k || -key   convert all the words into one lowercase/_-delimited key'
   STDERR.puts '-h || -help  print this message'
  exit 1
  return 0
end

def add_string(key,str)

  # Escape strings
  str = str.gsub /'/, "\\'"
  str = str.gsub /"/, "\\\""

  # Allow keys like '@string/request_a_feature'
  key = key.gsub /@string\//,''

  new_file = 'strings.xml'
  old_file = File.join 'res','values','strings.xml'
  File.open new_file,'w' do |out|
    IO.foreach old_file do |line|
      if line =~ /<\/resources>/
        new_str = '<string name="' + key + '">' + str + '</string>'
        out.puts '  ' + new_str
      else
        line.scan /name="([^"]+)"/ do |res|
          k = res[0]
          if k == key
            STDERR.puts 'This key already found!'
            return 1
          end
        end
        line.scan />([^<]+)</ do |res|
          s = res[0]
          if s == str
            STDERR.puts 'This string already found!'
          end
        end
      end
      out.puts line
    end
  end
  
  # Copy the file over
  FileUtils.mv new_file,old_file

  return 0
end

def main(argv)
  
  return print_help if argv.empty?

  options = argv.select {|x| x =~ /^-/}
  words = argv.reject {|x| x =~ /^-/}

  res = 0
  key_opt = false
  options.each do |opt|
    if opt =~ /^-+k/
      key_opt = true
    else
      STDERR.puts 'Unrecognized option ' + opt
      res = 1
    end
  end
  
  if key_opt
    key = words.join('_').downcase.gsub(/-\+./,'_').gsub(/[^a-z0-9_]/,'')
    str = words.join ' '
    puts 'R.string.' + key
  else 
    return print_help if argv.length < 2
    key = argv.shift
    str = argv.join ' '
  end

  exit add_string(key,str) && res
  
end

main ARGV