Current File : //usr/ddu/scripts/pkg_relate.sh

#!/bin/ksh93
#
# Copyright (c) 2008, 2013, Oracle and/or its affiliates. All rights reserved.
#
# Description: Perform pkg related function: 
#              list repo, add repo, search repo and modify
#              repo etc.
#
#              Usage:
#              pkg_relate.sh $1 $2
#              $1 can be "add","list","search" and "modify"
#              $2 is the sub-argument for $1
#              $3 is the sub-argument for $1
#

#
# Get all publishers info on the system
#
function fetch_repo
{
    pfexec pkg publisher -Ha > /tmp/repo_temp_store1 2>>$err_log
    if [ $? -ne 0 ]; then
        print -u2 "$0: Error found when getting repo list"
        exit 1
    fi
    cat /tmp/repo_temp_store1 | awk '{ if (NF>2) print $1,$NF}' \
        > /tmp/repo_temp_store
    repo_list=$(awk '{ print $1 }' /tmp/repo_temp_store \
        2>/dev/null)
    repo_url=$(awk '{ print $NF }' /tmp/repo_temp_store \
        2>/dev/null)
    repo_count=$(wc -l </tmp/repo_temp_store 2>/dev/null)
}

#
# Output the publisher info in the following format:
# opensolaris.org http://ipkg.sfbay.sun.com/dev/
#

function print_repo
{
    index=1
    while [ $index -le ${repo_count} ]; do
        repo_listi=$(echo ${repo_list} | nawk '{ print $'${index}' }')
        repo_urli=$(echo ${repo_url} | nawk '{ print $'${index}' }')
        printf "${repo_listi}\t${repo_urli}\n"
        index=$(($index + 1))
    done
}

#
# Output the publisher info with following format based on the input 
# parameter:
# opensolaris.org http://ipkg.sfbay.sun.com/dev/
#
# Input paramenters can be:
# all:            list all the publisher info.
# default:        list preferred publisher info.
# publisher name: list specified pulibsher info.
#

function list
{
    if [ "${min_argu_1}" == "all" ]; then #list all repo.
        fetch_repo
        print_repo
    elif [ "${min_argu_1}" == "default" ]; then #list default repo
        pfexec pkg publisher -HP > /tmp/repo_default_store 2>>$err_log
        if [  $? != 0 ]; then
            print -u2 "$0: Error executing pfexec pkg publisher -HP."
            exit 1
        fi
        repo_list=$(awk '{ print $1 }' /tmp/repo_default_store \
            2>/dev/null)
        repo_url=$(awk '{ print $NF }' /tmp/repo_default_store \
            2>/dev/null)
        repo_count=$(wc -l </tmp/repo_default_store 2>/dev/null)
        print_repo
    elif [ ! -z ${min_argu_1} ]; then #list a specific repo
	repo_name=${min_argu_1}
        pfexec pkg publisher ${repo_name} | grep "URI" \
            > /tmp/repo_spe_store 2>>$err_log
        if [  $? != 0 ]; then
            print -u2 "$0: ${repo_name} is not a valid repo name!"
            exit 1
        fi
        repo_url=$(awk '{ print $NF }'  /tmp/repo_spe_store \
            2>/dev/null)
	for repo_urli in ${repo_url}
	do
	    printf "${repo_name}\t${repo_urli}\n"
	done
    fi
}

