1 files changed, 6 insertions, 5 deletions
diff --git a/Timeline/ClientApp/src/app/user/user-login-success/user-login-success.component.ts b/Timeline/ClientApp/src/app/user/user-login-success/user-login-success.component.ts
index 99de5970..d141b3b6 100644
--- a/Timeline/ClientApp/src/app/user/user-login-success/user-login-success.component.ts
+++ b/Timeline/ClientApp/src/app/user/user-login-success/user-login-success.component.ts
@@ -1,5 +1,7 @@
 import { Component, OnInit, Input } from '@angular/core';
-import { UserInfo } from '../user-info';
+import { UserInfo } from '../entities';
+import { UserService } from '../user-service/user.service';
+import { ActivatedRoute } from '@angular/router';
 
 @Component({
   selector: 'app-user-login-success',
@@ -8,15 +10,14 @@ import { UserInfo } from '../user-info';
 })
 export class UserLoginSuccessComponent implements OnInit {
 
-  @Input()
   displayLoginSuccessMessage = false;
 
-  @Input()
   userInfo: UserInfo;
 
-  constructor() { }
+  constructor(private route: ActivatedRoute, private userService: UserService) { }
 
   ngOnInit() {
+    this.userInfo = this.userService.userInfo;
+    this.displayLoginSuccessMessage = this.route.snapshot.paramMap.get('reason') === 'login';
   }
-
 }