From 9835c106d8e4129dc5d96c4fac1765832f1c757e Mon Sep 17 00:00:00 2001
From: piotrpekala7 <31202938+piotrpekala7@users.noreply.github.com>
Date: Thu, 29 Apr 2021 18:15:22 +0200
Subject: [PATCH] Login guard added

---
 src/app/app-routing.module.ts                 | 71 ++++++++++++-------
 src/app/app.module.ts                         |  2 +
 src/app/components/login/login.component.html |  3 +-
 src/app/components/login/login.component.ts   |  9 ++-
 src/app/guards/login-guard.ts                 | 19 +++++
 5 files changed, 76 insertions(+), 28 deletions(-)

diff --git a/src/app/app-routing.module.ts b/src/app/app-routing.module.ts
index eeba6e83..7edf8dc4 100644
--- a/src/app/app-routing.module.ts
+++ b/src/app/app-routing.module.ts
@@ -53,6 +53,7 @@ import { ListOfSnapshotsComponent } from './components/snapshots/list-of-snapsho
 import { SystemStatusComponent } from './components/system-status/system-status.component';
 import { WebConsoleFullWindowComponent } from './components/web-console-full-window/web-console-full-window.component';
 import { ConsoleGuard } from './guards/console-guard';
+import { LoginGuard } from './guards/login-guard';
 import { DefaultLayoutComponent } from './layouts/default-layout/default-layout.component';
 import { ServerResolve } from './resolvers/server-resolve';
 
