threepipe/src/plugins/export/FileTransferPlugin.ts

A 3D viewer framework built on top of three.js in TypeScript with a focus on quality rendering, modularity and extensibility.

import {AViewerPluginSync, ThreeViewer} from '../../viewer'
import {downloadBlob} from 'ts-browser-helpers'

/**
 * File Transfer Plugin
 *
 * Provides a way to extend the viewer.export functionality with custom actions. Used in `AWSClientPlugin` to upload files directly to S3.
 *
 * @category Plugins
 */
export class FileTransferPlugin extends AViewerPluginSync<'transferFile'> {
    enabled = true

    static readonly PluginType = 'FileTransferPlugin'

    toJSON: any = undefined

    async exportFile(file: File|Blob, name?: string) {
        name = name || (file as File).name || 'file_export'
        this.dispatchEvent({type: 'transferFile', path: name, state: 'exporting', progress: 0})
        await this.actions.exportFile(file, name, ({state, progress})=>{
            this.dispatchEvent({type: 'transferFile', path: name, state: state ?? 'exporting', progress})
        })
        this.dispatchEvent({type: 'transferFile', path: name, state: 'done'})
    }

    readonly defaultActions = {
        exportFile: async(blob: Blob, name: string, _onProgress?: (d: {state?: string, progress?: number})=>void)=>{
            downloadBlob(blob, name)
        },
    }


    constructor() {
        super()
        this._updateProcessState = this._updateProcessState.bind(this)
    }

    onAdded(viewer: ThreeViewer) {
        super.onAdded(viewer)
        this.addEventListener('transferFile', this._updateProcessState as any)
    }
    onRemove(viewer: ThreeViewer) {
        this.removeEventListener('transferFile', this._updateProcessState as any)
        super.onRemove(viewer)
    }

    protected _updateProcessState(data: {path: string, state: string, progress?: number}) {
        if (!this._viewer) return
        this._viewer.assetManager.setProcessState(data.path, data.state !== 'done' ? {
            state: data.state,
            progress: data.progress ? data.progress * 100 : undefined,
        } : undefined)
    }

    actions = {...this.defaultActions}
}