From 422d0827f9fd808eec7912a144b95b5dff2acca5 Mon Sep 17 00:00:00 2001
From: Godzil <godzil@godzil.net>
Date: Sat, 13 Aug 2016 16:32:56 +0100
Subject: [PATCH] Correct file source extension for the merge pass.

---
 src/episode.ts     |  2 +-
 src/video/merge.ts | 12 ++++++++++--
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/src/episode.ts b/src/episode.ts
index f5963cf..ad82194 100644
--- a/src/episode.ts
+++ b/src/episode.ts
@@ -50,7 +50,7 @@ function download(config: IConfig, page: IEpisodePage, player: IEpisodePlayer, d
         if (err) return done(err);
         if (config.merge) return complete('Finished ' + fileName, now, done);
         var isSubtited = Boolean(player.subtitle);
-        video.merge(config, isSubtited, player.video.file, filePath, err => {
+        video.merge(config, isSubtited, player.video.file, filePath, player.mode, err => {
           if (err) return done(err);
           complete('Finished ' + fileName, now, done);
         });
diff --git a/src/video/merge.ts b/src/video/merge.ts
index 8cfc643..ea366d6 100644
--- a/src/video/merge.ts
+++ b/src/video/merge.ts
@@ -8,9 +8,17 @@ import subtitle from '../subtitle/index';
 /**
  * Merges the subtitle and video files into a Matroska Multimedia Container.
  */
- export default function(config: IConfig, isSubtitled: boolean, rtmpInputPath: string, filePath: string, done: (err: Error) => void) {
+ export default function(config: IConfig, isSubtitled: boolean, rtmpInputPath: string, filePath: string, streamMode: string, done: (err: Error) => void) {
   var subtitlePath = filePath + '.' + (subtitle.formats[config.format] ? config.format : 'ass');
-  var videoPath = filePath + path.extname(rtmpInputPath);
+  var videoPath = filePath;
+	if (streamMode == "RTMP")
+  {
+	  videoPath += path.extname(rtmpInputPath);
+  }
+  else
+  {
+    videoPath += ".mp4";
+  }
   childProcess.exec(command() + ' ' +
     '-o "' + filePath + '.mkv" ' +
     '"' + videoPath + '" ' +