Pratt 1.6.2 → 1.6.4

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/{array.rb → core_ext/array.rb}

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

data/lib/pratt/core_ext/float.rb

@@ -0,0 +1,28 @@
+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/{string.rb → core_ext/string.rb}

@@ -16,3 +16,16 @@ 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 { 
+      puts "pratt (#{Process.pid.to_s.yellow})"
+      app.pid = Process.pid
+      app.save!
+
+      gui
+      while(daemonized?)
+        sleep(app.interval)
+        gui
+      end
+      quit
+    }
+  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
+      defork do
+        command = "ruby views/pop.rb  --project '#{project.name}' --start '#{project.whences.last_unended.start_at}' --project_time '#{Pratt.totals(project.time_spent)}'"
+        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,24 @@
+class Pratt
+
+  @@color = true
+
+  class << self
+
+    def color
+      @@color
+    end
+
+    def color= color_bool
+      @@color = color_bool
+    end
+
+    def color?
+      @@color == true
+    end
+
+    # Calculate totals. I think this should be an instance method on Projects/?/Whences
+    def totals hr, fmt = false
+      "#{(hr / 24).format_integer.cyan} day #{(hr % 24).format_integer.yellow} hour #{(60*(hr -= hr.to_i)).format_integer.green} min"
+    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,127 @@
+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.whences.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
+      "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(scale, when_to)
+    else
+      @projects = (Project.all - [Project.primary, Project.off])
+      @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(scale, when_to)
+        total
+      end
+    end
+
+    @projects.each do |project| 
+      puts "<!-- #{project.name} #{project.payment.inspect} -->"
+    end
+    if @total > 0.0
+      process_template!
+    else
+      puts "No data to report"
+    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
+    `pgrep -fl 'bin/pratt'`.chomp.split(' ').first
+  end
+
+  def pid
+    self.template = 'pid'
+    process_template!
+  end
+
+  private
+    def process_template!
+      input = File.open(Pratt.root("views", "#{template}.eruby").first).read
+      erubis = Erubis::Eruby.new(input)
+      puts erubis.evaluate(self)
+    end
+
+end

data/models/app.rb

@@ -6,9 +6,8 @@ class App < ActiveRecord::Base
     save!
   end
 
-  def gui? which = '', log_err = false
+  def gui? which = ''
     match = (self.gui == which)
-#    $stderr.write("#{which} already being displayed\n") if log_err
     match
   end
 

data/models/customer.rb

@@ -1,10 +1,29 @@
 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 = '.'
+        self.split(/(\d{3})(\d{3})(\d{4})/)[1,3] * sep
+      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
@@ -12,7 +31,10 @@ class Customer < ActiveRecord::Base
         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

data/models/invoice.rb

@@ -0,0 +1,28 @@
+class Invoice < ActiveRecord::Base
+	belongs_to :customer
+	has_many :invoice_whences
+	has_many :whences, :through => :invoice_whences
+
+	validates_presence_of :customer_id, :weekday_start
+	validates_format_of :invoice_number, :with => /(0?[1-9]+[0-9]*)-([0-4][0-9]|5[0-4])-(0?[1-9]+[0-9]*)/, :on => :create
+
+	def before_validation_on_create
+	  self.invoice_number = "%s-%s-%s"% [customer.id, weekday_start.week, customer.invoices.count+1 ]
+	end
+
+    def self.migrate which = :up 
+      ActiveRecord::Schema.define do
+        if which == :up
+          create_table :invoices do |t|
+            t.references :customer
+			t.datetime :weekday_start
+			t.datetime :weekday_end
+            t.string 'invoice_number'
+			t.integer :total
+          end
+        elsif which == :down
+          drop_table :invoices
+        end
+      end
+    end
+end