fix: add x-application-name header, add python to server.ts

Author: o-santi

src/server/routes/generators/python.ts

@@ -1,25 +1,23 @@
 import type { FastifyInstance } from 'fastify'
 import { PostgresMeta } from '../../../lib/index.js'
-import { DEFAULT_POOL_CONFIG } from '../../constants.js'
-import { extractRequestForLogging } from '../../utils.js'
+import { createConnectionConfig, extractRequestForLogging } from '../../utils.js'
 import { apply as applyPyTemplate } from '../../templates/python.js'
 import { getGeneratorMetadata } from '../../../lib/generators.js'
 
 export default async (fastify: FastifyInstance) => {
   fastify.get<{
-    Headers: { pg: string }
+    Headers: { pg: string; 'x-pg-application-name'?: string }
     Querystring: {
       excluded_schemas?: string
       included_schemas?: string
     }
   }>('/', async (request, reply) => {
-    const connectionString = request.headers.pg
+    const config = createConnectionConfig(request)
     const excludedSchemas =
       request.query.excluded_schemas?.split(',').map((schema) => schema.trim()) ?? []
     const includedSchemas =
       request.query.included_schemas?.split(',').map((schema) => schema.trim()) ?? []
-
-    const pgMeta: PostgresMeta = new PostgresMeta({ ...DEFAULT_POOL_CONFIG, connectionString })
+    const pgMeta: PostgresMeta = new PostgresMeta(config)
     const { data: generatorMeta, error: generatorMetaError } = await getGeneratorMetadata(pgMeta, {
       includedSchemas,
       excludedSchemas,

src/server/server.ts

@@ -18,6 +18,7 @@ import {
 import { apply as applyTypescriptTemplate } from './templates/typescript.js'
 import { apply as applyGoTemplate } from './templates/go.js'
 import { apply as applySwiftTemplate } from './templates/swift.js'
+import { apply as applyPythonTemplate } from './templates/python.js'
 
 const logger = pino({
   formatters: {
@@ -143,6 +144,8 @@ async function getTypeOutput(): Promise<string | null> {
       })
     case 'go':
       return applyGoTemplate(config)
+    case 'python':
+      return applyPythonTemplate(config)
     default:
       throw new Error(`Unsupported language for GENERATE_TYPES: ${GENERATE_TYPES}`)
   }