Update index.js

Author: icebob

src/index.js

@@ -3,45 +3,55 @@ import IO from "socket.io-client";
 
 export default {
 
-	install(Vue, connection = "", opts) {
+	install(Vue, connection, opts) {
 
 		let socket;
 
 		if (typeof connection === "object")
 			socket = connection;
 		else
-			socket = IO(connection, opts);
+			socket = IO(connection || "", opts);
 
 		Vue.prototype.$socket = socket;
 
-		Vue.mixin({
+		let addListeners = function() {
+			if (this.$options.hasOwnProperty("socket")) {
+				var conf = this.$options.socket;
+				if (conf.namespace) {					
+					this.$socket = IO(conf.namespace, conf.options);
+				}
+
+				if (conf.events) {
+					let prefix = conf.prefix || "";
+					Object.keys(conf.events).forEach((key) => {
+						let func = conf.events[key].bind(this);
+						this.$socket.on(prefix + key, func);
+						conf.events[key].__binded = func;
+					});
+				}
+			}
+		}
+
+		let removeListeners = function() {
+			if (this.$options.hasOwnProperty("socket")) {
+				var conf = this.$options.socket;
 
-			beforeCompile() {
-				if (this.$options.hasOwnProperty("socket")) {
-					var conf = this.$options.socket;
-					if (conf.events) {
-						let prefix = conf.prefix || "";
-						Object.keys(conf.events).forEach((key) => {
-							let func = conf.events[key].bind(this);
-							socket.on(prefix + key, func);
-							conf.events[key].__binded = func;
-						});
-					}
+				if (conf.namespace) {
+					this.$socket.disconnect();
 				}
-			},
-
-			beforeDestroy() {
-				if (this.$options.hasOwnProperty("socket")) {
-					var conf = this.$options.socket;
-					if (conf.events) {
-						let prefix = conf.prefix || "";
-						Object.keys(conf.events).forEach((key) => {
-							socket.off(prefix + key, conf.events[key].__binded);
-						});
-					}
+
+				if (conf.events) {
+					let prefix = conf.prefix || "";
+					Object.keys(conf.events).forEach((key) => {
+						this.$socket.off(prefix + key, conf.events[key].__binded);
+					});
 				}
 			}
+		}
 
+		Vue.mixin({
+			beforeCompile: addListeners,
+			beforeDestroy: removeListeners
 		});
 
 	}