#!/usr/bin/env ruby
#
# $Id: mycal,v 1.23 2003/11/13 10:43:03 ianmacd Exp $
#
# simple program to retrieve one's calendar entries for the coming seven days
#
# Copyright (C) 2002 Ian Macdonald
#
#   This program is free software; you can redistribute it and/or modify
#   it under the terms of the GNU General Public License as published by
#   the Free Software Foundation; either version 2, or (at your option)
#   any later version.
#
#   This program is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty of
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#   GNU General Public License for more details.
#
#   You should have received a copy of the GNU General Public License
#   along with this program; if not, write to the Free Software Foundation,
#   Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.

begin
  require 'password'
rescue LoadError
  $stderr.puts "This program requires Ruby/Password: http://www.caliban.org/ruby/"
  exit 1
end

require 'ctime'
require 'date'
require 'getoptlong'

def usage(code = 0)
  prog_name = File::basename($0)
  $stderr.print "
Usage: #{prog_name} [-d|--days <days>] [-u|--user <user>]
       #{prog_name} -h|--help

-d, --days	retrieve calendar for this number of days (default: 7)
-h, --help	show this usage message
-u, --user	log on as this user (overrides ~/.mycal)
"
  exit code
end

server = nil; line_nr = __LINE__
user = nil
passwd = nil
days = 7	# default number of days to retrieve events for

# read ~/.mycal if it exists
dot_mycal = ENV['HOME'] + '/.mycal'
if FileTest.readable?(dot_mycal)
  File.open(dot_mycal) do |file|
    while line = file.gets
      eval line
    end
  end
end

if server.nil?
  $stderr.puts "Please define a CorporateTime server in either line " +
	       "#{line_nr} of this program or in a ~/.mycal file."
  exit 1
end

opt = GetoptLong.new(
  ['--days',		'-d', GetoptLong::REQUIRED_ARGUMENT],
  ['--help',		'-h', GetoptLong::NO_ARGUMENT],
  ['--user',		'-u', GetoptLong::REQUIRED_ARGUMENT])

begin
  opt.each_option do |name,arg|
    case name
    when '--days'
      days = arg.to_i
    when '--help'
      usage
    when '--user'
      user = arg
    end
  end
rescue
  usage 1
end

user ||= ( print "Username: "; gets.chomp ) 
passwd ||= Password.new(Password.getc("Calendar password"))

now = Time.now.gmtime - 1
today = Date.today
in_a_week = today + days

start_date = sprintf("%s%02d%02dT%02d%02d00Z",
		     today.year, today.month, today.day, now.hour, now.min)
end_date = sprintf("%s%02d%02dT%02d%02d00Z",
		   in_a_week.year, in_a_week.month, in_a_week.day,
		   now.hour, now.min)

ct = CTime.new(server, user, passwd)

unless ct.error == "CAPI_STAT_OK"
  printf("Failed to connect to %s as user %s\n", server, user)
  exit 2
end

ct.open_agenda(user)
events = ct.get_events(start_date, end_date)

puts "Calendar for user #{user}:\n\n"

events.sort {|a,b| a.start <=> b.start}.each do |e|
  begin
    s_year, s_month, s_day, s_hours, s_mins, s_secs =
      /^(\d{4})(\d\d)(\d\d)T(\d\d)(\d\d)(\d\d)Z$/.match(e.start)[1..6]
    start_time = Time.gm(s_year, s_month, s_day, s_hours, s_mins, s_secs)
  rescue NameError
    # likely an all-day event. Start time is of the form YYYYMMDD.
    s_year, s_month, s_day = /^(\d{4})(\d\d)(\d\d)$/.match(e.start)[1..3]
    start_time = Time.gm(s_year, s_month, s_day, 0, 0, 0)
  end

  begin
    # both of these matches will throw an exception for an all-day event,
    # because the end time will be of the format YYYYMMDD.
    if ! e.duration.nil?  # CAPI 2.0 returns DTDURATION, not DTEND
      d_days, d_hours, d_mins, d_secs =
        /^P(\d)DT(\d+)H(\d+)M(\d+)S$/.match(e.duration)[1..3]
      end_time = start_time + d_days.to_s.to_i * 86400 + 
		 d_hours.to_s.to_i * 3600 + d_mins.to_s.to_i * 60 +
		 d_secs.to_s.to_i
    else		  # CAPI 2.5 or later returns DTEND, not DTDURATION
      e_year, e_month, e_day, e_hours, e_mins, e_secs =
        /^(\d{4})(\d\d)(\d\d)T(\d\d)(\d\d)(\d\d)Z$/.match(e.end)[1..6]
      end_time = Time.gm(e_year, e_month, e_day, e_hours, e_mins, e_secs)
    end

    start_time.localtime
    end_time.localtime

    printf("%s until %s\n", start_time.strftime("%A, %B %d, %Y at %H:%M"),
	   end_time.strftime("%H:%M"))

  # catch all-day events
  rescue NameError
    printf("%s (all day event)\n", start_time.strftime("%A, %B %d, %Y"))

  ensure
    unless e.summary.nil?
      e.summary.gsub!(/\\,/, ',')
      printf("  %s\n", e.summary)
    end

    unless e.description.nil?
      e.description.gsub!(/\\n/, "\n")
      e.description.gsub!(/\\,/, ',')
      printf("  %s\n", e.description)
    end

    unless e.location.nil?
      e.location.gsub!(/\\,/, ',')
      printf("  %s\n", e.location)
    end

    puts

  end

end

ct.quit
