app/build.gradle

@@ -117,6 +117,8 @@ dependencies {
 
     implementation 'com.arthenica:mobile-ffmpeg-full-gpl:4.3.1.LTS'
 
+    implementation 'org.java-websocket:Java-WebSocket:1.5.1'
+
     compileOnly 'org.projectlombok:lombok:1.18.10'
     annotationProcessor 'org.projectlombok:lombok:1.18.10'
     annotationProcessor 'com.github.bumptech.glide:compiler:4.11.0'

app/src/main/java/io/lbry/browser/MainActivity.java

@@ -98,6 +98,8 @@ import androidx.appcompat.widget.Toolbar;
 import androidx.recyclerview.widget.LinearLayoutManager;
 import androidx.recyclerview.widget.RecyclerView;
 
+import org.java_websocket.client.WebSocketClient;
+import org.java_websocket.handshake.ServerHandshake;
 import org.json.JSONArray;
 import org.json.JSONException;
 import org.json.JSONObject;
@@ -107,6 +109,7 @@ import java.io.File;
 import java.io.FileInputStream;
 import java.io.InputStreamReader;
 import java.net.ConnectException;
+import java.net.URI;
 import java.text.DecimalFormat;
 import java.util.ArrayList;
 import java.util.Arrays;
@@ -199,6 +202,7 @@ import io.lbry.lbrysdk.ServiceHelper;
 import io.lbry.lbrysdk.Utils;
 import lombok.Getter;
 import lombok.Setter;
+import lombok.SneakyThrows;
 import okhttp3.OkHttpClient;
 
 public class MainActivity extends AppCompatActivity implements SdkStatusListener {
@@ -208,6 +212,7 @@ public class MainActivity extends AppCompatActivity implements SdkStatusListener
     private static final int REMOTE_NOTIFICATION_REFRESH_TTL = 300000; // 5 minutes
     public static final String SKU_SKIP = "lbryskip";
 
+    private boolean shuttingDown;
     private Date remoteNotifcationsLastLoaded;
     private Map<String, Class> specialRouteFragmentClassMap;
     @Getter
@@ -240,6 +245,7 @@ public class MainActivity extends AppCompatActivity implements SdkStatusListener
 
     @Setter
     private BackPressInterceptor backPressInterceptor;
+    private WebSocketClient webSocketClient;
 
     @Getter
     private String firebaseMessagingToken;
@@ -1002,11 +1008,15 @@ public class MainActivity extends AppCompatActivity implements SdkStatusListener
 
     @Override
     protected void onDestroy() {
+        shuttingDown = true;
         unregisterReceivers();
         NotificationManagerCompat notificationManager = NotificationManagerCompat.from(this);
         if (receivedStopService || !isServiceRunning(this, LbrynetService.class)) {
             notificationManager.cancelAll();
         }
+        if (webSocketClient != null) {
+            webSocketClient.close();
+        }
         if (dbHelper != null) {
             dbHelper.close();
         }
@@ -1072,10 +1082,40 @@ public class MainActivity extends AppCompatActivity implements SdkStatusListener
         task.executeOnExecutor(AsyncTask.THREAD_POOL_EXECUTOR);
     }
 
+
+
+    @SneakyThrows
+    private void checkWebSocketClient() {
+        if ((webSocketClient == null || webSocketClient.isClosed()) && !Helper.isNullOrEmpty(Lbryio.AUTH_TOKEN)) {
+            webSocketClient = new WebSocketClient(new URI(String.format("%s%s", Lbryio.WS_CONNECTION_BASE_URL, Lbryio.AUTH_TOKEN))) {
+                @Override
+                public void onOpen(ServerHandshake handshakedata) { }
+
+                @Override
+                public void onMessage(String message) {
+                    loadRemoteNotifications(false);
+                }
+
+                @Override
+                public void onClose(int code, String reason, boolean remote) {
+                    if (!shuttingDown) {
+                        // attempt to re-establish the connection if the app isn't being closed
+                        checkWebSocketClient();
+                    }
+                }
+
+                @Override
+                public void onError(Exception ex) { }
+            };
+            webSocketClient.connect();
+        }
+    }
+
     @Override
     protected void onResume() {
         super.onResume();
         checkPurchases();
+        checkWebSocketClient();
         enteringPIPMode = false;
 
         applyNavbarSigninPadding();
@@ -2649,6 +2689,7 @@ public class MainActivity extends AppCompatActivity implements SdkStatusListener
                 loadRemoteNotifications(false);
 
                 checkUrlIntent(getIntent());
+                checkWebSocketClient();
                 LbryAnalytics.logEvent(LbryAnalytics.EVENT_APP_LAUNCH);
                 appStarted = true;
             }

app/src/main/java/io/lbry/browser/utils/Lbryio.java

@@ -57,6 +57,7 @@ public final class Lbryio {
 
     public static final String TAG = "Lbryio";
     public static final String CONNECTION_STRING = "https://api.lbry.com";
+    public static final String WS_CONNECTION_BASE_URL = "wss://api.lbry.com/subscribe?auth_token=";
     public static final String AUTH_TOKEN_PARAM = "auth_token";
     public static List<Subscription> subscriptions = new ArrayList<>();
     public static List<Claim> cacheResolvedSubscriptions = new ArrayList<>();