fix undefined state issue

wildlifela · May 2, 2016 · ffb103c · ffb103c
1 parent 94a2232
commit ffb103c
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/.babelrc b/.babelrc
@@ -0,0 +1,3 @@
+{
+  "presets": ["es2015", "stage-2"]
+}
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "redux-persist-migrate",
-  "version": "3.1.1",
+  "version": "3.1.2",
   "description": "migrate your redux state",
   "main": "lib/index.js",
   "scripts": {
@@ -30,6 +30,8 @@
   "devDependencies": {
     "babel-cli": "^6.7.7",
     "babel-eslint": "^6.0.4",
+    "babel-preset-es2015": "^6.6.0",
+    "babel-preset-stage-2": "^6.5.0",
     "babel-register": "^6.7.2",
     "rimraf": "^2.5.2",
     "standard": "^6.0.8"

diff --git a/src/index.js b/src/index.js
@@ -3,9 +3,9 @@ import { REHYDRATE } from 'redux-persist/constants'
 export default function createMigration (manifest, versionSelector, versionSetter) {
   if (typeof versionSelector === 'string') {
     let versionString = versionSelector
-    versionSelector = (state) => state[versionString].version
+    versionSelector = (state) => state && state[versionString] && state[versionString].version
     versionSetter = (state, version) => {
-      state[versionString].version = version
+      if (state && state[versionString]) state[versionString].version = version
       return state
     }
   }