enhancement: docker-build-and-run.sh

docker-build-and-run.sh

@@ -0,0 +1,162 @@
+#!/usr/bin/env bash
+
+#? Description:
+#?   Build the Docker images and run the containers for testing.
+#?
+#?   Without any option, it will build the images and run the containers in the foreground.
+#?
+#? Usage:
+#?   docker-build-and-run.sh [-P] [-m MGT_PORT] [-p SS_PORTS] [-e ENCRYPT] -d DOMAIN [-N DNS] [-E DNS_ENV]
+#?   docker-build-and-run.sh [-h]
+#?
+#? Options:
+#?   [-P]
+#?
+#?   Enable to fetch the env `DOCKER_BUILD_ARGS_PROXY` and pass to docker build command.
+#?   If the env `DOCKER_BUILD_ARGS_PROXY` is not set or empty, exit with error.
+#?
+#?   [-m MGT_PORT]
+#?
+#?   Specify the management port for the shadowsocks manager.
+#?   Default is 6001.
+#?
+#?   [-p SS_PORTS]
+#?
+#?   Specify the shadowsocks ports to be exposed.
+#?   Default is 8381-8385.
+#?
+#?   [-e ENCRYPT]
+#?
+#?   Specify the encryption method for shadowsocks.
+#?   Default is aes-256-cfb.
+#?
+#?   -d DOMAIN
+#?
+#?   Specify the domain for the shadowsocks server.
+#?
+#?   [-N DNS]
+#?
+#?   Specify the DNS provider for the DOMAIN.
+#?
+#?   [-E DNS_ENV]
+#?
+#?   Specify the environment variables for the DNS provider.
+#?
+#?   [-h]
+#?
+#?   This help.
+#?
+#? Environment:
+#?   The following environment variables are used by this script conditionally:
+#?
+#?   - DOCKER_BUILD_ARGS_PROXY="--build-arg http_proxy=http://host.docker.internal:1086 --build-arg https_proxy=http://host.docker.internal:1086 --build-arg all_proxy=socks5://host.docker.internal:1086"
+#?
+#?     Optional, default is unset.
+#?     Set the proxy for the Docker build. Please replace the proxy port with your actual port.
+#?
+
+# exit on any error
+set -e -o pipefail
+
+function usage () {
+    awk '/^#\?/ {sub("^[ ]*#\\?[ ]?", ""); print}' "$0" \
+        | awk '{gsub(/^[^ ]+.*/, "\033[1m&\033[0m"); print}'
+}
+
+function check-vars () {
+    declare var ret=0
+    for var in "$@"; do
+        if [[ -z ${!var} ]]; then
+            echo "FATAL: environment variable $var is not set or empty." >&2
+            (( ret++ ))
+        fi
+    done
+    return $ret
+}
+
+function main () {
+    declare proxy_flag=0 \
+            mgt_port=6001 ss_ports=8381-8385 encrypt=aes-256-cfb domain dns dns_env \
+            OPTIND OPTARG opt
+
+    while getopts Pm:p:e:d:N:E:h opt; do
+        case $opt in
+            P)
+                proxy_flag=1
+                ;;
+            m)
+                mgt_port=$OPTARG
+                ;;
+            p)
+                ss_ports=$OPTARG
+                ;;
+            e)
+                encrypt=$OPTARG
+                ;;
+            d)
+                domain=$OPTARG
+                ;;
+            N)
+                dns=$OPTARG
+                ;;
+            E)
+                dns_env=$OPTARG
+                ;;
+            *)
+                usage
+                return 255
+                ;;
+        esac
+    done
+
+    check-vars domain mgt_port ss_ports encrypt
+
+    declare build_opts=() \
+            run_opts=(--restart=always) \
+            run_env_opts=(-e V2RAY=1 -e DOMAIN="$domain" -e DNS="$dns" -e DNS_ENV="$dns_env") \
+            run_port_opts=(-p "$mgt_port:$mgt_port/UDP" -p "$ss_ports:$ss_ports" -p "$ss_ports:$ss_ports/UDP") \
+            run_cmd_opts=( ss-manager
+                --manager-address "0.0.0.0:$mgt_port"
+                --executable /usr/local/bin/ss-server -m "$encrypt" -s 0.0.0.0 -u
+                --plugin v2ray-plugin --plugin-opts "server;tls;host=$domain"
+            )
+
+    if [[ $proxy_flag -eq 1 ]]; then
+        check-vars DOCKER_BUILD_ARGS_PROXY
+        # do not quote it
+        # shellcheck disable=SC2206
+        build_opts+=($DOCKER_BUILD_ARGS_PROXY)
+    fi
+
+    declare image_name=alexzhangs/shadowsocks-libev-v2ray
+
+    declare script_dir
+    script_dir=$(dirname "$0")
+
+    declare -a BUILD_OPTS RUN_OPTS
+
+    function __build_and_run__ () {
+        # build the image
+        echo ""
+        echo "INFO: docker build ${BUILD_OPTS[*]}"
+        docker build "${BUILD_OPTS[@]}"
+
+        # run the container
+        echo ""
+        echo "INFO: docker run ${RUN_OPTS[*]}"
+        docker run "${RUN_OPTS[@]}" || :
+    }
+
+    declare image_tag image container
+    image_tag=dev-$(date +%Y%m%d-%H%M)
+    image="$image_name:$image_tag"
+    container="ss-libev-$image_tag"
+    BUILD_OPTS=( "${build_opts[@]}" -t "$image" -f "$script_dir/Dockerfile" "$script_dir" )
+    RUN_OPTS=( "${run_opts[@]}" "${run_env_opts[@]}" "${run_port_opts[@]}" --name "$container" "$image" "${run_cmd_opts[@]}" )
+
+    __build_and_run__
+}
+
+main "$@"
+
+exit