Class: Money::Splitter

Inherits:
Object
  • Object
show all
Includes:
Enumerable
Defined in:
lib/money/splitter.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(money, num) ⇒ Splitter

Returns a new instance of Splitter.

Raises:

  • (ArgumentError) 


7
8
9
10
11
12
 
# File 'lib/money/splitter.rb', line 7

def initialize(money, num)
  @num = Integer(num)
  raise ArgumentError, "need at least one party" if num < 1
  @money = money
  @split = nil
end

Instance Attribute Details

#split ⇒ Object 



16
17
18
19
20
21
22
23
24
25
26
27
28
29
 
# File 'lib/money/splitter.rb', line 16

def split
  @split ||= begin
    subunits = @money.subunits
    low = Money.from_subunits(subunits / @num, @money.currency)
    high = Money.from_subunits(low.subunits + 1, @money.currency)

    num_high = subunits % @num

    split = {}
    split[high] = num_high if num_high > 0
    split[low] = @num - num_high
    split.freeze
  end
end

Instance Method Details

#[](index) ⇒ Object 



76
77
78
79
80
81
82
83
84
85
 
# File 'lib/money/splitter.rb', line 76

def [](index)
  offset = 0
  split.each do |money, count|
    offset += count
    if index < offset
      return money
    end
  end
  nil
end

#each(&block) ⇒ Object 



95
96
97
98
99
100
101
 
# File 'lib/money/splitter.rb', line 95

def each(&block)
  split.each do |money, count|
    count.times do
      yield money
    end
  end
end

#first(count = (count_undefined = true)) ⇒ Object 



33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
 
# File 'lib/money/splitter.rb', line 33

def first(count = (count_undefined = true))
  if count_undefined
    each do |money|
      return money
    end
  else
    if count >= size
      to_a
    else
      result = Array.new(count)
      index = 0
      each do |money|
        result[index] = money
        index += 1
        break if index == count
      end
      result
    end
  end
end

#last(count = (count_undefined = true)) ⇒ Object 



54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
 
# File 'lib/money/splitter.rb', line 54

def last(count = (count_undefined = true))
  if count_undefined
    reverse_each do |money|
      return money
    end
  else
    if count >= size
      to_a
    else
      result = Array.new(count)
      index = 0
      reverse_each do |money|
        result[index] = money
        index += 1
        break if index == count
      end
      result.reverse!
      result
    end
  end
end

#reverse ⇒ Object 



103
104
105
106
107
 
# File 'lib/money/splitter.rb', line 103

def reverse
  copy = dup
  copy.split = split.reverse_each.to_h.freeze
  copy
end

#reverse_each(&block) ⇒ Object 



87
88
89
90
91
92
93
 
# File 'lib/money/splitter.rb', line 87

def reverse_each(&block)
  split.reverse_each do |money, count|
    count.times do
      yield money
    end
  end
end

#size ⇒ Object 



109
110
111
112
113
 
# File 'lib/money/splitter.rb', line 109

def size
  count = 0
  split.each_value { |c| count += c }
  count
end