gateway: add gateway-backend plugin

Signed-off-by: Vincenzo Scamporlino <vincenzos@spotify.com>
This commit is contained in:
Vincenzo Scamporlino
2025-03-21 11:39:04 +01:00
parent 64278020b4
commit 918ecfd03b
8 changed files with 342 additions and 0 deletions
+16
View File
@@ -0,0 +1,16 @@
/*
* Copyright 2025 The Backstage Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
export { gatewayPlugin as default } from './plugin';
+137
View File
@@ -0,0 +1,137 @@
/*
* Copyright 2025 The Backstage Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import { createBackend } from '@backstage/backend-defaults';
import { Backend } from '@backstage/backend-app-api';
import { mockServices } from '@backstage/backend-test-utils';
import {
coreServices,
createBackendPlugin,
createServiceFactory,
} from '@backstage/backend-plugin-api';
import { Router } from 'express';
describe('gateway', () => {
let backend: Backend;
let anotherBackend: Backend;
const dummyPlugin = createBackendPlugin({
pluginId: 'dummy',
register(env) {
env.registerInit({
deps: {
httpRouter: coreServices.httpRouter,
},
async init({ httpRouter }) {
const router = Router();
router.get('/foo', (_req, res) => {
res.json({ foo: true });
});
httpRouter.use(router);
},
});
},
});
const discovery = mockServices.discovery.mock();
beforeAll(async () => {
backend = createBackend();
backend.add(mockServices.rootHttpRouter.factory());
backend.add(
mockServices.rootConfig.factory({
data: {
backend: { baseUrl: 'http://localhost:7777', listen: { port: 7777 } },
},
}),
);
backend.add(mockServices.auth.factory());
backend.add(mockServices.httpAuth.factory());
backend.add(
createServiceFactory({
service: coreServices.discovery,
deps: {},
factory: () => discovery,
}),
);
backend.add(dummyPlugin);
backend.add(import('./'));
anotherBackend = createBackend();
anotherBackend.add(mockServices.rootHttpRouter.factory());
anotherBackend.add(
mockServices.rootConfig.factory({
data: {
backend: { baseUrl: 'http://localhost:7778', listen: { port: 7778 } },
},
}),
);
anotherBackend.add(mockServices.auth.factory());
anotherBackend.add(mockServices.httpAuth.factory());
anotherBackend.add(mockServices.discovery.factory());
anotherBackend.add(
createBackendPlugin({
pluginId: 'external-plugin',
register(env) {
env.registerInit({
deps: {
rootHttpRouter: coreServices.rootHttpRouter,
httpRouter: coreServices.httpRouter,
},
async init({ httpRouter }) {
const router = Router();
router.get('/foo', (_req, res) => {
res.json({ bar: true });
});
httpRouter.use(router);
},
});
},
}),
);
await Promise.all([backend.start(), anotherBackend.start()]);
}, 15_000);
afterAll(async () => {
await backend.stop();
await anotherBackend.stop();
});
it('should invoke the endpoint of an installed plugin', async () => {
const response = await fetch('http://localhost:7777/api/dummy/foo');
expect(response.status).toBe(200);
const data = await response.json();
expect(data).toEqual({ foo: true });
});
it('should proxy requests for unknown plugins', async () => {
discovery.getBaseUrl.mockImplementation(async (pluginId: string) => {
if (pluginId === 'external-plugin') {
return 'http://localhost:7778/api/external-plugin';
}
return `http://localhost:7777/api/${pluginId}`;
});
const response = await fetch(
'http://localhost:7777/api/external-plugin/foo',
);
expect(response.status).toBe(200);
const data = await response.json();
expect(data).toEqual({ bar: true });
});
});
+51
View File
@@ -0,0 +1,51 @@
/*
* Copyright 2025 The Backstage Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import {
coreServices,
createBackendPlugin,
} from '@backstage/backend-plugin-api';
import { createRouter } from './router';
import { instanceMetadataServiceRef } from '@backstage/backend-plugin-api/alpha';
import { Handler } from 'express';
/**
* gateway backend plugin
*
* @public
*/
export const gatewayPlugin = createBackendPlugin({
pluginId: 'gateway',
register(env) {
env.registerInit({
deps: {
logger: coreServices.logger,
rootHttpRouter: coreServices.rootHttpRouter,
instanceMeta: instanceMetadataServiceRef,
discovery: coreServices.discovery,
},
async init({ logger, discovery, instanceMeta, rootHttpRouter }) {
rootHttpRouter.use(
'/api/:pluginId',
createRouter({
discovery,
instanceMeta,
logger,
}) as Handler,
);
},
});
},
});
+63
View File
@@ -0,0 +1,63 @@
/*
* Copyright 2025 The Backstage Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import { DiscoveryService, LoggerService } from '@backstage/backend-plugin-api';
import { InstanceMetadataService } from '@backstage/backend-plugin-api/alpha';
import { Request, Response, NextFunction } from 'express';
import { createProxyMiddleware } from 'http-proxy-middleware';
import { context } from '@opentelemetry/api';
import { getRPCMetadata } from '@opentelemetry/core';
export function createRouter({
discovery,
instanceMeta,
}: {
discovery: DiscoveryService;
instanceMeta: InstanceMetadataService;
logger: LoggerService;
}) {
const localPluginIds = new Set(
instanceMeta
.getInstalledFeatures()
.filter(f => f.type === 'plugin')
.map(f => f.pluginId),
);
const proxy = createProxyMiddleware({
changeOrigin: true,
router: async (req: Request<{ pluginId: string }>) => {
const pluginId = req.params.pluginId;
return discovery.getBaseUrl(pluginId);
},
});
return function proxyMiddleware(
req: Request<{ pluginId: string }>,
res: Response,
next: NextFunction,
) {
if (localPluginIds.has(req.params.pluginId)) {
next();
return;
}
const rpcMetadata = getRPCMetadata(context.active());
if (rpcMetadata) {
rpcMetadata.route = req.baseUrl;
}
proxy(req, res, next);
};
}
+16
View File
@@ -0,0 +1,16 @@
/*
* Copyright 2025 The Backstage Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
export {};