@code-plate/iran-cities 1.0.0 → 1.0.2

package/dist/react.cjs

@@ -5232,61 +5232,114 @@ var data_default = [
 ];
 
 // src/index.ts
-var provinces = data_default.map((p) => ({
-  en: p["province-en"],
-  fa: p["province-fa"],
-  cities: p.cities.map((c) => ({ en: c["city-en"], fa: c["city-fa"] }))
-}));
 function normalize(input) {
   return input.trim().toLowerCase().replace(/ي/g, "\u06CC").replace(/ك/g, "\u06A9").replace(/[\u200c\s-]+/g, "");
 }
-var _flatCache = null;
-function getFlatCitiesInternal() {
-  if (_flatCache) return _flatCache;
-  const flat = [];
-  for (const p of provinces) {
-    for (const c of p.cities) {
-      flat.push({ en: c.en, fa: c.fa, provinceEn: p.en, provinceFa: p.fa });
+function levenshtein(a, b) {
+  if (a === b) return 0;
+  const al = a.length;
+  const bl = b.length;
+  if (al === 0) return bl;
+  if (bl === 0) return al;
+  let prev = new Array(bl + 1);
+  for (let j = 0; j <= bl; j++) prev[j] = j;
+  for (let i = 1; i <= al; i++) {
+    const curr = new Array(bl + 1);
+    curr[0] = i;
+    for (let j = 1; j <= bl; j++) {
+      const cost = a.charCodeAt(i - 1) === b.charCodeAt(j - 1) ? 0 : 1;
+      curr[j] = Math.min(prev[j] + 1, curr[j - 1] + 1, prev[j - 1] + cost);
     }
+    prev = curr;
   }
-  _flatCache = flat;
-  return flat;
+  return prev[bl];
+}
+function matchScore(query, target, fuzzy, maxDistance) {
+  if (!target) return Infinity;
+  if (target === query) return 0;
+  if (target.startsWith(query)) return 1;
+  if (target.includes(query)) return 2;
+  if (fuzzy) {
+    const dist = levenshtein(query, target);
+    if (dist <= maxDistance) return 3 + dist;
+  }
+  return Infinity;
+}
+var provinces = data_default.map((p) => {
+  const provinceEn = p["province-en"];
+  const provinceFa = p["province-fa"];
+  return {
+    en: provinceEn,
+    fa: provinceFa,
+    cities: p.cities.map((c) => ({
+      en: c["city-en"],
+      fa: c["city-fa"],
+      slug: `${provinceEn}/${c["city-en"]}`
+    }))
+  };
+});
+var flatCities = [];
+var cityBySlug = /* @__PURE__ */ new Map();
+var cityByNormKey = /* @__PURE__ */ new Map();
+var provinceByNormKey = /* @__PURE__ */ new Map();
+var cityToProvinceByNormKey = /* @__PURE__ */ new Map();
+var searchableCities = [];
+var searchableProvinces = [];
+for (const p of provinces) {
+  const pNormEn = normalize(p.en);
+  const pNormFa = normalize(p.fa);
+  if (!provinceByNormKey.has(pNormEn)) provinceByNormKey.set(pNormEn, p);
+  if (!provinceByNormKey.has(pNormFa)) provinceByNormKey.set(pNormFa, p);
+  searchableProvinces.push({ item: p, normEn: pNormEn, normFa: pNormFa });
+  for (const c of p.cities) {
+    const flat = { en: c.en, fa: c.fa, slug: c.slug, provinceEn: p.en, provinceFa: p.fa };
+    flatCities.push(flat);
+    cityBySlug.set(flat.slug, flat);
+    const cNormEn = normalize(c.en);
+    const cNormFa = normalize(c.fa);
+    if (!cityByNormKey.has(cNormEn)) cityByNormKey.set(cNormEn, flat);
+    if (!cityByNormKey.has(cNormFa)) cityByNormKey.set(cNormFa, flat);
+    if (!cityToProvinceByNormKey.has(cNormEn)) cityToProvinceByNormKey.set(cNormEn, p);
+    if (!cityToProvinceByNormKey.has(cNormFa)) cityToProvinceByNormKey.set(cNormFa, p);
+    searchableCities.push({ item: flat, normEn: cNormEn, normFa: cNormFa });
+  }
+}
+function rankedSearch(pool, query, options) {
+  var _a, _b, _c;
+  const q = normalize(query);
+  if (!q) {
+    const all = pool.map((e) => e.item);
+    return options.limit ? all.slice(0, options.limit) : all;
+  }
+  const locale = (_a = options.locale) != null ? _a : "both";
+  const fuzzy = (_b = options.fuzzy) != null ? _b : false;
+  const maxDistance = (_c = options.maxDistance) != null ? _c : 2;
+  const scored = [];
+  for (const entry of pool) {
+    let score = Infinity;
+    if (locale !== "fa") score = Math.min(score, matchScore(q, entry.normEn, fuzzy, maxDistance));
+    if (locale !== "en") score = Math.min(score, matchScore(q, entry.normFa, fuzzy, maxDistance));
+    if (score !== Infinity) scored.push({ item: entry.item, score });
+  }
+  scored.sort((a, b) => a.score - b.score);
+  const results = scored.map((s) => s.item);
+  return options.limit ? results.slice(0, options.limit) : results;
 }
 function getProvincesList() {
   return provinces.map((p) => ({ en: p.en, fa: p.fa }));
 }
 function getProvince(query) {
-  const q = normalize(query);
-  return provinces.find((p) => normalize(p.en) === q || normalize(p.fa) === q);
+  return provinceByNormKey.get(normalize(query));
 }
 function getCities(provinceQuery) {
-  const province = getProvince(provinceQuery);
-  return province ? province.cities : [];
+  var _a, _b;
+  return (_b = (_a = getProvince(provinceQuery)) == null ? void 0 : _a.cities) != null ? _b : [];
 }
 function searchProvinces(query, options = {}) {
-  var _a;
-  const q = normalize(query);
-  if (!q) return options.limit ? provinces.slice(0, options.limit) : provinces;
-  const locale = (_a = options.locale) != null ? _a : "both";
-  const results = provinces.filter((p) => {
-    const matchEn = locale !== "fa" && normalize(p.en).includes(q);
-    const matchFa = locale !== "en" && normalize(p.fa).includes(q);
-    return matchEn || matchFa;
-  });
-  return options.limit ? results.slice(0, options.limit) : results;
+  return rankedSearch(searchableProvinces, query, options);
 }
 function searchCities(query, options = {}) {
-  var _a;
-  const q = normalize(query);
-  const all = getFlatCitiesInternal();
-  if (!q) return options.limit ? all.slice(0, options.limit) : all;
-  const locale = (_a = options.locale) != null ? _a : "both";
-  const results = all.filter((c) => {
-    const matchEn = locale !== "fa" && normalize(c.en).includes(q);
-    const matchFa = locale !== "en" && normalize(c.fa).includes(q);
-    return matchEn || matchFa;
-  });
-  return options.limit ? results.slice(0, options.limit) : results;
+  return rankedSearch(searchableCities, query, options);
 }
 
 // src/react.ts