#!/usr/bin/ruby 
#
# Create a pie chart based on the return codes encountered when crawling
# a website.  Limit depth with -r if neeed.
#
# Jon Hart <jhart@spoofed.org>
#
# Copyright (c) 2008, Jon Hart
# All rights reserved.
# 
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#     * Redistributions of source code must retain the above copyright
#       notice, this list of conditions and the following disclaimer.
#     * Redistributions in binary form must reproduce the above copyright
#       notice, this list of conditions and the following disclaimer in the
#       documentation and/or other materials provided with the distribution.
#     * Neither the name of the <organization> nor the
#       names of its contributors may be used to endorse or promote products
#       derived from this software without specific prior written permission.
# 
# THIS SOFTWARE IS PROVIDED BY Jon Hart ``AS IS'' AND ANY
# EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL <copyright holder> BE LIABLE FOR ANY
# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
require 'rubygems'
require 'hawler'
require 'hawleroptions'
require 'gruff'

def codemap (uri, referer, response)
  unless (response.nil?)
    if (@codes[response.code])
      @codes[response.code] += 1
    else
      @codes[response.code] = 1
    end

    puts "#{uri} returned a #{response.code}" if (@options.verbose)
  end
end

@options = HawlerOptions.parse(ARGV)
@codes = {}

if (ARGV.empty?)
  puts @options.help
else
  crawler = Hawler.new(ARGV[0], method(:codemap))
  @options.each_pair do |o,v|
    crawler.send("#{o}=",v)
  end
  crawler.start

  g = Gruff::Pie.new
  g.title = "Crawl of #{ARGV}"

  @codes = @codes.sort { |a,b| b[1] <=> a[1] }
  @codes.each do |f|
    g.data "#{f[0]} (#{f[1]})", f[1]
  end
    

  name = "#{ARGV[0]}"
  name.gsub!(/\//, '_')
  g.write("#{name}-codemap.png")
  puts"Done -- codemap is #{name}-codemap.png"
end