index.mjs

#!/usr/bin/env node

import fs from 'fs'
import path from 'path'
import { fileURLToPath } from 'url'

import Fuse from 'fuse.js'

const __filename = fileURLToPath(import.meta.url);
const __dirname = path.dirname(__filename);

const [, , ...args] = process.argv;
const flag = args[1];

const query = args[0] ?? '';
let fuzzySearch = () => {}
const dataPath = path.join(__dirname, 'data', 'data.json');
const countryCodeData = JSON.parse(
  fs.readFileSync(dataPath, 'utf-8')
)

if (query === '') {
  console.log('No query detected')
  console.log('Please provide a query to search for. Eg: \'cntry zim\'')
} else {
  const fuse = new Fuse(countryCodeData, {
    keys: ['name', 'alpha2', 'alpha3'], // Fields to search
    threshold: 0.3,                     // Adjust this for more/less fuzzy matches
  });

  fuzzySearch = (query) => {
    const results = fuse.search(query);
    if (results.length === 0) {
      console.log(`No matches found for "${query}"`);
      return;
    }

    results.forEach(({ item }) => {
      console.log(JSON.stringify(item, null, 2));
    });
  }
}

if (flag === '-r') {
  fuzzySearch = (query) => {
    const normalizedQuery = query.toUpperCase().trim()
    const found = countryCodeData.find(eachCountry => {
      const { alpha2, alpha3 } = eachCountry;
      if (alpha2 === normalizedQuery || alpha3 === normalizedQuery) {
        return true
      }
      return false
    })
    if (found === undefined) {
      console.log(`No matches found for "${query}"`);
    }
    console.log(JSON.stringify(found, null, 2));
  }
}

fuzzySearch(query)