ttrack 0.1.0

data/bin/ttrack

@@ -0,0 +1,23 @@
+#!/Users/alex/.rvm/rubies/ruby-1.9.2-p290/bin/ruby -w
+require 'ttrack'
+
+tt = TTrack.new
+case ARGV[0]
+  when 'start' then
+    tt.start ARGV[1], ARGV[2]
+  when 'stop' then
+    tt.stop
+  when 'status' then
+    tt.status ARGV[1]
+  when 'init' then
+    tt.init
+  when 'report' then
+    tt.report ARGV[1]
+  when 'beguin' then
+    timestamp = ARGV[3] ? "%s %s" % ARGV[2..3] : ARGV[2]
+    tt.setstart ARGV[1], timestamp
+  when 'end' then
+    timestamp = ARGV[3] ? "%s %s" % ARGV[2..3] : ARGV[2]
+    tt.setstop ARGV[1], timestamp
+  else tt.usage
+end

data/lib/ttrack/datastore.rb

@@ -0,0 +1,137 @@
+# -*- coding: utf-8 -*-
+require 'sqlite3'
+
+class TTrack::DataStore
+
+  def initialize dbname=".timetrackerdb", timezone="+02:00", version='v0.1.0', verbosity=0
+    @dbname = dbname
+    @timezone = timezone
+    @version = version
+    @verbosity = verbosity
+
+    if FileTest.zero?(dbname) or not FileTest.file?(dbname)
+      @db = SQLite3::Database.new( dbname )
+      createschema
+      init(timezone, version, verbosity)
+    else
+      @db = SQLite3::Database.open( dbname )
+    end
+  end
+
+  private
+  def createschema
+    @db.execute <<-SQL
+    CREATE TABLE system (
+      current,
+      latest,
+      timezone NOT NULL,
+      version INTEGER NOT NULL,
+      verbosity NOT NULL,
+      redmine
+    )
+    SQL
+
+    @db.execute <<-SQL
+    CREATE TABLE timesheet (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      name,
+      tstart DEFAULT CURRENT_TIMESTAMP,
+      tstop,
+      synced DEFAULT 'false',
+      notes
+    )
+    SQL
+  end
+
+  def init timezone, version, verbosity
+    @db.execute 'INSERT INTO system ( timezone, version, verbosity ) VALUES ( "%s", "%s", "%s" )' %
+      [timezone, version, verbosity]
+  end
+
+  def droptables
+    for table in ['system', 'timesheet']
+      @db.execute 'DROP TABLE IF EXISTS %s' % table
+    end
+  end
+
+  def setcurrent issueid
+    oldcurrent = @db.execute 'SELECT current, latest FROM system'
+    @db.execute "UPDATE system SET current = '%d', latest = '%s'" %
+      [issueid, oldcurrent[0][0] != '' ? oldcurrent[0][0] : oldcurrent[0][1]]
+  end
+
+  public
+  def getcurrent
+    current = @db.execute "SELECT current FROM system"
+    current[0][0] == "" ? nil : current[0][0]
+  end
+
+  def getstatus
+    current = getcurrent
+    if current
+      status = @db.execute "SELECT id,name FROM timesheet WHERE id=%d" % current
+      status[0]
+    else
+      nil
+    end
+  end
+
+  def startnew issuename, notes=''
+    @db.execute "INSERT INTO timesheet ( name, notes ) VALUES ( '%s', '%s' )" %
+      [issuename, notes]
+    latest = @db.execute "SELECT id FROM timesheet ORDER BY id DESC LIMIT 1"
+    setcurrent latest[0][0]
+  end
+
+  def stoprunning
+    current = getcurrent
+    if current
+      @db.execute "UPDATE timesheet set tstop = datetime('now') WHERE id = %d" % current
+      @db.execute "UPDATE system SET current = '', latest = '%d'" % current
+    else
+      p "Not tracking"
+    end
+  end
+
+  def gettime issueid=false
+    issueid = issueid ? issueid : getcurrent
+    if issueid
+      time = @db.execute "SELECT tstart,tstop FROM timesheet WHERE id=%d" % issueid
+      time[0]
+    else
+      false
+    end
+  end
+
+  def gettimesbyissuename issuename
+    @db.execute "SELECT tstart,tstop FROM timesheet WHERE name='%s'" % issuename
+  end
+
+  def getissuesandtimesbyname issuename
+    @db.execute "SELECT tstart,tstop FROM timesheet WHERE name='%s'" % issuename
+  end
+
+  def getissuesidsbyname issuename
+    @db.execute "SELECT id,tstart,tstop,notes FROM timesheet WHERE name='%s'" % issuename
+  end
+
+  def settstart id, tstart
+    @db.execute "UPDATE timesheet SET tstart=datetime('%s') WHERE id=%d" % [tstart, id]
+  end
+
+  def settstop id, tstop
+    @db.execute "UPDATE timesheet SET tstop=datetime('%s') WHERE id=%d" % [tstop, id]
+  end
+
+  def isissue? id
+    r = @db.execute "SELECT id from timesheet WHERE id=%d" % id
+    r[0].nil? ? false : true
+  end
+
+  def cleanup
+    droptables
+    createschema
+    init(@timezone, @version, @verbosity)
+  end
+
+end  # End DataStore class

