#!/bin/bash





enumAppInfoList() {
    appInfoList=()
    apps="/var/lib/apm"
    list=$(ls $apps 2>/dev/null)
    for appID in $list; do
        appInfoList+=("$appID")
    done
    echo "${appInfoList[@]}"
}
linkDir() {
    ensureTargetDir() {
        targetFile=$1
        t=$(dirname "$targetFile")
        mkdir -p "$t"
    }

    source=$1
    target=$2
    sourceDir=$(dirname "$source")
    targetDir=$(dirname "$target")
    find "$source" -type f | while read sourceFile; do
        targetFile="$targetDir/${sourceFile#$sourceDir/}"


        ensureTargetDir "$targetFile"
	sourceFile=$(realpath --relative-to="$(dirname $targetFile)" "$sourceFile" )
	if [  ! -e ${targetFile} ];then
        ln -sv "$sourceFile" "$targetFile"
        fi
    done
}


linkApp() {
    appID=$1
    appEntriesDir="/var/lib/apm/$appID/entries"
    appLibsDir="/var/lib/apm/$appID/files/lib"
    autoStartDir="$appEntriesDir/autostart"

    if [ -d "$autoStartDir" ]; then
        linkDir "$autoStartDir" "/etc/xdg/autostart"
    fi

    # link application
    sysShareDir="/usr/share"
    for folder in "$appEntriesDir/applications" "$appEntriesDir/icons" "$appEntriesDir/mime" "$appEntriesDir/glib-2.0" "$appEntriesDir/services" "$appEntriesDir/GConf" "$appEntriesDir/help" "$appEntriesDir/locale" "$appEntriesDir/fcitx"; do
        if [ ! -d "$folder" ]; then
            continue
        fi
        if [ "$folder" = "$appEntriesDir/polkit" ]; then
            linkDir "$folder" "/usr/share/polkit-1"
        elif [ "$folder" = "$appEntriesDir/fonts/conf" ]; then
            linkDir "$folder" "/etc/fonts/conf.d"
        else
            linkDir "$folder" "$sysShareDir/${folder##*/}"
        fi
    done
}

function exec_uos_package_link(){

for app in $(enumAppInfoList); do
    linkApp "$app" &

done
wait
}

function exec_v23_icon_link(){
# Fix v23 broken icon
if [ ! -d "/usr/share/icons/hicolor/scalable/apps" ];then
mkdir -p /usr/share/icons/hicolor/scalable/apps
fi

for icon_root_icon_path in $(ls /usr/share/icons/*.png /usr/share/icons/*.svg 2>/dev/null)
do
target_icon_path=/usr/share/icons/hicolor/scalable/apps/$(basename ${icon_root_icon_path})
if [  ! -e ${target_icon_path} ];then
ln -sv $(realpath --relative-to=/usr/share/icons/hicolor/scalable/apps ${icon_root_icon_path}) /usr/share/icons/hicolor/scalable/apps
fi
done
}

function exec_link_clean(){
# remove broken links in /usr/share

    find /usr/share/applications -xtype l -exec echo '{} is invalid now and going to be cleaned' \; -exec unlink  {} \; 2>/dev/null &
    find /usr/share/icons -xtype l -exec echo '{} is invalid now and going to be cleaned' \; -exec unlink  {} \; 2>/dev/null &
    find /usr/share/mime/packages -xtype l -exec echo '{} is invalid now and going to be cleaned' \; -exec unlink  {} \; 2>/dev/null &
    find /usr/share/glib-2.0 -xtype l -exec echo '{} is invalid now and going to be cleaned' \; -exec unlink  {} \; 2>/dev/null &
    find /usr/share/dbus-1/services -xtype l -exec echo '{} is invalid now and going to be cleaned' \; -exec unlink  {} \; 2>/dev/null &
    find /usr/share/fcitx -xtype l -exec echo '{} is invalid now and going to be cleaned' \; -exec unlink  {} \; 2>/dev/null &
    find /usr/share/help -xtype l -exec echo '{} is invalid now and going to be cleaned' \; -exec unlink  {} \; 2>/dev/null &
    find /usr/share/locale -xtype l -exec echo '{} is invalid now and going to be cleaned' \; -exec unlink  {} \; 2>/dev/null &
    find /usr/lib/`dpkg-architecture -qDEB_HOST_MULTIARCH`/fcitx -xtype l -exec echo '{} is invalid now and going to be cleaned' \; -exec unlink  {} \; 2>/dev/null &
    find /usr/lib/mozilla/plugins -xtype l -exec echo '{} is invalid now and going to be cleaned' \; -exec unlink  {} \; 2>/dev/null &
    find /usr/share/polkit-1/actions -xtype l -exec echo '{} is invalid now and going to be cleaned' \; -exec unlink  {} \; 2>/dev/null &
    find /usr/share/fonts -xtype l -exec echo '{} is invalid now and going to be cleaned' \; -exec unlink  {} \; 2>/dev/null &
    find /etc/fonts/conf.d -xtype l -exec echo '{} is invalid now and going to be cleaned' \; -exec unlink  {} \; 2>/dev/null &


}
function exec_uos_package_update(){
    update-icon-caches /usr/share/icons/* > /dev/null 2>&1 &
    update-desktop-database -q > /dev/null 2>&1 &
    update-mime-database -V /usr/share/mime > /dev/null 2>&1 &
    glib-compile-schemas /usr/share/glib-2.0/schemas/ > /dev/null 2>&1 &

}

function exec_debian_compatibile_links(){
# 源目录和目标目录定义
local SOURCE_DIR="/var/lib/apm"
local TARGET_DIR="/var/lib/apm/apm/files/ace-env/var/lib/apm"

# 检查目标目录是否存在
if [[ ! -d "$TARGET_DIR" ]]; then
    mkdir -p $TARGET_DIR
fi

# 第一部分：为缺失的目录创建软链接
echo "检查并创建缺失的软链接..."
for dir in "$SOURCE_DIR"/*/; do
    # 获取目录名（去掉路径和尾部斜杠）
    dirname=$(basename "$dir")
    
    # 跳过 apm 目录
    if [[ "$dirname" == "apm" ]]; then
        continue
    fi
    
    # 检查目标目录中是否已存在对应的软链接或目录
    target_link="$TARGET_DIR/$dirname"
    if [[ ! -e "$target_link" ]]; then
        echo "创建软链接: $target_link -> $dir"
        ln -sv "$dir" "$target_link"
    fi
done

# 第二部分：清理无效的软链接
echo "清理无效的软链接..."
for link in "$TARGET_DIR"/*; do
    # 检查是否为软链接
    if [[ -L "$link" ]]; then
        # 检查软链接是否有效（指向的目标是否存在）
        if [[ ! -e "$link" ]]; then
            echo "删除无效软链接: $link"
            rm "$link"
        fi
    fi
done
}
#########################################################################################
echo "----------------Running APM Dstore Patch----------------"

# execute linkApp function for each app and print output
exec_uos_package_link


#exec_v23_icon_link
exec_link_clean
wait
exec_uos_package_update
if [[ "${IS_APM_ENV}" = "" ]];then
exec_debian_compatibile_links
fi

echo "----------------Finished----------------"
