timetrap 1.1.1

data/lib/timetrap/models.rb

@@ -0,0 +1,80 @@
+module Timetrap
+  class Entry < Sequel::Model
+    plugin :schema
+
+    class << self
+    # a class level instance variable that controls whether or not all entries
+    # should respond to #start and #end with times rounded to 15 minute
+    # increments.
+      attr_accessor :round
+    end
+
+    def round?
+      !!self.class.round
+    end
+
+    def start= time
+      self[:start]= Chronic.parse(time) || time
+    end
+
+    def end= time
+      self[:end]= Chronic.parse(time) || time
+    end
+
+    def start
+      round? ? rounded_start : self[:start]
+    end
+
+    def end
+      round? ? rounded_end : self[:end]
+    end
+
+    def end_or_now
+      self.end || (round? ? round(Time.now) : Time.now)
+    end
+
+    def rounded_start
+      round(self[:start])
+    end
+
+    def rounded_end
+      round(self[:end])
+    end
+
+    def round time
+      return nil unless time
+      Time.at(
+        if (r = time.to_i % 900) < 450
+          time.to_i - r
+        else
+          time.to_i + (900 - r)
+        end
+      )
+    end
+
+    def self.sheets
+      map{|e|e.sheet}.uniq.sort
+    end
+
+    # do a quick pseudo migration.  This should only get executed on the first run
+    set_schema do
+      primary_key :id
+      column :note, :string
+      column :start, :timestamp
+      column :end, :timestamp
+      column :sheet, :string
+    end
+    create_table unless table_exists?
+  end
+
+  class Meta < Sequel::Model(:meta)
+    plugin :schema
+
+    set_schema do
+      primary_key :id
+      column :key, :string
+      column :value, :string
+    end
+    create_table unless table_exists?
+  end
+end

data/lib/timetrap.rb

@@ -0,0 +1,80 @@
+require 'rubygems'
+require 'chronic'
+require 'sequel'
+require 'sequel/extensions/inflector'
+require 'Getopt/Declare'
+require File.join(File.dirname(__FILE__), 'timetrap', 'helpers')
+require File.join(File.dirname(__FILE__), 'timetrap', 'cli')
+DB_NAME = defined?(TEST_MODE) ? nil : "#{ENV['HOME']}/.timetrap.db"
+# connect to database.  This will create one if it doesn't exist
+DB = Sequel.sqlite DB_NAME
+require File.join(File.dirname(__FILE__), 'timetrap', 'models')
+Dir["#{File.dirname(__FILE__)}/timetrap/formatters/*.rb"].each do |path|
+  require path
+end
+module Timetrap
+  extend self
+
+  def current_sheet= sheet
+    m = Meta.find_or_create(:key => 'current_sheet')
+    m.value = sheet
+    m.save
+  end
+
+  def current_sheet
+    unless Meta.find(:key => 'current_sheet')
+      Meta.create(:key => 'current_sheet', :value => 'default')
+    end
+    Meta.find(:key => 'current_sheet').value
+  end
+
+  def entries sheet = nil
+    Entry.filter(:sheet => sheet).order_by(:start)
+  end
+
+  def running?
+    !!active_entry
+  end
+
+  def active_entry
+    Entry.find(:sheet => Timetrap.current_sheet, :end => nil)
+  end
+
+  def stop time = nil
+    while a = active_entry
+      time ||= Time.now
+      a.end = time
+      a.save
+    end
+  end
+
+  def start note, time = nil
+    raise AlreadyRunning if running?
+    time ||= Time.now
+    Entry.create(:sheet => Timetrap.current_sheet, :note => note, :start => time).save
+  rescue => e
+    CLI.say e.message
+  end
+
+  def switch sheet
+    self.current_sheet = sheet
+  end
+
+  def kill_sheet sheet
+    Entry.filter(:sheet => sheet).destroy
+  end
+
+  def format format_klass, entries
+    format_klass.new(entries).output
+  end
+
+  class AlreadyRunning < StandardError
+    def message
+      "Timetrap is already running"
+    end
+  end
+
+  CLI.args = Getopt::Declare.new(<<-EOF)
+    #{CLI::USAGE}
+  EOF
+end