From 9bcc0adbc6f06eeb48d84b2557f236cab8b534fa Mon Sep 17 00:00:00 2001
From: James Barnsley <james@barnsley.nz>
Date: Sun, 26 Feb 2017 18:41:21 +1300
Subject: [PATCH] URIType fix for general use

---
 src/js/helpers.js | 42 +++++++++++++++---------------------------
 1 file changed, 15 insertions(+), 27 deletions(-)

diff --git a/src/js/helpers.js b/src/js/helpers.js
index 83db124a..3925d149 100755
--- a/src/js/helpers.js
+++ b/src/js/helpers.js
@@ -207,34 +207,22 @@ export let getFromUri = function(element,uri){
  * @return string
  **/
 export let uriType = function( uri ){
-    var exploded = uri.split(':');
+    var exploded = uri.split(':')
 
-    if (exploded[0] == 'spotify'){
-    	switch( exploded[1] ){
-    		case 'track':
-    			return 'track'
-    			break;
-    		case 'artist':
-    			return 'artist'
-    			break;
-    		case 'album':
-    			return 'album'
-    			break;
-    		case 'user':
-    			if( exploded[3] == 'playlist' ) return 'playlist'
-    			if( exploded.length == 3 ) return 'user'
-    			return null
-    			break;
-    	}
-    } else if (exploded[0] == 'local'){
-    	switch( exploded[1] ){
-    		case 'album':
-    			return 'album'
-    			break;
-    		case 'artist':
-    			return 'artist'
-    			break;
-    	}
+    switch (exploded[1]){
+    	case 'track':
+    	case 'artist':
+    	case 'album':
+    	case 'playlist':
+    		return exploded[1]
+    		break
+
+    	case 'user':
+    		if (exploded.length > 3 && exploded[3] == 'playlist'){
+    			return 'playlist'
+    		}
+    		return exploded[1]
+    		break
     }
 
     return null;