From 0fa94fc8679b45aa702068ad96097fe18b385a50 Mon Sep 17 00:00:00 2001
From: Nicolai Ort <info@nicolai-ort.com>
Date: Fri, 5 Mar 2021 18:37:51 +0100
Subject: [PATCH] Added MailTypes to responses

ref #5
---
 src/controllers/MailController.ts | 9 +++++----
 src/models/MailTypeEnum.ts        | 3 ++-
 src/models/SuccessResponse.ts     | 7 ++++++-
 3 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/src/controllers/MailController.ts b/src/controllers/MailController.ts
index e0a4a37..69e8d6c 100644
--- a/src/controllers/MailController.ts
+++ b/src/controllers/MailController.ts
@@ -2,6 +2,7 @@ import { Authorized, Body, JsonController, Post, QueryParam } from 'routing-cont
 import { OpenAPI } from 'routing-controllers-openapi';
 import { Mailer } from '../Mailer';
 import { locales } from '../models/LocaleEnum';
+import { MailTypes } from '../models/MailTypeEnum';
 import { ResetMail } from '../models/ResetMail';
 import { SuccessResponse } from '../models/SuccessResponse';
 import { WelcomeMail } from '../models/WelcomeMail';
@@ -28,7 +29,7 @@ export class MailController {
         } catch (error) {
             throw error;
         }
-        return new SuccessResponse(locale);
+        return new SuccessResponse(MailTypes.PASSWORD_RESET, locale);
     }
 
     @Post('/test')
@@ -44,7 +45,7 @@ export class MailController {
             console.log(error)
             throw error;
         }
-        return new SuccessResponse(locale);
+        return new SuccessResponse(MailTypes.TEST, locale);
     }
 
     @Post('/registration')
@@ -60,7 +61,7 @@ export class MailController {
             console.log(error)
             throw error;
         }
-        return new SuccessResponse(locale);
+        return new SuccessResponse(MailTypes.RUNNER_WELCOME, locale);
     }
 
     @Post('/registration_forgot')
@@ -76,6 +77,6 @@ export class MailController {
             console.log(error)
             throw error;
         }
-        return new SuccessResponse(locale);
+        return new SuccessResponse(MailTypes.RUNNER_FORGOT, locale);
     }
 }
diff --git a/src/models/MailTypeEnum.ts b/src/models/MailTypeEnum.ts
index 3231bd4..338f52c 100644
--- a/src/models/MailTypeEnum.ts
+++ b/src/models/MailTypeEnum.ts
@@ -1,5 +1,6 @@
 export enum MailTypes {
     PASSWORD_RESET = "PASSWORD_RESET",
     RUNNER_FORGOT = "RUNNER_FORGOT",
-    RUNNER_WELCOME = "RUNNER_WELCOME"
+    RUNNER_WELCOME = "RUNNER_WELCOME",
+    TEST = "TEST"
 }
\ No newline at end of file
diff --git a/src/models/SuccessResponse.ts b/src/models/SuccessResponse.ts
index d3104a8..48cc43e 100644
--- a/src/models/SuccessResponse.ts
+++ b/src/models/SuccessResponse.ts
@@ -1,5 +1,6 @@
 import { IsBoolean, IsString } from 'class-validator';
 import { locales } from './LocaleEnum';
+import { MailTypes } from './MailTypeEnum';
 
 /**
  * Simple success response class to make everyone happy :)
@@ -15,7 +16,11 @@ export class SuccessResponse {
     @IsString()
     locale: locales;
 
-    constructor(locale?: locales) {
+    @IsString()
+    type: MailTypes;
+
+    constructor(type: MailTypes, locale?: locales) {
+        this.type = type;
         this.locale = locale || locales.en;
     }
 }
\ No newline at end of file