From b3db2da3d84946222234bf3793be149a26f2fed1 Mon Sep 17 00:00:00 2001
From: LASER-Yi <ly0007@yeah.net>
Date: Sat, 26 Sep 2020 20:58:56 +0800
Subject: [PATCH 1/3] add mass upload feature

---
 bazarr/api.py       |  13 +-
 views/episodes.html | 390 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 402 insertions(+), 1 deletion(-)

diff --git a/bazarr/api.py b/bazarr/api.py
index 07c73d77e..6ad28c5d9 100644
--- a/bazarr/api.py
+++ b/bazarr/api.py
@@ -37,7 +37,7 @@ from scheduler import scheduler
 from subsyncer import subsync
 from filesystem import browse_bazarr_filesystem, browse_sonarr_filesystem, browse_radarr_filesystem
 
-from subliminal_patch.core import SUBTITLE_EXTENSIONS
+from subliminal_patch.core import SUBTITLE_EXTENSIONS, guessit
 
 from flask import Flask, jsonify, request, Response, Blueprint, url_for, make_response
 
@@ -549,6 +549,15 @@ class Episodes(Resource):
             item.update({"desired_languages": desired_languages})
         return jsonify(draw=draw, recordsTotal=row_count, recordsFiltered=row_count, data=result)
 
+class SubtitleNameInfo(Resource):
+    @authenticate
+    def get(self):
+        name = request.args.get('filename')
+        if name is not None:
+            result = guessit(name, options=None)
+            return jsonify(data=result)
+        else:
+            return '', 400
 
 class EpisodesSubtitlesDelete(Resource):
     @authenticate
@@ -1990,6 +1999,8 @@ api.add_resource(SystemProviders, '/systemproviders')
 api.add_resource(SystemStatus, '/systemstatus')
 api.add_resource(SystemReleases, '/systemreleases')
 
+api.add_resource(SubtitleNameInfo, '/subtitle_name_info')
+
 api.add_resource(Series, '/series')
 api.add_resource(SeriesEditor, '/series_editor')
 api.add_resource(SeriesEditSave, '/series_edit_save')
diff --git a/views/episodes.html b/views/episodes.html
index ab39dd6a9..37a2e40a6 100644
--- a/views/episodes.html
+++ b/views/episodes.html
@@ -62,6 +62,10 @@
 
 {% block bcright %}
     <div class="d-flex m-t-5 justify-content-end">
+        <button class="btn btn-outline" id="mass_upload_button">
+            <div><i class="fas fa-cloud-upload-alt align-top text-themecolor text-center font-20" aria-hidden="true"></i></div>
+            <div class="align-bottom text-themecolor small text-center">Upload</div>
+        </button>
         <button class="btn btn-outline" id="edit_button">
             <div><i class="fas fa-wrench align-top text-themecolor text-center font-20" aria-hidden="true"></i></div>
             <div class="align-bottom text-themecolor small text-center">Edit Series</div>
@@ -219,6 +223,56 @@
         </div>
     </div>
 
+    <div id="massUploadModal" class="modal" tabindex="-1" role="dialog">
+        <div class="modal-dialog modal-lg" role="document">
+            <div class="modal-content">
+                <div class="modal-header">
+                    <h5 class="modal-title"><span id="mass_upload_title_span"></span></h5><br>
+                    <button type="button" class="close" id="mass_upload_close_btn" data-dismiss="modal" aria-label="Close">
+                        <span aria-hidden="true">&times;</span>
+                    </button>
+                </div>
+                <form class="form" name="edit_form" id="mass_upload_form">
+                    <div class="modal-body">
+                        <div class="container-fluid">
+                            <div class="row form-group">
+                                <div class="custom-file">
+                                    <input type="file" multiple class="custom-file-input" id="mass_upload_file_list">
+                                    <label id="mass-upload-file-label" class="custom-file-label" for="upload">Choose files</label>
+                                </div>
+                            </div>
+                            <!-- Store episodes we previous collect -->
+                            <input type="hidden" id="mass-upload-exist-episodes" value=""/>
+
+                            <table id="upload_table" class="table table-striped" style="width:100%">
+                                <thead>
+                                <tr>
+                                    <th></th>
+                                    <th style="text-align: left;">Filename</th>
+                                    <th style="text-align: left;">Season</th>
+                                    <th style="text-align: left;">Episode</th>
+                                    <th style="text-align: left;">Action</th>
+                                </tr>
+                                </thead>
+                            </table>
+                        </div>
+                    </div>
+                    <div class="modal-footer justify-content-between">
+                        <div>
+                            <div>
+                                <select class="selectpicker" id="mass_upload_language_select" name="language"></select>
+                            </div>
+                        </div>
+                        <div>
+                            <button type="submit" id="mass_upload_save_button" class="btn btn-info">Upload</button>
+                            <button type="button" class="btn btn-secondary" id="mass_upload_cancel_btn" data-dismiss="modal">Cancel</button>
+                        </div>
+                    </div>
+                </form>
+            </div>
+        </div>
+    </div>
+
     <div id="editModal" class="modal" tabindex="-1" role="dialog">
         <div class="modal-dialog modal-lg" role="document">
             <div class="modal-content">
