Support preview mode on the root API endpoint

chaynHQ · Oct 27, 2020 · ee43c26 · ee43c26
1 parent a6cdab5
commit ee43c26
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/app.controller.ts b/src/app.controller.ts
@@ -1,15 +1,22 @@
-import { Controller, Get, HttpCode, Inject } from '@nestjs/common';
+import { Controller, Get, Inject, UseGuards } from '@nestjs/common';
 import StoryblokClient from 'storyblok-js-client';
+import { PreviewMode } from './preview-mode/preview-mode.decorator';
+import { PreviewModeGuard } from './preview-mode/preview-mode.guard';
 import { STORYBLOK_CLIENT } from './storyblok/storyblok-factory';
 
 @Controller()
 export class AppController {
   constructor(@Inject(STORYBLOK_CLIENT) private storyblok: StoryblokClient) {}
 
   @Get('/')
-  @HttpCode(204)
-  root(): void {
+  @UseGuards(PreviewModeGuard)
+  root(@PreviewMode() previewMode: boolean): any {
     // Root level API endpoint that we can use for things like uptime checks
+    if (previewMode) {
+      return { previewMode };
+    } else {
+      return {};
+    }
   }
 
   @Get('flush_cache')