Pratt 1.6.8-x86-linux

data/lib/pratt/core_ext.rb

@@ -0,0 +1,6 @@
+require 'lib/pratt/core_ext/array'
+require 'lib/pratt/core_ext/float'
+require 'lib/pratt/core_ext/nil'
+require 'lib/pratt/core_ext/numeric'
+require 'lib/pratt/core_ext/string'
+require 'lib/pratt/core_ext/time'

data/lib/pratt/core_ext/array.rb

@@ -0,0 +1,20 @@
+class Pratt
+  class Formats
+    module Array
+      ##
+      # Turns an array into a sentence
+      #
+      # @param [String] conjunction
+      # @return [String] 
+      def to_sentence conjunction = 'and'
+        if self.size >= 2
+          self[0..-2].join(", ") << (self.size > 2 ? ',' : '') << " #{conjunction} #{self.last}"
+        elsif self.size <= 1
+          self.first.to_s
+        end
+      end
+    end
+  end
+end
+
+Array.send :include, Pratt::Formats::Array

data/lib/pratt/core_ext/float.rb

@@ -0,0 +1,29 @@
+class Float
+  def pretty_print
+    "%0.2f"% self
+  end
+
+  def percentage of_total = 1
+    ( (self/of_total)*100.0 ).pretty_print << "%"
+  end
+
+  def to_money
+    Money.new(self)
+  end
+end
+
+class Money
+  include Comparable
+  
+  def <=> other
+    @f <=> other
+  end
+
+  def initialize f
+    @f = Float(f)
+  end
+
+  def pretty_print
+    "$#{@f.pretty_print}"
+  end
+end

data/lib/pratt/core_ext/nil.rb

@@ -0,0 +1,5 @@
+class NilClass
+	def pretty_print
+	  ''
+	end
+end

data/lib/pratt/core_ext/numeric.rb

@@ -0,0 +1,9 @@
+class Numeric
+  def format_integer
+    colored_integer = ("%02i"% self)
+  end
+
+  def with_label label
+    self.to_s.with_label label
+  end
+end

data/lib/pratt/core_ext/string.rb

@@ -0,0 +1,31 @@
+class Pratt
+  module ColorString
+    def or format_alternative
+      return self unless Pratt.color?
+      format_alternative
+    end
+  end
+
+  module Colored
+    def colorize(string, options = {})
+      return self unless Pratt.color?
+      super 
+    end
+  end
+end
+
+String.send :include, Pratt::ColorString
+String.send :include, Pratt::Colored
+
+class String
+  def with_label label
+    "#{padded_to_max(label)} #{self}"
+  end
+
+  private
+    # Pad the output string to the maximum Project name
+    def padded_to_max string
+      project_padding = Project.longest_project_name
+      "%#{project_padding}.#{project_padding}s"% string
+    end
+end

data/lib/pratt/core_ext/time.rb

@@ -0,0 +1,20 @@
+class Date
+  ##
+  # Return "zero hour" of the first day of the, possibly shifted, week.
+  #
+  # @return [Date]
+  def beginning_of_week
+    (self - wday_offset).beginning_of_day
+  end
+
+  ##
+  # Return "last second" of the last day of the, possibly shifted, week.
+  #
+  # @return [Date]
+  def end_of_week
+    (beginning_of_week+6).end_of_day
+  end
+end
+
+class DateTime < Date
+end

data/lib/pratt/dialogs.rb

