#!/bin/bash
# copyright (C) 2014 FUJITSU LIMITED All Rights Reserved

# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; version 2
# of the License.
# 
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# 
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  
# 02110-1301, USA.

# check distro
DISTRO=`/usr/lib64/lxcf/lxcf-distro`

# check root
if [ ${EUID:-${UID}} != 0 ]; then
    echo "error: Because you are not root, you cannot execute this command. "
    exit 1
fi

# check args
if [ $# -lt 2 ]; then
	echo "usage: lxcf rename LXCNAME NEWNAME"
	exit 1
fi

lsdir() {
  ls -f --ind=none $1 | sed '/^\.\{1,2\}$/d'
}


# When uuid is specified, the container name is returned. 
uuid2name() {
  NAME=$1
  if [ -d /opt/lxcf/${NAME} ]; then
    echo -n $NAME
    return
  fi
  for i in `lsdir /etc/lxcf/rsc`
  do
     if [ x$NAME == x`cat /etc/lxcf/rsc/${i}/uuid` ]; then
       echo -n $i
       return
     fi
  done
  echo -n $NAME
}

# rename one container
lxcf_rename1() {

  LXCNAME=`uuid2name ${1}`
  NEWNAME=$2

  if ! /usr/lib64/lxcf/lxcf-parmchk-cname $LXCNAME ; then
    echo "error:" $LXCNAME "is not a container name"
    echo "        The container name must be alphanumeric character, \"-\", and \"_\"."
    return
  fi

  if [ ! -e /opt/lxcf/${LXCNAME} ] ; then
    echo "error:" $LXCNAME "is the unknown container name"
    return
  fi

  if [ -e /opt/lxcf/${NEWNAME} ] ; then
    echo "error:" $NEWNAME "is duplicate container name"
    return
  fi

  # update xml file
  /usr/bin/virsh -c lxc:/// dumpxml ${LXCNAME}> /etc/lxcf/rsc/${LXCNAME}/${LXCNAME}.xml

  LSTATE=`LANG=C /usr/bin/virsh -c lxc:/// list --all | /usr/bin/awk '($2 == "'${LXCNAME}'"){print $3}'`

  if [ x${LSTATE} == x"" ] ; then
    echo "error:" $LXCNAME "is the unknown container name"
    return
  fi

  if [ x${LSTATE} == x"paused" ] ; then
    /usr/bin/virsh -c lxc:/// resume ${LXCNAME} >& /dev/null
  fi

  if [ x${LSTATE} == x"running" -o x${LSTATE} == x"paused" ] ; then
    /usr/bin/virsh -c lxc:/// shutdown ${LXCNAME} >& /dev/null
    /usr/bin/virsh -c lxc:/// destroy ${LXCNAME} >& /dev/null

    wait
    if [ x$DISTRO == xu ]; then
      /usr/lib64/lxcf/lxcf-kill-container $LXCNAME
    fi
  fi

  # erase a container
  /usr/bin/virsh -c lxc:/// undefine ${LXCNAME} >& /dev/null

  # move file tree
  mv /opt/lxcf/$LXCNAME /opt/lxcf/$NEWNAME
  mv /etc/lxcf/rsc/$LXCNAME /etc/lxcf/rsc/$NEWNAME
  mv /etc/lxcf/rsc/$NEWNAME/${LXCNAME}.xml /etc/lxcf/rsc/$NEWNAME/${NEWNAME}.xml
  sed -i -e 's:<name>'${LXCNAME}'</name>:<name>'${NEWNAME}'</name>:' \
		 /etc/lxcf/rsc/$NEWNAME/${NEWNAME}.xml
  sed -i -e "s:<source dir='/opt/lxcf/"${LXCNAME}"'/>:<source dir='/opt/lxcf/"${NEWNAME}"'/>:" \
		 /etc/lxcf/rsc/$NEWNAME/${NEWNAME}.xml
  sed -i -e 's/"name" : "'${LXCNAME}'"/"name" : "'${NEWNAME}'"/' \
 		/etc/lxcf/rsc/$NEWNAME/resource.val
  sed -i -e 's/'${LXCNAME}'/'${NEWNAME}'/' /etc/lxcf/rsc/$NEWNAME/container_name

  cp /etc/lxcf/rsc/${NEWNAME}/container_name /opt/lxcf/${NEWNAME}/etc/lxcf/.
  mv /opt/lxcf/${NEWNAME}/etc/lxcf/rsc/$LXCNAME \
		/opt/lxcf/${NEWNAME}/etc/lxcf/rsc/$NEWNAME

  echo $NEWNAME > /opt/lxcf/$NEWNAME/etc/hostname

  # reset /etc/hosts
  sed -i -e 's/	'$LXCNAME'$/	'$NEWNAME'/g' /etc/hosts
  for i in `lsdir /etc/lxcf/rsc`
  do
    sed -i -e 's/	'$LXCNAME'$/	'$NEWNAME'/g' /opt/lxcf/$i/etc/hosts
  done

  # define $NEWNAME container
  virsh -c lxc:/// define /etc/lxcf/rsc/${NEWNAME}/${NEWNAME}.xml

  if [ x${LSTATE} == x"paused" ] ; then
    /usr/bin/virsh -c lxc:/// start ${NEWNAME}
    /usr/bin/virsh -c lxc:/// suspend ${NEWNAME}
  fi

  if [ x${LSTATE} == x"running" ] ; then
    /usr/bin/virsh -c lxc:/// start ${NEWNAME}
  fi

  echo $LXCNAME "is renamed to" $NEWNAME

}

# rename container
lxcf_rename1 $1 $2

exit 0

