paperless-ngx/src-ui/src/app/services/open-documents.service.ts

import { Injectable } from '@angular/core'
import { Document } from '../data/document'
import { OPEN_DOCUMENT_SERVICE } from '../data/storage-keys'
import { DocumentService } from './rest/document.service'
import { NgbModal } from '@ng-bootstrap/ng-bootstrap'
import { ConfirmDialogComponent } from 'src/app/components/common/confirm-dialog/confirm-dialog.component'
import { Observable, Subject, of } from 'rxjs'
import { first } from 'rxjs/operators'

@Injectable({
  providedIn: 'root',
})
export class OpenDocumentsService {
  private MAX_OPEN_DOCUMENTS = 5

  constructor(
    private documentService: DocumentService,
    private modalService: NgbModal
  ) {
    if (sessionStorage.getItem(OPEN_DOCUMENT_SERVICE.DOCUMENTS)) {
      try {
        this.openDocuments = JSON.parse(
          sessionStorage.getItem(OPEN_DOCUMENT_SERVICE.DOCUMENTS)
        )
      } catch (e) {
        sessionStorage.removeItem(OPEN_DOCUMENT_SERVICE.DOCUMENTS)
        this.openDocuments = []
      }
    }
  }

  private openDocuments: Document[] = []
  private dirtyDocuments: Set<number> = new Set<number>()

  refreshDocument(id: number) {
    let index = this.openDocuments.findIndex((doc) => doc.id == id)
    if (index > -1) {
      this.documentService.get(id).subscribe({
        next: (doc) => {
          this.openDocuments[index] = doc
          this.save()
        },
        error: () => {
          this.openDocuments.splice(index, 1)
          this.save()
        },
      })
    }
  }

  getOpenDocuments(): Document[] {
    return this.openDocuments
  }

  getOpenDocument(id: number): Document {
    return this.openDocuments.find((d) => d.id == id)
  }

  openDocument(doc: Document): Observable<boolean> {
    if (this.openDocuments.find((d) => d.id == doc.id) == null) {
      if (this.openDocuments.length == this.MAX_OPEN_DOCUMENTS) {
        // at max, ensure changes aren't lost
        const docToRemove = this.openDocuments[this.MAX_OPEN_DOCUMENTS - 1]
        const closeObservable = this.closeDocument(docToRemove)
        closeObservable.pipe(first()).subscribe((closed) => {
          if (closed) this.finishOpenDocument(doc)
        })
        return closeObservable
      } else {
        // not at max
        this.finishOpenDocument(doc)
      }
    }
    return of(true)
  }

  private finishOpenDocument(doc: Document) {
    this.openDocuments.unshift(doc)
    this.dirtyDocuments.delete(doc.id)
    this.save()
  }

  setDirty(doc: Document, dirty: boolean) {
    if (!this.openDocuments.find((d) => d.id == doc.id)) return
    if (dirty) this.dirtyDocuments.add(doc.id)
    else this.dirtyDocuments.delete(doc.id)
  }

  hasDirty(): boolean {
    return this.dirtyDocuments.size > 0
  }

  isDirty(doc: Document): boolean {
    return this.dirtyDocuments.has(doc.id)
  }

  closeDocument(doc: Document): Observable<boolean> {
    let index = this.openDocuments.findIndex((d) => d.id == doc.id)
    if (index == -1) return of(true)
    if (!this.dirtyDocuments.has(doc.id)) {
      this.openDocuments.splice(index, 1)
      this.save()
      return of(true)
    } else {
      let modal = this.modalService.open(ConfirmDialogComponent, {
        backdrop: 'static',
      })
      modal.componentInstance.title = $localize`Unsaved Changes`
      modal.componentInstance.messageBold =
        $localize`You have unsaved changes to the document` +
        ' "' +
        doc.title +
        '".'
      modal.componentInstance.message = $localize`Are you sure you want to close this document?`
      modal.componentInstance.btnClass = 'btn-warning'
      modal.componentInstance.btnCaption = $localize`Close document`
      modal.componentInstance.confirmClicked.pipe(first()).subscribe(() => {
        modal.componentInstance.buttonsEnabled = false
        modal.close()
        this.openDocuments.splice(index, 1)
        this.dirtyDocuments.delete(doc.id)
        this.save()
      })
      const subject = new Subject<boolean>()
      modal.componentInstance.confirmSubject = subject
      return subject.asObservable()
    }
  }

  closeAll(): Observable<boolean> {
    if (this.dirtyDocuments.size) {
      let modal = this.modalService.open(ConfirmDialogComponent, {
        backdrop: 'static',
      })
      modal.componentInstance.title = $localize`Unsaved Changes`
      modal.componentInstance.messageBold = $localize`You have unsaved changes.`
      modal.componentInstance.message = $localize`Are you sure you want to close all documents?`
      modal.componentInstance.btnClass = 'btn-warning'
      modal.componentInstance.btnCaption = $localize`Close documents`
      modal.componentInstance.confirmClicked.pipe(first()).subscribe(() => {
        modal.componentInstance.buttonsEnabled = false
        modal.close()
        this.openDocuments.splice(0, this.openDocuments.length)
        this.dirtyDocuments.clear()
        this.save()
      })
      const subject = new Subject<boolean>()
      modal.componentInstance.confirmSubject = subject
      return subject.asObservable()
    } else {
      this.openDocuments.splice(0, this.openDocuments.length)
      this.dirtyDocuments.clear()
      this.save()
      return of(true)
    }
  }

  save() {
    try {
      sessionStorage.setItem(
        OPEN_DOCUMENT_SERVICE.DOCUMENTS,
        JSON.stringify(this.openDocuments)
      )
    } catch (e) {
      console.error('Error saving open documents to session storage', e)
    }
  }
}