@@ -0,0 +1,81 @@
+class Pratt
+  def daemonized?
+    !app.pid.blank? and ( cpid.to_i == app.pid )
+  end
+
+  def daemonize!
+    defork do
+      puts "pratt (#{Process.pid.to_s.yellow})"
+      app.pid = Process.pid
+      app.save!
+
+      while daemonized?
+        gui
+        sleep(app.interval)
+      end
+      quit
+    end
+  end
+
+  def gui
+    if Whence.last_unended
+      pop
+    else
+      main
+    end
+  end
+
+  def detect
+    if self.daemonized?
+      gui
+    else
+      daemonize!
+    end
+  end
+
+  private
+    def main
+      reload_and_detect_lock 'main'
+      projects = ([Project.primary, Project.off] | Project.rest).collect(&:name)
+      if Whence.count == 0 
+        # first run
+        project = Whence.new(:project => Project.new)
+        current_project_name = project.project.name
+      else
+        project = Whence.last_unended || Whence.last
+        current_project_name = ''
+      end
+
+      if Project.count > 0
+        projects = ([Project.primary, Project.off] | Project.rest).compact.collect(&:name)
+      else
+        projects = []
+      end
+      defork { 
+        command = "ruby views/main.rb  --projects '#{projects*"','"}' --current '#{current_project_name}'"
+        system command
+      } 
+    end
+
+    def pop
+      reload_and_detect_lock 'pop'
+      self.project = Whence.last_unended.project
+      project_time = project.formatted_time_spent_totals( project.time_spent(scale, when_to) )
+      defork do
+        command = "ruby views/pop.rb  --project '#{project.name}' --start '#{project.whences.last_unended.start_at}' --project_time '#{project_time}'"
+        system command
+      end
+    end
+
+    def reload_and_detect_lock of
+      self.app.reload
+      return if self.app.gui? of
+      self.app.log of
+    end
+
+    def defork &block
+      Process.detach(
+        fork &block 
+      )
+    end
+end

data/lib/pratt/formatting.rb

@@ -0,0 +1,19 @@
+class Pratt
+
+  @@color = true
+
+  class << self
+
+    def color
+      @@color
+    end
+
+    def color= color_bool
+      @@color = color_bool
+    end
+
+    def color?
+      @@color == true
+    end
+  end
+end

data/lib/pratt/project_actions.rb

@@ -0,0 +1,25 @@
+class Pratt
+  def begin
+    self.project.start! when_to
+  end
+  def restart
+    if project?
+      project.restart! when_to
+    else
+      Whence.last_unended.project.restart! when_to
+    end
+  end
+  def end
+    if project?
+      project.stop! when_to
+    else
+      Whence.last_unended.stop! when_to
+    end
+  end
+  def change
+    Whence.last.change! project.name
+  end
+  def destroy
+    project.destroy
+  end
+end

data/lib/pratt/reports.rb

@@ -0,0 +1,140 @@
+class Pratt
+
+  # TODO Rename
+  def graph
+    self.template = 'graph'
+
+    if project?
+      @projects = [ project ]
+    else
+      @projects = Project.all
+    end
+
+    process_template!
+  end
+
+  def proportions
+    @primary = @off_total = @rest_total = 0.0
+    self.template = 'proportions'
+
+    if project?
+      @projects = [project]
+
+      @primary = project.time_spent(scale, when_to)
+      @scaled_total = project.time_spent(scale, when_to)
+    else
+      @projects = Project.all
+
+      @projects.each do |proj| 
+        @primary     = proj.time_spent(scale, when_to) if proj.name == Project.primary.name
+        @off_total   = proj.time_spent(scale, when_to) if proj.name == Project.off.name
+        @rest_total += proj.time_spent(scale, when_to) if Project.rest.collect(&:name).include?(proj.name)
+      end
+      @scaled_total = Whence.time_spent(scale, when_to) - @off_total
+    end
+
+    if @primary + @off_total + @rest_total > 0.0
+      process_template!
+    else
+      $stdout.puts "No data to report"
+    end
+  end
+
+  # Generate an invoice for a given time period
+  def invoice
+    self.template ||= 'invoice'
+
+    if project?
+      @projects = [project]
+
+      @total = project.amount( project.time_spent(scale, when_to) )
+    else
+      @projects = (Project.all - [Project.primary, Project.off])
+      @projects = @projects.select {|proj| show_all or ( !show_all and proj.time_spent(scale, when_to) != 0.0 ) }
+
+      @total = @projects.inject 0.0 do |total, proj| 
+        total += proj.amount( proj.time_spent(scale, when_to) )
+        total
+      end
+    end
+
+    @projects.each do |project| 
+      puts "<!-- #{project.name} #{project.payment.inspect} -->"
+    end
+    if @total > 0.0
+      process_template!
+    else
+      $stdout.puts "No data to report in Pratt#invoice"
+    end
+  end
+
+  def current
+    self.template = 'current'
+    @last_whence = Whence.last_unended || Whence.last || Whence.new( :end_at => nil, :project => Project.new )
+
+    projects = ([Project.primary, Project.off] | Project.rest).compact
+    @project_names = projects.collect do |proj| 
+
+      if @last_whence.end_at.nil? && @last_whence.project.name == proj.name
+        colored_name = proj.name.green
+      else
+        colored_name = proj.name.magenta
+      end
+      "'" << colored_name << "'"
+    end
+    @time_til = ( app.interval - ( Time.now - @last_whence.start_at ) ) if @last_whence.end_at.nil?
+
+    process_template!
+  end
+
+  def raw
+    self.template = 'raw'
+
+    if project?
+      @whences = project.whences.all
+    else
+      case raw_conditions
+        when 'all'
+          @whences = Whence.find raw_conditions.to_sym
+        when /^last$/, 'first'
+          @whences = [Whence.find raw_conditions.to_sym]
+        when /last[\(\s]?(\d+)[\)\s]?/
+          @whences = Whence.all.last($1.to_i)
+        else
+          @whences = Whence.all :conditions => ["start_at > ?", Chronic.parse("today 00:00")]
+      end
+    end
+    @whences.sort_by(&:id)
+    process_template!
+  end
+
+  def cpid
+    rubies = %x(ps -C ruby -o pid=,cmd=)
+    pratts = rubies.select do |ruby_process|
+      pid, interp, cmd, args = ruby_process.chomp.split(' ')
+      cmd =~ /pratt/ && args == '-d'
+    end
+    return '' unless pratts.size == 1
+    pratts.first.split(' ').first
+  end
+
+  def pid
+    self.template = 'pid'
+    process_template!
+  end
+
+  private
+   def output
+     # for testing
+     @__output
+   end
+
+    def process_template!
+      input = File.open(Pratt.root("views", "#{template}.eruby").first).read
+      erubis = Erubis::Eruby.new(input)
+      @__output = erubis.evaluate(self)
+      $stdout.puts @__output
+      nil
+    end
+
+end

