import { WebApplication } from '@/UIModels/Application'
import { SNComponent, ClientDisplayableError, FeatureDescription } from '@standardnotes/snjs'
import { makeAutoObservable, observable } from 'mobx'

export class ExtensionsLatestVersions {
  static async load(application: WebApplication): Promise<ExtensionsLatestVersions | undefined> {
    const response = await application.getAvailableSubscriptions()

    if (response instanceof ClientDisplayableError) {
      return undefined
    }

    const versionMap: Map<string, string> = new Map()
    collectFeatures(response.CORE_PLAN?.features as FeatureDescription[], versionMap)
    collectFeatures(response.PLUS_PLAN?.features as FeatureDescription[], versionMap)
    collectFeatures(response.PRO_PLAN?.features as FeatureDescription[], versionMap)

    return new ExtensionsLatestVersions(versionMap)
  }

  constructor(private readonly latestVersionsMap: Map<string, string>) {
    makeAutoObservable<ExtensionsLatestVersions, 'latestVersionsMap'>(this, {
      latestVersionsMap: observable.ref,
    })
  }

  getVersion(extension: SNComponent): string | undefined {
    return this.latestVersionsMap.get(extension.package_info.identifier)
  }
}

function collectFeatures(
  features: FeatureDescription[] | undefined,
  versionMap: Map<string, string>,
) {
  if (features == undefined) {
    return
  }
  for (const feature of features) {
    versionMap.set(feature.identifier, feature.version as string)
  }
}