#!/bin/bash
# SPDX-Version: 3.0
# SPDX-CreationInfo: 2025-06-17; WEIDNER, Marc S.; <msw@coresecret.dev>
# SPDX-ExternalRef: GIT https://git.coresecret.dev/msw/CISS.debian.installer.git
# SPDX-FileContributor: WEIDNER, Marc S.; Centurion Intelligence Consulting Agency
# SPDX-FileCopyrightText: 2024-2025; WEIDNER, Marc S.; <msw@coresecret.dev>
# SPDX-FileType: SOURCE
# SPDX-License-Identifier: EUPL-1.2 OR LicenseRef-CCLA-1.0
# SPDX-LicenseComment: This file is part of the CISS.debian.installer.secure framework.
# SPDX-PackageName: CISS.debian.installer
# SPDX-Security-Contact: security@coresecret.eu

guard_sourcing

#######################################
# Function that generates each partition on each device according to the chosen recipe string.
# Globals:
#   ERR_PARTITIONTBL
#   ERR_READ_PARTTBL
#   HMP_UUID_PARTITION
#   RECIPE_STRING
#   VAR_RECIPE_TABLE
#   VAR_SETUP_PART
# Arguments:
#  None
#######################################
partitioning() {
  ### Declare Arrays and Variables.
  declare -Ag HMP_UUID_PARTITION
  declare var_dev var_part var_begin var_end var_fs var_boot var_pri var_uuid

  ### Iterate over all devices in the recipe.
  for var_dev in $(yq e ".recipe.${VAR_RECIPE_STRING}.dev | keys | .[]" "${VAR_SETUP_PART}"); do

    ### All current data for the respective device will be deleted.
    blkdiscard "/dev/${var_dev}" \
      || { do_log "fatal" "false" "Partition table deletion '/dev/${var_dev}' failed."; exit "${ERR_READ_PARTTBL}"; }
    do_log "info" "false" "Partition table of '/dev/${var_dev}' discarded."

    if [[ "${VAR_RECIPE_TABLE}" == "gpt" ]]; then

      parted -s "/dev/${var_dev}" mklabel gpt \
        || { do_log "fatal" "false" "Partition table creation '/dev/${var_dev}' failed."; exit "${ERR_READ_PARTTBL}"; }
      do_log "info" "false" "Partition table '${VAR_RECIPE_TABLE}' of '/dev/${var_dev}' generated."

    elif [[ "${VAR_RECIPE_TABLE}" == "mbr" ]]; then

      parted -s "/dev/${var_dev}" mklabel mbr \
        || { do_log "fatal" "false" "Partition table creation '/dev/${var_dev}' failed."; exit "${ERR_READ_PARTTBL}"; }
      do_log "info" "false" "Partition table '${VAR_RECIPE_TABLE}' of '/dev/${var_dev}' generated."

    else

      do_log "fatal" "false" "No valid partition table chosen. String was '${VAR_RECIPE_TABLE}'."
      exit "${ERR_PARTITIONTBL}"

    fi

    ### Iterate over all partitions for this device.
    for var_part in $(yq e ".recipe.${VAR_RECIPE_STRING}.dev.${var_dev} | keys | .[]" "${VAR_SETUP_PART}"); do

      ### Extract parameters from YAML.
      var_begin=$(yq e ".recipe.${VAR_RECIPE_STRING}.dev.${var_dev}.${var_part}.begin"           "${VAR_SETUP_PART}")
      var_end=$(yq e ".recipe.${VAR_RECIPE_STRING}.dev.${var_dev}.${var_part}.end"               "${VAR_SETUP_PART}")
      var_fs=$(yq e ".recipe.${VAR_RECIPE_STRING}.dev.${var_dev}.${var_part}.filesystem.version" "${VAR_SETUP_PART}")
      var_boot=$(yq e ".recipe.${VAR_RECIPE_STRING}.dev.${var_dev}.${var_part}.bootable"         "${VAR_SETUP_PART}")
      var_pri=$(yq e ".recipe.${VAR_RECIPE_STRING}.dev.${var_dev}.${var_part}.primary"           "${VAR_SETUP_PART}")

      ### Generate partition.
      if [[ "${var_end}" == "-1" ]]; then

        parted -s /dev/"${var_dev}" mkpart "${var_pri}" "${var_fs}" "${var_begin}" "100%" \
          || { do_log "fatal" "false" "Partition creation '/dev/${var_dev}${var_part}' failed."; exit "${ERR_READ_PARTTBL}"; }
        do_log "info" "false" "Partition generated: '${var_part}' | on device '/dev/${var_dev}' | begin: '${var_begin}' | end: 100 % of remaining disk."

      else

        parted -s /dev/"${var_dev}" mkpart "${var_pri}" "${var_fs}" "${var_begin}" "${var_end}" \
          || { do_log "fatal" "false" "Partition creation '/dev/${var_dev}${var_part}' failed."; exit "${ERR_READ_PARTTBL}"; }
        do_log "info" "false" "Partition generated: '${var_part}' | on device '/dev/${var_dev}' | begin: '${var_begin}' | end: '${var_end}'."

      fi

      ### Set the bootable flag if necessary.
      if [[ "${var_boot,,}" == "true" ]]; then
        parted -s "/dev/${var_dev}" set "${var_part}" boot on
        do_log "info" "false" "Partition: '/dev/${var_dev}${var_part}' marked as bootable."
      fi

      ### Read UUID of the newly created partition
      var_uuid=$(blkid -s UUID -o value "/dev/${var_dev}${var_part}")
      if [[ -z "${var_uuid}" ]]; then
        do_log "fatal" "false" "WARNING: could not read UUID for '/dev/${var_dev}${var_part}'."
        exit "${ERR_READ_PARTTBL}"
      else
        ### Store UUID in an associative array.
        HMP_UUID_PARTITION["UUID_${var_dev}${var_part}"]="${var_uuid}"
        do_log "info" "false" "Saved in HashMap HMP_UUID_PARTITION: 'UUID_${var_dev}${var_part}' -> '${var_uuid}'."
      fi

    done

  done
}
# vim: number et ts=2 sw=2 sts=2 ai tw=128 ft=sh