@aws/neptune-for-graphql/templates/Lambda4AppSyncHTTP/index.mjs

CLI utility to create and maintain a GraphQL API for Amazon Neptune

import axios from "axios";
import * as rax from 'retry-axios';
import { aws4Interceptor } from "aws4-axios";
import { initSchema, resolveGraphDBQueryFromAppSyncEvent } from './output.resolver.graphql.js';
import { queryNeptune } from "./queryHttpNeptune.mjs";
import { decompressGzipToString, injectAwsScalarDefinitions } from './util.mjs';

const LOGGING_ENABLED = process.env.LOGGING_ENABLED;

const {
    AWS_ACCESS_KEY_ID,
    AWS_SECRET_ACCESS_KEY,
    AWS_SESSION_TOKEN,
    AWS_REGION
} = process.env;

if (process.env.NEPTUNE_IAM_AUTH_ENABLED === 'true') {
    let serviceName;
    if (process.env.NEPTUNE_TYPE) {
        serviceName = process.env.NEPTUNE_TYPE;
    } else {
        console.log('NEPTUNE_TYPE environment variable is not set - defaulting to neptune-db');
        serviceName = 'neptune-db';
    }
    const interceptor = aws4Interceptor({
        options: {
            region: AWS_REGION,
            service: serviceName,
        },
        credentials: {
            accessKeyId: AWS_ACCESS_KEY_ID,
            secretAccessKey: AWS_SECRET_ACCESS_KEY,
            sessionToken: AWS_SESSION_TOKEN
        }
    });

    axios.interceptors.request.use(interceptor);
}

rax.attach();

const schemaDataModelJSON = await decompressGzipToString('output.resolver.schema.json.gz');
const schemaModel = JSON.parse(schemaDataModelJSON);
injectAwsScalarDefinitions(schemaModel);
initSchema(schemaModel);

export const handler = async (event) => {
    if (LOGGING_ENABLED) console.log("Event: ", event);
    if (event.selectionSetGraphQL.includes('...')) {
        throw new Error('Fragments are not supported');
    }
    try {
        // Create Neptune query from GraphQL query
        const resolver = resolveGraphDBQueryFromAppSyncEvent(event);
        if (LOGGING_ENABLED) console.log("Resolver: ", resolver);
        return queryNeptune(`https://${process.env.NEPTUNE_HOST}:${process.env.NEPTUNE_PORT}`, resolver, {loggingEnabled: LOGGING_ENABLED})
    } catch (err) {
        console.error(err);
        throw err;
    }

};