8997167ad7
* Update README.md Fixed the browser example which did not specify an import for the Spellchecker. Also fixed an infinite loop due to checkSpelling being called from within the resultHandler. * Fix and test for #20 - Runtime error when changing checkSpelling:maxEditDistance * Added a test for #19 - Exception when loading dictionary with non-ASCII characters
81 lines
3.3 KiB
JavaScript
81 lines
3.3 KiB
JavaScript
"use strict";
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
const fs_1 = require("fs");
|
|
const SpellCheckerBase_1 = require("../SpellCheckerBase");
|
|
const SuggestedItem_1 = require("../SuggestedItem");
|
|
/**
|
|
* This class provides the wrapper for the spellcheck-wasm.wasm functionality.
|
|
*/
|
|
class SpellcheckerWasm extends SpellCheckerBase_1.SpellcheckerBase {
|
|
constructor(resultHandler) {
|
|
super(resultHandler);
|
|
SuggestedItem_1.SuggestedItem.decodeString = (data) => Buffer.from(data).toString('utf8');
|
|
}
|
|
/**
|
|
* Prepares the spellcheck wasm for use.
|
|
*
|
|
* @param wasmLocation
|
|
* @param dictionaryLocation
|
|
* @param bigramLocation
|
|
* @param options
|
|
*/
|
|
async prepareSpellchecker(wasmLocation, dictionaryLocation, bigramLocation = null, options = SpellCheckerBase_1.defaultOptions) {
|
|
if (typeof wasmLocation !== 'string') {
|
|
throw new TypeError('The wasmLocation argument must be a string');
|
|
}
|
|
if (typeof dictionaryLocation !== 'string') {
|
|
throw new TypeError('The dictionaryLocation argument must be a string');
|
|
}
|
|
const wasmBytes = await fs_1.promises.readFile(wasmLocation);
|
|
const result = await WebAssembly.instantiate(wasmBytes, {
|
|
env: {
|
|
memoryBase: 0,
|
|
tableBase: 0,
|
|
memory: new WebAssembly.Memory({ initial: 1 }),
|
|
table: new WebAssembly.Table({ initial: 1, element: 'anyfunc' }),
|
|
result_handler: this.resultTrap
|
|
}
|
|
});
|
|
if (!result) {
|
|
throw new Error(`Failed to instantiate the parser.`);
|
|
}
|
|
const { symspell, write_to_dictionary, lookup, lookup_compound, memory } = result.instance.exports;
|
|
this.wasmSymSpell = { symspell, write_to_dictionary, lookup, lookup_compound, memory };
|
|
symspell(options.dictionaryEditDistance, options.countThreshold);
|
|
const newline = new Uint8Array([10]);
|
|
await new Promise(resolve => {
|
|
const dictionaryReadStream = fs_1.createReadStream(dictionaryLocation);
|
|
dictionaryReadStream.on('data', (chunk) => {
|
|
this.writeToBuffer(chunk, memory);
|
|
write_to_dictionary(0, chunk.length, false);
|
|
});
|
|
dictionaryReadStream.on('close', () => {
|
|
this.writeToBuffer(newline, memory); // Closes the stream
|
|
write_to_dictionary(0, 1, false);
|
|
resolve();
|
|
});
|
|
});
|
|
await new Promise(resolve => {
|
|
if (!bigramLocation) {
|
|
return resolve();
|
|
}
|
|
if (typeof bigramLocation !== 'string') {
|
|
throw new TypeError('The bigramLocation argument must be a string');
|
|
}
|
|
const bigramReadStream = fs_1.createReadStream(bigramLocation);
|
|
bigramReadStream.on('data', (chunk) => {
|
|
this.writeToBuffer(chunk, memory);
|
|
write_to_dictionary(0, chunk.length, true);
|
|
});
|
|
bigramReadStream.on('close', () => {
|
|
this.writeToBuffer(newline, memory); // Closes the stream
|
|
write_to_dictionary(0, 1, true);
|
|
resolve();
|
|
});
|
|
});
|
|
}
|
|
encodeString(str) {
|
|
return Buffer.from(str);
|
|
}
|
|
}
|
|
exports.SpellcheckerWasm = SpellcheckerWasm;
|