county.ts

Filename	Percent	Threshold	Total	Covered	Uncovered
src/db/schema/county.ts	100.00%	80%	166	166	0

import { boolean, integer, varchar } from 'drizzle-orm/pg-core';

import { absenteeMethod } from './absenteeMethod.js';
import { candidateVote } from './candidateVote.js';
import { candidateVoteByParty } from './candidateVoteByParty.js';
import { countyByJurisdiction } from './countyByJurisdiction.js';
import { countyDemographic } from './countyDemographic.js';
import { countyExpectedVote } from './countyExpectedVote.js';
import { countyExpectedVoteStratum } from './countyExpectedVoteStratum.js';
import { countyGeoStratum } from './countyGeoStratum.js';
import { countyPartyStratum } from './countyPartyStratum.js';
import { countyRegisteredVotersByParty } from './countyRegisteredVotersByParty.js';
import { delegateDistrictCounty } from './delegateDistrictCounty.js';
import { officeVote } from './officeVote.js';
import { pastCandidateVoteByCounty } from './pastCandidateVoteByCounty.js';
import { pastCountyVote } from './pastCountyVote.js';
import { presidentialPrimaryDistrictVoteByCounty } from './presidentialPrimaryDistrictVoteByCounty.js';
import { relations } from 'drizzle-orm';
import { schema } from './schema.js';
import { state } from './state.js';
import { totalExpectedVote } from './totalExpectedVote.js';
import { voteFlag } from './voteFlag.js';

/**
 * County has fields that aren't constrained as notNull b/c we get the data
 * for a county record from more than one endpoint.
 */
export const county = schema.table('County', {
  uid: varchar('uid').primaryKey(),
  fips: varchar('fips'),
  code: integer('code'),
  name: varchar('name'),
  abbr: varchar('abbr'),
  totalRegisteredVoters: integer('totalRegisteredVoters'),
  isTown: boolean('isTown'),
  isPseudoCounty: boolean('isPseudoCounty'),
  containingCounty: integer('containingCounty'),
  maxAbsenteeBallotsExpected: integer('maxAbsenteeBallotsExpected'),

  absenteeMethod: integer('absenteeMethod').references(
    () => absenteeMethod.uid
  ),
  countyExpectedVoteStratum: varchar('countyExpectedVoteStratum').references(
    () => countyExpectedVoteStratum.uid
  ),
  countyGeoStratum: varchar('countyGeoStratum').references(
    () => countyGeoStratum.uid
  ),
  countyPartyStratum: varchar('countyPartyStratum').references(
    () => countyPartyStratum.uid
  ),
  state: integer('state')
    .notNull()
    .references(() => state.uid, { onDelete: 'cascade' }),
});

export type NewCounty = typeof county.$inferInsert;

export const countyRelations = relations(county, ({ one, many }) => ({
  absenteeMethod: one(absenteeMethod, {
    fields: [county.absenteeMethod],
    references: [absenteeMethod.uid],
  }),
  candidateVotes: many(candidateVote),
  candidateVoteByParties: many(candidateVoteByParty),
  countyByJurisdictions: many(countyByJurisdiction),
  countyDemographics: one(countyDemographic),
  countyExpectedVotes: many(countyExpectedVote),
  countyExpectedVoteStratum: one(countyExpectedVoteStratum, {
    fields: [county.countyExpectedVoteStratum],
    references: [countyExpectedVoteStratum.uid],
  }),
  countyGeoStratum: one(countyGeoStratum, {
    fields: [county.countyGeoStratum],
    references: [countyGeoStratum.uid],
  }),
  countyPartyStratum: one(countyPartyStratum, {
    fields: [county.countyPartyStratum],
    references: [countyPartyStratum.uid],
  }),
  countyRegisteredVotersByParties: many(countyRegisteredVotersByParty),
  voteFlags: many(voteFlag),
  delegateDistrictCounties: many(delegateDistrictCounty),
  officeVotes: many(officeVote),
  pastCandidateVoteByCounties: many(pastCandidateVoteByCounty),
  pastCountyVotes: many(pastCountyVote),
  presidentialPrimaryDistrictVoteByCounties: many(
    presidentialPrimaryDistrictVoteByCounty
  ),
  state: one(state, {
    fields: [county.state],
    references: [state.uid],
  }),
  totalExpectedVotes: many(totalExpectedVote),
}));

[]

TypeScript coverage report