Start guessing countries

This change only recognises cases where the location is a country name or code.

Refs #31

package.json

@@ -10,6 +10,7 @@
     "d3-dsv": "^3.0.1",
     "d3-time-format": "^4.1.0",
     "effect": "^3.3.5",
+    "i18n-iso-countries": "^7.11.2",
     "iso-639-1": "^3.1.2"
   },
   "devDependencies": {

src/data/users.json.ts

@@ -1,7 +1,8 @@
 import { HttpClient, HttpClientRequest, HttpClientResponse, Terminal } from '@effect/platform'
 import { NodeTerminal } from '@effect/platform-node'
 import { Schema } from '@effect/schema'
-import { Config, Effect, Redacted } from 'effect'
+import { Array, Config, Effect, Option, Redacted } from 'effect'
+import * as Iso3166 from '../lib/Iso3166.js'
 import * as Temporal from '../lib/Temporal.js'
 
 const Users = Schema.Array(
@@ -12,6 +13,15 @@ const Users = Schema.Array(
   }),
 )
 
+const Output = Schema.Array(
+  Schema.Struct({
+    careerStage: Schema.OptionFromUndefinedOr(Schema.Literal('early', 'mid', 'late')),
+    location: Schema.OptionFromUndefinedOr(Schema.String),
+    country: Schema.OptionFromUndefinedOr(Iso3166.Alpha2CodeSchema),
+    timestamp: Temporal.InstantFromStringSchema,
+  }),
+)
+
 const program = Effect.gen(function* () {
   const terminal = yield* Terminal.Terminal
   const token = yield* Config.redacted('PREREVIEW_REVIEWS_DATA_TOKEN')
@@ -26,7 +36,12 @@ const program = Effect.gen(function* () {
     Effect.scoped,
   )
 
-  const encoded = yield* Schema.encode(Schema.parseJson(Users))(data)
+  const transformedData = Array.map(data, user => ({
+    ...user,
+    country: Option.flatMap(user.location, Iso3166.guessCountry),
+  }))
+
+  const encoded = yield* Schema.encode(Schema.parseJson(Output))(transformedData)
 
   yield* terminal.display(encoded)
 })

src/lib/Iso3166.ts

@@ -0,0 +1,13 @@
+import { Schema } from '@effect/schema'
+import { Option, type Predicate } from 'effect'
+import iso3166 from 'i18n-iso-countries'
+
+export type Alpha2Code = iso3166.Alpha2Code
+
+export const isAlpha2Code: Predicate.Refinement<unknown, Alpha2Code> = (u): u is Alpha2Code =>
+  typeof u === 'string' && u in iso3166.getAlpha2Codes()
+
+export const Alpha2CodeSchema: Schema.Schema<Alpha2Code, string> = Schema.String.pipe(Schema.filter(isAlpha2Code))
+
+export const guessCountry = (location: string): Option.Option<Alpha2Code> =>
+  Option.fromNullable(iso3166.getAlpha2Code(location, 'en')).pipe(Option.filter(isAlpha2Code))

src/users.md

@@ -135,8 +135,8 @@ function usersByCareerStage({ width } = {}) {
 ```js
 function usersByLocation() {
   return Inputs.table(
-    usersInTimePeriod.flatMap(user => (user.location ? { location: user.location } : [])),
-    { header: { location: 'Location' }, sort: 'location' },
+    usersInTimePeriod.flatMap(user => (user.location ? { location: user.location, country: user.country } : [])),
+    { header: { country: 'Country', location: 'Location' }, sort: 'location' },
   )
 }
 ```