@agentrhq/webcmd 0.1.2 → 0.2.0

package/clis/dianping/cityResolver.js

@@ -1,185 +0,0 @@
-/**
- * Async city → cityId resolver for dianping adapters.
- *
- * Wraps the synchronous static-map `resolveCityId` from utils.js and falls
- * back to a live lookup against www.dianping.com when the input is not in
- * the curated map. Resolves both pinyin slugs (e.g. "shantou") and Chinese
- * names (e.g. "汕头") by reading dianping itself, so the adapter no longer
- * has to ship a complete static city table.
- *
- * Strategy:
- *   1. Empty / null  → null  (let the cookie's default city stand).
- *   2. All-digits    → numeric cityId pass-through.
- *   3. Static map    → fast path, no network. Reuses utils.CITY_ID.
- *   4. Pinyin slug   → goto https://www.dianping.com/<slug>, parse the
- *                       cityId out of any /search/keyword/{id}/ link.
- *   5. Chinese name  → goto https://www.dianping.com/citylist, build a
- *                       Chinese-name → pinyin map, then resolve the slug
- *                       as in step 4.
- *
- * Resolved (input → cityId) pairs are memoized in a module-level Map so a
- * second search in the same process skips both navigations.
- */
-
-import { ArgumentError, CommandExecutionError } from '@agentrhq/webcmd/errors';
-import { CITY_ID, resolveCityId } from './utils.js';
-
-const CHINESE_RE = /^[一-龥]+$/;
-const PINYIN_RE = /^[a-z]+$/;
-
-const RESOLVE_CACHE = new Map();
-
-/**
- * Reset the in-process resolver cache. Exposed for tests so each case
- * starts from a clean slate; production code never needs to call this.
- */
-export function clearCityResolverCache() {
-    RESOLVE_CACHE.clear();
-}
-
-/**
- * Async resolver. Falls back to live dianping pages only when the input
- * is not in the static map.
- *
- * @param {{ goto: Function, evaluate: Function }} page  page handle from the adapter func
- * @param {string|number|null|undefined} cityArg         user-supplied city (name, pinyin, or numeric id)
- * @returns {Promise<number|null>}                       numeric cityId, or null to use the cookie default
- */
-export async function resolveCityIdAsync(page, cityArg) {
-    if (cityArg == null || cityArg === '') return null;
-    const raw = String(cityArg).trim();
-    if (!raw) return null;
-    if (/^\d+$/.test(raw)) return Number(raw);
-
-    const lowered = raw.toLowerCase();
-
-    // Fast path: reuse the synchronous static map. resolveCityId throws
-    // ArgumentError when the input is unknown — that's the trigger to fall
-    // back to dynamic resolution rather than surface the error to the user.
-    try {
-        const staticId = resolveCityId(raw);
-        if (staticId != null) return staticId;
-    } catch (err) {
-        if (err?.code !== 'ARGUMENT') throw err;
-    }
-
-    if (RESOLVE_CACHE.has(lowered)) return RESOLVE_CACHE.get(lowered);
-    if (RESOLVE_CACHE.has(raw)) return RESOLVE_CACHE.get(raw);
-
-    let pinyin = null;
-    if (PINYIN_RE.test(lowered)) {
-        pinyin = lowered;
-    } else if (CHINESE_RE.test(raw)) {
-        pinyin = await lookupPinyinFromCitylist(page, raw);
-        if (!pinyin) {
-            const known = Object.keys(CITY_ID).filter((k) => /^[a-z]+$/.test(k)).join(', ');
-            throw new ArgumentError(
-                'city',
-                `unknown city '${cityArg}'. pass a numeric cityId, a pinyin slug (e.g. shantou), `
-                + `a Chinese name listed on dianping.com/citylist, or one of: ${known}`,
-            );
-        }
-    } else {
-        const known = Object.keys(CITY_ID).filter((k) => /^[a-z]+$/.test(k)).join(', ');
-        throw new ArgumentError(
-            'city',
-            `unknown city '${cityArg}'. pass a numeric cityId, a pinyin slug (e.g. shantou), `
-            + `a Chinese name listed on dianping.com/citylist, or one of: ${known}`,
-        );
-    }
-
-    const cityId = await fetchCityIdByPinyin(page, pinyin);
-    if (!cityId) {
-        throw new CommandExecutionError(
-            `dianping could not resolve cityId for '${cityArg}' (pinyin=${pinyin}); `
-            + `the city page rendered without a /search/keyword/{id}/ link`,
-        );
-    }
-
-    RESOLVE_CACHE.set(lowered, cityId);
-    RESOLVE_CACHE.set(pinyin, cityId);
-    if (CHINESE_RE.test(raw)) RESOLVE_CACHE.set(raw, cityId);
-    return cityId;
-}
-
-/**
- * Read https://www.dianping.com/citylist and return a Chinese-name → pinyin
- * slug map for every city link present on the page. Used when the user
- * supplied a Chinese name that isn't in the static map.
- */
-async function lookupPinyinFromCitylist(page, chineseName) {
-    await page.goto('https://www.dianping.com/citylist');
-    const map = await page.evaluate(`(${buildCitylistMap.toString()})()`);
-    if (!map || typeof map !== 'object' || Object.keys(map).length === 0) {
-        throw new CommandExecutionError(
-            'dianping citylist did not render any city anchors; cannot resolve Chinese city names',
-        );
-    }
-    if (map && typeof map === 'object' && map[chineseName]) {
-        return String(map[chineseName]).toLowerCase();
-    }
-    return null;
-}
-
-/**
- * Pure DOM extractor for /citylist. Walks every anchor on the page and
- * keeps the ones whose href matches the per-city slug shape and whose
- * text is a pure-Chinese label. Defined at module scope so the same code
- * can be exercised from JSDOM tests via toString() injection.
- */
-export function buildCitylistMap() {
-    const map = {};
-    const anchors = document.querySelectorAll('a');
-    anchors.forEach((a) => {
-        const hrefRaw = a.getAttribute('href') || '';
-        const text = ((a.textContent || '').trim());
-        if (!text || !/^[一-龥]+$/.test(text)) return;
-        const href = hrefRaw.replace(/^https?:/, '');
-        const m = href.match(/^\/\/(?:www\.)?dianping\.com\/([a-z]+)\/?$/)
-            || href.match(/^\/([a-z]+)\/?$/);
-        if (!m) return;
-        const slug = m[1].toLowerCase();
-        // Filter out non-city slugs that share the shape (e.g. /citylist itself,
-        // /promo, /events). Only register the first slug per Chinese label.
-        if (slug === 'citylist' || slug === 'promo' || slug === 'events') return;
-        if (!map[text]) map[text] = slug;
-    });
-    return map;
-}
-
-/**
- * Visit https://www.dianping.com/<slug> and pull the cityId out of any
- * /search/keyword/{id}/ anchor. The PC city landing page renders these
- * links server-side for every category card, so a single goto + DOM read
- * is enough — no extra clicks or hydration wait.
- */
-async function fetchCityIdByPinyin(page, pinyin) {
-    await page.goto(`https://www.dianping.com/${pinyin}`);
-    const cityId = await page.evaluate(`(${extractCityIdFromPage.toString()})()`);
-    return Number.isInteger(cityId) && cityId > 0 ? cityId : null;
-}
-
-/**
- * Pure DOM extractor for the per-city landing page. Defined at module
- * scope so the same code is exercised in JSDOM tests via toString().
- */
-export function extractCityIdFromPage() {
-    const baseHref = (typeof location !== 'undefined' && location.href) || 'https://www.dianping.com/';
-    const anchors = document.querySelectorAll('a[href]');
-    for (const a of anchors) {
-        const hrefRaw = a.getAttribute('href') || '';
-        let url;
-        try {
-            url = new URL(hrefRaw, baseHref);
-        } catch {
-            continue;
-        }
-        if (url.protocol !== 'https:') continue;
-        if (url.hostname !== 'www.dianping.com' && url.hostname !== 'dianping.com') continue;
-        const m = url.pathname.match(/^\/search\/keyword\/(\d+)(?:\/|$)/);
-        if (!m) continue;
-        const n = Number(m[1]);
-        if (Number.isInteger(n) && n > 0) return n;
-    }
-    return null;
-}