mckoss / dawg Goto Github PK

const trie = new Trie(['yes', 'yes_no']);
trie.isWord('yes'); // => true (which is correct)
trie.isWord('no'); // => false (which is correct)
trie.isWord('yes_no'); // => false (which is incorrect)

import { createReadStream, createWriteStream } from 'fs';
import { resolve } from 'path';
import { createInterface, Interface } from 'readline';

var Trie = require('dawg-lookup').Trie;
var PTrie = require('dawg-lookup/lib/ptrie').PTrie;

var trie = new Trie();
var scrabbleDictionary = createInterface({
   input: createReadStream(resolve(__dirname, '../lib', 'Collins Scrabble Words (2019).txt'), 'utf8')
});
var output = createWriteStream(resolve(__dirname, '../lib', 'dictionary.txt'));

scrabbleDictionary.on('line', (line: string) => {
        trie.insert(line);
    }).on('close', () => {
        console.log(trie.isWord('FUN'));

        let packed = trie.pack();
        let ptrie = new PTrie(packed);

        console.log(ptrie.isWord('FUN'));
        console.log(ptrie.completions('F'));

        output.write(packed);
        output.close();
    });