Add --volume option support.
[src/xds/xds-server.git] / scripts / xds-docker-create-container.sh
old mode 100755 (executable)
new mode 100644 (file)
index 52dfda3..13ebb71
@@ -32,23 +32,42 @@ DOCKER_USER=devel
 DEFIMAGE=$REGISTRY/$REPO/$NAME-$FLAVOUR:$VERSION
 
 function usage() {
-       echo "Usage: $(basename $0) <instance ID> [image name]"  >&2
+       echo "Usage: $(basename $0) [-h] [-fr] [-v] <instance ID> [image name]"  >&2
        echo "Instance ID must be 0 or a positive integer (1,2,...)" >&2
        echo "Image name is optional: 'make show-image' is used by default to get image" >&2
        echo "Default image: $DEFIMAGE" >&2
+    echo ""
+    echo "Options:"
+    echo " -fr | --force-restart   Force restart of xds-server service"
+    echo " -nuu | --no-uid-update  Don't update user/group id within docker"
+    echo " -v | --volume           Additional docker volume to bind, syntax is -v /InDockerPath:/HostPath "
        exit 1
 }
 
 ID=""
-IMAGE=$DEFIMAGE
-FORCE=false
+IMAGE=""
+FORCE_RESTART=false
+UPDATE_UID=true
+USER_VOLUME_OPTION=""
 while [ $# -ne 0 ]; do
     case $1 in
         -h|--help|"")
             usage
             ;;
-        -fr|-force-restart)
-            FORCE=true
+        -fr|--force-restart)
+            FORCE_RESTART=true
+            ;;
+        -nuu|--no-uid-update)
+            UPDATE_UID=false
+            ;;
+        -v|--volume)
+            shift
+            if [[ "$1" =~ .*:.* ]]; then
+                USER_VOLUME_OPTION="-v $1"
+            else
+                echo "Invalid volume option, format must be /InDockerPath:/hostPath"
+                exit 1
+            fi
             ;;
         *)
             if [[ "$1" =~ ^[0-9]+$ ]]; then
@@ -63,6 +82,28 @@ done
 
 [ "$ID" = "" ] && ID=0
 
+# Dynamically retrieve image name
+if [ "$IMAGE" = "" ]; then
+
+    VER_NUM=`docker images $REGISTRY/$REPO/$NAME-$FLAVOUR:* --format {{.Tag}} | wc -l`
+    if [ $VER_NUM -gt 1 ]; then
+        echo "ERROR: more than one xds image found, please set explicitly the image to use !"
+        exit 1
+    elif [ $VER_NUM -lt 1 ]; then
+        echo "ERROR: cannot automatically retrieve image tag for $REGISTRY/$REPO/$NAME-$FLAVOUR"
+        exit 1
+    fi
+
+    VERSION=`docker images $REGISTRY/$REPO/$NAME-$FLAVOUR:* --format {{.Tag}}`
+    if [ "$VERSION" = "" ]; then
+        echo "ERROR: cannot automatically retrieve image tag for $REGISTRY/$REPO/$NAME-$FLAVOUR"
+        usage
+        exit 1
+    fi
+
+    IMAGE=$REGISTRY/$REPO/$NAME-$FLAVOUR:$VERSION
+fi
+
 USER=$(id -un)
 echo "Using instance ID #$ID (user $(id -un))"
 
@@ -71,16 +112,16 @@ NAME=agl-xds-$(hostname|cut -f1 -d'.')-$ID-$USER
 docker ps -a |grep "$NAME" > /dev/null
 [ "$?" = "0" ] && { echo "Image name already exist ! (use -h option to read help)"; exit 1; }
 
-MIRRORDIR=$HOME/ssd/localmirror_$ID
-XDTDIR=$HOME/ssd/xdt_$ID
-SHAREDDIR=$HOME/$DOCKER_USER/docker/share
+XDS_WKS=$HOME/xds-workspace
+XDTDIR=$XDS_WKS/.xdt_$ID
 
 SSH_PORT=$((2222 + ID))
 WWW_PORT=$((8000 + ID))
 BOOT_PORT=$((69 + ID))
 NBD_PORT=$((10809 + ID))
 
