paperless-ngx/src-ui/src/app/services/rest/abstract-paperless-service.ts

import { HttpClient, HttpParams } from '@angular/common/http'
import { Observable, of, Subject } from 'rxjs'
import { map, publishReplay, refCount } from 'rxjs/operators'
import { ObjectWithId } from 'src/app/data/object-with-id'
import { Results } from 'src/app/data/results'
import { environment } from 'src/environments/environment'

export abstract class AbstractPaperlessService<T extends ObjectWithId> {

  protected baseUrl: string = environment.apiBaseUrl

  constructor(protected http: HttpClient, private resourceName: string) { }

  protected getResourceUrl(id?: number, action?: string): string {
    let url = `${this.baseUrl}${this.resourceName}/`
    if (id) {
      url += `${id}/`
    }
    if (action) {
      url += `${action}/`
    }
    return url
  }

  private getOrderingQueryParam(sortField: string, sortDirection: string) {
    if (sortField && sortDirection) {
      return (sortDirection == 'des' ? '-' : '') + sortField
    } else if (sortField) {
      return sortField
    } else {
      return null
    }
  }

  list(page?: number, pageSize?: number, sortField?: string, sortDirection?: string, extraParams?): Observable<Results<T>> {
    let httpParams = new HttpParams()
    if (page) {
      httpParams = httpParams.set('page', page.toString())
    }
    if (pageSize) {
      httpParams = httpParams.set('page_size', pageSize.toString())
    }
    let ordering = this.getOrderingQueryParam(sortField, sortDirection)
    if (ordering) {
      httpParams = httpParams.set('ordering', ordering)
    }
    for (let extraParamKey in extraParams) {
      if (extraParams[extraParamKey] != null) {
        httpParams = httpParams.set(extraParamKey, extraParams[extraParamKey])
      }
    }
    return this.http.get<Results<T>>(this.getResourceUrl(), {params: httpParams})
  }

  private _listAll: Observable<Results<T>>

  listAll(ordering?: string, extraParams?): Observable<Results<T>> {
    if (!this._listAll) {
      this._listAll = this.list(1, 100000, ordering, extraParams).pipe(
        publishReplay(1),
        refCount()
      )
    }
    return this._listAll
  }

  getCached(id: number): Observable<T> {
    return this.listAll().pipe(
      map(list => list.results.find(o => o.id == id))
    )
  }

  getCachedMany(ids: number[]): Observable<T[]> {
    return this.listAll().pipe(
      map(list => ids.map(id => list.results.find(o => o.id == id)))
    )
  }

  get(id: number): Observable<T> {
    return this.http.get<T>(this.getResourceUrl(id))
  }

  create(o: T): Observable<T> {
    this._listAll = null
    return this.http.post<T>(this.getResourceUrl(), o)
  }

  delete(o: T): Observable<any> {
    this._listAll = null
    return this.http.delete(this.getResourceUrl(o.id))
  }

  update(o: T): Observable<T> {
    this._listAll = null
    return this.http.put<T>(this.getResourceUrl(o.id), o)
  }
}