-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'main' into kc/nested-filtering-and-sorting
- Loading branch information
Showing
8 changed files
with
343 additions
and
187 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,55 +1,142 @@ | ||
| import { CosmosClient, Database, Container, SqlQuerySpec } from "@azure/cosmos" | ||
| import * as sdk from '@hasura/ndc-sdk-typescript' | ||
| import { CosmosClient, Database, Container, SqlQuerySpec } from "@azure/cosmos"; | ||
| import { DefaultAzureCredential } from "@azure/identity"; | ||
| import { throwError } from "../../utils"; | ||
|
|
||
| export type ManagedIdentityConfig = { | ||
| type: "ManagedIdentity"; | ||
| // Name of the ENV var where the key can be found | ||
| fromEnvVar: string; | ||
| }; | ||
|
|
||
| export type CosmosKeyConfig = { | ||
| type: "Key"; | ||
| // Name of the ENV var where the key can be found | ||
| fromEnvVar: string; | ||
| }; | ||
|
|
||
| export type AzureCosmosAuthenticationConfig = | ||
| | CosmosKeyConfig | ||
| | ManagedIdentityConfig; | ||
|
|
||
| export type RawCosmosDbConfig = { | ||
| databaseName: string, | ||
| endpoint: string, | ||
| key: string | ||
| } | ||
| databaseName: string; | ||
| endpoint: string; | ||
| connectionConfig: AzureCosmosAuthenticationConfig; | ||
| }; | ||
|
|
||
| /* Creates a new cosmos DB client with which the specified database can be queried. */ | ||
| function getCosmosDbClient(rawDbConfig: RawCosmosDbConfig): Database { | ||
| const dbClient = new CosmosClient({ | ||
| key: rawDbConfig.key, | ||
| endpoint: rawDbConfig.endpoint, | ||
| }); | ||
| export function getCosmosDbClient( | ||
| endpoint: string, | ||
| databaseName: string, | ||
| connectionConfig: AzureCosmosAuthenticationConfig, | ||
| ): Database { | ||
| let dbClient: CosmosClient; | ||
| switch (connectionConfig.type) { | ||
| case "Key": | ||
| const key = | ||
| getEnvVariable(connectionConfig.fromEnvVar, true) ?? | ||
| throwError( | ||
| `Azure Cosmos Key not found in the env var "${connectionConfig.fromEnvVar}"`, | ||
| ); | ||
| dbClient = new CosmosClient({ | ||
| key, | ||
| endpoint, | ||
| }); | ||
| break; | ||
| case "ManagedIdentity": | ||
| const managedIdentityClientId = | ||
| getEnvVariable(connectionConfig.fromEnvVar, true) ?? | ||
| throwError( | ||
| `Azure Cosmos Key not found in the env var "${connectionConfig.fromEnvVar}"`, | ||
| ); | ||
|
|
||
| return dbClient.database(rawDbConfig.databaseName); | ||
| let credentials = new DefaultAzureCredential({ | ||
| managedIdentityClientId, | ||
| }); | ||
| dbClient = new CosmosClient({ | ||
| endpoint, | ||
| aadCredentials: credentials, | ||
| }); | ||
| break; | ||
| } | ||
|
|
||
| return dbClient.database(databaseName); | ||
| } | ||
|
|
||
| function getEnvVariable(envVarName: string): string { | ||
| const envVariable = process.env[envVarName]; | ||
| if (!envVariable) { | ||
| throw new Error(`${envVarName} environment variable is not defined.`); | ||
| function getEnvVariable( | ||
| envVarName: string, | ||
| isRequired?: undefined | boolean, | ||
| ): string | null { | ||
| const envVariable = process.env[envVarName]; | ||
| if (!envVariable) { | ||
| if (isRequired) { | ||
| throw new Error(`${envVarName} environment variable is not defined.`); | ||
| } else { | ||
| return null; | ||
| } | ||
| return envVariable; | ||
| } | ||
| return envVariable; | ||
| } | ||
|
|
||
| export function constructCosmosDbClient() { | ||
| const key = getEnvVariable("AZURE_COSMOS_KEY"); | ||
| const endpoint = getEnvVariable("AZURE_COSMOS_ENDPOINT"); | ||
| const databaseName = getEnvVariable("AZURE_COSMOS_DB_NAME"); | ||
|
|
||
| const dbClient = getCosmosDbClient({ | ||
| databaseName, endpoint, key | ||
| }); | ||
| function getConnectionConfig(): AzureCosmosAuthenticationConfig | null { | ||
| const key = getEnvVariable("AZURE_COSMOS_KEY"); | ||
| const managed_identity_client_id = getEnvVariable( | ||
| "AZURE_COSMOS_MANAGED_CLIENT_ID", | ||
| ); | ||
|
|
||
| const connectionDetails = { | ||
| endpoint, | ||
| key, | ||
| databaseName | ||
| if (key === null && managed_identity_client_id === null) { | ||
| throw new Error( | ||
| `Either the AZURE_COSMOS_KEY or the AZURE_COSMOS_MANAGED_CLIENT_ID env var is expected`, | ||
| ); | ||
| } else if (key && managed_identity_client_id) { | ||
| throw new Error( | ||
| `Both AZURE_COSMOS_KEY and the AZURE_COSMOS_MANAGED_CLIENT_ID cannot be set`, | ||
| ); | ||
| } else { | ||
| if (key) { | ||
| return { | ||
| type: "Key", | ||
| fromEnvVar: "AZURE_COSMOS_KEY", | ||
| }; | ||
| } else if (managed_identity_client_id) { | ||
| return { | ||
| type: "ManagedIdentity", | ||
| fromEnvVar: "AZURE_COSMOS_MANAGED_CLIENT_ID", | ||
| }; | ||
| } | ||
| } | ||
| return null; | ||
| } | ||
|
|
||
| return { | ||
| connectionDetails, | ||
| dbClient | ||
| } | ||
| export function constructCosmosDbClient() { | ||
| const endpoint = | ||
| getEnvVariable("AZURE_COSMOS_ENDPOINT", true) ?? | ||
| throwError("AZURE_COSMOS_ENDPOINT not found"); | ||
| const databaseName = | ||
| getEnvVariable("AZURE_COSMOS_DB_NAME", true) ?? | ||
| throwError("AZURE_COSMOS_DB_NAME not found"); | ||
| const connectionConfig = | ||
| getConnectionConfig() ?? | ||
| throwError("internal: could not get the connection config"); | ||
|
|
||
| } | ||
| const dbClient = getCosmosDbClient(endpoint, databaseName, connectionConfig); | ||
|
|
||
| const connectionDetails = { | ||
| endpoint, | ||
| databaseName, | ||
| connectionConfig, | ||
| }; | ||
|
|
||
| return { | ||
| connectionDetails, | ||
| dbClient, | ||
| }; | ||
| } | ||
|
|
||
| /* Runs the `sqlQuerySpec` in the specified `container` */ | ||
| export async function runSQLQuery<T>(sqlQuerySpec: SqlQuerySpec, container: Container): Promise<T[]> { | ||
| return (await container.items.query(sqlQuerySpec).fetchAll()).resources | ||
| export async function runSQLQuery<T>( | ||
| sqlQuerySpec: SqlQuerySpec, | ||
| container: Container, | ||
| ): Promise<T[]> { | ||
| return (await container.items.query(sqlQuerySpec).fetchAll()).resources; | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.