Function collectLocalSecretSources

dev/local/env-sync/plan.ts:427–474

(
  repoRoot: string,
  workerDirs: string[],
  envLocal: Map<string, string>,
  lanIp: string | undefined,
  dirUsesLanIp: Map<string, boolean>
)

  serviceFilter?: Set<string>
): SecretsStoreQuery[] {
  const queries = new Map<string, SecretsStoreQuery>();
  for (const [name, svc] of services) {
    if (svc.type !== 'worker') continue;
    if (serviceFilter && !serviceFilter.has(name)) continue;

    const bindings = extractSecretsStoreBindings(repoRoot, svc.dir);
    for (const binding of bindings) {
      const cacheKey = `${svc.dir}:${binding.store_id}`;
      if (queries.has(cacheKey)) continue;
      const wranglerDir = path.join(repoRoot, svc.dir, '.wrangler');
      let persistenceDir = wranglerDir;
      try {
        persistenceDir = fs.realpathSync(wranglerDir);
      } catch {
        // Wrangler will create this service-local persistence directory on demand.
      }
      queries.set(cacheKey, {
        cacheKey,
        persistenceDir,
        storeId: binding.store_id,
        workerDir: svc.dir,
      });
    }
  }
  return [...queries.values()];
}

async function loadLocalStoreSecrets(
  repoRoot: string,
  serviceFilter?: Set<string>,
  options: AsyncPlanOptions = {}
): Promise<Map<string, string>> {
  const groups = new Map<string, SecretsStoreQuery[]>();
  for (const query of getSecretsStoreQueries(repoRoot, serviceFilter)) {
    const group = groups.get(query.persistenceDir) ?? [];
    group.push(query);
    groups.set(query.persistenceDir, group);
  }

  const listSecrets = options.listSecrets ?? listLocalStoreSecretsAsync;
  const groupedResults = await mapWithConcurrency(
    [...groups.values()],
    options.concurrency ?? resolveEnvSyncConcurrency(),
    async queries => {
      const results: [string, string][] = [];
      // Store IDs sharing one Wrangler persistence directory use the same local SQLite state.
      for (const query of queries) {
        results.push([query.cacheKey, await listSecrets(repoRoot, query.workerDir, query.storeId)]);
      }
      return results;
    }