retry command

dependency/shell/run.sh

@@ -1,71 +1,228 @@
 #!/usr/local
-
 python_dir=/usr/local/PlayerCode/CAPI/python/PyAPI
 python_main_dir=/usr/local/PlayerCode/CAPI/python
-playback_dir=/usr/local/playback
+playback_dir=/usr/local/output
+map_dir=/usr/local/map
+mkdir -p $playback_dir
 
-if [ $EXPOSED -eq 1 ]; then
-    nice -10 ./Server --port 8888 --studentCount 4 --trickerCount 1 --resultFileName $playback_dir/result --gameTimeInSecond $TIME --mode $MODE --mapResource $MAP --url $URL --token $TOKEN --fileName $playback_dir/video --startLockFile $playback_dir/start.lock > $playback_dir/server.log 2>&1 &
-    server_pid=$!
-else
-    nice -10 ./Server --port 8888 --studentCount 4 --trickerCount 1 --resultFileName $playback_dir/result --gameTimeInSecond $TIME --mode $MODE --mapResource $MAP --notAllowSpectator --url $URL --token $TOKEN --fileName $playback_dir/video --startLockFile $playback_dir/start.lock > $playback_dir/server.log 2>&1 &
-    server_pid=$!
+# initialize
+if [[ "${MODE}" == "ARENA" ]]; then
+    MODE_NUM=0
+elif [[ "${MODE}" == "COMPETITION" ]]; then
+    MODE_NUM=1
 fi
-sleep 5
-for k in {1..2}
-do
-    pushd /usr/local/team$k
-    if [ $k -eq 1 ]; then
-        for i in {1..4}
+
+# set default value
+: "${TEAM_SEQ_ID:=0}"
+: "${TEAM_LABELS:=Student:Tricker}"
+: "${TEAM_LABEL:=Student}"
+: "${EXPOSED=1}"
+: "${MODE_NUM=0}"
+: "${GAME_TIME=10}"
+: "${CONNECT_IP=172.17.0.1}"
+
+
+get_current_team_label() {
+    if [ $TEAM_SEQ_ID -eq $2 ]; then
+        echo "find current team label: $1"
+        current_team_label=$1
+    fi
+}
+
+read_array() {
+    callback=$1
+    echo "read array: set callback command: $callback"
+
+    IFS=':' read -r -a fields <<< "$2"
+
+    count=0 # loop count
+
+    for field in "${fields[@]}"
+    do
+        echo "parse field: $field"
+        param0=$field
+
+        # call command
+        run_command="$callback $param0 $count"
+        echo "Call Command: $run_command"
+        $run_command
+
+        count=$((count+1))
+    done
+}
+
+
+function retry_command {
+    local command="$1"
+    local max_attempts=2
+    local attempt_num=1
+    local sleep_seconds=10
+
+    while [ $attempt_num -le $max_attempts ]; do
+        echo "Attempt $attempt_num / $max_attempts to run command: $command"
+
+        eval $command &
+        local PID=$!
+
+        sleep $sleep_seconds
+
+        if kill -0 $PID 2>/dev/null; then
+            echo "Connected to server successfully."
+            return 0
+        else
+            echo "Failed to connect to server. Retrying..."
+            ((attempt_num++))
+        fi
+    done
+
+    echo "Failed to connect to server after $max_attempts attempts."
+    return 1
+}
+
+
+
+if [ "$TERMINAL" = "SERVER" ]; then
+    map_path=$map_dir/$MAP_ID.txt
+    # allow spectator always.
+    nice -10 ./Server --port 8888 --studentCount 4 --trickerCount 1 --resultFileName $playback_dir/result --gameTimeInSecond $GAME_TIME --mode $MODE_NUM --mapResource $map_path --url $SCORE_URL --token $TOKEN --fileName $playback_dir/video --startLockFile $playback_dir/start.lock > $playback_dir/server.log 2>&1 &
+    server_pid=$!
+    echo "server pid: $server_pid"
+    ls $playback_dir
+
+    echo "SCORE URL: $SCORE_URL"
+    echo "FINISH URL: $FINISH_URL"
+
+    echo "waiting..."
+    sleep 30 # wait connection time
+    echo "watching..."
+
+    if [ -f $playback_dir/start.lock ]; then
+        ps -p $server_pid
+        while [ $? -eq 0 ]
         do
