countyStrata.ts

Filename	Percent	Threshold	Total	Covered	Uncovered
src/events/parsers/county-by-elec/countyStrata.ts	100.00%	80%	297	297	0

import type { Document } from './types.js';
import type { NewCounty } from 'db/schema/county.js';
import type { NewCountyExpectedVoteStratum } from 'db/schema/countyExpectedVoteStratum.js';
import type { NewCountyGeoStratum } from 'db/schema/countyGeoStratum.js';
import type { NewCountyPartyStratum } from 'db/schema/countyPartyStratum.js';
import { parseKey } from 'utils/documentKey.js';
import { truthyString } from 'utils/truthyString.js';
import uniqBy from 'lodash-es/uniqBy.js';

/**
 * Parses documents we get from querying the `events/county-by-elec` endpoint. It then transforms the data to fit the structure of the CountyGeoStratum model and returns an array of objects that can be upserted into the model.
 *
 * The returned array contains information on geographical strata scheme. A strata scheme partitions each state into different strata and assigns each county to a particular stratum. For example, New York state has these geo strata: New York City, Long Islandm Hudson Valley, Urban Upstate, and Rural Upstate. Each county is assigned to one of these strata via the geoStratum field.
 */
export const parseCountyGeoStrata = (documents: Document[]) => {
  let strata: NewCountyGeoStratum[] = [];
  for (const elec of documents) {
    // Skip non-statewide strata
    if (elec.key.jType !== 'SW') continue;
    const { state } = parseKey(elec.key);
    for (const schema of elec.countyGeoStrataScheme.strmSch) {
      const {
        strmSchAbbv4: schemaAbbr,
        strmSchName: schemaName,
        strmSchCde: schemaCode,
      } = schema;
      const instances: NewCountyGeoStratum[] = schema.stratum.map((strat) => ({
        uid: `${state}_${strat.strmCde}`,
        state,

        // scheme info
        schemaAbbr,
        schemaName,
        schemaCode,

        // stratum info
        code: strat.strmCde,
        name: strat.strmName,
        abbr: strat.strmAbbv4,

        isBlk: truthyString(strat.isBlk),
        totalPrecincts: strat.totalPrecincts,
      }));
      strata = [...strata, ...instances];
    }
  }
  return uniqBy(strata, (d) => d.uid);
};

export const parseCountyPartyStrata = (documents: Document[]) => {
  let strata: NewCountyPartyStratum[] = [];
  for (const elec of documents) {
    // Skip non-statewide strata
    if (elec.key.jType !== 'SW') continue;
    const { state } = parseKey(elec.key);

    for (const schema of elec.countyPartyStrataScheme.strmSch) {
      const {
        strmSchAbbv4: schemaAbbr,
        strmSchName: schemaName,
        strmSchCde: schemaCode,
      } = schema;
      const instances: NewCountyPartyStratum[] = schema.stratum.map(
        (strat) => ({
          uid: `${state}_${strat.strmCde}`,
          state, // 🔑

          // scheme info
          schemaAbbr,
          schemaName,
          schemaCode,

          // stratum info
          code: strat.strmCde,
          name: strat.strmName,
          abbr: strat.strmAbbv4,

          isBlk: truthyString(strat.isBlk),
          totalPrecincts: strat.totalPrecincts,
        })
      );
      strata = [...strata, ...instances];
    }
  }
  return uniqBy(strata, (d) => d.uid);
};

export const parseCountyExpectedVoteStrata = (documents: Document[]) => {
  let strata: NewCountyExpectedVoteStratum[] = [];
  for (const elec of documents) {
    // Skip non-statewide strata
    if (elec.key.jType !== 'SW') continue;
    const { state } = parseKey(elec.key);

    // Check if countyExpVoteStrataScheme exists
    if (!elec.countyExpVoteStrataScheme) continue;

    /**
     * 🚨 Inconsistent with ElectionTonight API documentation
     *
     * The ElectionTonight API documentation says `elec.countyExpVoteStrataScheme.strmSch` should be an array but in the 2020 data, it is an object.
     *
     * We need added checks to see if it is an array or an object and to handle it accordingly.
     */
    const strmSch = elec.countyExpVoteStrataScheme.strmSch as
      | typeof elec.countyExpVoteStrataScheme.strmSch
      | (typeof elec.countyExpVoteStrataScheme.strmSch)[];

    // Check if countyExpVoteStrataScheme.strmSch is an array or object
    if (Array.isArray(strmSch)) {
      for (const schema of strmSch) {
        const {
          strmSchAbbv4: schemaAbbr,
          strmSchName: schemaName,
          strmSchCde: schemaCode,
        } = schema;
        const instances: NewCountyExpectedVoteStratum[] = schema.stratum.map(
          (strat) => ({
            uid: `${state}_${strat.strmCde}`,
            state,

            // scheme info
            schemaAbbr,
            schemaName,
            schemaCode,

            // stratum info
            code: strat.strmCde,
            name: strat.strmName,
            abbr: strat.strmAbbv4,

            isBlk: truthyString(strat.isBlk),
            totalPrecincts: strat.totalPrecincts,
          })
        );
        strata = [...strata, ...instances];
      }
    } else if (typeof strmSch === 'object') {
      const {
        strmSchAbbv4: schemaAbbr,
        strmSchName: schemaName,
        strmSchCde: schemaCode,
        stratum,
      } = strmSch;
      for (const strat of stratum) {
        const instance: NewCountyExpectedVoteStratum = {
          uid: `${state}_${strat.strmCde}`,
          state,

          // scheme info
          schemaAbbr,
          schemaName,
          schemaCode,

          // stratum info
          code: strat.strmCde,
          name: strat.strmName,
          abbr: strat.strmAbbv4,

          isBlk: truthyString(strat.isBlk),
          totalPrecincts: strat.totalPrecincts,
        };
        strata.push(instance);
      }
    }
  }
  return uniqBy(strata, (d) => d.uid);
};

export const parseCountyStrata = (documents: Document[]) => {
  const instances: NewCounty[] = [];
  for (const elec of documents) {
    // Skip non-statewide strata
    if (elec.key.jType !== 'SW') continue;
    const { state } = parseKey(elec.key);

    for (const c of elec.counties) {
      const { expectedVoteStratum, geoStratum, partyStratum, cnty } = c;
      const uid = `${state}_${cnty}`;

      const instance: NewCounty = {
        uid,
        state,
        countyPartyStratum: `${state}_${partyStratum}`,
        countyGeoStratum: `${state}_${geoStratum}`,
        countyExpectedVoteStratum: expectedVoteStratum
          ? `${state}_${expectedVoteStratum}`
          : null,
      };
      instances.push(instance);
    }
  }
  return uniqBy(instances, (d) => d.uid);
};

[]

TypeScript coverage report