import { BusinessClient } from "datafiniti-sdk";

const sdk = BusinessClient.fromEnv();

const response = await sdk.searchByName("Starbucks", {
  city: "Seattle",
});

console.log("
=== NAME SEARCH TEST ===");
console.log("Matches:", response.num_found);

const records = response.records ?? [];

if (records.length) {
  const business = records[0];
  console.log("
Business:");
  console.log(business.name);
  console.log(business.address);
  console.log(business.city, business.province);
  console.log(business.categories);
} else {
  console.log("No business found.");
}

import { BusinessClient } from "datafiniti-sdk";

const sdk = BusinessClient.fromEnv();

const response = await sdk.searchByAddress("2200 E Cesar Chavez St", {
  postalCode: "78702",
});

console.log("
=== ADDRESS SEARCH TEST ===");
console.log("Matches:", response.num_found);

const records = response.records ?? [];

if (records.length) {
  const business = records[0];
  console.log("
Business:");
  console.log(business.name);
  console.log(business.address);
  console.log(business.city, business.province, business.postalCode);
} else {
  console.log("No business found.");
}

import { BusinessClient } from "datafiniti-sdk";

const sdk = BusinessClient.fromEnv();

const count = await sdk.count('country:US AND categories:"Coffee Shop"');

console.log("
=== COUNT TEST ===");
console.log(`Businesses found: ${count.toLocaleString()}`);

import { BusinessClient } from "datafiniti-sdk";

const sdk = BusinessClient.fromEnv();

// Search for businesses within 5 miles of downtown Austin, TX
// geoLocation format: [Longitude, Latitude, Distance, Unit]
// Supported units: m, mi, ft, yd, mm, km, NM, cm
const response = await sdk.geolocation({
  longitude: -97.7430600,
  latitude: 30.2671500,
  distance: 5,
  unit: "mi",
  numRecords: 5,
  view: [
    { name: "name" },
    { name: "address" },
    { name: "city" },
    { name: "province" },
    { name: "postalCode" },
    { name: "categories" },
    { name: "geoLocation" },
  ],
});

console.log("
=== GEOLOCATION SEARCH ===");
console.log(`Total matches: ${(response.num_found ?? 0).toLocaleString()}`);

const records = response.records ?? [];

for (const business of records) {
  console.log(
    `
${business.name} — ${business.address}, ${business.city} ${business.postalCode}` +
      `
--geoLocation: ${business.geoLocation}`
  );
}

import { BusinessClient } from "datafiniti-sdk";

const sdk = BusinessClient.fromEnv();

console.log("
=== PAGINATION TEST ===");

let recordsSeen = 0;

for await (const business of sdk.paginate({
  query: 'country:US AND categories:"Restaurant"',
  pageSize: 100,
  maxRecords: 250,
})) {
  recordsSeen += 1;
  console.log(`${recordsSeen}. ${business.name ?? "Unknown Business"}`);
}

console.log(`
Total Retrieved: ${recordsSeen}`);

/**
 * Business Views Example: Directory Enrichment
 *
 * Uses a custom view to fetch only the fields needed to build a business
 * directory — categories and phone numbers — for a set of businesses in a
 * city. Requesting a narrow view keeps responses small and fast.
 *
 * Reference: https://docs.datafiniti.co/docs/using-a-custom-business-view
 */

import { BusinessClient, type DatafinitiRecord } from "datafiniti-sdk";

// ---------------------------------------------------------------------------
// View options
//
// OPTION 1 (used below): Custom inline view
//   Pass an array of field objects directly in the request. No setup required.
//
// OPTION 2: Premade (named) view — pass a string instead of an array.
//   "default"  All fields (same as omitting view entirely).
//   Example: sdk.search({ query, view: "default", numRecords: 10 })
//   Reference: https://docs.datafiniti.co/docs/business-data-views
// ---------------------------------------------------------------------------

const BUSINESS_VIEW = [
  { name: "name" },
  { name: "address" },
  { name: "city" },
  { name: "province" },
  { name: "categories" },
  { name: "phones" },
];

function formatList(value: unknown): string {
  return Array.isArray(value) && value.length ? value.join(", ") : "N/A";
}

async function main() {
  const sdk = BusinessClient.fromEnv();

  // Find restaurants in Austin, TX and enrich each with categories + phones.
  const query = sdk
    .query()
    .country("US")
    .province("TX")
    .city("Austin")
    .categories("Restaurant")
    .build();

  const response = await sdk.search({ query, numRecords: 10, view: BUSINESS_VIEW });
  const records: DatafinitiRecord[] = response.records ?? [];

  console.log("
=== BUSINESS DIRECTORY ===
");

  for (const business of records) {
    console.log(`${business.name ?? "N/A"}`);
    console.log(`  Address:    ${business.address ?? "N/A"}, ${business.city ?? "N/A"}, ${business.province ?? "N/A"}`);
    console.log(`  Categories: ${formatList(business.categories)}`);
    console.log(`  Phones:     ${formatList(business.phones)}`);
    console.log();
  }
}

main();

import { BusinessClient, DatafinitiApiError } from "datafiniti-sdk";

const sdk = BusinessClient.fromEnv();

try {
  const response = await sdk.search({ query: 'country:US AND categories:"Restaurant"', numRecords: 5 });
  console.log(`Matches: ${response.num_found}`);
} catch (err) {
  if (err instanceof DatafinitiApiError) {
    console.error(`API error ${err.statusCode}: ${err.message}`, err.errors);
  } else {
    throw err;
  }
}