Register custom scalar codecs for type generation

This means we can gen correct types, not statically known,
for query files & inline queries.
These two processes (via `$.analyzeQuery`) use
the `tsType` property from the codec.

Since we set the codecs in the client's codec registry now,
we don't need to modify the SCALAR_CODECS global.

Author: CarsonF

src/core/edgedb/generator/run.ts

@@ -1,10 +1,11 @@
 import { createClient } from 'edgedb';
 import { IndentationText, Project, QuoteKind } from 'ts-morph';
+import { codecs, registerCustomScalarCodecs } from '../codecs';
 import { generateSchema } from './generate-schema';
 import { generateInlineQueries } from './inline-queries';
 import { generateQueryBuilder } from './query-builder';
 import { generateQueryFiles } from './query-files';
-import { changeScalarCodecsToOurCustomTypes } from './scalars';
+import { setTsTypesFromOurScalarCodecs } from './scalars';
 import { GeneratorParams } from './util';
 
 (async () => {
@@ -29,7 +30,8 @@ import { GeneratorParams } from './util';
     edgedbDir: project.addDirectoryAtPath('src/core/edgedb'),
   };
 
-  changeScalarCodecsToOurCustomTypes();
+  await registerCustomScalarCodecs(client, codecs);
+  setTsTypesFromOurScalarCodecs();
 
   try {
     await generateQueryBuilder(params);

src/core/edgedb/generator/scalars.ts

@@ -1,23 +1,16 @@
 import { scalarToLiteralMapping } from '@edgedb/generate/dist/genutil.js';
 import { mapEntries } from '@seedcompany/common';
-import { SCALAR_CODECS } from 'edgedb/dist/codecs/codecs.js';
-import { KNOWN_TYPENAMES } from 'edgedb/dist/codecs/consts.js';
 import { codecs } from '../codecs';
 
 export const customScalars = mapEntries(codecs, (codec) => [
   codec.info.ts,
   codec.info,
 ]).asMap;
 
-export function changeScalarCodecsToOurCustomTypes() {
-  for (const codec of codecs) {
-    const fqName = `${codec.info.module}::${codec.info.type}`;
-
-    // codecs are used for edgeql files & inline queries ($.analyzeQuery)
-    const id = KNOWN_TYPENAMES.get(fqName);
-    id && SCALAR_CODECS.set(id, new codec(id));
-
-    // this is used for schema interfaces
-    scalarToLiteralMapping[fqName].type = codec.info.ts;
+export function setTsTypesFromOurScalarCodecs() {
+  // this is used for schema interfaces
+  for (const { info } of codecs) {
+    const fqName = `${info.module}::${info.type}`;
+    scalarToLiteralMapping[fqName].type = info.ts;
   }
 }