From a3945bc7f1d028adb5db66cd7fe68e750d92a3f8 Mon Sep 17 00:00:00 2001
From: Dolan <dolan_miu@hotmail.com>
Date: Tue, 9 Jan 2018 21:57:10 +0000
Subject: [PATCH] Media addMedia method

---
 src/file/media/media.ts | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/file/media/media.ts b/src/file/media/media.ts
index 1e2e354298..2868e51bbc 100644
--- a/src/file/media/media.ts
+++ b/src/file/media/media.ts
@@ -19,13 +19,17 @@ export class Media {
         return data;
     }
 
-    public addMedia(key: string, filePath: string): void {
-        this.map.set(key, {
+    public addMedia(filePath: string): IMediaData {
+        const key = path.basename(filePath);
+        const imageData = {
             referenceId: this.map.values.length,
             stream: fs.createReadStream(filePath),
             path: filePath,
-            fileName: path.basename(filePath),
-        });
+            fileName: key,
+        };
+        this.map.set(key, imageData);
+
+        return imageData;
     }
 
     public get array(): IMediaData[] {