Prevent duplicate assignment of libraries and custom script

Author: kennethphough

src/Mvc/Controller/KyteLibraryAssignmentController.php

@@ -9,6 +9,13 @@ public function hook_response_data($method, $o, &$r = null, &$d = null) {
                 throw new \Exception("Cannot delete assignment for globally included libraries. This library is set to include on all pages. To remove it, either disable global inclusion in Site → Libraries or delete the library entirely.");
             }
         } else {
+            if ($method == 'new') {
+                $assignments = new \Kyte\Core\Model($this->model);
+                $assignments->retrieve('library', $d['library'], false, [['field' => 'page', 'value' => $d['page']], ['field' => 'site', 'value' => $d['site']], ['field' => 'kyte_account', 'value' => $this->account]]);
+                if ($assignments->count > 0) {
+                    throw new \Exception("This library is already assigned to the specified page or site.");
+                }
+            }
             if (isset($r['page'], $r['page']['header'], $r['page']['footer'])) {
                 $r['page']['header'] = null;
                 $r['page']['footer'] = null;

src/Mvc/Controller/KyteScriptAssignmentController.php

@@ -9,6 +9,13 @@ public function hook_response_data($method, $o, &$r = null, &$d = null) {
                 throw new \Exception("Cannot delete assignment for globally included script. This script is set to include on all pages. To remove it, either disable global inclusion in Site → Custom Scripts or delete the script entirely.");
             }
         } else {
+            if ($method == 'new') {
+                $assignments = new \Kyte\Core\Model($this->model);
+                $assignments->retrieve('script', $d['library'], false, [['field' => 'page', 'value' => $d['page']], ['field' => 'site', 'value' => $d['site']], ['field' => 'kyte_account', 'value' => $this->account]]);
+                if ($assignments->count > 0) {
+                    throw new \Exception("This custom script is already assigned to the specified page or site.");
+                }
+            }
             if (isset($r['script'])) {
                 $r['script']['content'] = '';
                 $r['script']['content_js_obfuscated'] = '';