-            j=$((i - 1))
-            if [ -f "./player$i.py" ]; then
-                cp -r $python_main_dir $python_main_dir$i
-                cp -f ./player$i.py $python_main_dir$i/PyAPI/AI.py
-                nice -0 python3 $python_main_dir$i/PyAPI/main.py -I 127.0.0.1 -P 8888 -p $j > $playback_dir/team$k-player$j.log 2>&1 &
-            elif [ -f "./capi$i" ]; then
-                nice -0 ./capi$i -I 127.0.0.1 -P 8888 -p $j > $playback_dir/team$k-player$j.log 2>&1 &
-            else
-                echo "ERROR. $i is not found."
-            fi
+            sleep 1
+            ps -p $server_pid > /dev/null 2>&1
         done
+
+        # update score by finish url
+
+        echo "Getting result score..."
+        result=$(cat $playback_dir/result.json)
+        score0=$(echo "$result" | grep -oP '(?<="Student":)\d+') # Student score
+        score1=$(echo "$result" | grep -oP '(?<="Tricker":)\d+') # Tricker score
+        echo "Result score: Student: $score0, Tricker: $score1"
+
+        # detect two team seqs:
+        echo "Parsing TEAM_LABELS: $TEAM_LABELS"
+        TEAM_SEQ_ID=0
+        read_array get_current_team_label $TEAM_LABELS
+        if [[ "${current_team_label}" == "Student" ]]; then
+            echo "Parse Success: 1st team is Student"
+            finish_payload='{"result": {"status": "Finished", "scores": ['${score0}', '${score1}']}}'
+        elif [[ "${current_team_label}" == "Tricker" ]]; then
+            echo "Parse Success: 1st team is Tricker"
+            finish_payload='{"result": {"status": "Finished", "scores": ['${score1}', '${score0}']}}'
+        else
+            echo "Parse Failure: 1st team is Unknown"
+            finish_payload='{"result": {"status": "Crashed", "scores": [0, 0]}}'
+        fi
+
+        if [[ -n $finish_payload ]]; then
+            echo "FINISH_URL: $FINISH_URL, payload: $finish_payload. Start update score..."
+            curl $FINISH_URL -X POST -H "Content-Type: application/json" -H "Authorization: Bearer $TOKEN" -d "${finish_payload}" > $playback_dir/send.log 2>&1
+        else
+            echo "Payload not set."
+        fi
+
+        # Congratulations! You have finished the competition!!!!!
+        # touch $playback_dir/finish.lock
+        echo "Finish!"
     else
