@newdash/newdash/string/distance.js

javascript/typescript utility library

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.closest = exports.distance = void 0;
const uniq_1 = require("../uniq");
/**
 * calculate distance of two string
 *
 * @since 5.16.0
 * @category String
 *
 * @param s1
 * @param s2
 */
function distance(s1 = "", s2 = "") {
    const s1Length = s1.length || 0;
    const s2Length = s2.length || 0;
    if (s1Length === s2Length) {
        if (s1 === s2) {
            return 0;
        }
    }
    const tmp = [];
    if (s1Length && s2Length) {
        let i1 = 0, i2 = 0, a, b, c;
        const tmp1 = tmp;
        while (i1 < s1Length) {
            tmp1[i1] = ++i1;
        }
        while (i2 < s2Length) {
            // @ts-ignore
            const c2 = s2.charCodeAt(i2);
            a = i2;
            ++i2;
            b = i2;
            for (i1 = 0; i1 < s1Length; ++i1) {
                // @ts-ignore
                c = a + (s1.charCodeAt(i1) !== c2 ? 1 : 0);
                a = tmp1[i1];
                b = b < a ? (b < c ? b + 1 : c) : (a < c ? a + 1 : c);
                tmp1[i1] = b;
            }
        }
        return b;
    }
    return s1Length + s2Length;
}
exports.distance = distance;
/**
 * get closest string from dict
 *
 * @since 5.16.0
 * @category string
 * @param input
 * @param dict
 *
 *
 * ```ts
  const dict = ['hello', 'haha', 'mama', 'moment', 'world', 'latest'];

  expect(closest('h', dict)).toBe('haha');
  expect(closest('he', dict)).toBe('hello');
  expect(closest('m', dict)).toBe('mama');
  expect(closest('mo', dict)).toBe('mama');
  expect(closest('mome', dict)).toBe('mama');
  expect(closest('latast', dict)).toBe('latest');
  ```
 */
function closest(input, dict) {
    if (dict === undefined || dict.length === 0) {
        return input;
    }
    dict = (0, uniq_1.uniq)(dict);
    if (input === undefined || input?.length === 0) {
        return dict[0];
    }
    if (dict.includes(input)) {
        return input;
    }
    const sortedResult = dict.map((s) => ({ value: s, distance: distance(input, s) })).sort((o1, o2) => o1.distance - o2.distance);
    return sortedResult[0].value;
}
exports.closest = closest;