@@ -507,6 +561,7 @@
             $('#series_nav').addClass("active");
 
             seriesDetailsRefresh();
+            episodesDetailsRefresh();
             getLanguages();
             getEnabledLanguages();
 
@@ -961,6 +1016,326 @@
                 });
             });
 
+            const UploadStatus = {
+                ERROR: 0,
+                VAILD: 1,
+                UPLOAD: 2,
+                DONE: 3
+            }
+
+            $('#mass_upload_button').on('click', function (e) {
+                e.preventDefault();
+
+                $('#upload_table').DataTable({
+                    destroy: true,
+                    processing: true,
+                    language: {
+                        zeroRecords: 'Select Subtitles to Get Started',
+                        processing: "Loading Subtitle..."
+                    },
+                    searching: false,
+                    ordering: false,
+                    lengthChange: false,
+                    serverSide: false,
+                    responsive: true,
+                    columns: [
+                        {
+                            data: null,
+                            render: function(data, type, row) {
+                                switch (data.status) {
+                                    case UploadStatus.VAILD:
+                                        return '<i class="fas fa-check px-1"></i>'
+                                    case UploadStatus.UPLOAD:
+                                        return '<i class="spinner-border spinner-border-sm px-1" role="status" />'
+                                    case UploadStatus.DONE:
+                                        return '<i class="far fa-check-circle px-1"></i>'
+                                    case UploadStatus.ERROR:
+                                    default:
+                                        return '<i class="fas fa-exclamation-triangle px-1"></i>'
+                                }
+                            }
+                        },
+                        {data: 'filename'},
+                        {
+                            data: "season",
+                            render: function(data, type, row) {
+                                let cls = []
+                                let readonly = false
+                                if (data <= 0) {
+                                    if (row.status !== UploadStatus.UPLOAD) {
+                                        cls.push('is-invalid');
+                                    } else {
+                                        readonly = true
+                                    }
+                                } else {
+                                    if (row.status === UploadStatus.UPLOAD) {
+                                        readonly = true;
+                                    }
+                                }
+                                return `<input type="text" ${readonly ? 'readonly' : ''} \
+                                    class="mass-upload-season-input form-control w-50 ${cls.join(' ')}" \
+                                    value="${data}" />`
+                            }
+                        },
+                        {
+                            data: "episode",
+                            render: function(data, type, row) {
+                                let cls = []
+                                let readonly = false
+                                if (data <= 0) {
+                                    if (row.status !== UploadStatus.UPLOAD) {
+                                        cls.push('is-invalid');
+                                    } else {
+                                        readonly = true
+                                    }
+                                } else {
+                                    if (row.status === UploadStatus.UPLOAD) {
+                                        readonly = true;
+                                    }
+                                }
+                                return `<input type="text" ${readonly ? 'readonly' : ''} \
+                                    class="mass-upload-episode-input form-control w-50 ${cls.join(' ')}" \
+                                    value="${data}" />`
+                            }
+                        },
+                        {
+                            data: null,
+                            render: function(data, type, row) {
+                                return `<a href="" class="mass-upload-del-button badge badge-secondary"><i class="far fa-trash-alt"></i></a>`
+                            }
+                        }
+                    ]
+                })
+
+                // reset
+                $('#upload_table').DataTable().table().clear().draw();
+                $('#mass_upload_file_list').val("")
+                $('#mass-upload-file-label').text('Choose files')
+
+                $("#mass_upload_title_span")
+                    .html(`${seriesDetails['title']} - Upload`);
+
+                $('#mass_upload_language_select')
+                    .empty();
+                $.each(enabledLanguages, function (i, item) {
+                    $('#mass_upload_language_select')
+                        .append(`<option value="${item.code2}">${item.name}</option>`);
+                });
+                $("#mass_upload_language_select")
+                    .selectpicker("refresh");
+
+                $('#mass_upload_forced_checkbox')
+                    .val(seriesDetails['forced'])
+                    .change();
+
+                $('#massUploadModal')
+                    .modal({
+                        focus: false
+                    });
+            });
+
+            $('#upload_table').on('click', '.mass-upload-del-button', function(e) {
+                e.preventDefault();
+                $('#upload_table').DataTable()
+                    .row($(this).parents('tr'))
+                    .remove()
+                    .draw();
+            });
+
+            $('#upload_table').on('change', '.mass-upload-season-input', function(e) {
+                const value = $(this).val();
+
+                let row = $('#upload_table').DataTable().row($(this).parents('tr'));
+                let data = row.data();
+                data.season = value;
+
+                data.status = UploadStatus.ERROR;
+                for(const exist of episodesDetails.data) {
+                    if (exist.episode == data.episode && exist.season == data.season) {
+                        data.status = UploadStatus.VAILD;
+                        data.exist = exist
+                        break;
+                    }
+                }
+
+                row.data(data).draw()
+            })
+
+            $('#upload_table').on('change', '.mass-upload-episode-input', function(e) {
+                const value = $(this).val();
+
+                let row = $('#upload_table').DataTable().row($(this).parents('tr'));
+                let data = row.data();
+                data.episode = value;
+
+                data.status = UploadStatus.ERROR;
+                for(const exist of episodesDetails.data) {
+                    if (exist.episode == data.episode && exist.season == data.season) {
+                        data.status = UploadStatus.VAILD;
+                        data.exist = exist
+                        break;
+                    }
+                }
+
+                row.data(data).draw()
+            })
+
+            $('#mass_upload_file_list').change(function() {
+                let filelist = $('#mass_upload_file_list').get(0).files
+                $('#mass-upload-file-label').text(`${filelist.length} Files`)
+
+                $('#mass_upload_save_button').prop('disabled', true)
+                $('#mass_upload_close_btn').prop('disabled', true);
+                $('#mass_upload_cancel_btn').prop('disabled', true);
+
+                let table = $('#upload_table').DataTable();
+
+                table.table().clear().draw();
+
+                const episodes = episodesDetails.data
+
+                let promiselist = []
+
+                for (const file of filelist) {
+                    const name = file.name;
+
+                    const object = {
+                        file: file,
+                        filename: name,
+                        season: 0,
+                        episode: 0,
+                        status: UploadStatus.UPLOAD,
+                        exist: null
+                    }
+
+                    const cacheRow = table.row.add(object)
+
+                    promiselist.push(Promise.resolve($.ajax({
+                        url: "{{ url_for('api.subtitlenameinfo') }}",
+                        type: "GET",
+                        dataType: "json",
+                        data: {
+                            filename: name
+                        },
+                        complete: function(data) {
+                            const response = data.responseJSON.data;
+                            const season = (response.season ?? 1);
+
+                            let existdata = null
+                            for(const exist of episodes) {
+                                if (exist.episode == response.episode && exist.season == season) {
+                                    existdata = exist;
+                                    break;
+                                }
+                            }
+
+                            let complete = {
+                                file: file,
+                                filename: name,
+                                season: season,
+                                episode: response.episode ?? 0,
+                                status: existdata != null ? UploadStatus.VAILD : UploadStatus.ERROR,
+                                exist: existdata,
+                                row: cacheRow
+                            };
+                            table.row(cacheRow).data(complete)
+                                .draw();
+
+                        },
+                        error: function(data) {
+                            let error = {
+                                file: file,
+                                filename: name,
+                                season: 0,
+                                episode: 0,
+                                status: UploadStatus.ERROR,
+                                exist: null,
+                                row: cacheRow
+                            };
+                            table.row(cacheRow).data(error)
+                                .draw();
+                        }
+                    }))) 
+                }
+
+                table.table().draw();
+
+                Promise.all(promiselist)
+                    .then(function(){
+                        $('#mass_upload_save_button').prop('disabled', false)
+                        $('#mass_upload_close_btn').prop('disabled', false);
+                        $('#mass_upload_cancel_btn').prop('disabled', false);
+                    })
+            })
+
+            $('#mass_upload_form').on('submit', function(e) {
+                e.preventDefault();
+                $('#mass_upload_save_button').html('<div class="spinner-border spinner-border-sm" role="status"></div>');
+
+                const formdata = new FormData(document.getElementById("mass_upload_form"));
+
+                const language = formdata.get("language");
+
+                let table = $('#upload_table').DataTable();
+
+                const uploadlist = table.data().toArray().filter(function(item) {
+                        return item.status === UploadStatus.VAILD
+                    });
+
+                const promiselist = uploadlist.map(function(item) {
+                    const data = {
+                            sonarrSeriesId: item.exist.sonarrSeriesId,
+                            sonarrEpisodeId: item.exist.sonarrEpisodeId,
+                            language: language,
+                            upload: item.file,
+                            episodePath: item.exist.mapped_path,
+                            // sceneName,
+                            title: item.exist.title,
+                            audioLanguage: item.exist.audio_language.name,
+                            forced: false
+                        }
+
+                    const form = new FormData()
+                    for(const key in data) {
+                        form.append(key, data[key])
+                    }
+
+                    const cacheRow = item.row ?? null
+                    
+                    item.status = UploadStatus.UPLOAD;
+
+                    let row = table.row(cacheRow);
+
+                    row.data(item).draw()
+
+                    return Promise.resolve($.ajax({
+                            url: "{{ url_for('api.episodessubtitlesupload') }}",
+                            data: form,
+                            processData: false,
+                            contentType: false,
+                            type: 'POST',
+                            complete: function(e) {
+                                item.status = UploadStatus.DONE;
+                                row.data(item).draw()
+                            },
+                            error: function(e) {
+                                item.status = UploadStatus.ERROR;
+                                row.data(item).draw()
+                            }
+                        }));
+                })
+
+                Promise.all(promiselist)
+                    .then(function(){
+                        $('#massUploadModal').modal('hide');
+                    })
+                    .catch(function() {
+                    })
+                    .finally(function() {
+                        $('#mass_upload_save_button').html('Upload');
+                    })
+            })
+
             $('#edit_button').on('click', function (e) {
                 e.preventDefault();
                 $("#edit_series_title_span").html(seriesDetails['title']);
@@ -1413,6 +1788,21 @@
 
         });
 
