#!/bin/sh
# copyright (C) 2013-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 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 1 ]; then
	echo "usage lxcf suspend LXCNAME ..."
	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
}

# suspend one container
lxcf_suspend1() {
  LXCNAME=`uuid2name ${1}`

  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

  # check container state
  LXCSTATE=`LANG=C /usr/bin/virsh -c lxc:/// list | /usr/bin/awk '($2 == "'${LXCNAME}'"){print $3}'`

  if [ x"running" != x${LXCSTATE} ] ; then
    echo $LXCNAME "is not running"
    return
  fi

  while :
  do
	/usr/bin/virsh -c lxc:/// suspend ${LXCNAME} >& /dev/null
	if [ $? -eq 0 ] ; then
		break;
	fi

	sleep 1
  done
  echo ${LXCNAME} "is suspended."
}

# suspend containers of args
for name in $*
do
  name=`uuid2name $name`
  LXCSTATE=`/usr/lib64/lxcf/lxcf-list | awk '{if ($1 == "'${name}'") print $3}'`
  if [ x"${LXCSTATE}" = x"running" ] ; then
    lxcf_suspend1 $name
  else
    if [ ! -e /opt/lxcf/${name} ] ; then
      echo $name "is the unknown container name"
      continue
    fi

    echo ${name} "is not running."
  fi
done

exit 0
