From 49537a2efe33a028fc0c06aca47ef74fe8885b41 Mon Sep 17 00:00:00 2001
From: Leonardo Galli <leonardo.galli@bluewin.ch>
Date: Sun, 8 Jan 2017 17:47:19 +0100
Subject: [PATCH] Update UI to display download status.

---
 src/NzbDrone.Api/Series/MovieResource.cs      |  8 ++++-
 src/UI/Cells/MovieDownloadStatusCell.js       |  6 ++++
 src/UI/Cells/MovieDownloadStatusTemplate.hbs  |  1 +
 src/UI/Handlebars/Helpers/Series.js           | 32 +++++++++++++++++++
 src/UI/Movies/Details/InfoViewTemplate.hbs    |  1 +
 src/UI/Movies/Index/MoviesIndexLayout.js      | 10 ++++++
 .../SeriesOverviewItemViewTemplate.hbs        |  2 ++
 7 files changed, 59 insertions(+), 1 deletion(-)
 create mode 100644 src/UI/Cells/MovieDownloadStatusCell.js
 create mode 100644 src/UI/Cells/MovieDownloadStatusTemplate.hbs

diff --git a/src/NzbDrone.Api/Series/MovieResource.cs b/src/NzbDrone.Api/Series/MovieResource.cs
index b1924629d..4038792d7 100644
--- a/src/NzbDrone.Api/Series/MovieResource.cs
+++ b/src/NzbDrone.Api/Series/MovieResource.cs
@@ -29,7 +29,7 @@ namespace NzbDrone.Api.Movie
         public DateTime? InCinemas { get; set; }
         public List<MediaCover> Images { get; set; }
         public string Website { get; set; }
-
+        public bool Downloaded { get; set; }
         public string RemotePoster { get; set; }
         public int Year { get; set; }
 
