require 'zuora/api'

class ZuoraInterface
  begin
    require 'system_timer'
    SysTimer = SystemTimer
  rescue LoadError
    require 'timeout'
    SysTimer = Timeout
  end

  def make_account(accountName)
    acc = ZUORA::Account.new
    acc.allowInvoiceEdit = 0
    acc.name = accountName
    acc.currency = "USD"
    acc.autoPay = 0
    acc.status = "Draft"
    acc.paymentTerm = "Due Upon Receipt"
    acc.batch = "Batch1"
    acc.billCycleDay = "01"
    return acc
  end

  def make_contact(accountId)
    con = ZUORA::Contact.new
    con.accountId = accountId
    con.address1 = '4901 Morena Blvd';
    con.city = 'San Diego';
    con.country = 'United States';
    con.firstName = 'Robert';
    con.lastName = 'Smith';
    con.postalCode = '92117';
    con.state = 'Virginia';
    con.workEmail = 'robert@smith.com';
    return con
  end

  def make_payment_method(accountId)
    pmt = ZUORA::PaymentMethod.new
    pmt.accountId = accountId
    pmt.creditCardAddress1 = '52 Vexford Lane';
    pmt.creditCardCity = 'Anaheim';
    pmt.creditCardCountry = 'United States';
    pmt.creditCardExpirationMonth = '12';
    pmt.creditCardExpirationYear = '2010';
    pmt.creditCardHolderName = 'Firstly Lastly';
    pmt.creditCardNumber = '4111111111111111';
    pmt.creditCardPostalCode = '22042';
    pmt.creditCardState = 'California';
    pmt.creditCardType = 'Visa';
    pmt.type = 'CreditCard';
    return pmt
  end

  def create_active_account(name)

    val = false
    session_start
    account = make_account(name)
    result = create([account])
#    puts result.first.inspect

    if result.first.success

      accountId = result.first.id
      payment = make_payment_method(accountId)
      result = create([payment])

      if result.first.success

        paymentId = result.first.id
        con = make_contact(accountId)
        result = create([con])

        if result.first.success
          conId = result.first.id

          account.id = accountId
          account.status = 'Active'
          account.billToId = conId
          account.soldToId = conId
          result = update([account])

          if result.first.success
            val = account
          else
            add_errors(result)
          end

        else
          add_errors(result)
        end
      else
        add_errors(result)
      end
    else
      add_errors(result)
    end
    return val
  end

  def api_call(call, *args)
    SysTimer.timeout(120) { self.send(call, *args) }
  end

  def query(query_string)
    @z.query(ZUORA::Query.new(query_string))
  end

  def query_more(query_locator)    
    @z.queryMore(ZUORA::QueryMore.new(query_locator))
  end

  def create(objs)
    return @z.create(objs)
  end

  def generate(objs)
    return @z.generate(objs)
  end

  def update(objs)
    return @z.update(objs)
  end

  def delete(type, ids)
    d = ZUORA::Delete.new
    d.type = type
    d.ids = ids
    return @z.delete(d)
  end

  def subscribe(sub)
    return @z.subscribe(sub)
  end

  def amend(amend)
    return @z.amend(amend)
  end

  def get_object(query)
    object = lookup(query)
    unless object.result.size == 0
      return object.result.records.first
    else
      return nil
    end
  end

  def get_object_array(query)
    object = lookup(query)
    unless object.result.size == 0
      return object.result.records
    else
      return []
    end
  end

  def get_driver
    @z ||= Zuora::Billing::Api.instance.driver
  end

  def driver
    get_driver
  end

  def login
    loginargs = ZUORA::Login.new
    loginargs.username = $ZUORA_USER
    loginargs.password = $ZUORA_PASSWORD
    @session = ZUORA::SessionHeader.new
    @session.session = @z.login(loginargs).result.session
  end

  def session_start(custom_fields={})
    get_driver
    @z.do_init(custom_fields)
    session_cleanup
    login
    @z.headerhandler.set @session
    @z.wiredump_dev = STDERR if $ZUORA_VERBOSE
  end

  def session_cleanup
    @z.headerhandler.delete(@session) if @session
  end
end

public

def run_tests

  t = ZuoraInterface.new
  t.session_start

  # create active account
  e = t.create_active_account("name" + String(Time.now.to_f))
  puts "Created Account: " + e.id

  # query it
  r = t.query("SELECT Id, Name FROM Account WHERE Name = '#{e.name}' and status = 'Active'")
  e = r.result.records[0]
  puts "Queried Account: " + e.to_s

  # delete it
  r = t.delete("Account", [e.id])
  puts "Deleted Account? " + r[0].success.to_s

  t.session_cleanup

end

if __FILE__ == $0
end