@@ -68,6 +69,7 @@ const routes: Routes = [
       {
         path: 'server/:server_id/projects',
         component: ProjectsComponent,
+        canActivate: [LoginGuard],
         resolve: { server: ServerResolve },
       },
       { path: 'help', component: HelpComponent },
@@ -75,120 +77,137 @@ const routes: Routes = [
       { path: 'settings', component: SettingsComponent },
       { path: 'settings/console', component: ConsoleComponent },
       { path: 'installed-software', component: InstalledSoftwareComponent },
-      { path: 'server/:server_id/systemstatus', component: SystemStatusComponent },
+      { path: 'server/:server_id/systemstatus', component: SystemStatusComponent, canActivate: [LoginGuard] },
 
-      { path: 'server/:server_ip/:server_port/project/:project_id', component: DirectLinkComponent },
+      { path: 'server/:server_ip/:server_port/project/:project_id', component: DirectLinkComponent, canActivate: [LoginGuard] },
       {
         path: 'server/:server_id/project/:project_id/snapshots',
         component: ListOfSnapshotsComponent,
+        canActivate: [LoginGuard],
         resolve: { server: ServerResolve },
       },
-      { path: 'server/:server_id/preferences', component: PreferencesComponent },
-      { path: 'server/:server_id/preferences/gns3vm', component: Gns3vmComponent },
+      { path: 'server/:server_id/preferences', component: PreferencesComponent, canActivate: [LoginGuard] },
+      { path: 'server/:server_id/preferences/gns3vm', component: Gns3vmComponent, canActivate: [LoginGuard] },
       // { path: 'server/:server_id/preferences/general', component: GeneralPreferencesComponent },
-      { path: 'server/:server_id/preferences/builtin', component: BuiltInPreferencesComponent },
+      { path: 'server/:server_id/preferences/builtin', component: BuiltInPreferencesComponent, canActivate: [LoginGuard] },
 
-      { path: 'server/:server_id/preferences/builtin/ethernet-hubs', component: EthernetHubsTemplatesComponent },
+      { path: 'server/:server_id/preferences/builtin/ethernet-hubs', component: EthernetHubsTemplatesComponent, canActivate: [LoginGuard] },
       {
         path: 'server/:server_id/preferences/builtin/ethernet-hubs/addtemplate',
         component: EthernetHubsAddTemplateComponent,
+        canActivate: [LoginGuard]
       },
       {
         path: 'server/:server_id/preferences/builtin/ethernet-hubs/:template_id',
         component: EthernetHubsTemplateDetailsComponent,
+        canActivate: [LoginGuard]
       },
 
       {
         path: 'server/:server_id/preferences/builtin/ethernet-switches',
         component: EthernetSwitchesTemplatesComponent,
+        canActivate: [LoginGuard]
       },
       {
         path: 'server/:server_id/preferences/builtin/ethernet-switches/addtemplate',
         component: EthernetSwitchesAddTemplateComponent,
+        canActivate: [LoginGuard]
       },
       {
         path: 'server/:server_id/preferences/builtin/ethernet-switches/:template_id',
         component: EthernetSwitchesTemplateDetailsComponent,
+        canActivate: [LoginGuard]
       },
 
-      { path: 'server/:server_id/preferences/builtin/cloud-nodes', component: CloudNodesTemplatesComponent },
+      { path: 'server/:server_id/preferences/builtin/cloud-nodes', component: CloudNodesTemplatesComponent, canActivate: [LoginGuard] },
       {
         path: 'server/:server_id/preferences/builtin/cloud-nodes/addtemplate',
         component: CloudNodesAddTemplateComponent,
+        canActivate: [LoginGuard]
       },
       {
         path: 'server/:server_id/preferences/builtin/cloud-nodes/:template_id',
         component: CloudNodesTemplateDetailsComponent,
+        canActivate: [LoginGuard]
       },
 
       //{ path: 'server/:server_id/preferences/dynamips', component: DynamipsPreferencesComponent },
-      { path: 'server/:server_id/preferences/dynamips/templates', component: IosTemplatesComponent },
-      { path: 'server/:server_id/preferences/dynamips/templates/addtemplate', component: AddIosTemplateComponent },
-      { path: 'server/:server_id/preferences/dynamips/templates/:template_id', component: IosTemplateDetailsComponent },
+      { path: 'server/:server_id/preferences/dynamips/templates', component: IosTemplatesComponent, canActivate: [LoginGuard] },
+      { path: 'server/:server_id/preferences/dynamips/templates/addtemplate', component: AddIosTemplateComponent, canActivate: [LoginGuard] },
+      { path: 'server/:server_id/preferences/dynamips/templates/:template_id', component: IosTemplateDetailsComponent, canActivate: [LoginGuard] },
       {
         path: 'server/:server_id/preferences/dynamips/templates/:template_id/copy',
         component: CopyIosTemplateComponent,
+        canActivate: [LoginGuard]
       },
 
       // { path: 'server/:server_id/preferences/qemu', component: QemuPreferencesComponent },
-      { path: 'server/:server_id/preferences/qemu/templates', component: QemuVmTemplatesComponent },
+      { path: 'server/:server_id/preferences/qemu/templates', component: QemuVmTemplatesComponent, canActivate: [LoginGuard] },
       {
         path: 'server/:server_id/preferences/qemu/templates/:template_id/copy',
         component: CopyQemuVmTemplateComponent,
+        canActivate: [LoginGuard]
       },
-      { path: 'server/:server_id/preferences/qemu/templates/:template_id', component: QemuVmTemplateDetailsComponent },
-      { path: 'server/:server_id/preferences/qemu/addtemplate', component: AddQemuVmTemplateComponent },
+      { path: 'server/:server_id/preferences/qemu/templates/:template_id', component: QemuVmTemplateDetailsComponent, canActivate: [LoginGuard] },
+      { path: 'server/:server_id/preferences/qemu/addtemplate', component: AddQemuVmTemplateComponent, canActivate: [LoginGuard] },
 
       // { path: 'server/:server_id/preferences/vpcs', component: VpcsPreferencesComponent },
-      { path: 'server/:server_id/preferences/vpcs/templates', component: VpcsTemplatesComponent },
-      { path: 'server/:server_id/preferences/vpcs/templates/:template_id', component: VpcsTemplateDetailsComponent },
-      { path: 'server/:server_id/preferences/vpcs/addtemplate', component: AddVpcsTemplateComponent },
+      { path: 'server/:server_id/preferences/vpcs/templates', component: VpcsTemplatesComponent, canActivate: [LoginGuard] },
+      { path: 'server/:server_id/preferences/vpcs/templates/:template_id', component: VpcsTemplateDetailsComponent, canActivate: [LoginGuard] },
+      { path: 'server/:server_id/preferences/vpcs/addtemplate', component: AddVpcsTemplateComponent, canActivate: [LoginGuard] },
 
       // { path: 'server/:server_id/preferences/virtualbox', component: VirtualBoxPreferencesComponent },
-      { path: 'server/:server_id/preferences/virtualbox/templates', component: VirtualBoxTemplatesComponent },
+      { path: 'server/:server_id/preferences/virtualbox/templates', component: VirtualBoxTemplatesComponent, canActivate: [LoginGuard] },
       {
         path: 'server/:server_id/preferences/virtualbox/templates/:template_id',
         component: VirtualBoxTemplateDetailsComponent,
+        canActivate: [LoginGuard]
       },
-      { path: 'server/:server_id/preferences/virtualbox/addtemplate', component: AddVirtualBoxTemplateComponent },
+      { path: 'server/:server_id/preferences/virtualbox/addtemplate', component: AddVirtualBoxTemplateComponent, canActivate: [LoginGuard] },
 
       // { path: 'server/:server_id/preferences/vmware', component: VmwarePreferencesComponent },
-      { path: 'server/:server_id/preferences/vmware/templates', component: VmwareTemplatesComponent },
+      { path: 'server/:server_id/preferences/vmware/templates', component: VmwareTemplatesComponent, canActivate: [LoginGuard] },
       {
         path: 'server/:server_id/preferences/vmware/templates/:template_id',
         component: VmwareTemplateDetailsComponent,
+        canActivate: [LoginGuard]
       },
-      { path: 'server/:server_id/preferences/vmware/addtemplate', component: AddVmwareTemplateComponent },
+      { path: 'server/:server_id/preferences/vmware/addtemplate', component: AddVmwareTemplateComponent, canActivate: [LoginGuard] },
 
-      { path: 'server/:server_id/preferences/docker/templates', component: DockerTemplatesComponent },
+      { path: 'server/:server_id/preferences/docker/templates', component: DockerTemplatesComponent, canActivate: [LoginGuard] },
       {
         path: 'server/:server_id/preferences/docker/templates/:template_id',
         component: DockerTemplateDetailsComponent,
+        canActivate: [LoginGuard]
       },
       {
         path: 'server/:server_id/preferences/docker/templates/:template_id/copy',
         component: CopyDockerTemplateComponent,
+        canActivate: [LoginGuard]
       },
-      { path: 'server/:server_id/preferences/docker/addtemplate', component: AddDockerTemplateComponent },
+      { path: 'server/:server_id/preferences/docker/addtemplate', component: AddDockerTemplateComponent, canActivate: [LoginGuard] },
 
-      { path: 'server/:server_id/preferences/iou/templates', component: IouTemplatesComponent },
-      { path: 'server/:server_id/preferences/iou/templates/:template_id', component: IouTemplateDetailsComponent },
-      { path: 'server/:server_id/preferences/iou/templates/:template_id/copy', component: CopyIouTemplateComponent },
-      { path: 'server/:server_id/preferences/iou/addtemplate', component: AddIouTemplateComponent },
+      { path: 'server/:server_id/preferences/iou/templates', component: IouTemplatesComponent, canActivate: [LoginGuard] },
+      { path: 'server/:server_id/preferences/iou/templates/:template_id', component: IouTemplateDetailsComponent, canActivate: [LoginGuard] },
+      { path: 'server/:server_id/preferences/iou/templates/:template_id/copy', component: CopyIouTemplateComponent, canActivate: [LoginGuard] },
+      { path: 'server/:server_id/preferences/iou/addtemplate', component: AddIouTemplateComponent, canActivate: [LoginGuard] },
     ],
   },
   {
     path: 'server/:server_id/project/:project_id',
     component: ProjectMapComponent,
+    canActivate: [LoginGuard],
     canDeactivate: [ConsoleGuard],
   },
   {
     path: 'server/:server_id/project/:project_id/nodes/:node_id',
     component: WebConsoleFullWindowComponent,
+    canActivate: [LoginGuard]
   },
   {
     path: 'static/web-ui/server/:server_id/project/:project_id/nodes/:node_id',
     component: WebConsoleFullWindowComponent,
+    canActivate: [LoginGuard]
   },
   {
     path: '**',
diff --git a/src/app/app.module.ts b/src/app/app.module.ts
index 9a5b8d01..4c84fae6 100644
--- a/src/app/app.module.ts
+++ b/src/app/app.module.ts
@@ -210,6 +210,7 @@ import { ProjectsFilter } from './filters/projectsFilter.pipe';
 import { SearchFilter } from './filters/searchFilter.pipe';
 import { TemplateFilter } from './filters/templateFilter.pipe';
 import { ConsoleGuard } from './guards/console-guard';
+import { LoginGuard } from './guards/login-guard';
 import { ProjectWebServiceHandler } from './handlers/project-web-service-handler';
 import { DefaultLayoutComponent } from './layouts/default-layout/default-layout.component';
 import { MATERIAL_IMPORTS } from './material.imports';
@@ -548,6 +549,7 @@ import { LoginService } from './services/login.service';
     GoogleAnalyticsService,
     NodeConsoleService,
     ServerResolve,
+    LoginGuard,
     ConsoleGuard,
     Title,
     ApplianceService,
diff --git a/src/app/components/login/login.component.html b/src/app/components/login/login.component.html
index f7953ec0..7ec95108 100644
--- a/src/app/components/login/login.component.html
+++ b/src/app/components/login/login.component.html
@@ -3,7 +3,8 @@
         <mat-card class="matCard">
             <div class="loginTitle">
                 <div class="loginIcon">
-                    <mat-icon class="mat-icon-login" svgIcon="gns3"></mat-icon>
+                    <mat-icon class="mat-icon-login" *ngIf="!isLightThemeEnabled" svgIcon="gns3"></mat-icon>
+                    <mat-icon class="mat-icon-login" *ngIf="isLightThemeEnabled" svgIcon="gns3black"></mat-icon>
                     <div>
                         <h1>GNS3</h1>
                         <h6>v{{version}}</h6>
diff --git a/src/app/components/login/login.component.ts b/src/app/components/login/login.component.ts
index fafc30fa..3a0eac53 100644
--- a/src/app/components/login/login.component.ts
+++ b/src/app/components/login/login.component.ts
@@ -9,6 +9,7 @@ import { ToasterService } from '../../services/toaster.service';
 import { AuthResponse } from '../../models/authResponse';
 import { VersionService } from '../../services/version.service';
 import { Version } from '../../models/version';
+import { ThemeService } from '../../services/theme.service';
 
 @Component({
     selector: 'app-login',
@@ -19,6 +20,7 @@ import { Version } from '../../models/version';
 export class LoginComponent implements OnInit {
     private server: Server;
     public version: string;
+    public isLightThemeEnabled: boolean = false;
 
     loginForm = new FormGroup({
         username: new FormControl('', [Validators.required]),
@@ -32,7 +34,8 @@ export class LoginComponent implements OnInit {
         private route: ActivatedRoute,
         private router: Router,
         private toasterService: ToasterService,
-        private versionService: VersionService
+        private versionService: VersionService,
+        private themeService: ThemeService
     ) {}
 
     async ngOnInit() {
@@ -44,6 +47,10 @@ export class LoginComponent implements OnInit {
                 this.version = version.version;
             });
         });
+
+        this.themeService.getActualTheme() === 'light'
+            ? (this.isLightThemeEnabled = true)
+            : (this.isLightThemeEnabled = false);
     }
 
     public login() {
diff --git a/src/app/guards/login-guard.ts b/src/app/guards/login-guard.ts
index e69de29b..f4182705 100644
--- a/src/app/guards/login-guard.ts
+++ b/src/app/guards/login-guard.ts
@@ -0,0 +1,19 @@
+import { Injectable } from '@angular/core';
+import { ActivatedRouteSnapshot, CanActivate, RouterStateSnapshot } from '@angular/router';
+import { Server } from 'app/models/server';
+import { ServerService } from '../services/server.service';
+
+@Injectable()
+export class LoginGuard implements CanActivate {
+    constructor(
+      private serverService: ServerService
+    ) {}
+
+    canActivate(next: ActivatedRouteSnapshot, state: RouterStateSnapshot) {
+        const server_id = next.paramMap.get('server_id');
+        return this.serverService.get(parseInt(server_id, 10)).then((server: Server) => {
+            if (server.authToken) return true;
+            return false
+        });
+    }
+}