#!/usr/bin/env ruby

require 'rubygems'
require 'open-uri'
require 'openssl'

# FIXME (2008-07-24, praveen): This is nasty. I don't know of any other
# simpler way to skip SSL certificate verification.
OpenSSL::SSL::VERIFY_PEER = OpenSSL::SSL::VERIFY_NONE


class Order
  attr_reader :order_number, :zip_code, :valid, :canceled, \
  :customer_name, :order_date, :shipped, :ship_date, :ship_carrier, \
  :tracking_number

  def initialize order_number, zip_code
    @order_number = order_number
    @zip_code = zip_code
    @valid = false
    @canceled = false

    order_status_uri = URI.parse "https://www.wireless.att.com/order_status/\
order_status_results.jsp?fromwhere=order_status&vMethod=ordernum&\
vNumber=#{order_number}&ZipCode=#{zip_code}&x=40&y=14"
    order_status = order_status_uri.read

    if !order_status.scan("We're sorry, but the information you entered \
was not recognized by our systems.").to_s.empty?
      @valid = false
      return
    end
    @valid = true

    if !order_status.scan("Canceled")[0].to_s.empty?
      @canceled = true
      return
    end
    @canceled = false

    @customer_name =
      order_status.scan(/Customer:\s*\w*/).to_s.gsub!(/Customer:\s*/,"")
    @order_date =
      order_status.scan(/Date Ordered:.*/).to_s.gsub!(/Date Ordered:\s*/, "")
    @order_date.gsub!(/\s/, "")
    @shipped =
      order_status.scan(/\<td width="5%".*/)[3].scan(/[01]/).to_s.to_i == 1 ?
    true : false
    if !@shipped
      return
    end
    @ship_date = order_status.scan(/\<td width="6%".*/)[5]
    @ship_date.gsub! /^.*\<p\>/, ""
    @ship_date.gsub! /\<\/p\>.*$/, ""
    @ship_carrier = order_status.scan(/\<td width="6%".*/)[7]
    @ship_carrier.gsub! /^.*\<p\>/, ""
    @ship_carrier.gsub! /\<\/p\>.*$/, ""
    @tracking_number =
      order_status.scan(/\http:\/\/fedex.com\/Tracking\?tracknumbers=\d*/).to_s
    @tracking_number.gsub! /\http:\/\/fedex.com\/Tracking\?tracknumbers=/, ""
  end

  def print
    if !@valid
      puts "#{@order_number.ljust(6)} ------------------------ NOT FOUND \
------------------------"
      return
    end
    if @canceled
      puts "#{@order_number.ljust(6)} ------------------------ CANCELLED \
------------------------"
      return
    end
    if @shipped
      puts "#{@order_number.ljust(6)} #{@customer_name.slice(0..14).ljust(15)} \
#{@order_date} #{'Yes'.ljust(5)} #{@ship_date} #{@ship_carrier.ljust(6)} \
#{@tracking_number}"
    else
      puts "#{@order_number.ljust(6)} #{@customer_name.slice(0..14).ljust(15)} \
#{@order_date} #{'No'.ljust(5)} #{'NA'.ljust(8)} #{'NA'.ljust(6)} \
#{'NA'.ljust(12)}"
    end
  end
end

if ARGV.length != 3
  puts "Usage: ./order-status.rb ZIP BEGIN END

  ZIP      ZIP code of the AT&T store where the order is placed.
  BEGIN    Order number to begin searching for.
  END      Order number to end searching for.

Example: ./order-status.rb 94025 12000 12100
"
else
  puts "Que Order# #{'Name'.ljust(15)} Order Dt Ship? Ship Dt  Carrie Tracking"
  puts "----------------------------------------------------------------------"
  queue = 0
  for order_number in ARGV[1]..ARGV[2]
    o = Order.new order_number, ARGV[0]
    if o.valid and !o.canceled
      queue += 1 if !o.shipped
      display_queue = o.shipped ? "NA" : queue
    else
      display_queue = "NA"
    end
    print "#{display_queue.to_s.ljust(3)} "
    o.print
  end
end