@@ -79,6 +79,8 @@ namespace NzbDrone.Api.Movie
         {
             if (model == null) return null;
 
+            long Size = model.MovieFile.Value != null ? model.MovieFile.Value.Size : 0;
+
             return new MovieResource
             {
                 Id = model.Id,
@@ -87,6 +89,8 @@ namespace NzbDrone.Api.Movie
                 //AlternateTitles
                 SortTitle = model.SortTitle,
                 InCinemas = model.InCinemas,
+
+                Downloaded = model.MovieFile.Value != null,
                 //TotalEpisodeCount
                 //EpisodeCount
                 //EpisodeFileCount
@@ -104,6 +108,8 @@ namespace NzbDrone.Api.Movie
                 
                 Monitored = model.Monitored,
 
+                SizeOnDisk = Size,
+
                 Runtime = model.Runtime,
                 LastInfoSync = model.LastInfoSync,
                 CleanTitle = model.CleanTitle,
diff --git a/src/UI/Cells/MovieDownloadStatusCell.js b/src/UI/Cells/MovieDownloadStatusCell.js
new file mode 100644
index 000000000..ba35657b4
--- /dev/null
+++ b/src/UI/Cells/MovieDownloadStatusCell.js
@@ -0,0 +1,6 @@
+var TemplatedCell = require('./TemplatedCell');
+
+module.exports = TemplatedCell.extend({
+    className : 'movie-title-cell',
+    template  : 'Cells/MovieDownloadStatusTemplate',
+});
diff --git a/src/UI/Cells/MovieDownloadStatusTemplate.hbs b/src/UI/Cells/MovieDownloadStatusTemplate.hbs
new file mode 100644
index 000000000..002a9fdc6
--- /dev/null
+++ b/src/UI/Cells/MovieDownloadStatusTemplate.hbs
@@ -0,0 +1 @@
+<span class="label label-{{DownloadedStatusColor}}">{{DownloadedStatus}}</span>
diff --git a/src/UI/Handlebars/Helpers/Series.js b/src/UI/Handlebars/Helpers/Series.js
index fbb3a23fc..41c2d641a 100644
--- a/src/UI/Handlebars/Helpers/Series.js
+++ b/src/UI/Handlebars/Helpers/Series.js
@@ -127,8 +127,40 @@ Handlebars.registerHelper('GetBannerStatus', function() {
   else if (!monitored) {
       return new Handlebars.SafeString('<div class="announced-banner"><i class="icon-sonarr-series-unmonitored grid-icon" title=""></i>&nbsp;Not Monitored</div>');
   }
+});
+
+Handlebars.registerHelper('DownloadedStatusColor', function() {
+  if (!this.monitored) {
+    if (this.downloaded) {
+      return "default";
+    }
+    return "warning";
+  }
+
+    if (this.downloaded) {
+      return "success";
+    }
+
+  if (this.status != "released") {
+    return "primary";
+  }
+
+  return "danger";
 })
 
+Handlebars.registerHelper('DownloadedStatus', function() {
+
+  if (this.downloaded) {
+    return "Downloaded";
+  }
+  if (!this.monitored) {
+    return "Not Monitored";
+  }
+
+
+  return "Missing";
+});
+
 
 Handlebars.registerHelper('inCinemas', function() {
   var monthNames = ["January", "February", "March", "April", "May", "June",
diff --git a/src/UI/Movies/Details/InfoViewTemplate.hbs b/src/UI/Movies/Details/InfoViewTemplate.hbs
index 1897c841a..602a13e72 100644
--- a/src/UI/Movies/Details/InfoViewTemplate.hbs
+++ b/src/UI/Movies/Details/InfoViewTemplate.hbs
@@ -26,6 +26,7 @@
         {{else}}
             <span class="label label-default">Announced</span>
         {{/if_eq}}
+        <span class="label label-{{DownloadedStatusColor}}">{{DownloadedStatus}}</span>
     </div>
     <div class="col-md-4">
         <span class="series-info-links">
diff --git a/src/UI/Movies/Index/MoviesIndexLayout.js b/src/UI/Movies/Index/MoviesIndexLayout.js
index 8e87390ba..abcfdfd30 100644
--- a/src/UI/Movies/Index/MoviesIndexLayout.js
+++ b/src/UI/Movies/Index/MoviesIndexLayout.js
@@ -12,6 +12,7 @@ var ProfileCell = require('../../Cells/ProfileCell');
 var MovieLinksCell = require('../../Cells/MovieLinksCell');
 var MovieActionCell = require('../../Cells/MovieActionCell');
 var MovieStatusCell = require('../../Cells/MovieStatusCell');
+var MovieDownloadStatusCell = require('../../Cells/MovieDownloadStatusCell');
 var FooterView = require('./FooterView');
 var FooterModel = require('./FooterModel');
 var ToolbarLayout = require('../../Shared/Toolbar/ToolbarLayout');
@@ -55,6 +56,11 @@ module.exports = Marionette.Layout.extend({
             cell      : MovieLinksCell,
             className : "movie-links-cell"
         },
+        {
+          name        : "this",
+          label       : "Status",
+          cell        : MovieDownloadStatusCell,
+        },
         {
             name     : 'this',
             label    : '',
@@ -134,6 +140,10 @@ module.exports = Marionette.Layout.extend({
                 {
                     title : 'In Cinemas',
                     name  : 'inCinemas'
+                },
+                {
+                  title : "Status",
+                  name : "status",
                 }
             ]
         };
diff --git a/src/UI/Movies/Index/Overview/SeriesOverviewItemViewTemplate.hbs b/src/UI/Movies/Index/Overview/SeriesOverviewItemViewTemplate.hbs
index c222b56cb..302d431ca 100644
--- a/src/UI/Movies/Index/Overview/SeriesOverviewItemViewTemplate.hbs
+++ b/src/UI/Movies/Index/Overview/SeriesOverviewItemViewTemplate.hbs
@@ -40,6 +40,8 @@
                     <span class="label label-default">{{inCinemas}}</span>
 
                     {{profile profileId}}
+
+                    <span class="label label-{{DownloadedStatusColor}}">{{DownloadedStatus}}</span>
                 </div>
                 <div class="col-md-4 col-xs-4">
                   <span class="movie-info-links">