fix failed to retrieve room_id

.gitignore

@@ -127,4 +127,6 @@ dist
 .yarn/unplugged
 .yarn/build-state.yml
 .yarn/install-state.gz
-.pnp.*
+.pnp.*
+
+.tests/*

package.json

@@ -1,6 +1,6 @@
 {
   "name": "tiktok-live-connector",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "description": "Node.js module to receive live stream chat events like comments and gifts from TikTok LIVE",
   "main": "index.js",
   "types": "./dist/index.d.ts",

src/index.js

@@ -337,10 +337,23 @@ class WebcastPushConnection extends EventEmitter {
     async #retrieveRoomId() {
         try {
             let mainPageHtml = await this.#httpClient.getMainPage(`@${this.#uniqueStreamerId}/live`);
-            let roomId = getRoomIdFromMainPageHtml(mainPageHtml);
 
-            this.#roomId = roomId;
-            this.#clientParams.room_id = roomId;
+            try {
+                let roomId = getRoomIdFromMainPageHtml(mainPageHtml);
+
+                this.#roomId = roomId;
+                this.#clientParams.room_id = roomId;
+            } catch (err) {
+                // Use fallback method
+                let roomData = await this.#httpClient.getJsonObjectFromTiktokApi('api-live/user/room/', {
+                    ...this.#clientParams,
+                    uniqueId: this.#uniqueStreamerId,
+                    sourceType: 54,
+                });
+
+                this.#roomId = roomData.data.user.roomId;
+                this.#clientParams.room_id = roomData.data.user.roomId;
+            }
         } catch (err) {
             throw new Error(`Failed to retrieve room_id from page source. ${err.message}`);
         }

src/lib/tiktokHttpClient.js

@@ -78,6 +78,12 @@ class TikTokHttpClient {
         let response = await this.#post(`${Config.TIKTOK_URL_WEBCAST}${path}`, params, formData, 'json');
         return response.data;
     }
+
+    async getJsonObjectFromTiktokApi(path, params, shouldSign) {
+        let url = await this.#buildUrl(Config.TIKTOK_URL_WEB, path, params, shouldSign);
+        let response = await this.#get(url, 'json');
+        return response.data;
+    }
 }
 
 module.exports = TikTokHttpClient;