Improve video upload guard a little bit

1 files changed, 8 insertions, 5 deletions

diff --git a/client/src/app/shared/can-deactivate-guard.service.ts b/client/src/app/shared/guards/can-deactivate-guard.service.ts
index 4239df737..15618f699 100644
--- a/client/src/app/shared/can-deactivate-guard.service.ts
+++ b/client/src/app/shared/guards/can-deactivate-guard.service.ts
@@ -1,10 +1,10 @@
 import { Injectable } from '@angular/core'
 import { CanDeactivate, ActivatedRouteSnapshot, RouterStateSnapshot } from '@angular/router'
 import { Observable } from 'rxjs/Observable'
-import { ConfirmService } from '../core'
+import { ConfirmService } from '../../core/index'
 
 export interface CanComponentDeactivate {
-  canDeactivate: () => Observable<boolean> | boolean
+  canDeactivate: () => { text?: string, canDeactivate: Observable<boolean> | boolean }
 }
 
 @Injectable()
@@ -16,9 +16,12 @@ export class CanDeactivateGuard implements CanDeactivate<CanComponentDeactivate>
     currentState: RouterStateSnapshot,
     nextState: RouterStateSnapshot
   ): Observable<boolean> | boolean {
-    return component.canDeactivate() || this.confirmService.confirm(
-      'All unsaved data will be lost, are you sure you want to leave ?',
-      'Unsaved Data'
+    const result = component.canDeactivate()
+    const text = result.text || 'All unsaved data will be lost, are you sure you want to leave this page?'
+
+    return result.canDeactivate || this.confirmService.confirm(
+      text,
+      'Warning'
     )
   }