Merge pull request #365 from jaulz/patch-1

fix: hide stage in URL in REST scenario

Author: mnapoli

src/constructs/aws/ServerSideWebsite.ts

@@ -128,13 +128,17 @@ export class ServerSideWebsite extends AwsConstruct {
                 ? acm.Certificate.fromCertificateArn(this, "Certificate", configuration.certificate)
                 : undefined;
 
+        // Hide the stage in the URL in REST scenario
+        const originPath = configuration.apiGateway === "rest" ? "/" + (provider.getStage() ?? "") : undefined;
+
         this.distribution = new Distribution(this, "CDN", {
             comment: `${provider.stackName} ${id} website CDN`,
             defaultBehavior: {
                 // Origins are where CloudFront fetches content
                 origin: new HttpOrigin(apiGatewayDomain, {
                     // API Gateway only supports HTTPS
                     protocolPolicy: OriginProtocolPolicy.HTTPS_ONLY,
+                    originPath,
                 }),
                 // For a backend app we all all methods
                 allowedMethods: AllowedMethods.ALLOW_ALL,

src/providers/AwsProvider.ts

@@ -141,6 +141,15 @@ export class AwsProvider implements ProviderInterface {
         return this.serverless.service.provider.vpc ?? null;
     }
 
+    /**
+     * Returns name of the stage.
+     *
+     * @internal
+     */
+    getStage(): string | undefined {
+        return this.serverless.service.provider.stage;
+    }
+
     /**
      * Resolves the value of a CloudFormation stack output.
      */

test/unit/serverSideWebsite.test.ts

@@ -260,6 +260,7 @@ describe("server-side website", () => {
                                     [{ Ref: "ApiGatewayRestApi" }, "execute-api.us-east-1.amazonaws.com"],
                                 ],
                             },
+                            OriginPath: "/dev",
                         },
                     ],
                 },