#
# Search for driver package via publisher based on the device compatible string 
# 
function search
{
    if [ -z ${min_argu_2} ]; then
        package_found=0
        fetch_repo
    
        index=1
        while [ $index -le ${repo_count} ]; do
            repo_listi=$(echo ${repo_list} | nawk '{ print $'${index}' }')
            repo_urli=$(echo ${repo_url} | nawk '{ print $'${index}' }')

            pfexec pkg search -p -s "${repo_urli}" "${min_argu_1}" \
               | nawk -F"@" '{ print $1 }' | \
               nawk -F"pkg:/" '{ print $2 }' | uniq > /tmp/pkg_candi 

            #
            # Filter packages based on "driver" in their name
            #
            pkgname=$(grep driver /tmp/pkg_candi | tail -1 2>/dev/null)
            if [ -z ${pkgname} ]; then
                pkgname=$(tail -1 /tmp/pkg_candi 2>/dev/null)
            fi

            if [[ $? != 0  ||  -z ${pkgname} ]]; then
                index=$(($index + 1))
                continue
            fi
		
	    depend_str="depend:incorporate:${pkgname}" 
	    pkg_ver=$(pfexec pkg search -l ${depend_str} | tail -1 2>>/dev/null)

	    if [ $? = 0 ]; then
                pkgvalue=$(echo ${pkg_ver} | nawk '{ print $3 }')
                if [ -z $(echo ${pkgvalue} | grep "pkg:/") ]; then
		    pkgcondidate=${pkgvalue}
                else
                    pkgcondidate=$(echo ${pkgvalue} | nawk -F"pkg:/" \
                        '{ print $2 }')
                fi
	    fi

            if [ -z ${pkgcondidate} ]; then
                pkgcondidate=${pkgname}
            fi
            pfexec pkg info -r pkg://${repo_listi}/${pkgcondidate} \
                >/dev/null 2>>$err_log
            if [ $? = 0 ]; then
                /usr/bin/printf "${repo_listi}\t${pkgcondidate}\n"
                package_found=1
            fi    
            index=$(($index + 1))
        done
    else
        repo_listi=${min_argu_2}
	package_found=0
        repo_urli=$(pfexec pkg publisher ${repo_listi} 2>>$err_log \
            | grep "URI" | /usr/bin/awk  '{ print $NF }')

        if [ $? != 0 ]; then
            print "$0: Error executing pfexec pkg publisher" \
                " ${repo_listi}" >>$err_log
            exit 1
        fi

        for repo_urli_i in $repo_urli; do

            pfexec pkg search -p -s "${repo_urli_i}" "${min_argu_1}" \
                | nawk -F"@" '{ print $1 }' | \
                nawk -F"pkg:/" '{ print $2 }' | uniq > /tmp/pkg_candi

            if [ $? != 0 ]; then
                continue
            fi

            #
            # Filter packages based on "driver" in their name
            #
            pkgname=$(grep driver /tmp/pkg_candi | tail -1 2>/dev/null)
            if [ -z ${pkgname} ]; then
                pkgname=$(tail -1 /tmp/pkg_candi 2>/dev/null)
            fi
            if [ ! -z ${pkgname} ]; then
                break
            fi
        done
        
        if [[ $? != 0  ||  -z ${pkgname} ]]; then
            exit 1
        fi

        depend_str="depend:incorporate:${pkgname}" 
        pkg_ver=$(pfexec pkg search -l ${depend_str} | tail -1 2>>/dev/null)

        if [ $? -eq 0 ]; then
            pkgvalue=$(echo ${pkg_ver} | nawk '{ print $3 }')
            if [ -z "$(echo ${pkgvalue} | grep "pkg:/")" ]; then
	        pkgcondidate=${pkgvalue}
            else
                pkgcondidate=$(echo ${pkgvalue} | nawk -F"pkg:/" '{ print $2 }')
            fi
	fi

        if [ -z ${pkgcondidate} ]; then
            pkgcondidate=${pkgname}
        fi
        pfexec pkg info -r pkg://${repo_listi}/${pkgcondidate} \
            >/dev/null 2>>$err_log
        if [ $? -eq 0 ]; then
            printf "${repo_listi}\t${pkgcondidate}\n"
            package_found=1
        fi    
    fi
    if [ ${package_found} -eq 0 ]; then
        exit 1
    fi
}

#
# Add publisher on the system
#

function add
{
    #
    #add a repo and set it to default one if set.
    #
    if [ -z ${min_argu_2} ]; then
        print -u2 "\n$0: Third argument can not be null..."
        exit 1
    fi      
    publisher_name=${min_argu_1}
    publisher_url=${min_argu_2}
    pfexec pkg set-publisher -O ${publisher_url} ${publisher_name} \
        2>>$err_log 1>&2
    if [ $? -ne 0 ]; then
        print -u2 "$0: Error found when adding repo: "
        print -u2 "${publisher_name} ${publisher_url}"
        exit 1
    fi
}

#
# Modify the existing publisher on the system via
# pkg set-publisher and pkg unset-publisher.
#
function modify
{
    #
    #modify a repo and set it to default one if set.
    #
    if [ -z ${min_argu_2} ]; then
        print -u2 "\n$0: Third argument can not be null..."
        exit 1
    fi      
    publisher_name=${min_argu_1}
    publisher_url=${min_argu_2}

    #
    # Set LC_ALL="C', because does not want the output
    # to be translated. Like the word primary, mirror etc.
    #
     LC_ALL="C" pfexec pkg publisher ${publisher_name} > \
        /tmp/spe_repo_temp_store 2>>$err_log

    if [ $? != 0 ]; then
        print "$0: Error executing pfexec pkg publisher" \
            " ${publisher_name}" >>$err_log
        exit 1
    fi

    origin_uri=$(grep "Origin URI" /tmp/spe_repo_temp_store | \
                nawk -F ": " '{ print $2 }' 2>>$err_log)

    mirror_uris=$(grep "Mirror URI" /tmp/spe_repo_temp_store | \
                nawk -F ": " '{ print $2 }' 2>>$err_log)

    store_repo=" -O "${origin_uri}
    extra_repo=" "
    unset_extra_repo=" "

    for mirror_uri in ${mirror_uris}
    do
        extra_repo=${extra_repo}" -m "${mirror_uri}
        unset_extra_repo=${unset_extra_repo}" -M "${mirror_uri}
    done
    
    extra_repo=${extra_repo}" "${publisher_name}

    #
    # Try to unset publisher first before set publisher
    #
    pfexec pkg unset-publisher ${publisher_name} 2>>/dev/null 1>&2
    if [ $? -ne 0 ]; then
        print -u2 "Warning: Can not remove repo ${publisher_name}"
    fi

    pfexec pkg set-publisher -O ${publisher_url} \
                                ${unset_extra_repo} \
                                ${publisher_name} \
                                2>>$err_log 1>&2
    if [ $? -ne 0 ]; then
        print -u2 "$0: Error found when adding repo:"
        print -u2 "${publisher_name} ${publisher_url}"
        exit 1
    else
        echo ${store_repo} > /tmp/spe_repo_store
        echo ${extra_repo} >> /tmp/spe_repo_store
    fi
}

