chore: refactor rxjs

client/lib/source/poll-source.ts

@@ -1,13 +1,14 @@
-import { BehaviorSubject, concat, from, Observable, of } from 'rxjs'
 import {
-  concatMap,
-  delay,
-  map,
-  skip,
-  switchMap,
-  take,
-  tap,
-} from 'rxjs/operators'
+  BehaviorSubject,
+  concat,
+  from,
+  ignoreElements,
+  Observable,
+  of,
+  repeatWhen,
+  timer,
+} from 'rxjs'
+import { concatMap, map, switchMap, take, tap } from 'rxjs/operators'
 import { Store } from '../store'
 import { Http, Update } from '../types'
 import { Source } from './source'
@@ -23,33 +24,14 @@ export class PollSource<T> implements Source<T> {
     private readonly http: Http<T>,
   ) {}
 
-  watch$(store: Store<T>): Observable<RPCResponse<Update<T>>> {
-    const polling$ = new BehaviorSubject('')
-
-    const updates$ = of({}).pipe(
-      concatMap(_ => store.sequence$),
+  watch$({ sequence$ }: Store<T>): Observable<RPCResponse<Update<T>>> {
+    return sequence$.pipe(
       concatMap(seq => this.http.getRevisions(seq)),
       take(1),
-    )
-
-    const delay$ = of([]).pipe(
-      delay(this.pollConfig.cooldown),
-      tap(_ => polling$.next('')),
-      skip(1),
-    )
-
-    const poll$ = concat(updates$, delay$)
-
-    return polling$.pipe(
-      switchMap(_ => poll$),
-      concatMap(res => {
-        if (Array.isArray(res)) {
-          return from(res) // takes Revision[] and converts it into Observable<Revision>
-        } else {
-          return of(res) // takes Dump<T> and converts it into Observable<Dump<T>>
-        }
-      }),
-      map(result => ({ result, jsonrpc: '2.0' })),
+      // Revision[] converted it into Observable<Revision>, Dump<T> into Observable<Dump<T>>
+      concatMap(res => (Array.isArray(res) ? from(res) : of(res))),
+      map(result => ({ result, jsonrpc: '2.0' as const })),
+      repeatWhen(() => timer(this.pollConfig.cooldown)),
     )
   }
 }

client/lib/source/ws-source.ts

@@ -1,27 +1,24 @@
 import { Observable } from 'rxjs'
-import {
-  webSocket,
-  WebSocketSubject,
-  WebSocketSubjectConfig,
-} from 'rxjs/webSocket'
+import { webSocket, WebSocketSubject } from 'rxjs/webSocket'
 import { Update } from '../types'
 import { Source } from './source'
 
 export class WebsocketSource<T> implements Source<T> {
-  private websocket$: WebSocketSubject<RPCResponse<Update<T>>> | undefined
+  private websocket$: WebSocketSubject<RPCResponse<Update<T>>> = webSocket({
+    url: this.url,
+    openObserver: {
+      next: () => {
+        this.websocket$.next(this.document.cookie as any)
+      },
+    },
+  })
 
-  constructor(private readonly url: string) {}
+  constructor(
+    private readonly url: string,
+    private readonly document: Document,
+  ) {}
 
   watch$(): Observable<RPCResponse<Update<T>>> {
-    const fullConfig: WebSocketSubjectConfig<RPCResponse<Update<T>>> = {
-      url: this.url,
-      openObserver: {
-        next: () => {
-          this.websocket$!.next(document.cookie as any)
-        },
-      },
-    }
-    this.websocket$ = webSocket(fullConfig)
     return this.websocket$
   }
 }