cmpstr 1.0.0 → 1.0.1

package/README.md

@@ -24,16 +24,38 @@ Sample of how to use the package in your code:
 let str1 = 'kitten';
 let str2 = 'sitting';
 
+/**
+ * levenshteinDistance
+ * expected: 3
+ */
 let distance = cmpstr.levenshteinDistance( str1, str2 );
-// expected 3
 
+/**
+ * diceCoefficient
+ * expected: 0.3636363636363636
+ */
 let dice = cmpstr.diceCoefficient( str1, str2 );
-// expected 0.3636363636363636
 
+/**
+ * diceClosest
+ * expected: bestest
+ */
 let closest = cmpstr.diceClosest( 'best', [
   'better', 'bestest', 'well', 'good'
 ] );
-// expected bestest
+
+/**
+ * levenshteinMatch
+ * expected: [
+ *   { target: 'bestest', match: 0.5714285714285714 },
+ *   { target: 'better', match: 0.5 },
+ *   { target: 'well', match: 0.25 },
+ *   { target: 'good', match: 0 }
+ * ]
+ */
+let matches = cmpstr.levenshteinMatch( 'best', [
+  'better', 'bestest', 'well', 'good'
+] );
 ```
 
 ## API
@@ -59,6 +81,10 @@ Returns the match percentage of two strings ``a`` and ``b``. The output value is
 
 Returns the best match of the string ``str`` against the array ``arr`` of passed strings. The function returns the most closely matched string found in the array.
 
+#### ``levenshteinMatch( str, arr )``
+
+Calculates the similarity of all strings contained in the array ``arr`` according to Levenshtein compared to ``str`` and returns an array of all samples sorted by matching in descending order.
+
 ### Sørensen-Dice coefficient
 
 #### ``diceCoefficient( a, b )``
@@ -68,3 +94,7 @@ This function evaluates the similarity of two given strings ``a`` and ``b`` as p
 #### ``diceClosest( str, arr )``
 
 As another way to find the best match between the string ``str`` and a given array ``arr`` of samples, this function uses the Sørensen-Dice coefficient. It returns the most matching string as well.
+
+#### ``diceMatch( str, arr )``
+
+Calculates the similarity of all strings contained in the array ``arr`` according to Sørensen-Dice coefficient compared to ``str`` and returns an array of all samples sorted by matching in descending order.

package/index.js

@@ -3,7 +3,7 @@
  * lightweight npm package to calculate string similarity
  * 
  * @author komed3 (Paul Köhler)
- * @version 1.0.0
+ * @version 1.0.1
  * @license MIT
  */
 
@@ -30,7 +30,7 @@ const normalize = ( str ) => {
  * @param {String} str string
  * @returns bigrams
  */
-const bbigrams = ( str ) => {
+const str2bigrams = ( str ) => {
 
     let bigrams = new Set();
 
@@ -46,11 +46,35 @@ const bbigrams = ( str ) => {
 
 };
 
+/**
+ * 
+ * @param {String} algo algorithm to use
+ * @param {String} a string 1
+ * @param {String} b string 2
+ * @returns similarity
+ */
+const cpmByAlgo = ( algo, a, b ) => {
+
+    switch( algo ) {
+
+        case 'levenshtein':
+            return levenshtein( a, b );
+
+        case 'diceCoefficient':
+            return diceCoefficient( a, b );
+
+        default:
+            return 0;
+
+    }
+
+};
+
 /**
  * search for closest string
  * @param {String} algo algorithm to use
  * @param {String} test test string
- * @param  {Array} arr targets to test
+ * @param {Array} arr targets to test
  * @returns closest target
  */
 const findClosest = ( algo, test, arr ) => {
@@ -63,21 +87,7 @@ const findClosest = ( algo, test, arr ) => {
 
     arr.forEach( ( str, i ) => {
 
-        switch( algo ) {
-
-            case 'levenshtein':
-                pct = levenshtein( test, str );
-                break;
-
-            case 'diceCoefficient':
-                pct = diceCoefficient( test, str );
-                break;
-
-            default:
-                pct = 0;
-                break;
-
-        }
+        pct = cpmByAlgo( algo, test, str );
 
         if( pct > best ) {
 
@@ -96,6 +106,43 @@ const findClosest = ( algo, test, arr ) => {
 
 };
 
+/**
+ * sort best matches to test string
+ * @param {String} algo algorithm to use
+ * @param {String} test test string
+ * @param {Array} arr targets to test
+ * @returns sorted matches
+ */
+const bestMatch = ( algo, test, arr ) => {
+
+    let matches = [],
+        pct;
+
+    /* calculate similarity for each arr items */
+
+    arr.forEach( ( str ) => {
+
+        pct = cpmByAlgo( algo, test, str );
+
+        matches.push( {
+            target: str,
+            match: pct
+        } );
+
+    } );
+
+    /* sort by highest similarity */
+
+    let sorted = matches.sort( ( a, b ) => {
+        return b.match - a.match;
+    } );
+
+    /* return sorted matches */
+
+    return sorted;
+
+};
+
 /**
  * similarity calculations
  * @public
@@ -233,7 +280,7 @@ const levenshteinDistance = ( a, b ) => {
 /**
  * search for closest target to test string
  * @param {String} test test string
- * @param  {Array} arr targets to test
+ * @param {Array} arr targets to test
  * @returns closest target
  */
 const levenshteinClosest = ( test, arr ) => {
@@ -242,6 +289,18 @@ const levenshteinClosest = ( test, arr ) => {
 
 };
 
+/**
+ * sort best matches to test string
+ * @param {String} test test string
+ * @param {Array} arr targets to test
+ * @returns sorted matches
+ */
+const levenshteinMatch = ( test, arr ) => {
+
+    return bestMatch( 'levenshtein', test, arr );
+
+};
+
 /**
  * calculate dice coefficient
  * @param {String} a string 1
@@ -271,8 +330,8 @@ const diceCoefficient = ( a, b ) => {
 
         /* get bigrams */
 
-        let setA = bbigrams( a ),
-            setB = bbigrams( b );
+        let setA = str2bigrams( a ),
+            setB = str2bigrams( b );
 
         /* calculate dice coefficient */
 
@@ -292,7 +351,7 @@ const diceCoefficient = ( a, b ) => {
 /**
  * search for closest target to test string
  * @param {String} test test string
- * @param  {Array} arr targets to test
+ * @param {Array} arr targets to test
  * @returns closest target
  */
 const diceClosest = ( test, arr ) => {
@@ -301,6 +360,18 @@ const diceClosest = ( test, arr ) => {
 
 };
 
+/**
+ * sort best matches to test string
+ * @param {String} test test string
+ * @param {Array} arr targets to test
+ * @returns sorted matches
+ */
+const diceMatch = ( test, arr ) => {
+
+    return bestMatch( 'diceCoefficient', test, arr );
+
+};
+
 /**
  * export module functions
  */
@@ -308,6 +379,8 @@ module.exports = {
     levenshtein,
     levenshteinDistance,
     levenshteinClosest,
+    levenshteinMatch,
     diceCoefficient,
-    diceClosest
+    diceClosest,
+    diceMatch
 };

package/package.json

@@ -7,7 +7,7 @@
     "url" : "https://komed3.de"
   },
   "homepage": "https://github.com/komed3/cmpstr#readme",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "license": "MIT",
   "keywords": [
     "string",