#
# Restore the original publisher on the system 
# for those changed publisher during installaion
#
function restore
{
    pfexec pkg set-publisher ${min_argu_1} ${min_argu_2} \
        2>>$err_log 1>&2
    if [ $? -ne 0 ]; then
        print -u2 "$0: Error found when restoring repo!"
        exit 1
    fi
}

#
# Save system publisher info
#
function save_p
{
    #
    # Set LC_ALL="C', because does not want the output
    # to be translated. Like the word primary, mirror etc.
    #
    LC_ALL=C pfexec pkg publisher -H > $pub_file 
    chmod 666 $pub_file
}

#
# Restore system publisher info
#
function restore_p
{
    typeset p_name
    typeset p_type
    typeset p_prefer
    typeset p_url

    IFS='
'

    if [ -s $pub_file ]; then
        #
        # Set LC_ALL="C', because does not want the output
        # to be translated. Like the word primary, mirror etc.
        #
        LC_ALL=C pfexec pkg publisher -H > ${pub_file}_tmp
        if [ $? -ne 0 ]; then
            print -u2 "$0: Error executing pfexec pkg publisher -H"
            rm -f $pub_file ${pub_file}_tmp 
            exit 1
        fi

        diff $pub_file ${pub_file}_tmp >/dev/null 2>&1
        if [ $? != 0 ]; then
            for i in $(cat $pub_file); do
                #
                # Check whether the publisher already exists.
                #
                pfexec pkg publisher -H | grep "$i" >/dev/null 2>&1
                if [ $? = 0 ]; then
                    continue
                fi
                p_name=$(echo $i | awk '{print $1}')
                p_url=$(echo $i | awk '{print $NF}')
                if [ ! -z $(echo $i | grep "origin") ] ; then
                    p_type="-O"
                fi 
                if [ ! -z $(echo $i | grep "mirror") ] ; then
                    p_type="-m"
                fi 
                if [ ! -z $(echo $i | grep "preferred") ] ; then
                    p_prefer="-P"
                fi 
                pfexec pkg set-publisher $p_prefer $p_type $p_url $p_name >>$err_log 2>&1
                if [ $? -ne 0 ]; then
                    print -u2 "$0: Error found when restoring repo!"
                fi
                p_prefer=""
                p_type=""
            done
                
        fi
        rm -f $pub_file ${pub_file}_tmp 
    else
        #
        # $pub_file does not exist, User may not use the add repo function.
        #
        print -u2 "$0: $pub_file does not exist."
    fi
} 
   

#######################################################################
# clean_up
#       This function attempts to clean up any resources this script
#       could generate. Depending on where in the script this function
#       is involved some resources may not be there to cleanup, but
#       that will not adversely effect anything.
#
#       This function is not defined using the function keyword
#       to avoid an exit loop.
#
# Input: none
# Returns: none
#
#######################################################################
clean_up ()
{
    {
        rm -f  /tmp/repo_temp_store /tmp/repo_default_store \
               /tmp/repo_spe_store /tmp/repo_temp_store1 \
               /tmp/spe_repo_temp_store /tmp/pkg_candi 
    } >/dev/null 2>&1
}


#Main()

PATH=/usr/bin:/usr/sbin:$PATH; export PATH
LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/ddu/lib; export LD_LIBRARY_PATH

trap 'clean_up;exit 10' KILL INT
trap 'clean_up' EXIT


#
# Confirm the correct number of commnad line argument
#
if (( $# < 1 )) || (( $# > 3 )); then
    print -u2 "\n$0: Wrong number of arguments provided..."
    exit 1
fi

typeset main_argu=$1
typeset min_argu_1=$2
typeset min_argu_2=$3
typeset repo_list repo_url repo_count
typeset repo_listi repo_urli repo_name pkgname 
typeset depend_str pkg_ver pkgcondidate package_found
typeset index
typeset  err_log=/tmp/ddu_err.log
typeset  pub_file=/tmp/ddu_publisher.info

#
# Copy /var/pkg to /tmp if booted off a ramdisk
# Needed by pkg utility so it can expand its bootkeeping files in swap space.
#
if [ -x /usr/lib/install/live_img_pkg5_prep ] ; then
    pfexec /usr/lib/install/live_img_pkg5_prep 2>>$err_log
    if [ $? != 0 ]; then
        print -u2 "Error executing: /usr/lib/install/live_img_pkg5_prep."
        exit 1
    fi
fi

case ${main_argu} in
list)   list;;
search) search;;
add)    add;;
modify) modify;;
restore) restore;;
save_p) save_p;;
restore_p) restore_p;;
*)      print -u2 "\n$0: ${main_argu} is not valid"
        exit 1;;
esac