sproutkit-code 0.5.0 → 0.6.0

package/dist/index.js

@@ -80585,6 +80585,7 @@ var require_places = __commonJS({
     Object.defineProperty(exports2, "__esModule", { value: true });
     exports2.PLACES_COST_PER_REQUEST_GBP = void 0;
     exports2.scoreLead = scoreLead;
+    exports2.searchPlacesStream = searchPlacesStream;
     exports2.searchPlaces = searchPlaces;
     exports2.geocodeForward = geocodeForward;
     exports2.geocodeReverse = geocodeReverse;
@@ -80617,33 +80618,90 @@ var require_places = __commonJS({
       return Math.max(0, Math.round(score));
     }
     var COST_PER_REQUEST_GBP = 0.025;
-    async function searchPlaces(args) {
-      const results = [];
-      const seen = /* @__PURE__ */ new Set();
-      for (const type of args.types) {
-        if (results.length >= args.maxResults)
-          break;
-        const body = {
-          textQuery: `${type} in ${args.city}`,
-          maxResultCount: 20
+    async function searchOneType(args) {
+      const body = {
+        textQuery: `${args.type} in ${args.city}`,
+        maxResultCount: 20
+      };
+      if (args.center && args.radiusMeters) {
+        body.locationBias = {
+          circle: {
+            center: { latitude: args.center.lat, longitude: args.center.lng },
+            radius: args.radiusMeters
+          }
         };
-        if (args.center && args.radiusMeters) {
-          body.locationBias = {
-            circle: {
-              center: { latitude: args.center.lat, longitude: args.center.lng },
-              radius: args.radiusMeters
-            }
-          };
-        }
+      }
+      try {
         const r = await axios_1.default.post("https://places.googleapis.com/v1/places:searchText", body, {
           headers: {
             "X-Goog-Api-Key": args.apiKey,
             "X-Goog-FieldMask": "places.id,places.displayName,places.formattedAddress,places.nationalPhoneNumber,places.websiteUri,places.rating,places.userRatingCount,places.location,places.priceLevel,places.businessStatus,places.regularOpeningHours"
           }
         });
-        for (const p of r.data.places ?? []) {
-          if (results.length >= args.maxResults)
-            break;
+        return { type: args.type, places: r.data.places ?? [] };
+      } catch {
+        return { type: args.type, places: [] };
+      }
+    }
+    function processRawPlaces(type, places, center, radiusMeters) {
+      const out = [];
+      for (const p of places) {
+        if (p.websiteUri)
+          continue;
+        if (center && radiusMeters && p.location) {
+          const d = haversineMeters(center.lat, center.lng, p.location.latitude, p.location.longitude);
+          if (d > radiusMeters)
+            continue;
+        }
+        const phone = p.nationalPhoneNumber ?? "";
+        const rating = p.rating ?? null;
+        const review_count = p.userRatingCount ?? null;
+        const price_level = p.priceLevel ?? null;
+        const business_status = p.businessStatus ?? null;
+        const has_hours = !!p.regularOpeningHours;
+        out.push({
+          place_id: p.id,
+          name: p.displayName?.text ?? "Unknown",
+          type,
+          address: p.formattedAddress ?? "",
+          phone,
+          website: p.websiteUri ?? null,
+          rating,
+          review_count,
+          lat: p.location?.latitude ?? null,
+          lng: p.location?.longitude ?? null,
+          price_level,
+          business_status,
+          has_hours,
+          score: scoreLead({ rating, review_count, price_level, phone, has_hours, business_status })
+        });
+      }
+      return out;
+    }
+    async function searchPlacesStream(args, onBatch) {
+      await Promise.all(args.types.map(async (type) => {
+        const { places } = await searchOneType({
+          apiKey: args.apiKey,
+          city: args.city,
+          type,
+          center: args.center,
+          radiusMeters: args.radiusMeters
+        });
+        await onBatch(processRawPlaces(type, places, args.center, args.radiusMeters));
+      }));
+    }
+    async function searchPlaces(args) {
+      const batches = await Promise.all(args.types.map((type) => searchOneType({
+        apiKey: args.apiKey,
+        city: args.city,
+        type,
+        center: args.center,
+        radiusMeters: args.radiusMeters
+      })));
+      const results = [];
+      const seen = /* @__PURE__ */ new Set();
+      for (const { type, places } of batches) {
+        for (const p of places) {
           if (seen.has(p.id))
             continue;
           seen.add(p.id);
@@ -80687,7 +80745,7 @@ var require_places = __commonJS({
         }
       }
       results.sort((a, b) => b.score - a.score);
-      return results;
+      return results.slice(0, args.maxResults);
     }
     function haversineMeters(lat1, lon1, lat2, lon2) {
       const R = 6371e3;
@@ -80726,9 +80784,58 @@ var require_lead_search = __commonJS({
   "../desktop/dist/main/lib/lead-search.js"(exports2) {
     "use strict";
     Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.searchAndSaveLeadsStream = searchAndSaveLeadsStream;
     exports2.searchAndSaveLeads = searchAndSaveLeads;
     var db_1 = require_db();
     var places_1 = require_places();
+    async function searchAndSaveLeadsStream(args, onLead) {
+      const db2 = (0, db_1.getDb)();
+      const cap = args.maxResults ?? 50;
+      const seen = /* @__PURE__ */ new Set();
+      let saved = 0;
+      let found = 0;
+      await (0, places_1.searchPlacesStream)({
+        apiKey: args.placesKey,
+        city: args.city,
+        types: args.types,
+        center: args.center,
+        radiusMeters: args.radiusMeters
+      }, (batch) => {
+        for (const r of batch) {
+          found++;
+          if (saved >= cap)
+            continue;
+          if (seen.has(r.place_id))
+            continue;
+          seen.add(r.place_id);
+          const existing = db2.all("SELECT id FROM leads WHERE place_id = ?", [r.place_id]);
+          if (existing.length > 0)
+            continue;
+          db2.run(`INSERT INTO leads
+            (place_id, name, type, city, address, phone, website, rating, review_count, lat, lng, score, price_level, business_status, status)
+           VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, 'discovered')`, [
+            r.place_id,
+            r.name,
+            r.type,
+            args.city,
+            r.address,
+            r.phone,
+            r.website,
+            r.rating,
+            r.review_count,
+            r.lat,
+            r.lng,
+            r.score,
+            r.price_level,
+            r.business_status
+          ]);
+          const row = db2.get("SELECT id FROM leads WHERE place_id = ?", [r.place_id]);
+          saved++;
+          onLead({ id: row?.id ?? 0, place_id: r.place_id, name: r.name, type: r.type, city: args.city });
+        }
+      });
+      return { found, saved };
+    }
     async function searchAndSaveLeads(args) {
       const db2 = (0, db_1.getDb)();
       const results = await (0, places_1.searchPlaces)({
@@ -81096,15 +81203,20 @@ Interested?`;
           });
           const matches = await (0, places_1.geocodeForward)(location, 1);
           const place = matches[0];
-          const result = await (0, lead_search_1.searchAndSaveLeads)({
-            placesKey,
-            // searchPlaces uses `city` as the text query, so keep the user's words.
-            city: location,
-            types: categories,
-            maxResults: Math.min(Number(input.maxResults ?? 25), 100),
-            center: place ? { lat: place.lat, lng: place.lng } : void 0,
-            radiusMeters: Number(input.radiusMeters ?? 3e3)
-          });
+          const result = await (0, lead_search_1.searchAndSaveLeadsStream)(
+            {
+              placesKey,
+              // searchPlaces uses `city` as the text query, so keep the user's words.
+              city: location,
+              types: categories,
+              maxResults: Math.min(Number(input.maxResults ?? 25), 100),
+              center: place ? { lat: place.lat, lng: place.lng } : void 0,
+              radiusMeters: Number(input.radiusMeters ?? 3e3)
+            },
+            // Emit each lead the instant it's saved, so the UI/CLI streams results
+            // in rather than waiting for every category to finish.
+            (lead) => ctx.emitProgress?.({ channel: "leads:found", payload: { lead } })
+          );
           ctx.emitProgress?.({
             channel: "leads:changed",
             payload: { reason: "find_leads", ...result }
@@ -82272,7 +82384,7 @@ var require_package = __commonJS({
   "package.json"(exports2, module2) {
     module2.exports = {
       name: "sproutkit-code",
-      version: "0.5.0",
+      version: "0.6.0",
       description: "Sproutkit in your terminal \u2014 find leads, build sites, run outreach, all via chat.",
       bin: {
         sproutkit: "./dist/index.js"
@@ -83234,6 +83346,7 @@ async function runModels() {
 // src/commands.ts
 function makeCtx(opts) {
   const emit = (channel, payload) => {
+    opts.onEvent?.(channel, payload);
     if (opts.json) return;
     const p = payload;
     if (p?.msg && opts.status) opts.status.update(p.msg);
@@ -83277,8 +83390,18 @@ async function cmdFind(argv) {
     return;
   }
   const categories = String(flags.categories ?? "").split(",").map((s) => s.trim()).filter(Boolean);
-  const status = json ? void 0 : new Status();
-  if (status) status.start(`Finding leads in ${location}`);
+  if (!json) say(c.dim(`  Searching ${location}${categories.length ? ` for ${categories.join(", ")}` : ""}\u2026`));
+  let streamed = 0;
+  const onEvent = (channel, payload) => {
+    if (json) return;
+    if (channel === "leads:found") {
+      const lead = payload.lead;
+      if (lead) {
+        streamed++;
+        say(`  ${c.green("+")} ${c.bold(lead.name)} ${c.dim(`#${lead.id} \xB7 ${lead.type}`)}`);
+      }
+    }
+  };
   try {
     const result = await runTool(
       "find_leads",
@@ -83288,14 +83411,18 @@ async function cmdFind(argv) {
         maxResults: flags.max ? Number(flags.max) : void 0,
         radiusMeters: flags.radius ? Number(flags.radius) : void 0
       },
-      makeCtx({ json, status })
+      makeCtx({ json, onEvent })
     );
-    if (status) status.done(true, `Found ${result.found ?? 0}, saved ${result.saved ?? 0} new`);
     autoSync();
-    if (json) emitJson(result);
-    else if (result.note) say(c.dim("  " + result.note));
+    if (json) {
+      emitJson(result);
+    } else {
+      say();
+      say(c.green(`  Saved ${result.saved ?? streamed} new lead${(result.saved ?? streamed) === 1 ? "" : "s"}.`));
+      if (result.note) say(c.dim("  " + result.note));
+      say();
+    }
   } catch (e) {
-    if (status) status.done(false, "Find failed");
     json ? emitJson({ ok: false, error: e.message }) : say(c.red("  " + e.message));
     process.exitCode = 1;
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sproutkit-code",
-  "version": "0.5.0",
+  "version": "0.6.0",
   "description": "Sproutkit in your terminal — find leads, build sites, run outreach, all via chat.",
   "bin": {
     "sproutkit": "./dist/index.js"