require 'fileutils'
require 'net/https'
require 'logger'
require 'rake/clean'

require 'rubygems'
gem 'rake'
require 'rake/gempackagetask'
gem 'soap4r'
require 'wsdl/soap/wsdl2ruby'
require 'xsd/codegen/classdef'

require 'lib/adwords4r/services'

# Determine the current version of the software

CLOBBER.include('pkg')

CURRENT_VERSION = '13.0.0'
PKG_VERSION = ENV['REL'] ? ENV['REL'] : CURRENT_VERSION

SRC_RB = FileList['lib/**/*.rb']

WSDLDIR = 'wsdl'
LIBDIR = 'lib'
GENDIR = LIBDIR + '/adwords4r'

logger = Logger.new(STDERR)

CLEAN.include(WSDLDIR)
AdWords::Service.getVersions.each do |v|
  vname = "v#{v}"
  CLEAN.include(File.join(GENDIR, vname))
end

desc "gets the wsdl and generates the classes"
task :default => [:getwsdl, :generate]

desc "gets the wsdl files for AdWords services"
task :getwsdl do
  AdWords::Service.getVersions.each do |v|
    vname = "v#{v}"
    mkdir_p File.join(WSDLDIR, vname)
    AdWords::Service.getServices(v).each {|s|
      save(getfile("adwords.google.com",
		   "/api/adwords/#{vname}/#{s}Service?wsdl"),
           getWsdlFileName(vname,s))}
  end
end

desc "generates AdWords classes from the wsdl files"
task :generate do
  AdWords::Service.getVersions.each do |v|
    vname = "v#{v}"
    gendir = "#{LIBDIR}/adwords4r/#{vname}"
    mkdir_p gendir
    AdWords::Service.getServices(v).each do |name|
      worker = WSDL::SOAP::WSDL2Ruby.new
      worker.logger = logger
      worker.location = getWsdlFileName(vname,name)
      worker.basedir = gendir
      worker.opt.update(getWsdlOpt(name))
      worker.run
      fixImport(v, File.join(gendir, "#{name}Driver.rb"))
      fixImport(v, File.join(gendir, "#{name}MappingRegistry.rb"))
      fixImport(v, File.join(gendir, "#{name}.rb"))
    end
  end
end

def fixImport(version, file)
  vname = "v#{version}"
  tempfile = file + '.tmp'
  outfile = File.new(tempfile,"w")
  File.open(file, "r") do |infile|
    infile.each do |l|
      if (l =~ /require.*Service.*\.rb/) then
        outfile.puts l.gsub(/require '(.*)Service(.*)\.rb'/,
                            "require 'adwords4r/#{vname}/\\1Service\\2'")
      else
        outfile.puts l
      end
    end
  end
  outfile.close
  File.rename(tempfile, file)
end 

def getWsdlOpt(s)
  optcmd= {}
  s << "Service"
  optcmd['classdef'] = s
  optcmd['force'] = true
  optcmd['mapping_registry'] = true
  optcmd['driver'] = nil
  
  # Causes soap4r to wrap the classes it outputs into the given modules
  optcmd['module_path'] = ['AdWords', s]
  return optcmd
end

def getWsdlFileName(v,s)
  "#{WSDLDIR}/#{v}/#{s}.wsdl"
end

def getfile(host, path)
  puts "getting https//#{host}#{path}"
  https = Net::HTTP.new(host, 443)
  https.use_ssl = true
  https.verify_mode = OpenSSL::SSL::VERIFY_NONE
  https.start { |w| w.get2(path).body }
end

def fix_attribute(text, name, type)
  text.gsub!(Regexp.new("#{name}=\"#{type}\""),"#{name}=\"xsd:#{type}\"")
end

def fixWsdl(wsdl)
  ['type', 'base'].each {|name| 
    ['long', 'string', 'date', 'int', 'boolean'].each {|type|
      fix_attribute(wsdl, name, type)}}
  schema_ns = "xmlns:xsd=\"http://www.w3.org/2001/XMLSchema\""
  if wsdl !~ Regexp.new(schema_ns)
    wsdl.gsub!(/(<wsdl:definitions[^>]*)>/, '\1 ' + schema_ns + '>')
  end
  return wsdl
  #wsdl.gsub(/type=\"long\"/, 'type="xsd:long"')
end

# Saves this document to the specified @var path.
#doesn't create the file if contains markup for google 404 page
def save(content, path)
  if content !~ /<H2>Error 404<\/H2>/
    File::open(path, 'w') {|f| f.write(fixWsdl(content))}
  end
end

# ====================================================================
# Create a task that will package the Rake software into distributable
# gem files.

PKG_FILES = FileList[
  '*.*', 
  'Rakefile', 
  'lib/**/*.rb', 
  'examples/**/*.rb', 
  #  'test/**/*.rb',
  'scripts/**/*.rb'
]

#puts PKG_FILES

PKG_FILES.exclude(/\._/)

if ! defined?(Gem)
  puts "Package Target requires RubyGems"
else
  spec = Gem::Specification.new do |s|
    
    #### Basic information.
    
    s.name = 'adwords4r'
    s.version = PKG_VERSION
    s.summary = "Client library for the AdWords API."
    s.description = %{\
Adwords4r provides an easy to use way to access the AdWords API in ruby.\
Currently the following AdWords API versions are supported:\
\
* V10\
* V11\
}
    
    s.files = PKG_FILES.to_a
    s.require_path = 'lib'
    s.autorequire = 'adwords4r'
    
    #s.test_files = PKG_FILES.select { |fn| fn =~ /^test\/test/ }
    
    #s.has_rdoc = true
    s.has_rdoc = false
    #s.extra_rdoc_files = rd.rdoc_files.reject { |fn| fn =~ /\.rb$/ }.to_a
    #s.rdoc_options <<
    #      '--title' <<  'Builder -- Easy XML Building' <<
    #      '--main' << 'README' <<
    #      '--line-numbers'
    
    s.author = "Jeffrey Posnick, Patrick Chanezon, Ryan Leavengood"
    s.email = "jeffy@google.com"
    s.homepage = "http://rubyforge.org/projects/google4r/"
    s.requirements << 'soap4r v1.5.8 or greater'
    s.requirements << 'httpclient v2.1.2 or greater'
    s.rubyforge_project = 'google4r'
    s.add_dependency('soap4r', '>= 1.5.8')
    s.add_dependency('httpclient', '>= 2.1.2')
  end
  
  Rake::GemPackageTask.new(spec) do |t|
    t.need_tar = true
  end
end