terriajs/lib/ModelMixins/ExportableMixin.ts

Geospatial data visualization platform.

import Model from "../Models/Definition/Model";
import Constructor from "../Core/Constructor";
import ExportableTraits from "../Traits/TraitsClasses/ExportableTraits";
import { computed } from "mobx";

export type ExportData = string | { name: string; file: Blob };

function ExportableMixin<T extends Constructor<Model<ExportableTraits>>>(
  Base: T
) {
  abstract class ExportableMixin extends Base {
    protected abstract get _canExportData(): boolean;

    /**
     * Indicates if model is able to export data (will turn on/off UI elements)
     */
    @computed get canExportData() {
      return !this.disableExport && this._canExportData;
    }

    protected abstract _exportData(): Promise<ExportData | undefined>;

    /**
     * @returns an async function which returns a URL (to download) or a Blob with filename
     */
    exportData() {
      if (this.canExportData) {
        return this._exportData();
      }
    }
  }

  return ExportableMixin;
}

namespace ExportableMixin {
  export interface Instance
    extends InstanceType<ReturnType<typeof ExportableMixin>> {}
  export function isMixedInto(model: any): model is Instance {
    return model && "exportData" in model && "canExportData" in model;
  }
}

export default ExportableMixin;