-mkdir -p $MIRRORDIR $XDTDIR $SHAREDDIR || exit 1
+### Create the new container
+mkdir -p $XDS_WKS $XDTDIR  || exit 1
 docker run \
        --publish=${SSH_PORT}:22 \
        --publish=${WWW_PORT}:8000 \
@@ -89,25 +130,17 @@ docker run \
        --detach=true \
        --hostname=$NAME --name=$NAME \
        --privileged -v /sys/fs/cgroup:/sys/fs/cgroup:ro \
-       -v $MIRRORDIR:/home/$DOCKER_USER/mirror \
-       -v $SHAREDDIR:/home/$DOCKER_USER/share \
+       -v $XDS_WKS:/home/$DOCKER_USER/xds-workspace \
        -v $XDTDIR:/xdt \
+    $USER_VOLUME_OPTION \
        -it $IMAGE
 if [ "$?" != "0" ]; then
     echo "An error was encountered while creating docker container."
     exit 1
 fi
 
-if ($FORCE); then
-    echo "Stopping xds-server..."
-    docker exec -t ${NAME} bash -c "systemctl stop xds-server" || exit 1
-    sleep 1
-    echo "Starting xds-server..."
-    docker exec -t ${NAME} bash -c "systemctl start xds-server" || exit 1
-fi
-
+### Ssh key
 echo "Copying your identity to container $NAME"
-#wait ssh service
 echo -n wait ssh service .
 res=3
 max=30
@@ -121,11 +154,32 @@ while [ $res -ne 0 ] && [ $count -le $max ]; do
 done
 echo
 
-ssh-keygen -R [$(hostname)]:$SSH_PORT -f ~/.ssh/known_hosts
-docker exec ${NAME} bash -c "mkdir -p /home/devel/.ssh"
-docker cp ~/.ssh/id_rsa.pub ${NAME}:/home/devel/.ssh/authorized_keys
-docker exec ${NAME} bash -c "chown devel:devel -R /home/devel/.ssh ;chmod 0700 /home/devel/.ssh;chmod 0600 /home/devel/.ssh/*"
-ssh -o StrictHostKeyChecking=no -p $SSH_PORT devel@$(hostname) exit
+ssh-keygen -R [localhost]:$SSH_PORT -f ~/.ssh/known_hosts
+docker exec ${NAME} bash -c "mkdir -p /home/$DOCKER_USER/.ssh"
+docker cp ~/.ssh/id_rsa.pub ${NAME}:/home/$DOCKER_USER/.ssh/authorized_keys
+docker exec ${NAME} bash -c "chown $DOCKER_USER:$DOCKER_USER -R /home/$DOCKER_USER/.ssh ;chmod 0700 /home/$DOCKER_USER/.ssh; chmod 0600 /home/$DOCKER_USER/.ssh/*"
+ssh -o StrictHostKeyChecking=no -p $SSH_PORT $DOCKER_USER@localhost exit
 
 echo "You can now login using:"
-echo "   ssh -p $SSH_PORT $DOCKER_USER@$(hostname)"
+echo "   ssh -p $SSH_PORT $DOCKER_USER@localhost"
+
+### User / Group id
+if ($UPDATE_UID); then
+    echo "Setup docker user and group id to match yours"
+    docker exec -t ${NAME} bash -c "systemctl stop xds-server" || exit 1
+    docker exec -t ${NAME} bash -c "usermod -u $(id -u) $DOCKER_USER && groupmod -g $(id -g) $DOCKER_USER" || exit 1
+    docker exec -t ${NAME} bash -c "chown -R $DOCKER_USER:$DOCKER_USER /home/$DOCKER_USER /tmp/xds*" || exit 1
+    docker exec -t ${NAME} bash -c "systemctl start xds-server" || exit 1
+    docker exec -t ${NAME} bash -c "systemctl start xds-server" || exit 1
+fi
+
+### Force xds-server restart
+if ($FORCE_RESTART); then
+    echo "Stopping xds-server..."
+    docker exec -t ${NAME} bash -c "systemctl stop xds-server" || exit 1
+    sleep 1
+    echo "Starting xds-server..."
+    docker exec -t ${NAME} bash -c "systemctl start xds-server" || exit 1
+fi
+
+echo "Done, docker container $NAME is ready to be used."