tsg-tsg-os-buildimage/installer/app_bundle_install.sh

#!/bin/sh

TRAFFIC_ENGINE_HELM_CHART_TAR="%%TRAFFIC_ENGINE_HELM_CHART_TAR%%"
INJECT_HELM_CHART_TAR="%%INJECT_HELM_CHART_TAR%%"
MRZCPD_HELM_CHART_TAR="%%MRZCPD_HELM_CHART_TAR%%"
TREX_HELM_CHART_TAR="%%TREX_HELM_CHART_TAR%%"
DP_TRACE_HELM_CHART_TAR="%%DP_TRACE_HELM_CHART_TAR%%"
TSG_CM_HELM_CHART_TAR="%%TSG_CM_HELM_CHART_TAR%%"

charts_dest_dir="/var/lib/rancher/k3s/server/static/charts"
images_dest_dir="/var/lib/rancher/k3s/agent/images"
manifests_dest_dir="/var/lib/rancher/k3s/server/manifests"
clixon_conf_path="/opt/tsg/clixon/etc/mgnt-srv.conf"
load_images_switch="on"

src_dir=$(dirname $0)

#function define start
function read_charts_dest_dir_from_env()
{
    if [ ! -z "${APP_BUNDLE_CHARTS_DEST_DIR}" ]; then
        charts_dest_dir="${APP_BUNDLE_CHARTS_DEST_DIR}"
    fi
}

function read_images_dest_dir_from_env()
{
    if [ ! -z "${APP_BUNDLE_IMAGES_DEST_DIR}" ]; then
        images_dest_dir="${APP_BUNDLE_IMAGES_DEST_DIR}"
    fi
}

function read_manifests_dest_dir_from_env()
{
    if [ ! -z "${APP_BUNDLE_MANIFESTS_DEST_DIR}" ]; then
        manifests_dest_dir="${APP_BUNDLE_MANIFESTS_DEST_DIR}"
    fi
}

function read_clixon_conf_path_from_env()
{
    if [ ! -z "${APP_BUNDLE_CLIXON_CONF_PATH}" ]; then
        clixon_conf_path="${APP_BUNDLE_CLIXON_CONF_PATH}"
    fi
}

function read_load_images_switch_from_env()
{
    if [ ! -z "${APP_BUNDLE_LOAD_IMAGES_SWITCH}" ]; then
        load_images_switch="${APP_BUNDLE_LOAD_IMAGES_SWITCH}"
    fi
}

function install_helmcharts()
{
    if [ -d "${charts_dest_dir}" ]; then
        mkdir -p "${charts_dest_dir}"
    fi

    cp -r ${src_dir}/helmcharts/* ${charts_dest_dir}
}

function install_images()
{
    if [ -d "${images_dest_dir}" ]; then
        mkdir -p "${images_dest_dir}"
    fi

    cp -r ${src_dir}/images/* ${images_dest_dir}
}

function install_mainfests()
{
    if [ -d "${manifests_dest_dir}" ]; then
        mkdir -p "${manifests_dest_dir}"
    fi
    cp -arf ${src_dir}/manifests/* ${manifests_dest_dir}
}

function replace_clixon_chart_name()
{
    if [ ! -f "${clixon_conf_path}" ]; then
        echo "Error: ${clixon_conf_path} is not exist!"
        exit 1
    fi

    sed -ie "s/^chart_name=.*/chart_name=${TRAFFIC_ENGINE_HELM_CHART_TAR}/g" ${clixon_conf_path}
    sed -ie "s/^inject_adapter_chart_name=.*/inject_adapter_chart_name=${INJECT_HELM_CHART_TAR}/g" ${clixon_conf_path}
    sed -ie "s/^mrzcpd_chart_name=.*/mrzcpd_chart_name=${MRZCPD_HELM_CHART_TAR}/g" ${clixon_conf_path}
    sed -ie "s/^trex_chart_name=.*/trex_chart_name=${TREX_HELM_CHART_TAR}/g" ${clixon_conf_path}
    sed -ie "s/^datapath_trace_chart_name=.*/datapath_trace_chart_name=${DP_TRACE_HELM_CHART_TAR}/g" ${clixon_conf_path}
    sed -ie "s/^tsg_cm_chart_name=.*/tsg_cm_chart_name=${TSG_CM_HELM_CHART_TAR}/g" ${clixon_conf_path}
}

function load_images()
{
    if [ "${load_images_switch}" = "on" ]; then
        for image_tar_name in $(ls ${src_dir}/images/*.tar 2>/dev/null | sed 's#.*/##'); do
            /usr/bin/k3s ctr image import ${images_dest_dir}/${image_tar_name}
        done
    fi
}

#function exec start
read_charts_dest_dir_from_env
read_images_dest_dir_from_env
read_manifests_dest_dir_from_env
read_clixon_conf_path_from_env
read_load_images_switch_from_env

replace_clixon_chart_name
install_helmcharts
install_images
install_mainfests
load_images
#function exec end