Support base paths in api specs

Author: Victor Moreno

src/bin.ts

@@ -27,6 +27,11 @@ async function main(): Promise<void> {
             type: 'string',
             default: 'SimpleRest',
             choices: ['SimpleRest'] as const
+          })
+          .option('base-path', {
+            describe: 'The base path of your API. Include the path portion of the url only. ' +
+            'If your API spec has a `servers` entry, this parameter must align with one of the servers in the api spec.',
+            type: 'string'
           });
       })
       .command('generate-policies', 'Generate policies for a Cedar schema', (yargs) => {
@@ -47,6 +52,7 @@ async function main(): Promise<void> {
       const apiSpecFile = argv['api-spec'] as string;
       const namespace = argv.namespace as string;
       const mappingType = argv['mapping-type'] as MappingType;
+      const serverBasePath = typeof argv['base-path'] === 'string' ? argv['base-path'] : undefined;
 
       // Check if API spec file exists
       if (!fs.existsSync(apiSpecFile)) {
@@ -63,11 +69,12 @@ async function main(): Promise<void> {
         process.exit(1);
       }
 
-      const authMapping = Tools.generateApiMappingSchemaFromOpenAPISpec(
+      const authMapping = Tools.generateApiMappingSchemaFromOpenAPISpec({
         openApiSpec,
         namespace,
-        mappingType
-      );
+        mappingType,
+        serverBasePath,
+    });
 
       const fileNames = ['v2.cedarschema.json', 'v4.cedarschema.json'];
       console.log(`Cedar schema successfully generated. Your schema files are named: ${fileNames.join(', ')}.`);

src/tools.ts

@@ -15,13 +15,25 @@ export interface SimpleRestAuthMapping {
     schemaV4: string;
 }
 
+export interface GenerageSchemaFromOpenApiSpecOptions {
+    openApiSpec: OpenAPIV3.Document;
+    namespace: string;
+    mappingType: MappingType;
+    serverBasePath?: string;
+}
+
 export class Tools {
     private static openAPIToCedarPrimitiveTypeMap = {
         string: {type: 'String' as const},
         number: {type: 'Long' as const},
         integer: {type: 'Long' as const},
         boolean: {type: 'Boolean' as const},
     }
+    private static sanitizePath(pathStr: string): string {
+        const trimmed = pathStr.split('/')
+            .filter(segment => segment !== '');
+        return `/${trimmed.join('/')}`;
+    }
     /**
      * How action names are computed:
      *  - If your API spec has operation id's then those are used as cedar actions
@@ -32,14 +44,42 @@ export class Tools {
      * @param namespace cedar namespace for your application
      * @returns 
      */
-    public static generateApiMappingSchemaFromOpenAPISpec(openApiSpec: OpenAPIV3.Document, namespace: string, mappingType: MappingType): AuthMapping {
+    public static generateApiMappingSchemaFromOpenAPISpec(options: GenerageSchemaFromOpenApiSpecOptions): AuthMapping {
+        const {openApiSpec, namespace, mappingType} = options;
         if (!openApiSpec.paths) {
             throw new Error('Invalid OpenAPI spec - missing paths object');
         }
 
         if (!namespace) {
             throw new Error('Invalid input - missing namespace');
         }
+        const servers = openApiSpec.servers;
+
+        if (options.serverBasePath && Array.isArray(servers)) {
+            const basePathExistsInServersArray = servers
+                .map(server => server.url || '')
+                .some(serverUrl => {
+                    const normalizedBasePath = this.sanitizePath(options.serverBasePath || '');
+                    return serverUrl.endsWith(normalizedBasePath) || serverUrl.endsWith(`${normalizedBasePath}/`)
+                });
+            if (!basePathExistsInServersArray) {
+                throw new Error('Base Path option was provided but it does not match any of the `servers` entries in the API spec.');
+            }
+        }
+
+        let basePath = '';
+        if (Array.isArray(servers)) {
+            if (servers.length > 1) {
+                if (!options.serverBasePath) {
+                    throw new Error('Invalid input. API spec specifies more than one `server` entry. Server Base Path parameter required for disambiguation.');
+                }
+                basePath = this.sanitizePath(options.serverBasePath);
+            } else if (servers.length === 1) {
+                const fullBaseUrl = new URL(servers[0].url);
+                basePath = this.sanitizePath(fullBaseUrl.pathname);
+            }
+        }
+        
 
         const RESERVED_WORDS = ['if', 'in', 'is', '__cedar']; 
         const schemaNamespaceRegex = /^[_a-zA-Z][_a-zA-Z0-9]*(?:::(?:[_a-zA-Z][_a-zA-Z0-9]*))*$/;
@@ -100,6 +140,7 @@ export class Tools {
                 if (!operationObject) {
                     continue;
                 }
+                const httpPathTemplateWithBasePath = `${basePath}${httpPathTemplate}`;
                 const {actionName, actionDefinition} = Tools.generateActionDefinitionFromOperationObject(
                     httpVerb,
                     httpPathTemplate,
@@ -112,7 +153,7 @@ export class Tools {
                         ...actionDefinition,
                         annotations: {
                             httpVerb,
-                            httpPathTemplate,
+                            httpPathTemplate: httpPathTemplateWithBasePath,
                         }
                     },
 
@@ -231,6 +272,8 @@ export class Tools {
                 cedarExtension.appliesToResourceTypes.every((value) => typeof value === 'string');
             if (isValidValue) {
                 resourceTypes = cedarExtension.appliesToResourceTypes;
+            } else {
+                throw new Error(`Invalid x-cedar extension in operation definition for ${httpVerb} ${httpPathTemplate}`);
             }
         }
         let attributes = {};

tests/schemaGenPropTests.test.ts

@@ -122,7 +122,11 @@ describe('schema generation proptests', () => {
                             };
                         }
                     }
-                    const generatedCedarSchemaStr = Tools.generateApiMappingSchemaFromOpenAPISpec(openApiSpec, 'NS', 'SimpleRest').schemaV4;
+                    const generatedCedarSchemaStr = Tools.generateApiMappingSchemaFromOpenAPISpec({
+                        openApiSpec,
+                        namespace: 'NS',
+                        mappingType: 'SimpleRest',
+                    }).schemaV4;
                     const generatedCedarSchema = JSON.parse(generatedCedarSchemaStr);
                     expect(Object.keys(generatedCedarSchema['NS'].actions).length).to.equal(numActionsInApiSpec);
                 }
@@ -167,7 +171,11 @@ describe('schema generation proptests', () => {
                         }
                     } as OpenAPIV3.Document;
 
-                    const generatedCedarSchemaStr = Tools.generateApiMappingSchemaFromOpenAPISpec(openApiSpec, 'NS', 'SimpleRest').schemaV4;
+                    const generatedCedarSchemaStr = Tools.generateApiMappingSchemaFromOpenAPISpec({
+                        openApiSpec,
+                        namespace: 'NS',
+                        mappingType: 'SimpleRest',
+                    }).schemaV4;
                     const generatedCedarSchema = JSON.parse(generatedCedarSchemaStr);
 
                     expect(generatedCedarSchema.NS.commonTypes).toBeDefined();