From efc336e60cf1331bf5f3213d296981b87b8b2a6c Mon Sep 17 00:00:00 2001 From: George Hazan Date: Sun, 4 Jun 2023 19:24:05 +0300 Subject: =?UTF-8?q?fixes=20#3537=20(Telegram:=2032-=D1=80=D0=B0=D0=B7?= =?UTF-8?q?=D1=80=D1=8F=D0=B4=D0=BD=D0=B0=D1=8F=20=D0=B2=D0=B5=D1=80=D1=81?= =?UTF-8?q?=D0=B8=D1=8F=20=D0=BF=D0=B0=D0=B4=D0=B0=D0=B5=D1=82=20=D0=B2=20?= =?UTF-8?q?64-=D1=80=D0=B0=D0=B7=D1=80=D1=8F=D0=B4=D0=BD=D0=BE=D0=B9=20Win?= =?UTF-8?q?dows)=20+=20update=20to=20the=20fresh=20TDLIB?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../tdlib/td/example/web/tdweb/src/worker.js | 1034 ++++++++++++++++++++ 1 file changed, 1034 insertions(+) create mode 100644 protocols/Telegram/tdlib/td/example/web/tdweb/src/worker.js (limited to 'protocols/Telegram/tdlib/td/example/web/tdweb/src/worker.js') diff --git a/protocols/Telegram/tdlib/td/example/web/tdweb/src/worker.js b/protocols/Telegram/tdlib/td/example/web/tdweb/src/worker.js new file mode 100644 index 0000000000..3323a05025 --- /dev/null +++ b/protocols/Telegram/tdlib/td/example/web/tdweb/src/worker.js @@ -0,0 +1,1034 @@ +import localforage from 'localforage'; +import log from './logger.js'; +import { instantiateAny } from './wasm-utils.js'; + +import td_wasm_release from './prebuilt/release/td_wasm.wasm'; +import td_asmjs_mem_release from './prebuilt/release/td_asmjs.js.mem'; + +const tdlibVersion = 6; +const localForageDrivers = [ + localforage.INDEXEDDB, + localforage.LOCALSTORAGE, + 'memoryDriver' +]; + +async function initLocalForage() { + // Implement the driver here. + const memoryDriver = { + _driver: 'memoryDriver', + _initStorage: function(options) { + const dbInfo = {}; + if (options) { + for (const i in options) { + dbInfo[i] = options[i]; + } + } + this._dbInfo = dbInfo; + this._map = new Map(); + }, + clear: async function() { + this._map.clear(); + }, + getItem: async function(key) { + const value = this._map.get(key); + console.log('getItem', this._map, key, value); + return value; + }, + iterate: async function(iteratorCallback) { + log.error('iterate is not supported'); + }, + key: async function(n) { + log.error('key n is not supported'); + }, + keys: async function() { + return this._map.keys(); + }, + length: async function() { + return this._map.size(); + }, + removeItem: async function(key) { + this._map.delete(key); + }, + setItem: async function(key, value) { + const originalValue = this._map.get(key); + console.log('setItem', this._map, key, value); + this._map.set(key, value); + return originalValue; + } + }; + + // Add the driver to localForage. + localforage.defineDriver(memoryDriver); +} + +async function loadTdlibWasm(onFS, wasmUrl) { + console.log('loadTdlibWasm'); + const td_module = await import('./prebuilt/release/td_wasm.js'); + const createTdwebModule = td_module.default; + log.info('receive td_wasm.js', td_module, createTdwebModule); + let td_wasm = td_wasm_release; + if (wasmUrl) { + td_wasm = wasmUrl; + } + let module = createTdwebModule({ + onRuntimeInitialized: () => { + log.info('runtime intialized'); + onFS(module.FS); + }, + instantiateWasm: (imports, successCallback) => { + log.info('start instantiateWasm', td_wasm, imports); + const next = instance => { + log.info('finish instantiateWasm'); + successCallback(instance); + }; + instantiateAny(tdlibVersion, td_wasm, imports).then(next); + return {}; + }, + ENVIROMENT: 'WORKER' + }); + log.info('Wait module'); + module = await module; + log.info('Loaded module', module); + //onFS(module.FS); + return module; +} + +async function loadTdlibAsmjs(onFS) { + console.log('loadTdlibAsmjs'); + const createTdwebModule = (await import('./prebuilt/release/td_asmjs.js')) + .default; + console.log('Loaded td_asm.js', createTdwebModule); + const fromFile = 'td_asmjs.js.mem'; + const toFile = td_asmjs_mem_release; + let module = createTdwebModule({ + onRuntimeInitialized: () => { + console.log('runtime intialized'); + onFS(module.FS); + }, + locateFile: name => { + if (name === fromFile) { + return toFile; + } + return name; + }, + ENVIROMENT: 'WORKER' + }); + log.info('Wait module'); + module = await module; + log.info('Loaded module', module); + //onFS(module.FS); + return module; +} + +async function loadTdlib(mode, onFS, wasmUrl) { + const wasmSupported = (() => { + try { + if ( + typeof WebAssembly === 'object' && + typeof WebAssembly.instantiate === 'function' + ) { + const module = new WebAssembly.Module( + Uint8Array.of(0x0, 0x61, 0x73, 0x6d, 0x01, 0x00, 0x00, 0x00) + ); + if (module instanceof WebAssembly.Module) + return ( + new WebAssembly.Instance(module) instanceof WebAssembly.Instance + ); + } + } catch (e) {} + return false; + })(); + if (!wasmSupported) { + if (mode === 'wasm') { + log.error('WebAssembly is not supported, trying to use it anyway'); + } else { + log.warn('WebAssembly is not supported, trying to use asm.js'); + mode = 'asmjs'; + } + } + + if (mode === 'asmjs') { + return loadTdlibAsmjs(onFS); + } + return loadTdlibWasm(onFS, wasmUrl); +} + +class OutboundFileSystem { + constructor(root, FS) { + this.root = root; + this.nextFileId = 0; + this.FS = FS; + this.files = new Set(); + FS.mkdir(root); + } + blobToPath(blob, name) { + const dir = this.root + '/' + this.nextFileId; + if (!name) { + name = 'blob'; + } + this.nextFileId++; + this.FS.mkdir(dir); + this.FS.mount( + this.FS.filesystems.WORKERFS, + { + blobs: [{ name: name, data: blob }] + }, + dir + ); + const path = dir + '/' + name; + this.files.add(path); + return path; + } + + forgetPath(path) { + if (this.files.has(path)) { + this.FS.unmount(path); + this.files.delete(path); + } + } +} + +class InboundFileSystem { + static async create(dbName, root, FS_promise) { + const start = performance.now(); + try { + const ifs = new InboundFileSystem(); + ifs.pending = []; + ifs.pendingHasTimeout = false; + ifs.persistCount = 0; + ifs.persistSize = 0; + ifs.pendingI = 0; + ifs.inPersist = false; + ifs.totalCount = 0; + + ifs.root = root; + + //ifs.store = localforage.createInstance({ + //name: dbName, + //driver: localForageDrivers + //}); + log.debug('IDB name: ' + dbName); + ifs.idb = new Promise((resolve, reject) => { + const request = indexedDB.open(dbName); + request.onsuccess = () => resolve(request.result); + request.onerror = () => reject(request.error); + request.onupgradeneeded = () => { + request.result.createObjectStore('keyvaluepairs'); + }; + }); + + ifs.load_pids(); + + const FS = await FS_promise; + await ifs.idb; + ifs.FS = FS; + ifs.FS.mkdir(root); + const create_time = (performance.now() - start) / 1000; + log.debug('InboundFileSystem::create ' + create_time); + return ifs; + } catch (e) { + log.error('Failed to init Inbound FileSystem: ', e); + } + } + + async load_pids() { + const keys_start = performance.now(); + log.debug('InboundFileSystem::create::keys start'); + //const keys = await this.store.keys(); + + let idb = await this.idb; + let read = idb + .transaction(['keyvaluepairs'], 'readonly') + .objectStore('keyvaluepairs'); + const keys = await new Promise((resolve, reject) => { + const request = read.getAllKeys(); + request.onsuccess = () => resolve(request.result); + request.onerror = () => reject(request.error); + }); + + const keys_time = (performance.now() - keys_start) / 1000; + log.debug( + 'InboundFileSystem::create::keys ' + keys_time + ' ' + keys.length + ); + this.pids = new Set(keys); + } + + has(pid) { + if (!this.pids) { + return true; + } + + return this.pids.has(pid); + } + + forget(pid) { + if (this.pids) { + this.pids.delete(pid); + } + } + + async doPersist(pid, path, arr, resolve, reject, write) { + this.persistCount++; + let size = arr.length; + this.persistSize += size; + try { + //log.debug('persist.do start', pid, path, arr.length); + //await this.store.setItem(pid, new Blob([arr])); + await new Promise((resolve, reject) => { + const request = write.put(new Blob([arr]), pid); + request.onsuccess = () => resolve(request.result); + request.onerror = () => reject(request.error); + }); + if (this.pids) { + this.pids.add(pid); + } + this.FS.unlink(path); + } catch (e) { + log.error('Failed persist ' + path + ' ', e); + } + //log.debug('persist.do finish', pid, path, arr.length); + this.persistCount--; + this.persistSize -= size; + resolve(); + + this.tryFinishPersist(); + } + + async flushPersist() { + if (this.inPersist) { + return; + } + log.debug('persist.flush'); + this.inPersist = true; + let idb = await this.idb; + this.writeBegin = performance.now(); + let write = idb + .transaction(['keyvaluepairs'], 'readwrite') + .objectStore('keyvaluepairs'); + while ( + this.pendingI < this.pending.length && + this.persistCount < 20 && + this.persistSize < 50 << 20 + ) { + var q = this.pending[this.pendingI]; + this.pending[this.pendingI] = null; + // TODO: add to transaction + this.doPersist(q.pid, q.path, q.arr, q.resolve, q.reject, write); + this.pendingI++; + this.totalCount++; + } + log.debug( + 'persist.flush transaction cnt=' + + this.persistCount + + ', size=' + + this.persistSize + ); + this.inPersist = false; + this.tryFinishPersist(); + } + + async tryFinishPersist() { + if (this.inPersist) { + return; + } + if (this.persistCount !== 0) { + return; + } + log.debug('persist.finish ' + (performance.now() - this.writeBegin) / 1000); + if (this.pendingI === this.pending.length) { + this.pending = []; + this.pendingHasTimeout = false; + this.pendingI = 0; + log.debug('persist.finish done'); + return; + } + log.debug('persist.finish continue'); + this.flushPersist(); + } + + async persist(pid, path, arr) { + if (!this.pendingHasTimeout) { + this.pendingHasTimeout = true; + log.debug('persist set timeout'); + setTimeout(() => { + this.flushPersist(); + }, 1); + } + await new Promise((resolve, reject) => { + this.pending.push({ + pid: pid, + path: path, + arr: arr, + resolve: resolve, + reject: reject + }); + }); + } + + async unlink(pid) { + log.debug('Unlink ' + pid); + try { + this.forget(pid); + //await this.store.removeItem(pid); + let idb = await this.idb; + await new Promise((resolve, reject) => { + let write = idb + .transaction(['keyvaluepairs'], 'readwrite') + .objectStore('keyvaluepairs'); + const request = write.delete(pid); + request.onsuccess = () => resolve(request.result); + request.onerror = () => reject(request.error); + }); + } catch (e) { + log.error('Failed unlink ' + pid + ' ', e); + } + } +} + +class DbFileSystem { + static async create(root, FS_promise, readOnly = false) { + const start = performance.now(); + try { + const dbfs = new DbFileSystem(); + dbfs.root = root; + const FS = await FS_promise; + dbfs.FS = FS; + dbfs.syncfs_total_time = 0; + dbfs.readOnly = readOnly; + dbfs.syncActive = 0; + FS.mkdir(root); + FS.mount(FS.filesystems.IDBFS, {}, root); + + await new Promise((resolve, reject) => { + FS.syncfs(true, err => { + resolve(); + }); + }); + + const rmrf = path => { + log.debug('rmrf ', path); + let info; + try { + info = FS.lookupPath(path); + } catch (e) { + return; + } + log.debug('rmrf ', path, info); + if (info.node.isFolder) { + for (const key in info.node.contents) { + rmrf(info.path + '/' + info.node.contents[key].name); + } + log.debug('rmdir ', path); + FS.rmdir(path); + } else { + log.debug('unlink ', path); + FS.unlink(path); + } + }; + //const dirs = ['thumbnails', 'profile_photos', 'secret', 'stickers', 'temp', 'wallpapers', 'secret_thumbnails', 'passport']; + const dirs = []; + const root_dir = FS.lookupPath(root); + for (const key in root_dir.node.contents) { + const value = root_dir.node.contents[key]; + log.debug('node ', key, value); + if (!value.isFolder) { + continue; + } + dirs.push(root_dir.path + '/' + value.name); + } + for (const i in dirs) { + const dir = dirs[i]; + rmrf(dir); + //FS.mkdir(dir); + //FS.mount(FS.filesystems.MEMFS, {}, dir); + } + dbfs.syncfsInterval = setInterval(() => { + dbfs.sync(); + }, 5000); + const create_time = (performance.now() - start) / 1000; + log.debug('DbFileSystem::create ' + create_time); + return dbfs; + } catch (e) { + log.error('Failed to init DbFileSystem: ', e); + } + } + async sync(force) { + if (this.readOnly) { + return; + } + if (this.syncActive > 0 && !force) { + log.debug('SYNC: skip'); + return; + } + this.syncActive++; + const start = performance.now(); + await new Promise((resolve, reject) => { + this.FS.syncfs(false, () => { + const syncfs_time = (performance.now() - start) / 1000; + this.syncfs_total_time += syncfs_time; + log.debug('SYNC: ' + syncfs_time); + log.debug('SYNC total: ' + this.syncfs_total_time); + resolve(); + }); + }); + this.syncActive--; + } + async close() { + clearInterval(this.syncfsInterval); + await this.sync(true); + } + async destroy() { + clearInterval(this.syncfsInterval); + if (this.readOnly) { + return; + } + this.FS.unmount(this.root); + const req = indexedDB.deleteDatabase(this.root); + await new Promise((resolve, reject) => { + req.onsuccess = function(e) { + log.info('SUCCESS'); + resolve(e.result); + }; + req.onerror = function(e) { + log.info('ONERROR'); + reject(e.error); + }; + req.onblocked = function(e) { + log.info('ONBLOCKED'); + reject('blocked'); + }; + }); + } +} + +class TdFileSystem { + static async init_fs(prefix, FS_promise) { + const FS = await FS_promise; + FS.mkdir(prefix); + return FS; + } + static async create(instanceName, FS_promise, readOnly = false) { + try { + const tdfs = new TdFileSystem(); + const prefix = '/' + instanceName; + tdfs.prefix = prefix; + FS_promise = TdFileSystem.init_fs(prefix, FS_promise); + + //MEMFS. Store to IDB and delete files as soon as possible + const inboundFileSystem = InboundFileSystem.create( + instanceName, + prefix + '/inboundfs', + FS_promise + ); + + //IDBFS. MEMFS which is flushed to IDB from time to time + const dbFileSystem = DbFileSystem.create( + prefix + '/dbfs', + FS_promise, + readOnly + ); + + const FS = await FS_promise; + tdfs.FS = FS; + + //WORKERFS. Temporary stores Blobs for outbound files + tdfs.outboundFileSystem = new OutboundFileSystem( + prefix + '/outboundfs', + tdfs.FS + ); + + tdfs.inboundFileSystem = await inboundFileSystem; + tdfs.dbFileSystem = await dbFileSystem; + return tdfs; + } catch (e) { + log.error('Failed to init TdFileSystem: ', e); + } + } + async destroy() { + await this.dbFileSystem.destroy(); + } +} + +class TdClient { + constructor(callback) { + log.info('Start worker'); + this.pendingQueries = []; + this.isPending = true; + this.callback = callback; + this.wasInit = false; + } + + async testLocalForage() { + await initLocalForage(); + const DRIVERS = [ + localforage.INDEXEDDB, + 'memoryDriver', + localforage.LOCALSTORAGE, + localforage.WEBSQL, + localForageDrivers + ]; + for (const driverName of DRIVERS) { + console.log('Test ', driverName); + try { + await localforage.setDriver(driverName); + console.log('A'); + await localforage.setItem('hello', 'world'); + console.log('B'); + const x = await localforage.getItem('hello'); + console.log('receive ', x); + await localforage.clear(); + console.log('C'); + } catch (error) { + console.log('Error', error); + } + } + } + + async init(options) { + if (this.wasInit) { + return; + } + //await this.testLocalForage(); + log.setVerbosity(options.jsLogVerbosityLevel); + this.wasInit = true; + + options = options || {}; + const mode = options.mode || 'wasm'; + + const FS_promise = new Promise(resolve => { + this.onFS = resolve; + }); + + const tdfs_promise = TdFileSystem.create( + options.instanceName, + FS_promise, + options.readOnly + ); + + this.useDatabase = true; + if ('useDatabase' in options) { + this.useDatabase = options.useDatabase; + } + + log.info('load TdModule'); + this.TdModule = await loadTdlib(mode, this.onFS, options.wasmUrl); + log.info('loaded TdModule'); + this.td_functions = { + td_create: this.TdModule.cwrap( + 'td_emscripten_create_client_id', + 'number', + [] + ), + td_send: this.TdModule.cwrap('td_emscripten_send', null, [ + 'number', + 'string' + ]), + td_execute: this.TdModule.cwrap('td_emscripten_execute', 'string', [ + 'string' + ]), + td_receive: this.TdModule.cwrap('td_emscripten_receive', 'string', []), + td_set_verbosity: verbosity => { + this.td_functions.td_execute( + JSON.stringify({ + '@type': 'setLogVerbosityLevel', + new_verbosity_level: verbosity + }) + ); + }, + td_get_timeout: this.TdModule.cwrap( + 'td_emscripten_get_timeout', + 'number', + [] + ) + }; + //this.onFS(this.TdModule.FS); + this.FS = this.TdModule.FS; + this.TdModule['websocket']['on']('error', error => { + this.scheduleReceiveSoon(); + }); + this.TdModule['websocket']['on']('open', fd => { + this.scheduleReceiveSoon(); + }); + this.TdModule['websocket']['on']('listen', fd => { + this.scheduleReceiveSoon(); + }); + this.TdModule['websocket']['on']('connection', fd => { + this.scheduleReceiveSoon(); + }); + this.TdModule['websocket']['on']('message', fd => { + this.scheduleReceiveSoon(); + }); + this.TdModule['websocket']['on']('close', fd => { + this.scheduleReceiveSoon(); + }); + + // wait till it is allowed to start + this.callback({ '@type': 'inited' }); + await new Promise(resolve => { + this.onStart = resolve; + }); + this.isStarted = true; + + log.info('may start now'); + if (this.isClosing) { + return; + } + log.info('FS start init'); + this.tdfs = await tdfs_promise; + log.info('FS inited'); + this.callback({ '@type': 'fsInited' }); + + // no async initialization after this point + if (options.logVerbosityLevel === undefined) { + options.logVerbosityLevel = 2; + } + this.td_functions.td_set_verbosity(options.logVerbosityLevel); + this.client_id = this.td_functions.td_create(); + + this.savingFiles = new Map(); + this.send({ + '@type': 'setOption', + name: 'store_all_files_in_files_directory', + value: { + '@type': 'optionValueBoolean', + value: true + } + }); + this.send({ + '@type': 'setOption', + name: 'language_pack_database_path', + value: { + '@type': 'optionValueString', + value: this.tdfs.dbFileSystem.root + '/language' + } + }); + this.send({ + '@type': 'setOption', + name: 'ignore_background_updates', + value: { + '@type': 'optionValueBoolean', + value: !this.useDatabase + } + }); + + this.flushPendingQueries(); + + this.receive(); + } + + prepareQueryRecursive(query) { + if (query['@type'] === 'inputFileBlob') { + return { + '@type': 'inputFileLocal', + path: this.tdfs.outboundFileSystem.blobToPath(query.data, query.name) + }; + } + for (const key in query) { + const field = query[key]; + if (field && typeof field === 'object') { + query[key] = this.prepareQueryRecursive(field); + } + } + return query; + } + + prepareQuery(query) { + if (query['@type'] === 'setTdlibParameters') { + query.database_directory = this.tdfs.dbFileSystem.root; + query.files_directory = this.tdfs.inboundFileSystem.root; + + const useDb = this.useDatabase; + query.use_file_database = useDb; + query.use_chat_info_database = useDb; + query.use_message_database = useDb; + query.use_secret_chats = useDb; + } + if (query['@type'] === 'getLanguagePackString') { + query.language_pack_database_path = + this.tdfs.dbFileSystem.root + '/language'; + } + return this.prepareQueryRecursive(query); + } + + onStart() { + //nop + log.info('ignore on_start'); + } + + deleteIdbKey(query) { + try { + } catch (e) { + this.callback({ + '@type': 'error', + '@extra': query['@extra'], + code: 400, + message: e + }); + return; + } + this.callback({ + '@type': 'ok', + '@extra': query['@extra'] + }); + } + + readFilePart(query) { + let res; + try { + //const file_size = this.FS.stat(query.path).size; + const stream = this.FS.open(query.path, 'r'); + const buf = new Uint8Array(query.count); + this.FS.read(stream, buf, 0, query.count, query.offset); + this.FS.close(stream); + res = buf; + } catch (e) { + this.callback({ + '@type': 'error', + '@extra': query['@extra'], + code: 400, + message: e.toString() + }); + return; + } + this.callback( + { + '@type': 'filePart', + '@extra': query['@extra'], + data: res + }, + [res.buffer] + ); + } + + send(query) { + if (this.isClosing) { + return; + } + if (this.wasFatalError) { + if (query['@type'] === 'destroy') { + this.destroy({ '@type': 'ok', '@extra': query['@extra'] }); + } + return; + } + if (query['@type'] === 'init') { + this.init(query.options); + return; + } + if (query['@type'] === 'start') { + log.info('on_start'); + this.onStart(); + return; + } + if (query['@type'] === 'setJsLogVerbosityLevel') { + log.setVerbosity(query.new_verbosity_level); + return; + } + if (this.isPending) { + this.pendingQueries.push(query); + return; + } + if ( + query['@type'] === 'setLogVerbosityLevel' || + query['@type'] === 'getLogVerbosityLevel' || + query['@type'] === 'setLogTagVerbosityLevel' || + query['@type'] === 'getLogTagVerbosityLevel' || + query['@type'] === 'getLogTags' + ) { + this.execute(query); + return; + } + if (query['@type'] === 'readFilePart') { + this.readFilePart(query); + return; + } + if (query['@type'] === 'deleteIdbKey') { + this.deleteIdbKey(query); + return; + } + query = this.prepareQuery(query); + this.td_functions.td_send(this.client_id, JSON.stringify(query)); + this.scheduleReceiveSoon(); + } + + execute(query) { + try { + const res = this.td_functions.td_execute(JSON.stringify(query)); + const response = JSON.parse(res); + this.callback(response); + } catch (error) { + this.onFatalError(error); + } + } + receive() { + this.cancelReceive(); + if (this.wasFatalError) { + return; + } + try { + while (true) { + const msg = this.td_functions.td_receive(); + if (!msg) { + break; + } + const response = this.prepareResponse(JSON.parse(msg)); + if ( + response['@type'] === 'updateAuthorizationState' && + response.authorization_state['@type'] === 'authorizationStateClosed' + ) { + this.close(response); + break; + } + this.callback(response); + } + + this.scheduleReceive(); + } catch (error) { + this.onFatalError(error); + } + } + + cancelReceive() { + if (this.receiveTimeout) { + clearTimeout(this.receiveTimeout); + delete this.receiveTimeout; + } + delete this.receiveSoon; + } + scheduleReceiveSoon() { + if (this.receiveSoon) { + return; + } + this.cancelReceive(); + this.receiveSoon = true; + this.scheduleReceiveIn(0.001); + } + scheduleReceive() { + if (this.receiveSoon) { + return; + } + this.cancelReceive(); + const timeout = this.td_functions.td_get_timeout(); + this.scheduleReceiveIn(timeout); + } + scheduleReceiveIn(timeout) { + //return; + log.debug('Scheduler receive in ' + timeout + 's'); + this.receiveTimeout = setTimeout(() => this.receive(), timeout * 1000); + } + + onFatalError(error) { + this.wasFatalError = true; + this.asyncOnFatalError(error); + } + + async close(last_update) { + // close db and cancell all timers + this.isClosing = true; + if (this.isStarted) { + log.debug('close worker: start'); + await this.tdfs.dbFileSystem.close(); + this.cancelReceive(); + log.debug('close worker: finish'); + } + this.callback(last_update); + } + + async destroy(result) { + try { + log.info('destroy tdfs ...'); + await this.tdfs.destroy(); + log.info('destroy tdfs ok'); + } catch (e) { + log.error('Failed destroy', e); + } + this.callback(result); + this.callback({ + '@type': 'updateAuthorizationState', + authorization_state: { + '@type': 'authorizationStateClosed' + } + }); + } + + async asyncOnFatalError(error) { + await this.tdfs.dbFileSystem.sync(); + this.callback({ '@type': 'updateFatalError', error: error }); + } + + saveFile(pid, file) { + const isSaving = this.savingFiles.has(pid); + this.savingFiles.set(pid, file); + if (isSaving) { + return file; + } + try { + const arr = this.FS.readFile(file.local.path); + if (arr) { + file = Object.assign({}, file); + file.arr = arr; + this.doSaveFile(pid, file, arr); + } + } catch (e) { + log.error('Failed to readFile: ', e); + } + return file; + } + + async doSaveFile(pid, file, arr) { + await this.tdfs.inboundFileSystem.persist(pid, file.local.path, arr); + file = this.savingFiles.get(pid); + file.idb_key = pid; + this.callback({ '@type': 'updateFile', file: file }); + + this.savingFiles.delete(pid); + } + + prepareFile(file) { + const pid = file.remote.unique_id ? file.remote.unique_id : file.remote.id; + if (!pid) { + return file; + } + + if (file.local.is_downloading_active) { + this.tdfs.inboundFileSystem.forget(pid); + } else if (this.tdfs.inboundFileSystem.has(pid)) { + file.idb_key = pid; + return file; + } + + if (file.local.is_downloading_completed) { + file = this.saveFile(pid, file); + } + return file; + } + + prepareResponse(response) { + if (response['@type'] === 'file') { + return this.prepareFile(response); + } + for (const key in response) { + const field = response[key]; + if (field && typeof field === 'object') { + response[key] = this.prepareResponse(field); + } + } + return response; + } + + flushPendingQueries() { + this.isPending = false; + for (const query of this.pendingQueries) { + this.send(query); + } + } +} + +const client = new TdClient((e, t = []) => postMessage(e, t)); + +onmessage = function(e) { + try { + client.send(e.data); + } catch (error) { + client.onFatalError(error); + } +}; -- cgit v1.2.3