magicsheet 0.1

data/lib/magicsheet.rb

@@ -0,0 +1,184 @@
+require 'yahoo_finance'
+
+module MagicSheet
+  class Currency
+    attr_reader :name
+    alias :to_s :name
+    
+    LIST = %w(USD EUR)
+    
+    def self.[](str)
+      name = LIST.find do |name|
+        str.upcase.strip.end_with?(name)
+      end
+      return name ? self.new(name) : nil
+    end
+    
+    def initialize(name)
+      @name = name
+    end
+    
+    def ==(other)
+      self.name == other.name
+    end
+    
+    def conversion_rate_to(other)
+      symbol = "#{self.name}#{other.name}=X".upcase
+      YahooFinance.quotes(symbol).first.last_trade_price.to_f
+    end
+  end
+  
+  class Cost
+    attr_reader :currency, :amount
+    
+    def self.[](arg)
+      arg = arg.to_s.strip
+      currency = Currency[arg]
+      if currency
+        amount = (arg[/\d+(\.\d+)?/,0] || 0).to_f
+        return self.new(amount,currency)
+      else
+        return nil
+      end
+    end
+    
+    def initialize(amount,currency)
+      @amount = amount
+      @currency = currency
+    end
+    
+    def to_s
+      "#{"%.2f" % @amount} #{@currency}"
+    end
+  end
+  
+  class Sheet
+    DELEGATE = %w(rate fees convert_to chunk width log) # Which methods to make available in the global scope
+    DEFAULT_WIDTH = 20
+    
+    def initialize
+      @fees = []
+      @chunk = 1
+      @width = DEFAULT_WIDTH
+    end
+    
+    def width(w)
+      @width = w
+    end
+    
+    def convert_to(currency_name)
+      @convert_to = Currency[currency_name]
+    end
+    
+    def rate(rate)
+      @rate = Cost[rate]
+    end
+    
+    def fees(*list)
+      @fees = list.map {|f| Cost[f]}.compact
+    end
+    
+    def chunk(val)
+      unless val.between?(1,60)
+        raise ArgumentError, "Chunk must be between 1 (minute-wise calculation) and 60 (full-hour calculation)"
+      end
+      @chunk = val
+    end
+    
+    def scan
+      STDERR.puts "Scanning sheet..."
+      @hours, @minutes = 0, 0;
+      if defined?(DATA)
+        DATA.read.scan(/(\d+)\:(\d+)/) do |h,m|
+          @hours += h.to_i
+          @minutes += m.to_i
+        end
+        @hours += @minutes / 60
+        @minutes = @minutes % 60
+      end
+      return self
+    end
+    
+    def log(str)
+      @log << ((str == '-') ? ''.rjust(@width,'-') : str.to_s.rjust(@width))
+    end
+    
+    def process
+      @log = []
+      log "#{"%.2d" % @hours}h:#{"%.2d" % @minutes}m"
+      
+      if @chunk > 1 && @minutes > 0
+        @minutes = (@minutes.to_f/@chunk).ceil * @chunk
+        if @minutes == 60
+          @hours += 1
+          @minutes = 0
+        end
+      end
+      
+      log "= #{"%.2d" % @hours}h:#{"%.2d" % @minutes}m" if @chunk > 1
+      
+      #         if @minutes == 60
+      #           @hours += 1
+      #           @minutes = 0
+      #         end
+      #       end
+
+      log "x #{@rate}"
+      total = @hours * @rate.amount + (@minutes * @rate.amount/60.0)
+      log '-'
+      log "= #{Cost.new(total,@rate.currency)}"
+      
+      fees = false
+      
+      @fees.select {|f| f.currency == @rate.currency}.each do |f|
+        fees = true
+        total -= f.amount
+        log "- #{f}"
+      end
+      
+      if fees
+        #log '-'
+        log "= #{Cost.new(total,@rate.currency)}"
+      end
+      
+      
+      # Convert to target currency, if any
+      if @convert_to && (@convert_to != @rate.currency)
+        STDERR.puts "Getting conversion rate..."
+        c = @rate.currency.conversion_rate_to(@convert_to)
+        log ''
+        log "x #{c}"
+        log "(1h = #{Cost.new(@rate.amount * c,@convert_to)})"
+        log ''
+        total *= c
+        log "= #{Cost.new(total,@convert_to)}"
+        
+        fees = false
+        # Substract fees in target currency
+        @fees.select {|f| f.currency == @convert_to}.each do |f|
+          total -= f.amount
+          log "- #{f}"
+          fees = true
+        end
+        
+        if fees
+          log '-'
+          log "= #{Cost.new(total,@convert_to)}"
+        end
+      end
+      
+      puts
+      puts @log.join("\n")
+    end # process
+  end
+  
+  def self.current
+    @current ||= Sheet.new.scan
+  end
+end
+
+MagicSheet::Sheet::DELEGATE.each { |name| eval "def #{name}(*args); MagicSheet.current.#{name}(*args); end" }
+
+at_exit do
+  MagicSheet.current.process
+end

metadata

@@ -0,0 +1,74 @@
+--- !ruby/object:Gem::Specification 
+name: magicsheet
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  version: "0.1"
+platform: ruby
+authors: 
+- Esad Hajdarevic
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-03-06 00:00:00 +01:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: yahoo-finance
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        - 0
+        - 2
+        version: 0.0.2
+  type: :runtime
+  version_requirements: *id001
+description: Write your timesheets in ruby after the __END__. Supports currencies and tells you how much you've earned so far to keep you motivated
+email: esad@eigenbyte.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/magicsheet.rb
+has_rdoc: true
+homepage: http://github.com/esad/magicsheet
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: Timesheets in ruby
+test_files: []
+