1 files changed, 37 insertions, 2 deletions
diff --git a/frontend/src/app/_modals/save-experiment-dialog/save-experiment-dialog.component.ts b/frontend/src/app/_modals/save-experiment-dialog/save-experiment-dialog.component.ts
index ca01f57e..3c19e732 100644
--- a/frontend/src/app/_modals/save-experiment-dialog/save-experiment-dialog.component.ts
+++ b/frontend/src/app/_modals/save-experiment-dialog/save-experiment-dialog.component.ts
@@ -1,5 +1,12 @@
 import { Component, OnInit } from '@angular/core';
-import { MatDialogRef } from '@angular/material/dialog';
+import { MatDialogRef, MAT_DIALOG_DATA } from '@angular/material/dialog';
+import Experiment from 'src/app/_data/Experiment';
+import { ExperimentsService } from 'src/app/_services/experiments.service';
+import { Inject} from '@angular/core';
+
+interface DialogData {
+  experiment: Experiment;
+}
 
 @Component({
   selector: 'app-save-experiment-dialog',
@@ -9,8 +16,13 @@ import { MatDialogRef } from '@angular/material/dialog';
 export class SaveExperimentDialogComponent implements OnInit {
 
   selectedName: string = '';
+  wrongAlreadyExists: boolean = false;
+  wrongEmptyName: boolean = false;
 
-  constructor(public dialogRef: MatDialogRef<SaveExperimentDialogComponent>) { }
+  constructor(public dialogRef: MatDialogRef<SaveExperimentDialogComponent>, @Inject(MAT_DIALOG_DATA) public data: DialogData, private experimentService: ExperimentsService) { 
+    this.wrongAlreadyExists = false;
+    this.wrongEmptyName = false;
+  }
 
   ngOnInit(): void {
   }
@@ -18,4 +30,27 @@ export class SaveExperimentDialogComponent implements OnInit {
   onNoClick() {
     this.dialogRef.close();
   }
+
+  saveWithEnterKey(keyboardEvent: KeyboardEvent) {
+    if (keyboardEvent.code == "Enter" || keyboardEvent.code == "NumpadEnter") 
+      this.onYesClick();
+  }
+  onYesClick() {
+    if (this.selectedName == "") {
+      this.wrongEmptyName = true;
+      return;
+    }
+    this.wrongEmptyName = false;
+    
+    this.data.experiment.name = this.selectedName;
+    this.experimentService.addExperiment(this.data.experiment).subscribe((response) => {
+      this.wrongAlreadyExists = false;
+      this.data.experiment = response;
+      this.dialogRef.close(this.data.experiment);
+    }, (error) => {
+      if (error.error == "Experiment with this name exists") {
+        this.wrongAlreadyExists = true;
+      }
+    });
+  }
 }