wip for ruby

gistfile1.rb

def add(*numbers)
  numbers.inject(0) {|sum, number| sum += number}
end

def subtract(*numbers)
  numbers.inject {|result, number| result -= number}
end

def calculate(*args)
  options = {'add'=>true}
  if(args.last.is_a?(Hash))
    options = args.last
    args.delete_at(-1)
  end
  (return add(*args)) if options['add']
  (return subtract(*args)) if options['subtract']
end

# p add(1,2)
# p add([1,2])
p calculate(1,2,{'subtract'=>true})
p calculate(1,2,3,{'subtract'=>true})
p calculate(1,2, {'add'=>true})
p calculate(1,2,3,4, {'add'=>true})