#!/usr/bin/env bash

# Tell build process to exit if there are any errors.
set -euo pipefail

# Copy included systemd unit files to system
SYSTEM_UNIT_INCLUDE="$CONFIG_DIRECTORY"/systemd/system
USER_UNIT_INCLUDE="$CONFIG_DIRECTORY"/systemd/user
SYSTEM_UNIT_DIR="/usr/lib/systemd/system"
USER_UNIT_DIR="/usr/lib/systemd/user"

if [[ -d "$SYSTEM_UNIT_INCLUDE" ]]; then
  if [[ -n $(find "$SYSTEM_UNIT_INCLUDE" -type f) ]]; then
    echo "Copying 'system' systemd units to system directory"
    cp -r "$SYSTEM_UNIT_INCLUDE"/* "$SYSTEM_UNIT_DIR"
  fi
fi
if [[ -d "$USER_UNIT_INCLUDE" ]]; then
  if [[ -n $(find "$USER_UNIT_INCLUDE" -type f) ]]; then
    echo "Copying 'user' systemd units to system directory"  
    cp -r "$USER_UNIT_INCLUDE"/* "$USER_UNIT_DIR"
  fi  
fi  

# Systemd units configuration (enable, disable, unmask & mask)
get_json_array ENABLED 'try .["system"].["enabled"][]' "$1"
get_json_array DISABLED 'try .["system"].["disabled"][]' "$1"
get_json_array UNMASKED 'try .["system"].["unmasked"][]' "$1"
get_json_array MASKED 'try .["system"].["masked"][]' "$1"
get_json_array USER_ENABLED 'try .["user"].["enabled"][]' "$1"
get_json_array USER_DISABLED 'try .["user"].["disabled"][]' "$1"
get_json_array USER_UNMASKED 'try .["user"].["unmasked"][]' "$1"
get_json_array USER_MASKED 'try .["user"].["masked"][]' "$1"

if [[ ${#ENABLED[@]} -gt 0 ]]; then
    for unit in "${ENABLED[@]}"; do
        unit=$(printf "$unit")
        systemctl -f enable $unit
    done
fi
if [[ ${#DISABLED[@]} -gt 0 ]]; then
    for unit in "${DISABLED[@]}"; do
        unit=$(printf "$unit")
        systemctl disable $unit
    done
fi
if [[ ${#UNMASKED[@]} -gt 0 ]]; then
    for unit in "${UNMASKED[@]}"; do
        unit=$(printf "$unit")
        systemctl unmask $unit
    done
fi
if [[ ${#MASKED[@]} -gt 0 ]]; then
    for unit in "${MASKED[@]}"; do
        unit=$(printf "$unit")
        systemctl mask $unit
    done
fi
if [[ ${#USER_ENABLED[@]} -gt 0 ]]; then
    for unit in "${USER_ENABLED[@]}"; do
        unit=$(printf "$unit")
        systemctl --global -f enable $unit
    done
fi
if [[ ${#USER_DISABLED[@]} -gt 0 ]]; then
    for unit in "${USER_DISABLED[@]}"; do
        unit=$(printf "$unit")
        systemctl --global disable $unit
    done
fi
if [[ ${#USER_UNMASKED[@]} -gt 0 ]]; then
    for unit in "${USER_UNMASKED[@]}"; do
        unit=$(printf "$unit")
        systemctl --global unmask $unit
    done
fi
if [[ ${#USER_MASKED[@]} -gt 0 ]]; then
    for unit in "${USER_MASKED[@]}"; do
        unit=$(printf "$unit")
        systemctl --global mask $unit
    done
fi