data/models/app.rb

@@ -0,0 +1,39 @@
+class App < ActiveRecord::Base
+  has_many :logs
+
+  def log which, overwrite = false
+    self.gui = which if (overwrite and gui?(which) ) || gui?('')
+    save!
+  end
+
+  def gui? which = ''
+    match = (self.gui == which)
+    match
+  end
+
+  def unlock
+    self.gui = ''
+    self.save!
+  end
+
+  def interval= intvl
+    write_attribute(:interval, intvl*60)
+  end
+
+  class << self
+    def migrate which = :up
+      ActiveRecord::Schema.define do
+        if which == :up
+          create_table :apps do |t|
+            t.integer :pid
+            t.string  :gui,      :default => ''
+            t.float   :interval, :default => 15.0*60
+          end
+          App.create!(:gui => '', :interval => 15.0)
+        elsif which == :down
+          drop_table :apps
+        end
+      end
+    end
+  end
+end

data/models/customer.rb

@@ -0,0 +1,50 @@
+class Customer < ActiveRecord::Base
+  has_many :projects
+  has_many :invoices
+  has_one :payment, :as => :billable
+
+  validates_presence_of :name
+
+  def amount
+    payment.rate / 100.0
+  end
+
+  def phone
+    phone = read_attribute(:phone)
+    class << phone
+      def pretty_print sep = '.'
+        if self.blank?
+          ""
+        else
+          self.split(/(\d{3})(\d{3})(\d{4})/)[1,3] * sep
+        end
+      end
+    end
+    phone
+  end
+
+  def city_state_zip
+    a_zip = Zip.find_by_zip(self.zip)
+    return "#{a_zip.city}, #{a_zip.ST} #{a_zip.zip}" if a_zip
+    return self.zip
+  end
+ 
+  class << self
+    def migrate which = :up
+      ActiveRecord::Schema.define do
+        if which == :up
+          create_table :customers do |t|
+            t.string :name, :null => false
+            t.string :company_name
+            t.string :address
+            t.string :phone
+            t.string :email
+            t.string :zip, :limit => 5
+          end
+        elsif which == :down
+          drop_table :customers
+        end
+      end
+    end
+  end
+end