#!/bin/bash

. /var/packages/AIConsole/scripts/pkg_utils

CreateLogDir()
{
  local PKG_LOG_DIR="${PKG_DIR}/var/log"
  MakeDir "${PKG_LOG_DIR}"
  $CHMOD_BIN 755 "${PKG_LOG_DIR}"
  ChangeOwnToPackage "${PKG_LOG_DIR}"
}

CreateDeidConfigDir()
{
  MakeDir "${DEID_CONFIG_DIR}"
  ChangeOwnToPackage "${DEID_CONFIG_DIR}"
}

CreatePresidioDir()
{
  MakeDir "${PRESIDIO_ANALYZER_DIR}"
  ChangeOwnToPackage "${PRESIDIO_ANALYZER_DIR}"
}

CreatePromptsDir()
{
  MakeDir "${PKG_PROMPTS_DIR}"
  ChangeOwnToPackage "${PKG_PROMPTS_DIR}"
}

CreateSettingDir()
{
  MakeDir "${SETTING_DIR}"
  ChangeOwnToPackage "${SETTING_DIR}"
}

CreateImageDir()
{
  MakeDir "${IMAGE_DIR}"
  ChangeOwnToPackage "${IMAGE_DIR}"

  MakeDir "${LOG_IMAGE_DIR}"
  ChangeOwnToPackage "${LOG_IMAGE_DIR}"

  MakeDir "${COMPLETION_IMAGE_DIR}"
  ChangeOwnToPackage "${COMPLETION_IMAGE_DIR}"
}

MigrateKey()
{
  MakeDir "${KEY_DIR}"
  # Restrict directory permissions to 700 so only the owner has access.
  # This follows security best practices for directories storing sensitive keys.
  ChangeOwnToPackage "${KEY_DIR}"
  $CHMOD_BIN 700 "${KEY_DIR}"
  "${TOOL_DIR}/synoaiproxy" --migrate_key_pair
}

RemoveDeprecatedPortConfig()
{
  local path="/usr/local/etc/services.d/AIConsole.sc"
  if [ -f "${path}" ]; then
    servicetool --remove-configure-file --package AIConsole.sc
  fi
}

CreateLogDir
CreateDeidConfigDir
MigrateTables
MigrateEntitiesFromPresidioJSON
CreatePresidioDir
CreatePromptsDir
CreateSettingDir
CreateImageDir
MigrateKey
RemoveDeprecatedPortConfig

exit 0