#!/usr/bin/env ruby

require 'rubygems'
require 'mechanize'

HELP_STRING =<<EOS

Tool for fetching wordpress.com weblog statistics. Usage:

    wls.rb [username] [pwd]

where 'user' is your wordpress user name and 'pwd' is your
password respectively.

EOS

if not ARGV.grep(/-h|--help/).empty?
    puts HELP_STRING
    exit(0)
end

# try to access the weblog statistics page
user = 'muharem'
password = nil  # set your password here if you dislike being prompted for it

if ARGV[0]
    user = ARGV[0]
end
if ARGV[1]
    password = ARGV[1]
end

stats_url = "http://#{user}.wordpress.com/wp-admin/index.php?page=stats"

# instantiate/initialise web agent ..
agent = WWW::Mechanize.new
agent.user_agent_alias = 'Mac Safari'
# .. and get the weblog statistics page
page = agent.get(stats_url)

# did we get back the login form?
if (page.title.strip.split[-1] == 'Login')
    # yes, fill it in and submit it
    loginf = page.form('loginform')
    loginf.log = user
    if not password
        print "Enter your wordpress.com password: "
        password = $stdin.gets.chomp
    end
    loginf.pwd = password
    agent.submit(loginf, loginf.buttons.first)
end

# now get the actual weblog statistics page
page = agent.get_file(stats_url)
# parse it!
doc = Hpricot(page)

# search for the div elements that contain the statistics data
stats_divs = doc.search("//div[@class='statsdiv']")
stats_divs.each do |div|
    heading = div.search("h3/a/text()")
    # we are only interested in the statistics for today
    day = div.search("h4/text()").first
    if (heading and day)
        heading = "==== #{heading} (#{day.inner_text.downcase}) ====".center(50)
        puts "\n#{heading}\n"
        # find the table with today's statistics data
        tab = div.search("table").first
        if tab
            # extract the statistics data from the <tr> elements
            tab.search("tr").each do |tr|
                what = tr.search("td[@class='label']")
                views = tr.search("td[@class='views']")
                whats = what.inner_text.strip()
                if not whats.empty?
                    views = views.inner_text.strip()
                    printf("%s -- %5s\n", whats.center(45), views)
                end
            end
        end
    end
end
# grab the div with the general (weblog level) statistics data
gbdiv = doc.search("//div[@id='generalblog']")
# find the <p> element with the number of views today
vtoday =  gbdiv.search("p").find { |p| p.inner_text.index('Views today') }
if vtoday
    printf("\n%s\n\n", "=> #{vtoday.inner_text.strip} <=".center(45))
else
    puts "\n\n!! No weblog statistics data found."
    puts "   Did you enter a wrong user name and/or password?"
end