From d01b24a91fac432813516f3da435adaf0da3e019 Mon Sep 17 00:00:00 2001
From: Josef Citrine <josefcitrine@googlemail.com>
Date: Sun, 15 May 2016 23:03:49 +0100
Subject: [PATCH] Admins can assign tracks to accounts

---
 app/Commands/EditTrackCommand.php                 | 15 +++++++++++++++
 public/templates/directives/track-editor.html     |  7 +++++++
 .../scripts/app/directives/track-editor.coffee    |  1 +
 resources/assets/scripts/app/services/auth.coffee |  2 +-
 4 files changed, 24 insertions(+), 1 deletion(-)

diff --git a/app/Commands/EditTrackCommand.php b/app/Commands/EditTrackCommand.php
index 5cdd8af2..900c94d5 100644
--- a/app/Commands/EditTrackCommand.php
+++ b/app/Commands/EditTrackCommand.php
@@ -146,6 +146,15 @@ class EditTrackCommand extends CommandBase
             }
         }
 
+        $oldid = null;
+
+        if (isset($this->_input['user_id'])) {
+          if ($track->user_id != $this->_input['user_id']) {
+            $oldid = $track->user_id;
+            $track->user_id = $this->_input['user_id'];
+          }
+        }
+
         $track->updateTags();
         $track->save();
 
@@ -153,6 +162,12 @@ class EditTrackCommand extends CommandBase
             'track_count' => DB::raw('(SELECT COUNT(id) FROM tracks WHERE deleted_at IS NULL AND published_at IS NOT NULL AND user_id = ' . $this->_track->user_id . ')')
         ]);
 
+        if ($oldid != null) {
+          User::whereId($oldid)->update([
+              'track_count' => DB::raw('(SELECT COUNT(id) FROM tracks WHERE deleted_at IS NULL AND published_at IS NOT NULL AND user_id = ' . $oldid . ')')
+          ]);
+        }
+
         return CommandResponse::succeed(['real_cover_url' => $track->getCoverUrl(Image::NORMAL)]);
     }
 
diff --git a/public/templates/directives/track-editor.html b/public/templates/directives/track-editor.html
index 83406389..eae3f1bb 100644
--- a/public/templates/directives/track-editor.html
+++ b/public/templates/directives/track-editor.html
@@ -107,6 +107,13 @@
                 <label for="is_listed"><input ng-disabled="isSaving" ng-change="touchModel()" id="is_listed" type="checkbox" ng-model="track.is_listed" /> Is Listed</label>
             </div>
         </div>
+        <div class="row-fluid" ng-show="isAdmin">
+          <div class="span6 form-row">
+            <label for="title" class="strong">User:</label>
+            <input ng-disabled="isSaving" ng-change="touchModel()" placeholder="Username" type="text" id="username" ng-model="track.user_id" />
+            <div class="error">{{errors.title}}</div>
+          </div>
+        </div>
         <div class="form-row">
             <label class="strong">Choose a License:</label>
             <ul class="license-grid">
diff --git a/resources/assets/scripts/app/directives/track-editor.coffee b/resources/assets/scripts/app/directives/track-editor.coffee
index 10c5909d..404d8091 100644
--- a/resources/assets/scripts/app/directives/track-editor.coffee
+++ b/resources/assets/scripts/app/directives/track-editor.coffee
@@ -30,6 +30,7 @@ module.exports = angular.module('ponyfm').directive 'pfmTrackEditor', () ->
             $scope.selectedSongs = {}
             $scope.albums = []
             $scope.selectedAlbum = null
+            $scope.isAdmin = auth.data.isAdmin
             albumsDb = {}
 
             $scope.selectAlbum = (album) ->
diff --git a/resources/assets/scripts/app/services/auth.coffee b/resources/assets/scripts/app/services/auth.coffee
index 70d390e3..d092cc8a 100644
--- a/resources/assets/scripts/app/services/auth.coffee
+++ b/resources/assets/scripts/app/services/auth.coffee
@@ -17,7 +17,7 @@
 module.exports = angular.module('ponyfm').factory('auth', [
     '$rootScope'
     ($rootScope) ->
-        data: {isLogged: window.pfm.auth.isLogged, user: window.pfm.auth.user}
+        data: {isLogged: window.pfm.auth.isLogged, user: window.pfm.auth.user, isAdmin: window.pfm.auth.user.roles[0].id == 2}
         login: (email, password, remember) ->
             def = new $.Deferred()
             $.post('/api/web/auth/login', {email: email, password: password, remember: remember})