/** * Simple client that streams object info from a Speckle Server. * TODO: This should be split from the viewer into its own package. */ export default class ObjectLoader { constructor( { serverUrl, streamId, token, objectId } ) { this.INTERVAL_MS = 20 this.TIMEOUT_MS = 180000 // three mins this.serverUrl = serverUrl || window.location.origin this.streamId = streamId this.objectId = objectId this.token = token || localStorage.getItem( 'AuthToken' ) this.headers = { 'Accept': 'text/plain' } if( token ) { this.headers['Authorization'] = `Bearer ${this.token}` } this.requestUrl = `${this.serverUrl}/objects/${this.streamId}/${this.objectId}` this.promises = [] this.intervals = {} this.buffer = [] } dispose() { this.buffer = [] this.intervals.forEach( i => clearInterval( i.interval ) ) } async getObject( id ){ if ( this.buffer[id] ) return this.buffer[id] let promise = new Promise( ( resolve, reject ) => { this.promises.push( { id, resolve, reject } ) // Only create a new interval checker if none is already present! if ( this.intervals[id] ) { this.intervals[id].elapsed = 0 // reset elapsed } else { let intervalId = setInterval( this.tryResolvePromise.bind( this ), this.INTERVAL_MS, id ) this.intervals[id] = { interval: intervalId, elapsed: 0 } } } ) return promise } tryResolvePromise( id ) { this.intervals[id].elapsed += this.INTERVAL_MS if ( this.buffer[id] ) { for ( let p of this.promises.filter( p => p.id === id ) ) { p.resolve( this.buffer[id] ) } clearInterval( this.intervals[id].interval ) delete this.intervals[id] // this.promises = this.promises.filter( p => p.id !== p.id ) // clearing out promises too early seems to nuke loading return } if ( this.intervals[id].elapsed > this.TIMEOUT_MS ) { console.warn( `Timeout resolving ${id}. HIC SVNT DRACONES.` ) clearInterval( this.intervals[id].interval ) this.promises.filter( p => p.id === id ).forEach( p => p.reject() ) this.promises = this.promises.filter( p => p.id !== p.id ) // clear out } } async * getObjectIterator( ) { for await ( let line of this.getRawObjectIterator() ) { let { id, obj } = this.processLine( line ) this.buffer[ id ] = obj yield obj } } processLine( chunk ) { var pieces = chunk.split( '\t' ) return { id: pieces[0], obj: JSON.parse( pieces[1] ) } } async * getRawObjectIterator() { const decoder = new TextDecoder() const response = await fetch( this.requestUrl, { headers: this.headers } ) const reader = response.body.getReader() let { value: chunk, done: readerDone } = await reader.read() chunk = chunk ? decoder.decode( chunk ) : '' let re = /\r\n|\n|\r/gm let startIndex = 0 while ( true ) { let result = re.exec( chunk ) if ( !result ) { if ( readerDone ) break let remainder = chunk.substr( startIndex ); ( { value: chunk, done: readerDone } = await reader.read() ) chunk = remainder + ( chunk ? decoder.decode( chunk ) : '' ) startIndex = re.lastIndex = 0 continue } yield chunk.substring( startIndex, result.index ) startIndex = re.lastIndex } if ( startIndex < chunk.length ) { yield chunk.substr( startIndex ) } } }