data/lib/ttrack.rb

@@ -0,0 +1,164 @@
+# -*- coding: utf-8 -*-
+class TTrack
+
+  def initialize
+    @db = DataStore.new
+  end
+
+  def start issuename, notes
+    unless issuename.nil?
+      if @db.getcurrent
+        @db.stoprunning
+      end
+      @db.startnew issuename, notes
+    else
+      usage
+    end
+  end
+
+  def stop
+    @db.stoprunning
+  end
+
+  def status issuename
+    unless issuename
+      status = @db.getstatus
+      if status
+        delta = Time.now - gettstart(@db.getcurrent)
+        puts "Task '%s' run time: %d seconds (%.2f hours)" % [status[1], delta, delta / 3600]
+      else
+        puts "Not tracking"
+      end
+    else
+      total = gettotalissueduration(issuename)
+      unless total == 0
+        puts "Total seconds for task %s: %d (%.2f hours)" % 
+          [
+            issuename,
+            total,
+            total / 3600
+          ]
+      else
+        puts "No entry for issue '%s'" % issuename
+      end
+    end
+  end
+
+  def report issuename
+    r = @db.getissuesidsbyname issuename
+    unless r[0].nil?
+      puts "id | tstart | tstop | notes"
+      r.each do |x|
+        puts "%s | %s | %s | %s" % x
+      end
+    else
+      usage
+    end
+  end
+
+  def setstart issueid, timestamp
+    if validateissueid issueid
+      t = validatetimestamp timestamp
+      @db.settstart issueid, t
+    end
+  end
+
+  def setstop issueid, timestamp
+    if validateissueid issueid
+      t = validatetimestamp timestamp
+      @db.settstop issueid, t
+    end
+  end
+
+  def init
+    @db.cleanup
+  end
+
+  def usage
+    puts "Usage: ttrack [start|stop|status|init|report|begin|end] issue_name"
+  end
+
+  private
+  def gettimeobject timestamp
+    if timestamp
+      Time.utc(
+        timestamp[0..3],
+        timestamp[5..6],
+        timestamp[8..9],
+        timestamp[11..12],
+        timestamp[14..15],
+        timestamp[17..18]
+      )
+    else
+      Time.now
+    end
+  end
+
+  def validatetimestamp timestamp
+    t = Time.utc(
+        timestamp[0..3],
+        timestamp[5..6],
+        timestamp[8..9],
+        timestamp[11..12],
+        timestamp[14..15],
+        timestamp[17..18]
+      )
+    "%.4d-%.2d-%.2d %.2d:%.2d:%.2d" % [
+      t.year,
+      t.month,
+      t.day,
+      t.hour,
+      t.min,
+      t.sec
+    ]
+  end
+
+  def validateissueid issueid
+    unless @db.isissue? issueid
+      puts "Issueid invalid"
+      false
+    else
+      true
+    end
+  end
+
+  def gettotalissueduration issuename
+    r = @db.getissuesandtimesbyname(issuename)
+    total = 0
+    r.each do |timestamps|
+      if timestamps[0]
+        t0, t1 = gettimeobject(timestamps[0]), gettimeobject(timestamps[1])
+        total = total + (t1 - t0)
+      end
+    end
+    total
+  end
+
+  def getissueduration issueid
+    timestamps = @db.gettime issueid
+    if timestamps and timestamps[1]
+      t0, t1 = gettimeobject(timestamps[0]), gettimeobject(timestamps[1])
+      t1 - t0
+    else
+      false
+    end
+  end
+
+  def gettstart issueid
+    timestamps = @db.gettime issueid
+    unless timestamps[0].nil?
+      gettimeobject timestamps[0]
+    end
+  end
+
+  def gettstop issueid
+    timestamps = @db.gettime issueid
+    unless timestamps[1].nil?
+      gettimeobject timestamps[1]
+    end
+  end
+
+end  # End TTrack class
+
+require 'ttrack/datastore'
+# TODO: add support for timezone

metadata

@@ -0,0 +1,59 @@
+--- !ruby/object:Gem::Specification
+name: ttrack
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Alexander Fortin
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-09-30 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: &70137978508080 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.3.6
+  type: :runtime
+  prerelease: false
+  version_requirements: *70137978508080
+description: A simple CLI time tracker
+email: alexander.fortin@gmail.com
+executables:
+- ttrack
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/ttrack.rb
+- lib/ttrack/datastore.rb
+- bin/ttrack
+homepage: https://rubygems.org/gems/ttrack
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: Time Tracker
+test_files: []