From 9bc26146397acb5a216e20d5eb55bb2a582fdd3e Mon Sep 17 00:00:00 2001
From: Szymon Szukalski <szymon@skas.io>
Date: Thu, 24 Oct 2024 21:10:57 +1100
Subject: Implement key data model

- Added classes for Person, Gender, Family, FamilyTree
- Replaced FamilyTreeManager with FamilyTree
- Add FamilyFactory for seeding the initial FamilyTree for King Arthur and Queen Margaret
- Added a RelationshipManager for linking spouses correctly
- Refactored ActionFileExecutor for readability
- More test coverage
---
 lib/family.rb | 43 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 lib/family.rb

(limited to 'lib/family.rb')

diff --git a/lib/family.rb b/lib/family.rb
new file mode 100644
index 0000000..2c1acca
--- /dev/null
+++ b/lib/family.rb
@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+require_relative 'person'
+
+class Family
+  attr_reader :mother, :father, :children
+
+  def initialize(mother = NilPerson.new, father = NilPerson.new, children = [])
+    raise ArgumentError, 'Mother must be female' if !mother.is_a?(NilPerson) && mother.gender != Gender::FEMALE
+    raise ArgumentError, 'Father must be male' if !father.is_a?(NilPerson) && father.gender != Gender::MALE
+
+    @mother = mother
+    @father = father
+    @children = children
+  end
+
+  def assign_mother(mother)
+    raise ArgumentError, 'Mother must be female' if mother.gender != Gender::FEMALE
+
+    @mother = mother
+  end
+
+  def assign_father(father)
+    raise ArgumentError, 'Father must be male' if father.gender != Gender::MALE
+
+    @father = father
+  end
+
+  def add_child(child)
+    @children << child unless @children.include?(child)
+  end
+
+  def get_siblings(person)
+    @children.reject { |child| child == person }
+  end
+
+  def to_s
+    mother_str = mother.is_a?(NilPerson) ? 'UNKNOWN' : mother.name
+    father_str = father.is_a?(NilPerson) ? 'UNKNOWN' : father.name
+    children_str = children.empty? ? 'NONE' : children.map(&:name).join(', ')
+    "Family: Mother: #{mother_str}, Father: #{father_str}, Children: #{children_str}"
+  end
+end
-- 
cgit v1.2.3