Updated index.mjs file to be modified for Apollo subgraph Federation 2 features

Author: flora-jin

CHANGELOG.md

@@ -116,4 +116,6 @@ permissions and limitations under the License.
   option ([#117](https://github.com/aws/amazon-neptune-for-graphql/pull/117))
 * Fixed invalid schema generation when AWS AppSync scalar types are used in 
   an input 
-  schema ([#118](https://github.com/aws/amazon-neptune-for-graphql/pull/118))
+  schema ([#118](https://github.com/aws/amazon-neptune-for-graphql/pull/118))
+* Updated index.mjs file to be modified for Apollo subgraph Federation 2 
+  features ([#126](https://github.com/aws/amazon-neptune-for-graphql/pull/126))

src/zipPackage.js

@@ -49,7 +49,7 @@ async function createZip({targetZipFilePath, includePaths = [], includeContent =
         output.on('close', () => resolve());
         archive.on('error', err => reject(err));
     });
-    
+
     archive.pipe(output);
 
     includePaths.forEach(includePath => {

templates/ApolloServer/index.mjs

@@ -8,7 +8,15 @@ import dotenv from 'dotenv';
 
 dotenv.config();
 
-const typeDefs = parse(readFileSync('./output.schema.graphql', 'utf-8'));
+const fileContent = readFileSync('./output.schema.graphql', 'utf-8');
+let schema = fileContent;
+if (process.env.SUBGRAPH === 'true') {
+    schema = `extend schema @link(
+        url: "https://specs.apollo.dev/federation/v2.0"
+        import: ["@key", "@shareable"]
+    )${fileContent}`;
+}
+const typeDefs = parse(schema);
 const queryDefinition = typeDefs.definitions.find(
     definition => definition.kind === 'ObjectTypeDefinition' && definition.name.value === 'Query'
 );

test/testLib.js

@@ -226,6 +226,9 @@ async function testApolloArtifacts(outputFolderPath, testDbInfo, subgraph = fals
         ];
         const actualEnvContent = fs.readFileSync(path.join(outputFolderPath, 'unzipped', '.env'), 'utf8');
         expect(actualEnvContent).toEqual(expectedEnvContent.join('\n'));
+
+        const actualIndexContent = fs.readFileSync(path.join(outputFolderPath, 'unzipped', 'index.mjs'), 'utf8');
+        expect(actualIndexContent).toContain('https://specs.apollo.dev/federation/v2.0');
     });
 }