set COI headers based on vscode-coi query (#34)

microsoft · Jul 20, 2022 · cd5eb1c · cd5eb1c
1 parent bc48dc9
commit cd5eb1c
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/src/server/app.ts b/src/server/app.ts
@@ -44,6 +44,20 @@ export default async function createApp(config: IConfig): Promise<Koa> {
 		return next();
 	});
 
+	// COI
+	app.use((ctx, next) => {
+		const value = ctx.query['vscode-coi'];
+		if (value === '1') {
+			ctx.set('Cross-Origin-Opener-Policy', 'same-origin');
+		} else if (value === '2') {
+			ctx.set('Cross-Origin-Embedder-Policy', 'require-corp');
+		} else if (value === '3' || value === '') {
+			ctx.set('Cross-Origin-Opener-Policy', 'same-origin');
+			ctx.set('Cross-Origin-Embedder-Policy', 'require-corp');
+		}
+		return next()
+	})
+
 	// shift the line numbers of source maps in extensions by 2 as the content is wrapped by an anonymous function
 	app.use(async (ctx, next) => {
 		await next();
@@ -84,4 +98,3 @@ export default async function createApp(config: IConfig): Promise<Koa> {
 
 	return app;
 }
-