+        function episodesDetailsRefresh() {
+            $.ajax({
+                url: "{{ url_for('api.episodes') }}",
+                type: "GET",
+                dataType: "json",
+                data: {
+                    seriesid: "{{id}}",
+                },
+                complete: function(data) {
+                    const response = data.responseJSON;
+                    episodesDetails = response
+                }
+            })
+        }
+
         function seriesDetailsRefresh() {
             $.ajax({
                 url: "{{ url_for('api.series') }}?seriesid={{id}}"

From 38732ca47ed404c483f376672b3f01833441d8aa Mon Sep 17 00:00:00 2001
From: LASER-Yi <ly0007@yeah.net>
Date: Sun, 27 Sep 2020 19:47:00 +0800
Subject: [PATCH 2/3] bug fixed and ui tweak

---
 bazarr/api.py       | 6 +++++-
 views/episodes.html | 8 ++++----
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/bazarr/api.py b/bazarr/api.py
index 6ad28c5d9..55659871a 100644
--- a/bazarr/api.py
+++ b/bazarr/api.py
@@ -554,7 +554,11 @@ class SubtitleNameInfo(Resource):
     def get(self):
         name = request.args.get('filename')
         if name is not None:
-            result = guessit(name, options=None)
+            opts = dict()
+            opts['type'] = 'episode'
+            result = guessit(name, options=opts)
+            if 'subtitle_language' in result:
+                result['subtitle_language'] = str(result['subtitle_language'])
             return jsonify(data=result)
         else:
             return '', 400
diff --git a/views/episodes.html b/views/episodes.html
index 37a2e40a6..d972bfad7 100644
--- a/views/episodes.html
+++ b/views/episodes.html
@@ -224,7 +224,7 @@
     </div>
 
     <div id="massUploadModal" class="modal" tabindex="-1" role="dialog">
-        <div class="modal-dialog modal-lg" role="document">
+        <div class="modal-dialog modal-xl" role="document">
             <div class="modal-content">
                 <div class="modal-header">
                     <h5 class="modal-title"><span id="mass_upload_title_span"></span></h5><br>
@@ -251,7 +251,7 @@
                                     <th style="text-align: left;">Filename</th>
                                     <th style="text-align: left;">Season</th>
                                     <th style="text-align: left;">Episode</th>
-                                    <th style="text-align: left;">Action</th>
+                                    <th style="text-align: center;">Action</th>
                                 </tr>
                                 </thead>
                             </table>
@@ -1073,7 +1073,7 @@
                                     }
                                 }
                                 return `<input type="text" ${readonly ? 'readonly' : ''} \
-                                    class="mass-upload-season-input form-control w-50 ${cls.join(' ')}" \
+                                    class="mass-upload-season-input form-control ${cls.join(' ')}" \
                                     value="${data}" />`
                             }
                         },
