Shorter case statement

This commit is contained in:
Tomi Belan 2015-07-22 10:50:43 +02:00
parent 5085fa368b
commit c3efade5c4

47
bocker
View File

@ -1,7 +1,7 @@
#!/usr/bin/env bash #!/usr/bin/env bash
set -o errexit -o nounset -o pipefail; shopt -s nullglob set -o errexit -o nounset -o pipefail; shopt -s nullglob
btrfs_path='/var/bocker'; btrfs_path='/var/bocker';
function CHECK() { function bocker_check() {
[[ "$1" == 'img' ]] && TYPE='image' [[ "$1" == 'img' ]] && TYPE='image'
[[ "$1" == 'ps' ]] && TYPE='container' [[ "$1" == 'ps' ]] && TYPE='container'
[[ "$1" == '' ]] && TYPE='container or image' [[ "$1" == '' ]] && TYPE='container or image'
@ -12,7 +12,7 @@ if [[ "$2" == "$1"* ]]; then
fi fi
echo "No $TYPE named '$2' exists" && exit 1 echo "No $TYPE named '$2' exists" && exit 1
} }
function INIT() { function bocker_init() {
if [[ -d "$1" ]]; then if [[ -d "$1" ]]; then
uuid="img_$(shuf -i 10000-99999 -n 1)" uuid="img_$(shuf -i 10000-99999 -n 1)"
btrfs subvolume create "$btrfs_path/$uuid" > /dev/null btrfs subvolume create "$btrfs_path/$uuid" > /dev/null
@ -22,26 +22,27 @@ else
echo "No directory named '$1' exists" echo "No directory named '$1' exists"
fi fi
} }
function RM() { function bocker_rm() {
CHECK '' "$1" bocker_check '' "$1"
btrfs subvolume delete "$btrfs_path/$1" > /dev/null btrfs subvolume delete "$btrfs_path/$1" > /dev/null
echo "Removed: $1" echo "Removed: $1"
} }
function IMAGES() { function bocker_images() {
echo -e "IMAGE_ID" echo -e "IMAGE_ID"
for img in "$btrfs_path"/img_*; do for img in "$btrfs_path"/img_*; do
basename "$img" basename "$img"
done done
} }
function PS() { function bocker_ps() {
echo -e "CONTAINER_ID\t\tCOMMAND" echo -e "CONTAINER_ID\t\tCOMMAND"
for ps in "$btrfs_path"/ps_*; do for ps in "$btrfs_path"/ps_*; do
ps=$(basename "$ps") ps=$(basename "$ps")
echo -e "$ps\t\t$(cat "$btrfs_path/$ps/$ps.cmd")" echo -e "$ps\t\t$(cat "$btrfs_path/$ps/$ps.cmd")"
done done
} }
function RUN() { function bocker_run() {
CHECK 'img' "$1" bocker_check 'img' "$1"
cmd=${@:2}
uuid="ps_$(shuf -i 10000-99999 -n 1)" uuid="ps_$(shuf -i 10000-99999 -n 1)"
ip link add dev veth0_"$uuid" type veth peer name veth1_"$uuid" ip link add dev veth0_"$uuid" type veth peer name veth1_"$uuid"
ip link set dev veth0_"$uuid" up ip link set dev veth0_"$uuid" up
@ -54,23 +55,23 @@ ip netns exec netns_"$uuid" ip link set dev veth1_"$uuid" up
ip netns exec netns_"$uuid" ip route add default via 10.0.0.1 ip netns exec netns_"$uuid" ip route add default via 10.0.0.1
btrfs subvolume snapshot "$btrfs_path/$1" "$btrfs_path/$uuid" > /dev/null btrfs subvolume snapshot "$btrfs_path/$1" "$btrfs_path/$uuid" > /dev/null
echo 'nameserver 8.8.8.8' > "$btrfs_path/$uuid"/etc/resolv.conf echo 'nameserver 8.8.8.8' > "$btrfs_path/$uuid"/etc/resolv.conf
echo "$2" > "$btrfs_path/$uuid/$uuid.cmd" echo "$cmd" > "$btrfs_path/$uuid/$uuid.cmd"
ip netns exec netns_"$uuid" "unshare" -fp --mount-proc "chroot" \ ip netns exec netns_"$uuid" "unshare" -fp --mount-proc "chroot" \
"$btrfs_path/$uuid" /bin/sh -c "/bin/mount -t proc proc /proc && $2" \ "$btrfs_path/$uuid" /bin/sh -c "/bin/mount -t proc proc /proc && $cmd" \
2>&1 | tee "$btrfs_path/$uuid/$uuid.log" || true 2>&1 | tee "$btrfs_path/$uuid/$uuid.log" || true
ip link del dev veth0_"$uuid" ip link del dev veth0_"$uuid"
ip netns del netns_"$uuid" ip netns del netns_"$uuid"
} }
function LOGS() { function bocker_logs() {
CHECK 'ps' "$1" bocker_check 'ps' "$1"
cat "$btrfs_path/$1/$1.log" cat "$btrfs_path/$1/$1.log"
} }
function COMMIT() { function bocker_commit() {
CHECK 'ps' "$1" && CHECK 'img' "$2" && RM "$2" bocker_check 'ps' "$1" && bocker_check 'img' "$2" && bocker_rm "$2"
btrfs subvolume snapshot "$btrfs_path/$1" "$btrfs_path/$2" > /dev/null btrfs subvolume snapshot "$btrfs_path/$1" "$btrfs_path/$2" > /dev/null
echo "Created: $2" echo "Created: $2"
} }
function HELP() { function bocker_help() {
echo -e "Create an image: \n\t./bocker init <image_directory>\n" echo -e "Create an image: \n\t./bocker init <image_directory>\n"
echo -e "List images: \n\t./bocker images\n" echo -e "List images: \n\t./bocker images\n"
echo -e "Create a container: \n\t./bocker run <image_id> <command>\n" echo -e "Create a container: \n\t./bocker run <image_id> <command>\n"
@ -79,18 +80,8 @@ echo -e "View logs from a container: \n\t./bocker logs <container_id>\n"
echo -e "Delete an image or container: \n\t./bocker rm <image_or_container_id>" echo -e "Delete an image or container: \n\t./bocker rm <image_or_container_id>"
exit 0 exit 0
} }
[[ -z "${1-}" ]] && HELP [[ -z "${1-}" ]] && bocker_help
case $1 in case $1 in
init) INIT "$2" ;; init|rm|images|ps|run|logs|commit) bocker_"$1" "${@:2}" ;;
rm) RM "$2" ;; *) bocker_help ;;
images) IMAGES ;;
ps) PS ;;
run)
IMAGE="$2"
shift && shift
RUN "$IMAGE" "$*"
;;
logs) LOGS "$2" ;;
commit) COMMIT "$2" "$3" ;;
*) HELP ;;
esac esac