class EdgecastCdnReport

# File lib/edgecast_cdn_report.rb, line 20
def initialize(config_file, report_date=Date.today)
  if config_file.nil?
    raise "Please provide a config file #{config_file} "
  end
  @config = YAML.load_file(config_file)

  self.customer_id = @config['customer_id']
  self.token = @config['token']
  self.cname_groups = @config['cname_groups']
  self.report_date = report_date
end

# File lib/edgecast_cdn_report.rb, line 32
def base_uri
  BASE_URI+'/'+customer_id
end

# File lib/edgecast_cdn_report.rb, line 46
def data_transfer_by_cname(start_date, end_date)
  result = self.get "/media/3/cnamereportcodes?begindate=#{start_date}&enddate=#{end_date}"
  data_transfers = {}

  result.each do |cname_data|
    data_transfers[cname_data['Description']] = cname_data['Bytes'].to_f
  end

  data_transfers
end

# File lib/edgecast_cdn_report.rb, line 131
def f(bytes)
  if bytes.nil?
    "Na"
  elsif(bytes/TB_INBYTES > 1)
    (bytes.to_f/TB_INBYTES).round(2).to_s + ' TB'
  elsif(bytes/GB_INBYTES > 1)
    (bytes.to_f/GB_INBYTES).round(2).to_s + ' GB'
  else
    (bytes.to_f/MB_INBYTES).round(2).to_s + ' MB'
  end
end

# File lib/edgecast_cdn_report.rb, line 36
def get uri
  response = HTTParty.get(base_uri + uri, :headers => {"Authorization" => "TOK:#{token}"})
  JSON.parse(response.body)
end

# File lib/edgecast_cdn_report.rb, line 98
def grouped_cname_rows
  cnames = [mtd_data_by_cname, lmtd_data_by_cname, lm_data_by_cname].flat_map(&:keys).uniq

  cname_data = {}
  cnames.each do |cname|
    cname_data[cname] = [mtd_data_by_cname[cname], lmtd_data_by_cname[cname], lm_data_by_cname[cname]]
  end

  grouped_cnames_data = []
  cname_groups.each do |group, cnames|
    group_cnames = cname_data.extract!(*cnames)

    grouped_cnames_data << [group.upcase, *sum_rows(group_cnames.values)]

    grouped_cnames_data += hash_to_row(group_cnames)
  end

  if (cname_data.length > 0)
    grouped_cnames_data << ['Others', *sum_rows(cname_data.values)]
    grouped_cnames_data += hash_to_row(cname_data)
  end

  grouped_cnames_data
end

# File lib/edgecast_cdn_report.rb, line 92
def hash_to_row h
    h.map do |cname, data|
      [cname, *data.map{|x| f(x)}]
    end
end

# File lib/edgecast_cdn_report.rb, line 63
def lm_data_by_cname
  @lm_data_by_cname ||= data_transfer_by_cname(report_date.last_month.beginning_of_month, report_date.last_month.end_of_month)
end

# File lib/edgecast_cdn_report.rb, line 68
def lmtd_data_by_cname
  @lmtd_data_by_cname ||= data_transfer_by_cname(report_date.last_month.beginning_of_month, report_date.last_month)
end

# File lib/edgecast_cdn_report.rb, line 58
def mtd_data_by_cname
  @mtd_data_by_cname ||= data_transfer_by_cname(report_date.beginning_of_month, report_date)
end

# File lib/edgecast_cdn_report.rb, line 143
def output
  header = ['CNAME', 'Cur MTD', 'Last MTD', 'Last Total']
  cname_group_names = cname_groups.keys.map(&:downcase)

  table = Terminal::Table.new headings: header do |t|
    t << total_row
    t << :separator
    t << unaccounted_row

    grouped_cname_rows.each do |row|
      t << :separator if (cname_group_names.include? row[0].downcase)
      t << row
    end
  end

  puts "Date : #{report_date} \n"
  puts table
end

# File lib/edgecast_cdn_report.rb, line 86
def sum_rows rows
  rows.transpose.map do |row|
    f(row.inject{ |sum, i| sum.to_f + i.to_f})
  end
end

# File lib/edgecast_cdn_report.rb, line 73
def tcm_data
  @tcm_data ||= total_data_transfer(report_date.strftime("%Y-%m"));
end

# File lib/edgecast_cdn_report.rb, line 78
def tlm_data
  @tlm_data ||= total_data_transfer(report_date.last_month.strftime("%Y-%m"))
end

# File lib/edgecast_cdn_report.rb, line 41
def total_data_transfer(month)
  result = self.get "/media/3/region/-1/units/2/trafficusage?begindate=#{month}-01"
  result['UsageResult'].to_f * GB_INBYTES
end

# File lib/edgecast_cdn_report.rb, line 82
def total_row
  ['Total', f(tcm_data), '',  f(tlm_data)]
end

# File lib/edgecast_cdn_report.rb, line 123
def unaccounted_row
  total_todate_cname     = mtd_data_by_cname.values.reduce(:+)
  total_last_month_cname = lm_data_by_cname.values.reduce(:+)

  ['Unaccounted', f(tcm_data-total_todate_cname), '', f(tlm_data-total_last_month_cname)]
end