standardnotes-app-web/packages/web/src/javascripts/Controllers/Moments/MomentsService.ts

import { addToast, dismissToast, ToastType } from '@standardnotes/toast'
import { ApplicationEvent, InternalEventBus, StorageKey } from '@standardnotes/services'
import { isDev } from '@/Utils'
import { FileItem, PrefKey, sleep, SNTag } from '@standardnotes/snjs'
import { FilesController } from '../FilesController'
import { action, makeObservable, observable } from 'mobx'
import { AbstractViewController } from '@/Controllers/Abstract/AbstractViewController'
import { WebApplication } from '@/Application/Application'
import { dateToStringStyle1 } from '@/Utils/DateUtils'
import { PhotoRecorder } from './PhotoRecorder'

const EVERY_HOUR = 1000 * 60 * 60
const EVERY_TEN_SECONDS = 1000 * 10
const DEBUG_MODE = isDev && false

const DELAY_AFTER_STARTING_CAMERA_TO_ALLOW_MOBILE_AUTOFOCUS = 2000

export class MomentsService extends AbstractViewController {
  isEnabled = false
  private intervalReference: ReturnType<typeof setInterval> | undefined

  constructor(application: WebApplication, private filesController: FilesController, eventBus: InternalEventBus) {
    super(application, eventBus)

    this.disposers.push(
      application.addEventObserver(async () => {
        this.isEnabled = (this.application.getValue(StorageKey.MomentsEnabled) as boolean) ?? false
        if (this.isEnabled) {
          void this.beginTakingPhotos()
        }
      }, ApplicationEvent.LocalDataLoaded),

      application.addEventObserver(async () => {
        this.pauseMoments()
      }, ApplicationEvent.BiometricsSoftLockEngaged),

      application.addEventObserver(async () => {
        this.resumeMoments()
      }, ApplicationEvent.BiometricsSoftLockDisengaged),
    )

    makeObservable(this, {
      isEnabled: observable,

      enableMoments: action,
      disableMoments: action,
    })
  }

  override deinit() {
    super.deinit()
    ;(this.application as unknown) = undefined
    ;(this.filesController as unknown) = undefined
  }

  public enableMoments = (): void => {
    this.application.setValue(StorageKey.MomentsEnabled, true)

    this.isEnabled = true

    void this.beginTakingPhotos()
  }

  public disableMoments = (): void => {
    this.application.setValue(StorageKey.MomentsEnabled, false)

    this.isEnabled = false

    clearInterval(this.intervalReference)
  }

  private pauseMoments(): void {
    clearInterval(this.intervalReference)
  }

  private resumeMoments(): void {
    if (!this.isEnabled) {
      return
    }

    void this.beginTakingPhotos()
  }

  private beginTakingPhotos() {
    if (this.intervalReference) {
      clearInterval(this.intervalReference)
    }

    void this.takePhoto()

    this.intervalReference = setInterval(
      () => {
        void this.takePhoto()
      },
      DEBUG_MODE ? EVERY_TEN_SECONDS : EVERY_HOUR,
    )
  }

  private getDefaultTag(): SNTag | undefined {
    const defaultTagId = this.application.getPreference(PrefKey.MomentsDefaultTagUuid)

    if (defaultTagId) {
      return this.application.items.findItem(defaultTagId)
    }
  }

  public takePhoto = async (): Promise<FileItem | undefined> => {
    const isAppLocked = await this.application.isLocked()

    if (isAppLocked) {
      return
    }

    const isAppInForeground = document.visibilityState === 'visible'

    let toastId: string | undefined

    if (isAppInForeground) {
      toastId = addToast({
        type: ToastType.Regular,
        message: 'Capturing Moment...',
        pauseOnWindowBlur: false,
      })
    }

    if (this.application.desktopDevice) {
      const granted = await this.application.desktopDevice.askForMediaAccess('camera')
      if (!granted) {
        if (toastId) {
          dismissToast(toastId)
        }
        addToast({
          type: ToastType.Error,
          message: 'Please enable Camera permissions for Standard Notes to enable Moments.',
          duration: 3000,
        })

        return
      }
    }

    const filename = `Moment ${dateToStringStyle1(new Date())}.png`
    const camera = new PhotoRecorder()
    await camera.initialize()

    if (this.application.isMobileDevice) {
      await sleep(DELAY_AFTER_STARTING_CAMERA_TO_ALLOW_MOBILE_AUTOFOCUS)
    }

    let file = await camera.takePhoto(filename)
    if (!file) {
      await sleep(1000)
      file = await camera.takePhoto(filename)
      if (!file) {
        return undefined
      }
    }

    if (toastId) {
      dismissToast(toastId)
    }

    const uploadedFile = await this.filesController.uploadNewFile(file)

    if (uploadedFile) {
      if (isAppInForeground) {
        void this.application.linkingController.linkItemToSelectedItem(uploadedFile)
      }

      const defaultTag = this.getDefaultTag()
      if (defaultTag) {
        void this.application.linkingController.linkItems(uploadedFile, defaultTag)
      }
    }

    camera.finish()

    return uploadedFile
  }
}