blob: ebb295539f486bfb0f1f4c75ff2d469a46cdbdce (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
# frozen_string_literal: true
require 'singleton'
require_relative 'person'
require_relative 'family_factory'
class FamilyTree
include Singleton
attr_accessor :families, :people
def initialize
@families = FamilyFactory.new.create_families
@people = []
end
def add_family(family)
@families << family unless @families.include?(family)
end
def add_child(*params)
puts "Adding Child with params: #{params.join(', ')}"
end
def get_relationship(name, relationship)
family = find_family(name)
return [] if family.nil?
if child_of_family?(family, name)
case relationship.downcase
when 'mother'
mother = find_mother(family)
return mother.is_a?(NilPerson) ? [] : [mother]
when 'father'
father = find_father(family)
return father.is_a?(NilPerson) ? [] : [father]
when 'siblings'
return find_siblings(family, name)
else
return []
end
end
[]
end
def find_family(name)
families.each do |family|
family.children.each do |child|
return family if child.name.casecmp(name).zero?
end
end
nil
end
def find_mother(family)
mother = family.mother
mother || NilPerson.new
end
def find_father(family)
father = family.father
father || NilPerson.new
end
def find_siblings(family, name)
family.children.reject { |child| child.name.casecmp(name).zero? }
end
def child_of_family?(family, name)
return false unless family.is_a?(Family)
family.children.any? do |child|
child.name.casecmp(name).zero?
end
end
end
|
