formula_eval 0.0.1

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,21 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage
+rdoc
+pkg
+
+## PROJECT::SPECIFIC

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Mike Harris
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,17 @@
+= formula_eval
+
+Description goes here.
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Copyright (c) 2010 Mike Harris. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,45 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "formula_eval"
+    gem.summary = %Q{formula_eval}
+    gem.description = %Q{formula_eval}
+    gem.email = "mharris717@gmail.com"
+    gem.homepage = "http://github.com/mharris717/formula_eval"
+    gem.authors = ["Mike Harris"]
+    gem.add_development_dependency "rspec", ">= 1.2.9"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end
+
+require 'spec/rake/spectask'
+Spec::Rake::SpecTask.new(:spec) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.spec_files = FileList['spec/**/*_spec.rb']
+end
+
+Spec::Rake::SpecTask.new(:rcov) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.pattern = 'spec/**/*_spec.rb'
+  spec.rcov = true
+end
+
+task :spec => :check_dependencies
+
+task :default => :spec
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "formula_eval #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+0.0.1

data/lib/formula_eval.rb

@@ -0,0 +1,168 @@
+require 'mharris_ext'
+require 'active_support'
+
+class Object
+  attr_accessor :calculating_formula
+end
+
+
+  
+
+class FormulaEval
+  attr_accessor :row, :formula, :row_index, :rows
+  include FromHash
+  def current_user
+    $current_user
+  end
+  def wrapped_row
+    Wrapper.new(row)
+  end
+  def method_missing(sym,*args,&b)
+    res = 'errored'
+    res = if row.respond_to?(sym)
+      wrapped_row.send(sym,*args,&b)
+    elsif row.respond_to?('[]')
+      wrapped_row[sym.to_s]
+    else
+      super
+    end
+  ensure
+    # puts "Formula mm #{sym} #{args.inspect} #{res.inspect}"
+  end
+  def safe_eval_result
+    instance_eval(formula).to_unwrapped
+  # rescue => exp
+  #   t = exp.backtrace.join("\n").gsub("/Users/mharris/.rvm/gems/ruby-1.9.1-p378/gems","gems").gsub("/Users/mharris/Code/smartlist/vendor/mongo_ui","mongo_ui")
+  #   t = t.gsub("/Users/mharris/Code/smartlist","smartlist")
+  #   mylog "formula_eval", :formula => formula, :row => row, :message => exp.message, :trace => t
+  #   "Error #{exp.message}"
+  rescue => exp
+    # puts "error evaling #{formula} against #{row.inspect}, #{exp.message}"
+    raise exp
+  end
+  def result
+    self.formula = formula[1..-1] if formula[0..0] == '='
+    res = safe_eval_result
+    eat_exceptions { res.calculating_formula = formula }
+    res
+  end
+  def call(row)
+    self.row = row
+    result
+  end
+end
+
+class FormulaEval
+  def next_row
+    res = rows[row_index+1] || {}
+    HashWrapper.new(:hash => res)
+  end
+  def prev_row_inner
+    return {} if row_index == 0
+    rows[row_index-1] || {}
+  end
+  def prev_row
+    HashWrapper.new(:hash => prev_row_inner)
+  end
+end
+
+load File.dirname(__FILE__) + "/formula_eval/wrapper.rb"
+load File.dirname(__FILE__) + "/formula_eval/calculating_collection.rb"
+load File.dirname(__FILE__) + "/formula_eval/multi_eval.rb"
+
+def mylog(*args)
+  yield if block_given?
+  #puts args.inspect if args.first == 'enriched_doc' or args.first == 'dot_set'
+end
+
+class Object
+  def klass
+    self.class
+  end
+end
+
+class Object
+  def blank?
+    to_s.strip == ''
+  end
+  def present?
+    !blank?
+  end
+end
+
+class Object
+  def dot_get(str)
+    str = str.split(".") if str.is_a?(String)
+    res = self
+    last_f = last_res = nil
+    str.each do |f|
+      if f.num? && !res.kind_of?(Array)
+        last_res[last_f] = res = []
+      end
+      last_res = res
+      if res.kind_of?(Array)
+        temp = res[f.safe_to_i]
+        if !temp
+          res << {}
+          temp = res.last
+          raise "can only add new row at end" unless res.size-1 == f.safe_to_i
+        end
+        res = temp
+      else
+        res = res[f]
+      end
+      last_f = f
+    end
+    res
+  end
+  def dot_set(str,val)
+    mylog 'dot_set', :str => str, :val => val, :self => self do
+      return self[str] = val if str.split(".").size == 1
+      strs = str.split(".")[0..-2]
+      lst = str.split(".")[-1]
+      obj = dot_get(strs)
+      return obj unless obj
+      #puts "dot_set, obj is #{obj.inspect}, str is #{str}, val is #{val}, lst is #{lst}"
+      obj.nested_set(lst,val)
+    end
+  end
+end
+
+class Object
+  def nested_set(k,v)
+    self[k] = v
+  end
+end
+
+class Array
+  def nested_set(k,v)
+    mylog 'dot_set', :context => 'nested', :klass => klass, :self => self, :k => k, :v => v do
+      each { |x| x.nested_set(k,v) } 
+    end
+  end
+end
+
+class String
+  def num?
+    size > 0 && self =~ /^[\d\.]*$/
+  end
+  def date?
+    matches = (self =~ /\/\d+\//) || (self =~ /-\d+-/)
+    matches2 = self =~ /^[ \d\-\/:]+$/
+    !!(matches && matches2 && Time.parse(self))
+  rescue
+    return false
+  end
+  def to_time
+    Time.parse(self)
+  end
+  def tmo
+    if num? 
+      to_f.tmo 
+    elsif blank?
+      nil
+    else
+      self
+    end
+  end
+end

data/lib/formula_eval/calculating_collection.rb

@@ -0,0 +1,55 @@
+class CalculatingCollection
+  attr_accessor :coll
+  include FromHash
+  fattr(:column_hash) do
+    res = {}
+    res[:zzz] = lambda { |doc| (doc['abc']||0).to_i + 7 }
+    res[:xyz] = FormulaEval.new(:formula => '=(abc||0)+8')
+    res[:percs] = FormulaEval.new(:formula => '=pension.perc')
+    res[:percs2] = FormulaEval.new(:formula => '=pension.perc * 2.0')
+    res['pension.just_perc'] = FormulaEval.new(:formula => '=perc')
+    res['pension.just_perc_plus'] = FormulaEval.new(:formula => '=perc+1')
+    res['pension.perc_plus_salary'] = FormulaEval.new(:formula => '=perc+salary')
+    res
+  end
+  def enriched_doc(doc)
+    column_hash.each do |col,blk|
+      # mylog 'enriched_doc', :col => col, :blk_class => blk.class, :doc => doc
+      if col.to_s.split('.').size == 1
+        val = blk.call(doc)
+        mylog 'dot_set',:col => col, :val => val, :doc => doc
+        doc.dot_set(col.to_s,val)  
+      else
+        base = col.to_s.split('.')[0..-2].join(".")
+        obj = MultiEval.get_nested(doc,base)
+        val = blk.call(obj)
+        mylog 'dot_set',:col => col, :val => val, :doc => doc, :obj => obj
+        doc.dot_set(col.to_s,val)
+      end
+    end
+    doc.to_unwrapped
+  end
+  def save(doc,ops={})
+    mylog 'calc_save', :doc => doc, :doc_class => doc.class, :unwrapped => doc.to_unwrapped.class
+    coll.save(enriched_doc(doc),ops)
+  end
+  def update_calcs!
+    find({},{:limit => 9999}).each do |row|
+      save(row)
+    end
+  end
+  def update_row(row_id,fields)
+    row = (row_id == 'NEW') ? {} : find_by_id(row_id)
+    raise "can't find row #{row_id} #{row_id.class} in coll #{name}.  Count is #{find.count} IDs are "+find.to_a.map { |x| x['_id'] }.inspect + "Trying to update with #{fields.inspect}" unless row
+    fields.each do |k,v|
+      row.dot_set(k,mongo_value(v))
+      row.delete(k) if v.blank?
+    end
+    row = enriched_doc(row)
+    save(row)
+    row
+  end
+  def method_missing(sym,*args,&b)
+    coll.send(sym,*args,&b)
+  end
+end

data/lib/formula_eval/multi_eval.rb

@@ -0,0 +1,56 @@
+class MultiEval
+  attr_accessor :objs
+  include FromHash
+  def methodf_missing(sym,*args,&b)
+    # puts "multi_eval mm #{sym}"
+    objs.each do |obj|
+      begin
+        res = obj.send(sym,*args,&b)
+        if res
+          #other = objs - [obj]
+          # puts "success sending #{sym} to #{obj}"
+          return MultiEval.new(:objs => [res])
+        end
+      rescue => exp
+        # puts "error sending #{sym} to #{obj} #{exp.message}"
+      end
+    end
+    objs.first.send(sym,*args,&b)
+  end
+  def without_obj(obj)
+    res = objs.reject { |x| x == obj }
+    raise "didn't find" unless objs.size == res.size+1
+    res
+  rescue => exp
+    puts "#{objs.inspect} #{obj.inspect}"
+    raise exp
+  end
+  def method_missing(sym,*args,&b)
+    objs.each do |obj|
+      if obj.respond_to?(sym)
+        res = obj.send(sym,*args,&b)
+        return MultiEval.new(:objs => [res]+without_obj(obj)) if res
+      end
+    end
+    raise 'none respond'
+    objs.first.send(sym,*args,&b)
+  end
+  def respond_to?(sym)
+    objs.any? { |x| x.respond_to?(sym) }
+  end
+  def coerce(x)
+    objs.first.coerce(x)
+  end
+  def +(x)
+    objs.first + x
+  end
+  def *(x)
+    objs.first * x
+  end
+  def self.get_nested(obj,method_str)
+    other = obj.instance_eval(method_str)
+    mylog 'get_nested', :obj => obj, :method_str => method_str, :other => other
+    arr = [other,obj]
+    new(:objs => arr)
+  end
+end

data/lib/formula_eval/wrapper.rb

@@ -0,0 +1,111 @@
+# class Wrapper
+#   attr_accessor :obj
+#   include FromHash
+#   def method_missing(sym,*args,&b)
+#     obj.send(sym,*args,&b).to_wrapped
+#   end
+# end
+
+class Wrapper
+  def respond_to?(sym)
+    obj.respond_to?(sym)
+  end
+end
+
+class HashWrapper 
+  attr_accessor :hash
+  include FromHash
+  def method_missing(sym,*args,&b)
+    if obj.keys.include?(sym.to_s)
+      self[sym.to_s]
+    else
+      super
+    end
+  end
+  def [](x)
+    raise 'dot_set' if x.to_s == 'dot_set'
+    res = hash[x.to_s]
+    res = nil if res.blank?
+    # raise "nil key #{x}" unless res
+    res.to_wrapped
+  end
+  def []=(k,v)
+    obj[k.to_s] = v
+  end
+  def obj; hash; end
+  def to_unwrapped
+    hash.to_unwrapped
+  end
+  def respond_to?(sym)
+    obj.keys.include?(sym.to_s)
+  end
+end
+
+class ArrayWrapper
+  attr_accessor :obj
+  include FromHash
+  def hash_mm(sym)
+    map { |h| h[sym.to_s] }.select { |x| x }.flatten.to_wrapped
+  end
+  def method_missing(sym,*args,&b)
+    if obj.respond_to?(sym)
+      obj.send(sym,*args,&b).to_wrapped
+    elsif contains_all_hashes?
+      hash_mm(sym)
+    else
+      obj.send(sym,*args,&b).to_wrapped
+    end
+  end
+  def *(arg)
+    map { |x| x * arg }
+  end
+  def to_unwrapped
+    obj.to_unwrapped
+  end
+end
+
+class Array
+  def to_unwrapped
+    map { |x| x.to_unwrapped } 
+  end
+end
+
+class Hash
+  def to_unwrapped
+    map_value { |v| v.to_unwrapped }
+  end
+end
+
+class Wrapper
+  def self.new_inner(obj)
+    if obj.kind_of?(Array)
+      ArrayWrapper.new(:obj => obj)
+    elsif obj.kind_of?(Hash)
+      HashWrapper.new(:hash => obj)
+    else
+      obj
+    end
+  end
+  def self.new(obj)
+    new_inner(obj).tap { |x| mylog 'wrapper', :obj => obj, :wrapper_class => x.class }
+  end
+  def self.wrapped?(obj)
+    [ArrayWrapper,HashWrapper].any? { |cls| obj.kind_of?(cls) }
+  end
+end
+
+class Object
+  def to_wrapped
+    Wrapper.new(self)
+  end
+  def to_unwrapped_inner
+    if Wrapper.wrapped?(self)
+      obj.to_unwrapped
+    else
+      self
+    end
+  end
+  def to_unwrapped
+    to_unwrapped_inner.tap { |x| mylog 'unwrap', :klass => self.klass, :unwrapped => x.class }
+  end
+end

data/lib/spec.opts

@@ -0,0 +1 @@
+--color

data/spec/calculating_collection_spec.rb

@@ -0,0 +1,26 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe "CalculatingCollection" do
+  fattr(:row) { {'pension' => {'year' => 2025, 'perc' => 0.65}, 'salary' => 42000, 'year' => 2010} }
+  fattr(:wrapped_row) { Wrapper.new(row) }
+  fattr(:formula) { FormulaEval.new(:row => row) }
+  fattr(:column) { 'pension.perc_plus_salary' }
+  fattr(:str) { '=perc*salary' }
+  fattr(:calc) do
+    CalculatingCollection.new(:column_hash => {column => FormulaEval.new(:formula => str)})
+  end
+  fattr(:enriched) do
+    calc.enriched_doc(wrapped_row)
+  end
+  it 'enriched' do
+    enriched['pension'].should == {'year' => 2025, 'perc' => 0.65, 'perc_plus_salary' => 42000.0*0.65}
+  end
+  it 'enriched 2' do
+    calc.column_hash['double_year'] = FormulaEval.new(:formula => '=year*2')
+    enriched['double_year'].should == 4020
+  end
+  it 'uses sub first' do
+    self.str = '=year+1'
+    enriched['pension']['perc_plus_salary'].should == 2026
+  end
+end

data/spec/formula_eval_spec.rb

@@ -0,0 +1,60 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+class A
+  def a
+    1
+  end
+  def [](k)
+    #puts "sending #{k} to #{self}"
+    send(k)
+  end
+end
+
+class B
+  def b
+    2
+  end
+  def [](k)
+    #puts "sending #{k} to #{self}"
+    send(k)
+  end
+end
+
+def bt
+  raise 'foo'
+rescue => exp
+  puts exp.message
+  puts exp.backtrace.join("\n")
+end
+
+describe "FormulaEval" do
+  fattr(:str) { '=2+2' }
+  fattr(:row) { {'year' => 2010} }
+  fattr(:formula) { FormulaEval.new(:row => row, :formula => str) }
+  it '2+2' do
+    formula.result.should == 4
+  end
+  it 'double_year' do
+    self.str = '=year*2'
+    formula.result.should == 4020
+  end
+  it 'nested' do
+    
+  end
+  it 'double' do
+    self.row = MultiEval.new(:objs => [A.new,B.new])
+    self.str = '=a+b'
+    formula.result.should == 3
+  end
+  it 'double 2' do
+    hash = {'year' => 2010, 'pension' => {'start_year' => 2025, 'perc' => 0.65}}
+    self.row = MultiEval.get_nested(Wrapper.new(hash),'pension')
+    self.str = '=perc*year + perc*year'
+    formula.result.should == 0.65 * 2010 * 2
+  end
+  it 'with hash' do
+    self.row = {'year' => 2010, 'pension' => {'start_year' => 2025, 'perc' => 0.65}}
+    self.str = '=year'
+    formula.result.should == 2010
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,9 @@
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'formula_eval'
+require 'spec'
+require 'spec/autorun'
+
+Spec::Runner.configure do |config|
+  
+end

metadata

@@ -0,0 +1,91 @@
+--- !ruby/object:Gem::Specification 
+name: formula_eval
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Mike Harris
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-05-20 00:00:00 -04:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 1
+        - 2
+        - 9
+        version: 1.2.9
+  type: :development
+  version_requirements: *id001
+description: formula_eval
+email: mharris717@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- .document
+- .gitignore
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- lib/formula_eval.rb
+- lib/formula_eval/calculating_collection.rb
+- lib/formula_eval/multi_eval.rb
+- lib/formula_eval/wrapper.rb
+- lib/spec.opts
+- spec/calculating_collection_spec.rb
+- spec/formula_eval_spec.rb
+- spec/spec_helper.rb
+has_rdoc: true
+homepage: http://github.com/mharris717/formula_eval
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+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: formula_eval
+test_files: 
+- spec/calculating_collection_spec.rb
+- spec/formula_eval_spec.rb
+- spec/spec_helper.rb