Dodao hvatanje notifikacija o treniranoj epohi / procesovanom datasetu na frontendu.

3 files changed, 32 insertions, 29 deletions

diff --git a/frontend/src/app/_elements/notifications/notifications.component.html b/frontend/src/app/_elements/notifications/notifications.component.html
index 0b87e4fc..4e33f77c 100644
--- a/frontend/src/app/_elements/notifications/notifications.component.html
+++ b/frontend/src/app/_elements/notifications/notifications.component.html
@@ -1,6 +1,5 @@
-<div *ngIf="notifications" class="position-fixed card card-body p-1 m-3" style="bottom: 0; right: 0; width: 18rem;">
-    <h2 class="m-auto" (click)="closed = !closed;" data-bs-toggle="collapse" href="#collapseNotifs" role="button"
-        aria-expanded="true" aria-controls="collapseNotifs">Notifikacije
+<div *ngIf="notifications && notifications.length > 0" class="position-fixed card card-body p-1 m-3" style="bottom: 0; right: 0; width: 18rem;">
+    <h2 class="m-auto" (click)="closed = !closed;" data-bs-toggle="collapse" href="#collapseNotifs" role="button" aria-expanded="true" aria-controls="collapseNotifs">Notifikacije
         <button class="border-0 bg-white">
             <mat-icon class="position-absolute" style="top: 8px; right: 12px;">{{closed ? 'keyboard_arrow_up' :
                 'keyboard_arrow_down'}}</mat-icon>
diff --git a/frontend/src/app/_elements/notifications/notifications.component.ts b/frontend/src/app/_elements/notifications/notifications.component.ts
index 5863f669..cad8a95f 100644
--- a/frontend/src/app/_elements/notifications/notifications.component.ts
+++ b/frontend/src/app/_elements/notifications/notifications.component.ts
@@ -9,19 +9,25 @@ import Notification from 'src/app/_data/Notification';
 })
 export class NotificationsComponent implements OnInit {
 
-  notifications?: Notification[];
+  notifications: Notification[] = [];
   closed: boolean = false;
 
-  constructor(private signalRService:SignalRService) {
-    this.notifications = [
-      new Notification("Titanik (Preziveli)", "79768456867", 0.2),
-      new Notification("Test Prediktor 1", "56758768678", 0.4),
-      new Notification("Test Prediktor 2", "11344556425", 0.7)
-    ]
+  constructor(private signalRService: SignalRService) {
+
   }
 
   ngOnInit(): void {
-    // this.wsService.send('test');
+    if (this.signalRService.hubConnection) {
+      this.signalRService.hubConnection.on("NotifyDataset", (message: string) => {
+        this.notifications.push(new Notification(message, "datasetIDOvde!!!", 1.0));
+      });
+
+      this.signalRService.hubConnection.on("NotifyEpoch", (message: string) => {
+        this.notifications.push(new Notification(message, "predictorIDOvde!!!", 0.5 /*(epoch / model.epochs)*/));
+      });
+    } else {
+      console.warn("Notifications: No connection!");
+    }
   }
 
 }
diff --git a/frontend/src/app/_services/signal-r.service.ts b/frontend/src/app/_services/signal-r.service.ts
index 5eca48d3..2a6e5d78 100644
--- a/frontend/src/app/_services/signal-r.service.ts
+++ b/frontend/src/app/_services/signal-r.service.ts
@@ -6,25 +6,23 @@ import { CookieService } from 'ngx-cookie-service';
   providedIn: 'root'
 })
 export class SignalRService {
-private hubConnection?:signalR.HubConnection;
-public startConnection=()=>{
+  public hubConnection?: signalR.HubConnection;
+  public startConnection = () => {
 
-  this.hubConnection= new signalR.HubConnectionBuilder()
-            .withUrl('http://localhost:5283/chatHub', {
-                accessTokenFactory: () => this.cookie.get("token"),
-                withCredentials: false
-            }).build();
+    this.hubConnection = new signalR.HubConnectionBuilder()
+      .withUrl('http://localhost:5283/chatHub', {
+        accessTokenFactory: () => this.cookie.get("token"),
+        withCredentials: false
+      }).build();
 
-  this.hubConnection.on("Notify",(message:string) =>{
-    console.log(" "+message);
-  });
+    this.hubConnection.on("Notify", (message: string) => {
+      console.log(" " + message);
+    });
 
-
-
-  this.hubConnection
-    .start()
-    .then(()=>console.log("con Started"))
-    .catch(err=>console.log("Error"+err))
-}
-  constructor(private cookie:CookieService) { }
+    this.hubConnection
+      .start()
+      .then(() => console.log("con Started"))
+      .catch(err => console.log("Error" + err))
+  }
+  constructor(private cookie: CookieService) { }
 }