From b2b6b3231dbec1de09b14a8cf81d7dd191970215 Mon Sep 17 00:00:00 2001
From: zarazaex69 <zarazaex@tuta.io>
Date: Mon, 20 Apr 2026 04:45:39 +0300
Subject: [PATCH] feat: add wb_stream provider support to cnc and srv scripts

---
 script/cnc.sh |  6 +++++-
 script/srv.sh | 41 +++++++++++++++++++++++++++++++++++++++--
 2 files changed, 44 insertions(+), 3 deletions(-)

diff --git a/script/cnc.sh b/script/cnc.sh
index bff90bd..9736882 100755
--- a/script/cnc.sh
+++ b/script/cnc.sh
@@ -69,12 +69,16 @@ echo ""
 echo "Select provider:"
 echo "  1) telemost"
 echo "  2) jazz"
-read -p "Enter choice [1-2, default: 1]: " PROVIDER_CHOICE
+echo "  3) wb_stream"
+read -p "Enter choice [1-3, default: 1]: " PROVIDER_CHOICE
 
 case "$PROVIDER_CHOICE" in
     2)
         PROVIDER="jazz"
         ;;
+    3)
+        PROVIDER="wb_stream"
+        ;;
     *)
         PROVIDER="telemost"
         ;;
diff --git a/script/srv.sh b/script/srv.sh
index 894d46b..993b4ee 100755
--- a/script/srv.sh
+++ b/script/srv.sh
@@ -60,12 +60,16 @@ echo ""
 echo "Select provider:"
 echo "  1) telemost"
 echo "  2) jazz"
-read -p "Enter choice [1-2, default: 1]: " PROVIDER_CHOICE
+echo "  3) wb_stream"
+read -p "Enter choice [1-3, default: 1]: " PROVIDER_CHOICE
 
 case "$PROVIDER_CHOICE" in
     2)
         PROVIDER="jazz"
         ;;
+    3)
+        PROVIDER="wb_stream"
+        ;;
     *)
         PROVIDER="telemost"
         ;;
@@ -93,6 +97,25 @@ if [ "$PROVIDER" = "jazz" ]; then
             echo "[*] Will auto-generate Jazz room"
             ;;
     esac
+elif [ "$PROVIDER" = "wb_stream" ]; then
+    echo "WB Stream room options:"
+    echo "  1) Auto-generate new room (recommended)"
+    echo "  2) Use specific room ID"
+    read -p "Enter choice [1-2, default: 1]: " WB_CHOICE
+    
+    case "$WB_CHOICE" in
+        2)
+            read -p "Enter Room ID: " ROOM_ID
+            if [ -z "$ROOM_ID" ]; then
+                echo "[X] Room ID cannot be empty"
+                exit 1
+            fi
+            ;;
+        *)
+            ROOM_ID="any"
+            echo "[*] Will auto-generate WB Stream room"
+            ;;
+    esac
 else
     read -p "Enter Room ID: " ROOM_ID
     if [ -z "$ROOM_ID" ]; then
@@ -189,6 +212,20 @@ if [ "$PROVIDER" = "jazz" ] && [ "$ROOM_ID" = "any" ]; then
     else
         echo "[+] Jazz room created: $ACTUAL_ROOM_ID"
     fi
+elif [ "$PROVIDER" = "wb_stream" ] && [ "$ROOM_ID" = "any" ]; then
+    echo "[*] Waiting for WB Stream room creation..."
+    sleep 2
+    LOGS=$(podman logs $CONTAINER_NAME 2>&1)
+    ACTUAL_ROOM_ID=$(echo "$LOGS" | grep -oP 'WB Stream room created: \K[^\s]+' | head -1)
+    
+    if [ -z "$ACTUAL_ROOM_ID" ]; then
+        echo "[!] WARNING: Could not extract WB Stream room ID from logs"
+        echo "[*] Full logs:"
+        podman logs $CONTAINER_NAME
+        ACTUAL_ROOM_ID="(check logs above)"
+    else
+        echo "[+] WB Stream room created: $ACTUAL_ROOM_ID"
+    fi
 fi
 
 echo ""
@@ -212,4 +249,4 @@ echo "  podman stop $CONTAINER_NAME"
 echo ""
 echo "Client command:"
 echo "  ./olcrtc -mode cnc -provider \"$PROVIDER\" -id \"$ACTUAL_ROOM_ID\" -key \"$KEY\" -socks-port 1080"
-echo ""
\ No newline at end of file
+echo ""