root/src/vfs/extfs/helpers/uwim.in

/*   */

#! /bin/sh
# Midnight Commander - WIM support
#
# Written by:
#  Vadim Kalinnikov <moose@ylsoftware.com>
#
# This file is part of the Midnight Commander.
#
# Requires wimtools: https://wimlib.net
# On Debian/Ubuntu wimtools can be installed via:
# apt install wimtools
 
# Let the test framework hook in:
WIM=${MC_TEST_WIMLIB_IMAGEX:-`command -v wimlib-imagex 2>/dev/null`}
AWK=@AWK@
 
[ -n "$2" ] || exit 1
 
ACTION=$1
WIMFILENAME=$2
 
mcwim_list() {
        # Here we can use "Image count" from output,
        # but on some broken images we can get garbage, instead of number
        IMAGECOUNT=`${WIM} info "${WIMFILENAME}" | grep  Index: | grep -v -c Boot `
        IMGNUM=1
        VUID=`id -nu`
        VGID=`id -ng`
        while [ "${IMGNUM}" -le "${IMAGECOUNT}" ]; do
                ${WIM} dir "${WIMFILENAME}" "${IMGNUM}" --detailed | \
                        ${AWK} -v "uid=${VUID}" -v "gid=${VGID}" -v "imgnum=${IMGNUM}" '
                                /^----------------------------------------------------------------------------/,/^$/ {
                                        if (match($0, /^Full Path/)) {
                                                split($0, namesrc, "\"");
                                                name=namesrc[2];
                                        }
                                        if (match($0, /FILE_ATTRIBUTE_DIRECTORY is set/)) {
                                                attr="drwxr-xr-x"
                                        }
                                        if (match($0, /^Uncompressed size/)) {
                                                size=$4;
                                        }
                                        if (match($0, /^Last Write Time/)) {
                                                months["Jan"] = "01";
                                                months["Feb"] = "02";
                                                months["Mar"] = "03";
                                                months["Apr"] = "04";
                                                months["May"] = "05";
                                                months["Jun"] = "06";
                                                months["Jul"] = "07";
                                                months["Aug"] = "08";
                                                months["Sep"] = "09";
                                                months["Oct"] = "10";
                                                months["Nov"] = "11";
                                                months["Dec"] = "12";
                                                split($0, mtimesrc, " ");
                                                mtime=sprintf("%s/%s/%s %s", months[mtimesrc[6]], mtimesrc[7], mtimesrc[9], mtimesrc[8]);
                                        }
                                        
                                        if (match($0, /^$/)) {
                                                printf("%s 1 %s %s % 20s % 24s IMAGE%s%s\n",
                                                        attr, uid, gid, size, mtime, imgnum, name);
                                                name = size = mtime = "";
                                                attr="-rw-r--r--";
                                        }
                                }
                        '
                
                IMGNUM=$((IMGNUM+1))
        done
        
        # Virtual files
        echo "-r-xr-xr-x 1 ${VUID} ${VGID}                    1      01/01/2020 00:00:00 OPTIMIZE"
        echo "-r-xr-xr-x 1 ${VUID} ${VGID}                    1      01/01/2020 00:00:00 VERIFY"
}
 
mcwim_copyout() {
        # Virtual files
        if [ "${FILENAMESRC}" = "OPTIMIZE" ]; then
                echo "#/bin/sh" > "${FILENAMEDST}"
                echo "# Run this to optimize archive" >> "${FILENAMEDST}"
                echo "${WIM} optimize \"${WIMFILENAME}\"" >> "${FILENAMEDST}"
                chmod a+x "${FILENAMEDST}"
                exit 0;
        elif [ "${FILENAMESRC}" = "VERIFY" ]; then
                echo "#/bin/sh" > "${FILENAMEDST}"
                echo "# Run this to verify archive" >> "${FILENAMEDST}"
                echo "${WIM} verify \"${WIMFILENAME}\"" >> "${FILENAMEDST}"
                chmod a+x "${FILENAMEDST}"
                exit 0;
        fi
 
        # Filename must contain imgnum
        echo "${FILENAMESRC}" | grep -E '^IMAGE[0-9]+/' || exit 1
 
        IMGNUM=`echo "${FILENAMESRC}" | cut -d '/' -f1 | sed "s/IMAGE//"`
        REALFILENAME=`echo "${FILENAMESRC}" | sed "s/IMAGE${IMGNUM}//"`
        ${WIM} extract "${WIMFILENAME}" "${IMGNUM}" "${REALFILENAME}" --to-stdout > "${FILENAMEDST}"
}
 