-        for i in {5..5}
-        do
-            j=$((i - 1))
-            if [ -f "./player$i.py" ]; then
+        echo "Failed to start game."
+        touch $playback_dir/finish.lock
+        touch temp.lock
+        mv -f temp.lock $playback_dir/video.thuaipb
+        kill -9 $server_pid
+        finish_payload='{"result": {"status": "Crashed", "scores": [0, 0]}}'
+        curl $FINISH_URL -X POST -H "Content-Type: application/json" -H "Authorization: Bearer $TOKEN" -d "${finish_payload}" > $playback_dir/send.log 2>&1
+    fi
+
+
+elif [ "$TERMINAL" = "CLIENT" ]; then
+    echo "Client Mode! Team Label data - $TEAM_LABEL"
+
+    # parse team label name
+    current_team_label=$TEAM_LABEL
+
+    # k is an enum (1,2), 1 = Student, 2 = Tricker
+    if [ "$current_team_label" = "Student" ]; then
+        k=1
+    elif [ "$current_team_label" = "Tricker" ]; then
+        k=2
+    else
+        echo "Error: Invalid Team Label"
+        exit
+    fi
+    pushd /usr/local/code
+        if [ $k -eq 1 ]; then
+            for i in {1..4}
+            do
+                j=$((i - 1)) # student player id from 0 to 3
+                code_name=Student$i
+                if [ -f "./$code_name.py" ]; then
+                    echo "find ./$code_name.py"
+                    cp -r $python_main_dir $python_main_dir$i
+                    cp -f ./$code_name.py $python_main_dir$i/PyAPI/AI.py
+
+                    command="nice -0 python3 $python_main_dir$i/PyAPI/main.py -I $CONNECT_IP -P $PORT -p $j > $playback_dir/team$k-player$j.log 2>&1"
+
+                    retry_command "$command" &
+
+                    ps -aux |grep main.py
+
+                elif [ -f "./$code_name" ]; then
+                    echo "find ./$code_name"
+
+                    command="nice -0 ./$code_name -I $CONNECT_IP -P $PORT -p $j > $playback_dir/team$k-player$j.log 2>&1"
+
+                    ps -aux |grep $code_name
+
+                else
+                    echo "ERROR. $code_name is not found."
+                fi
+            done
+        else
+            i=5
+            j=4 # tricker id is 4
+            code_name=Tricker
+            if [ -f "./$code_name.py" ]; then
+                echo "find ./$code_name.py"
                 cp -r $python_main_dir $python_main_dir$i
-                cp -f ./player$i.py $python_main_dir$i/PyAPI/AI.py
-                nice -0 python3 $python_main_dir$i/PyAPI/main.py -I 127.0.0.1 -P 8888 -p $j > $playback_dir/team$k-player$j.log 2>&1 &
-            elif [ -f "./capi$i" ]; then
-                nice -0 ./capi$i -I 127.0.0.1 -P 8888 -p $j > $playback_dir/team$k-player$j.log 2>&1 &
+                cp -f ./$code_name.py $python_main_dir$i/PyAPI/AI.py
+
+                command="nice -0 python3 $python_main_dir$i/PyAPI/main.py -I $CONNECT_IP -P $PORT -p $j > $playback_dir/team$k-player$j.log 2>&1"
+
+                retry_command "$command"
+
+                ps -aux |grep main.py
+            elif [ -f "./$code_name" ]; then
+                echo "find ./$code_name"
+
+                command="nice -0 ./$code_name -I $CONNECT_IP -P $PORT -p $j > $playback_dir/team$k-player$j.log 2>&1"
+
+                retry_command "$command"
+
+                ps -aux |grep $code_name
             else
-                echo "ERROR. $i is not found."
+                echo "ERROR. $code_name is not found."
             fi
-        done
-    fi
-    popd
-done
+        fi
 
-sleep 10
+        # curl $CONNECT_IP:$PORT
 
-if [ -f $playback_dir/start.lock ]; then
-    ps -p $server_pid
-    while [ $? -eq 0 ]
-    do
-        sleep 1
-        ps -p $server_pid > /dev/null 2>&1
-    done
-    # result=$(cat /usr/local/playback/result.json)
-    # score0=$(echo "$result" | grep -oP '(?<="Student":)\d+')
-    # score1=$(echo "$result" | grep -oP '(?<="Tricker":)\d+')
-    # curl $URL -X PUT -H "Content-Type: application/json" -H "Authorization: Bearer $TOKEN" -d '{"result":[{"team_id":0, "score":'${score0}'}, {"team_id":1, "score":'${score1}'}], "mode":'${MODE}'}'> $playback_dir/send.log 2>&1
-    touch $playback_dir/finish.lock
-    echo "Finish"
+        sleep $((GAME_TIME * 2))
+
+    popd
 else
-    echo "Failed to start game."
-    touch $playback_dir/finish.lock
-    touch temp.lock
-    mv -f temp.lock $playback_dir/video.thuaipb
-    kill -9 $server_pid
+    echo "VALUE ERROR: TERMINAL is neither SERVER nor CLIENT."
 fi