From 8de127066d307b547817707f4c98a44f0befed8e Mon Sep 17 00:00:00 2001
From: Fakhreev Rafael <fahreevrafael@yandex.ru>
Date: Thu, 23 Feb 2017 12:31:42 +0300
Subject: [PATCH] on event dispatch store option

---
 src/Main.js     |  4 ++--
 src/Observer.js | 34 +++++++++++++++++++++++++++++-----
 2 files changed, 31 insertions(+), 7 deletions(-)

diff --git a/src/Main.js b/src/Main.js
index f19a820..4586cf6 100755
--- a/src/Main.js
+++ b/src/Main.js
@@ -3,11 +3,11 @@ import Emitter from './Emitter'
 
 export default {
 
-    install(Vue, connection, store){
+    install(Vue, connection, store, useActions = false){
 
         if(!connection) throw new Error("[Vue-Socket.io] cannot locate connection")
 
-        let observer = new Observer(connection, store)
+        let observer = new Observer(connection, store, useActions)
 
         Vue.prototype.$socket = observer.Socket;
 
diff --git a/src/Observer.js b/src/Observer.js
index 76a71ce..cf55b8c 100755
--- a/src/Observer.js
+++ b/src/Observer.js
@@ -3,7 +3,7 @@ import Socket from 'socket.io-client'
 
 export default class{
 
-    constructor(connection, store) {
+    constructor(connection, store, useActions) {
 
         if(typeof connection == 'string'){
             this.Socket = Socket(connection);
@@ -11,7 +11,10 @@ export default class{
             this.Socket = connection
         }
 
-        if(store) this.store = store;
+        if(store) {
+            this.store = store;
+            this.useActions = useActions;
+        }
 
         this.onEvent()
 
@@ -21,17 +24,27 @@ export default class{
         this.Socket.onevent = (packet) => {
             Emitter.emit(packet.data[0], packet.data[1]);
 
-            if(this.store) this.commitStore('SOCKET_'+packet.data[0], packet.data[1])
+            if(this.store)
+                this.useActions
+                    ? this.dispatchStore('SOCKET_'+packet.data[0], packet.data[1])
+                    : this.commitStore('SOCKET_'+packet.data[0], packet.data[1])
 
         };
 
         let _this = this;
-
         ["connect", "error", "disconnect", "reconnect", "reconnect_attempt", "reconnecting", "reconnect_error", "reconnect_failed", "connect_error", "connect_timeout", "connecting", "ping", "pong"]
             .forEach((value) => {
                 _this.Socket.on(value, (data) => {
                     Emitter.emit(value, data);
-                    if(_this.store) _this.commitStore('SOCKET_'+value.toUpperCase(), data)
+                    if(_this.store) {
+                        const toCamelCase = (str) => str.split('_')
+                            .map(word => word.charAt(0).toUpperCase() + word.slice(1))
+                            .join('')
+
+                        _this.useActions
+                            ? _this.dispatchStore('socket' + toCamelCase(value), data)
+                            : _this.commitStore('SOCKET_' + value.toUpperCase(), data)
+                    }
                 })
             })
     }
@@ -48,4 +61,15 @@ export default class{
 
     }
 
+    dispatchStore(type, payload){
+
+        if(type.search(/socket/) === 0){
+
+            if(this.store._actions[type])
+                this.store.dispatch(type, payload)
+
+        }
+
+    }
+
 }