mcwim_copyin() {
        # Skip virtual files
        [ "${FILENAMEDST}" = "OPTIMIZE" ] && exit 1;
        [ "${FILENAMEDST}" = "VERIFY" ] && exit 1;
 
        # Filename must contain imgnum
        echo "${FILENAMEDST}" | grep -E '^IMAGE[0-9]+/' || exit 1
 
        IMGNUM=`echo "${FILENAMEDST}" | cut -d '/' -f1 | sed "s/IMAGE//"`
        REALFILENAME=`echo "${FILENAMEDST}" | sed "s/IMAGE${IMGNUM}//"`
        echo "add \"${FILENAMESRC}\" \"${REALFILENAME}\"" | ${WIM} update "${WIMFILENAME}" "${IMGNUM}" > /dev/null
}
 
 
mcwim_rm() {
        # Skip virtual files
        [ "${FILENAMESRC}" = "OPTIMIZE" ] && exit 0;
        [ "${FILENAMESRC}" = "VERIFY" ] && exit 0;
 
        # Filename must contain imgnum
        echo "${FILENAMESRC}" | grep -E '^IMAGE[0-9]+/' || exit 1
        
        IMGNUM=`echo "${FILENAMESRC}" | cut -d '/' -f1 | sed "s/IMAGE//"`
        REALFILENAME=`echo "${FILENAMESRC}" | sed "s/IMAGE${IMGNUM}//"`
 
        if [ -z "${REALFILENAME}" ]; then
                # If user want to remove image
                ${WIM} delete "${WIMFILENAME}" "${IMGNUM}"
        else
                # remove regular file or directory
                echo "delete \"${REALFILENAME}\"" | ${WIM} update "${WIMFILENAME}" "${IMGNUM}" --force --recursive > /dev/null
        fi
}
 
mcwim_run() {
        case "${RUNFILENAME}" in
                OPTIMIZE)
                        ${WIM} optimize "${WIMFILENAME}"
                        exit 0;
                ;;
                VERIFY)
                        ${WIM} verify "${WIMFILENAME}"
                        exit 0;
                ;;
        esac
        exit 1;
}
 
 
mcwim_mkdir() {
        # New dirname must contain imgnum
        echo "${NEWDIRNAME}" | grep -E '^IMAGE[0-9]+/' || exit 1
        IMGNUM=`echo "${NEWDIRNAME}" | cut -d '/' -f1 | sed "s/IMAGE//"`
        REALDIRNAME=`echo "${NEWDIRNAME}" | sed "s/IMAGE${IMGNUM}//"`
        [ -z "${REALDIRNAME}" ] && exit 1
        
        TMPDIR=`mktemp -d`
        DSTBASENAME=`basename "${REALDIRNAME}"`
        SRCDIRNAME="${TMPDIR}/${DSTBASENAME}"
        mkdir -p "${SRCDIRNAME}"
        echo "add \"${SRCDIRNAME}\" \"${REALDIRNAME}\"" | ${WIM} update "${WIMFILENAME}" "${IMGNUM}" > /dev/null
        rm -rf "${TMPDIR}"
}
 
#echo "'$1' '$2' '$3' '$4' '$5'" >> /tmp/mcdebug
 
case "$ACTION" in
        list)
                mcwim_list
        ;;
        
        copyout)
                [ -n "$4" ] || exit 1
                FILENAMESRC="$3"
                FILENAMEDST="$4"
                mcwim_copyout
        ;;
        
        copyin)
                [ -n "$4" ] || exit 1
                FILENAMEDST="$3"
                FILENAMESRC="$4"
                mcwim_copyin
        ;;
        
        rm|rmdir)
                [ -n "$3" ] || exit 1
                FILENAMESRC="$3"
                mcwim_rm
        ;;
        
        run)
                [ -n "$3" ] || exit 1
                RUNFILENAME="$3"
                mcwim_run
        ;;
        
        mkdir)
                [ -n "$3" ] || exit 1
                NEWDIRNAME="$3"
                mcwim_mkdir
        ;;
        
 
        *)
                exit 1
        ;;
esac

/*   */