Aam-Digital/ndb-core

/*
 *     This file is part of ndb-core.
 *
 *     ndb-core is free software: you can redistribute it and/or modify
 *     it under the terms of the GNU General Public License as published by
 *     the Free Software Foundation, either version 3 of the License, or
 *     (at your option) any later version.
 *
 *     ndb-core is distributed in the hope that it will be useful,
 *     but WITHOUT ANY WARRANTY; without even the implied warranty of
 *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *     GNU General Public License for more details.
 *
 *     You should have received a copy of the GNU General Public License
 *     along with ndb-core.  If not, see <http://www.gnu.org/licenses/>.
 */

import { Entity } from "../../../../core/entity/model/entity";
import { DatabaseEntity } from "../../../../core/entity/database-entity.decorator";
import { DatabaseField } from "../../../../core/entity/database-field.decorator";
import { WarningLevel } from "../../../warning-level";
import { Child } from "../../model/child";
import { PLACEHOLDERS } from "../../../../core/entity/schema/entity-schema-field";

/**
 * Model Class for the Health Checks that are taken for a Child.
 * It stores the Child's ID in a String and both, the height and weight in cm as a number, and the Date
 */
@DatabaseEntity("HealthCheck")
export class HealthCheck extends Entity {
  static override hasPII = true;

  static create(contents: Partial<HealthCheck>) {
    return Object.assign(new HealthCheck(), contents);
  }

  @DatabaseField({
    dataType: "entity",
    additional: Child.ENTITY_TYPE,
    entityReferenceRole: "composite",
    anonymize: "retain",
  })
  child: string;

  @DatabaseField({
    label: $localize`:Label for date of a health check:Date`,
    anonymize: "retain-anonymized",
    defaultValue: {
      mode: "dynamic",
      value: PLACEHOLDERS.NOW,
    },
  })
  date: Date;

  /** height measurement in cm **/
  @DatabaseField({
    label: $localize`:Label for height in cm of a health check:Height [cm]`,
    viewComponent: "DisplayUnit",
    additional: "cm",
  })
  height: number;

  /** weight measurement in kg **/
  @DatabaseField({
    label: $localize`:Label for weight in kg of a health check:Weight [kg]`,
    viewComponent: "DisplayUnit",
    additional: "kg",
  })
  weight: number;

  /**
   * dynamically calculated BMI value based on the height and weight, rounded to 2 decimal digits
   */
  get bmi(): number {
    const bmi = this.weight / ((this.height / 100) * (this.height / 100));
    return Math.round(bmi * 100) / 100;
  }

  getWarningLevel(): WarningLevel {
    if (this.bmi <= 16 || this.bmi >= 30) {
      return WarningLevel.URGENT;
    } else if (this.bmi >= 18 && this.bmi <= 25) {
      return WarningLevel.OK;
    } else {
      return WarningLevel.WARNING;
    }
  }

  getColor(): string {
    if (!this.height) {
      return "#DEDEDE";
    } else {
      return super.getColor();
    }
  }
}