@@ -1094,7 +1094,7 @@
                                     }
                                 }
                                 return `<input type="text" ${readonly ? 'readonly' : ''} \
-                                    class="mass-upload-episode-input form-control w-50 ${cls.join(' ')}" \
+                                    class="mass-upload-episode-input form-control ${cls.join(' ')}" \
                                     value="${data}" />`
                             }
                         },

From 76c50dc721a6a8556e7ec169bd71408d78f7bacf Mon Sep 17 00:00:00 2001
From: LASER-Yi <ly0007@yeah.net>
Date: Fri, 2 Oct 2020 09:10:22 +0800
Subject: [PATCH 3/3] Fix typo

---
 views/episodes.html | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/views/episodes.html b/views/episodes.html
index d972bfad7..9d5038b34 100644
--- a/views/episodes.html
+++ b/views/episodes.html
@@ -1018,7 +1018,7 @@
 
             const UploadStatus = {
                 ERROR: 0,
-                VAILD: 1,
+                VALID: 1,
                 UPLOAD: 2,
                 DONE: 3
             }
@@ -1043,7 +1043,7 @@
                             data: null,
                             render: function(data, type, row) {
                                 switch (data.status) {
-                                    case UploadStatus.VAILD:
+                                    case UploadStatus.VALID:
                                         return '<i class="fas fa-check px-1"></i>'
                                     case UploadStatus.UPLOAD:
                                         return '<i class="spinner-border spinner-border-sm px-1" role="status" />'
@@ -1152,7 +1152,7 @@
                 data.status = UploadStatus.ERROR;
                 for(const exist of episodesDetails.data) {
                     if (exist.episode == data.episode && exist.season == data.season) {
-                        data.status = UploadStatus.VAILD;
+                        data.status = UploadStatus.VALID;
                         data.exist = exist
                         break;
                     }
@@ -1171,7 +1171,7 @@
                 data.status = UploadStatus.ERROR;
                 for(const exist of episodesDetails.data) {
                     if (exist.episode == data.episode && exist.season == data.season) {
-                        data.status = UploadStatus.VAILD;
+                        data.status = UploadStatus.VALID;
                         data.exist = exist
                         break;
                     }
@@ -1234,7 +1234,7 @@
                                 filename: name,
                                 season: season,
                                 episode: response.episode ?? 0,
-                                status: existdata != null ? UploadStatus.VAILD : UploadStatus.ERROR,
+                                status: existdata != null ? UploadStatus.VALID : UploadStatus.ERROR,
                                 exist: existdata,
                                 row: cacheRow
                             };
@@ -1279,7 +1279,7 @@
                 let table = $('#upload_table').DataTable();
 
                 const uploadlist = table.data().toArray().filter(function(item) {
-                        return item.status === UploadStatus.VAILD
+                        return item.status === UploadStatus.VALID
                     });
 
                 const promiselist = uploadlist.map(function(item) {