module Timetrap
module CLI
extend Helpers
attr_accessor :args
extend self
USAGE = <<-EOF
Timetrap - Simple Time Tracking
Usage: #{File.basename $0} COMMAND [OPTIONS] [ARGS...]
where COMMAND is one of:
* alter - alter an entry's note, start, or end time. Defaults to the active entry
usage: t alter [--id ID] [--start TIME] [--end TIME] [NOTES]
-i, --id <id:i> Alter entry with id <id> instead of the running entry
-s, --start <time:qs> Change the start time to <time>
-e, --end <time:qs> Change the end time to <time>
* backend - open an sqlite shell to the database
usage: t backend
* display - display the current timesheet or a specific. Pass `all' as
timesheet to display all sheets.
usage: t display [--ids] [--start DATE] [--end DATE] [TIMESHEET | all ]
-v, --ids Print database ids (for use with alter)
-s, --start <date:qs> Include entries that start on this date or later
-e, --end <date:qs> Include entries that start on this date or earlier
* format - export a sheet to csv format
usage: t format [--start DATE] [--end DATE] FORMATTER
FORMATTER Currently only supports 'ical'
-s, --start <date:qs> Include entries that start on this date or later
-e, --end <date:qs> Include entries that start on this date or earlier
* in - start the timer for the current timesheet
usage: t in [--at TIME] [NOTES]
-a, --at <time:qs> Use this time instead of now
* kill - delete a timesheet
usage: t kill [--id ID] [TIMESHEET]
-i, --id <id:i> Alter entry with id <id> instead of the running entry
* list - show the available timesheets
usage: t list
* now - show the status of the current timesheet
usage: t now
* out - stop the timer for the current timesheet
usage: t out [--at TIME]
-a, --at <time:qs> Use this time instead of now
* running - show all running timesheets
NOT IMPLEMENTED
* switch - switch to a new timesheet
usage: t switch TIMESHEET
OTHER OPTIONS
-h, --help Display this help
EOF
def parse arguments
args.parse arguments
end
def invoke
args['-h'] ? say(USAGE) : invoke_command_if_valid
end
def commands
Timetrap::CLI::USAGE.scan(/\* \w+/).map{|s| s.gsub(/\* /, '')}
end
def say *something
puts *something
end
def invoke_command_if_valid
command = args.unused.shift
case (valid = commands.select{|name| name =~ %r|^#{command}|}).size
when 0 then say "Invalid command: #{command}"
when 1 then send valid[0]
else; say "Ambigous command: #{command}"; end
end
def alter
entry = args['-i'] ? Entry[args['-i']] : Timetrap.active_entry
say "can't find entry" && return unless entry
entry.update :start => args['-s'] if args['-s'] =~ /.+/
entry.update :end => args['-e'] if args['-e'] =~ /.+/
entry.update :note => unused_args if unused_args =~ /.+/
end
def backend
exec "sqlite3 #{DB_NAME}"
end
def in
Timetrap.start unused_args, args['-a']
end
def out
Timetrap.stop args['-a']
end
def kill
if e = Entry[args['-i']]
out = "are you sure you want to delete entry #{e.id}? "
out << "(#{e.note}) " if e.note.to_s =~ /.+/
print out
if $stdin.gets =~ /\Aye?s?\Z/i
e.destroy
say "it's dead"
else
say "will not kill"
end
elsif (sheets = Entry.map{|e| e.sheet }.uniq).include?(sheet = unused_args)
victims = Entry.filter(:sheet => sheet).count
print "are you sure you want to delete #{victims} entries on sheet #{sheet.inspect}? "
if $stdin.gets =~ /\Aye?s?\Z/i
Timetrap.kill_sheet sheet
say "killed #{victims} entries"
else
say "will not kill"
end
else
victim = args['-i'] ? args['-i'].to_s.inspect : sheet.inspect
say "can't find #{victim} to kill", 'sheets:', *sheets
end
end
def display
sheets = if (sheet = sheet_name_from_string(unused_args)) == 'all'
Timetrap::Entry.sheets
elsif sheet =~ /.+/
[sheet]
else
[Timetrap.current_sheet]
end
entries = []
sheets.each do |sheet|
say "Timesheet: #{sheet}"
id_heading = args['-v'] ? 'Id' : ' '
say "#{id_heading} Day Start End Duration Notes"
last_start = nil
from_current_day = []
ee = Timetrap::Entry.filter(:sheet => sheet).order(:start)
ee = ee.filter(:start >= Date.parse(args['-s'])) if args['-s']
ee = ee.filter(:start <= Date.parse(args['-e']) + 1) if args['-e']
entries += ee.all
ee.each_with_index do |e, i|
from_current_day << e
e_end = e.end || Time.now
say "%-4s%16s%11s -%9s%10s %s" % [
(args['-v'] ? e.id : ''),
format_date_if_new(e.start, last_start),
format_time(e.start),
format_time(e.end),
format_duration(e.start, e_end),
e.note
]
nxt = ee.map[i+1]
if nxt == nil or !same_day?(e.start, nxt.start)
say "%52s" % format_total(from_current_day)
from_current_day = []
else
end
last_start = e.start
end
say <<-OUT
---------------------------------------------------------
OUT
say " Total%43s" % format_total(ee)
say "\n" unless sheet == sheets.last
end
if sheets.size > 1
say <<-OUT
-------------------------------------------------------------
OUT
say "Grand Total%41s" % format_total(entries)
end
end
# TODO: Consolidate display and format
def format
begin
fmt_klass = Timetrap::Formatters.const_get("#{unused_args.classify}")
rescue
say "Invalid format specified `#{unused_args}'"
return
end
ee = Timetrap::Entry.filter(:sheet => Timetrap.current_sheet)
ee = ee.filter(:start >= Date.parse(args['-s'])) if args['-s']
ee = ee.filter(:start <= Date.parse(args['-e']) + 1) if args['-e']
say Timetrap.format(fmt_klass,ee.all)
end
def switch
sheet = unused_args
if not sheet =~ /.+/ then say "No sheet specified"; return end
say "Switching to sheet " + Timetrap.switch(sheet)
end
def list
sheets = Entry.sheets.map do |sheet|
sheet_atts = {:total => 0, :running => 0, :today => 0}
DB[:entries].filter(:sheet => sheet).inject(sheet_atts) do |m, e|
e_end = e[:end] || Time.now
m[:name] ||= sheet
m[:total] += (e_end.to_i - e[:start].to_i)
m[:running] += (e_end.to_i - e[:start].to_i) unless e[:end]
m[:today] += (e_end.to_i - e[:start].to_i) if same_day?(Time.now, e[:start])
m
end
end
width = sheets.sort_by{|h|h[:name].length }.last[:name].length + 4
say " %-#{width}s%-12s%-12s%s" % ["Timesheet", "Running", "Today", "Total Time"]
sheets.each do |sheet|
star = sheet[:name] == Timetrap.current_sheet ? '*' : ' '
say "#{star}%-#{width}s%-12s%-12s%s" % [
sheet[:running],
sheet[:today],
sheet[:total]
].map(&method(:format_seconds)).unshift(sheet[:name])
end
end
def now
if Timetrap.running?
out = "#{Timetrap.current_sheet}: #{format_duration(Timetrap.active_entry.start, Time.now)}".gsub(/ /, ' ')
out << " (#{Timetrap.active_entry.note})" if Timetrap.active_entry.note =~ /.+/
say out
else
say "#{Timetrap.current_sheet}: not running"
end
end
def running
say "Sorry not implemented yet :-("
end
private
def unused_args
args.unused.join(' ')
end
end
end