From d9453c132348ca7e2cd2d5c8fc7ea7c10f480626 Mon Sep 17 00:00:00 2001
From: naomi <naomi@happycatsin.space>
Date: Fri, 12 Nov 2021 09:11:00 +0000
Subject: [PATCH] Added amount field to case fee, total fee to case, and code
 to calculate the total

---
 modules/opencase_cases/opencase_cases.install | 12 +++++
 modules/opencase_cases/opencase_cases.module  | 46 +++++++++++++++++++
 modules/opencase_cases/src/Entity/OCCase.php  | 10 ++++
 3 files changed, 68 insertions(+)

diff --git a/modules/opencase_cases/opencase_cases.install b/modules/opencase_cases/opencase_cases.install
index fbf8e6b..7025221 100644
--- a/modules/opencase_cases/opencase_cases.install
+++ b/modules/opencase_cases/opencase_cases.install
@@ -72,3 +72,15 @@ function opencase_cases_update_8029() {
   \Drupal::entityDefinitionUpdateManager()
     ->installFieldStorageDefinition('notes', 'oc_case', 'oc_case', $storage_definition);
 }
+
+function opencase_cases_update_8030() {
+    $storage_definition =  \Drupal\Core\Field\BaseFieldDefinition::create('decimal')
+    ->setLabel(t('Total Fee'))
+    ->setRevisionable(TRUE)
+    ->setSettings([
+      'prefix' => '£',
+    ])
+    ->SetDisplayConfigurable("view", true);
+  \Drupal::entityDefinitionUpdateManager()
+    ->installFieldStorageDefinition('total_fee', 'oc_case', 'oc_case', $storage_definition);
+}
diff --git a/modules/opencase_cases/opencase_cases.module b/modules/opencase_cases/opencase_cases.module
index 3a0c302..781a599 100644
--- a/modules/opencase_cases/opencase_cases.module
+++ b/modules/opencase_cases/opencase_cases.module
@@ -1,4 +1,5 @@
 <?php
+declare(strict_types = 1);
 
 /**
  * @file
@@ -64,6 +65,51 @@ function opencase_cases_theme() {
   ];
 }
 
+/**
+ * @return Drupal\opencase_cases\entity\OCCaseFee[]
+ */
+function opencase_cases_get_case_fees(Drupal\opencase_cases\entity\OCCase $case): array {
+  $query = \Drupal::entityQuery('oc_case_fee')
+    ->condition('oc_case.target_id', $case->id());
+  $case_fee_ids =  $query->execute();
+  $case_fees = [];
+  foreach($case_fee_ids as $id) {
+    $case_fees[] = Drupal\opencase_cases\entity\OCCaseFee::load($id);
+  }
+  return $case_fees;
+}
+
+function opencase_cases_get_amount(Drupal\opencase_cases\entity\OCCaseFee $case_fee): string {
+  return $case_fee->amount->first()->value;
+}
+
+function opencase_cases_get_case(Drupal\opencase_cases\entity\OCCaseFee $case_fee): Drupal\opencase_cases\entity\OCCase {
+  return $case_fee->oc_case->referencedEntities()[0];
+}
+
+function opencase_cases_update_total_fee_for_case_belonging_to_case_fee(Drupal\opencase_cases\entity\OCCaseFee $case_fee): void {
+  $total = 0;
+  $case = opencase_cases_get_case($case_fee);
+  $case_fees = opencase_cases_get_case_fees($case);
+  foreach($case_fees as $case_fee) {
+    $total += opencase_cases_get_amount($case_fee);
+  }
+  $case->set('total_fee', $total);  
+  $case->save();
+}
+
+function opencase_cases_oc_case_fee_update(Drupal\opencase_cases\entity\OCCaseFee $case_fee): void {
+  opencase_cases_update_total_fee_for_case_belonging_to_case_fee($case_fee);
+}
+
+function opencase_cases_oc_case_fee_insert(Drupal\opencase_cases\entity\OCCaseFee $case_fee): void {
+  opencase_cases_update_total_fee_for_case_belonging_to_case_fee($case_fee);
+}
+
+function opencase_cases_oc_case_fee_delete(Drupal\opencase_cases\entity\OCCaseFee $case_fee): void {
+  opencase_cases_update_total_fee_for_case_belonging_to_case_fee($case_fee);
+}
+
 function opencase_cases_entity_base_field_info($entity_type) {
   $fields = array();
   
diff --git a/modules/opencase_cases/src/Entity/OCCase.php b/modules/opencase_cases/src/Entity/OCCase.php
index f0c22d4..3baee8b 100644
--- a/modules/opencase_cases/src/Entity/OCCase.php
+++ b/modules/opencase_cases/src/Entity/OCCase.php
@@ -305,6 +305,16 @@ class OCCase extends RevisionableContentEntityBase implements OCCaseInterface {
       ))
       ->setDisplayConfigurable('form', true)
       ->setDisplayConfigurable('view', true);
+
+
+    $fields['total_fee'] = BaseFieldDefinition::create('decimal')
+    ->setLabel(t('Total Fee'))
+    ->setRevisionable(TRUE)
+    ->setSettings([
+      'prefix' => '£',
+    ])
+    ->SetDisplayConfigurable("view", true);
+
     return $fields;
   }
 }