Fix the issue that the modifying responde body's rule cannot be added.

当存在旧规则数据库时添加 receiveBody 字段报错，升级数据库版本号，触发 onupgradeneeded 事件后添加。
ajksdhfueisde · Nov 11, 2019 · c51080a · c51080a
1 parent 5276e18
commit c51080a
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/src/core/rules.js b/src/core/rules.js
@@ -214,8 +214,11 @@ function init() {
 		if (cache.receiveHeader === null) {
 			queue.push(updateCache('receiveHeader'));
 		}
+		if (cache.receiveBody === null) {
+			queue.push(updateCache('receiveBody'));
+		}
 		Promise.all(queue).then(() => {
-			if (cache.request === null || cache.sendHeader === null || cache.receiveHeader === null) {
+			if (cache.request === null || cache.sendHeader === null || cache.receiveHeader === null || cache.receiveBody === null) {
 				init();
 			}
 		});

diff --git a/src/core/storage.js b/src/core/storage.js
@@ -6,7 +6,7 @@ import notify from './notify';
 
 function getDatabase() {
 	return new Promise((resolve, reject) => {
-		const dbOpenRequest = window.indexedDB.open("headereditor", 3);
+		const dbOpenRequest = window.indexedDB.open("headereditor", 4);
 		dbOpenRequest.onsuccess = function(e) {
 			resolve(e.target.result);
 		};
@@ -23,6 +23,10 @@ function getDatabase() {
 			} else {
 				utils.TABLE_NAMES.forEach(k => {
 					const tx = event.target.transaction;
+					if(!tx.objectStoreNames.contains(k)){
+						event.target.result.createObjectStore(k, {keyPath: 'id', autoIncrement: true});
+						return;
+					}
 					const os = tx.objectStore(k);
 					os.openCursor().onsuccess = function(e) {
 						const cursor = e.target.result;

diff --git a/src/core/utils.js b/src/core/utils.js
@@ -20,7 +20,7 @@ export default {
 	IS_CHROME: IS_CHROME,
 	CHROME_VERSION: CHROME_VERSION,
 	FIREFOX_VERSION: FIREFOX_VERSION,
-	TABLE_NAMES: ['request', 'sendHeader', 'receiveHeader'],
+	TABLE_NAMES: ['request', 'sendHeader', 'receiveHeader', 'receiveBody'],
 	getExportName(additional) {
 		return 'HE_' + dateFormat(new Date(), 'isoUtcDateTime').replace(/\:/g, '-') + (additional ? "_" + additional : "") + '.json';
 	},
@@ -93,6 +93,9 @@ export default {
 		if (ruleType === 'modifyReceiveHeader') {
 			return 'receiveHeader';
 		}
+		if (ruleType === 'modifyReceiveBody') {
+			return 'receiveBody';
+		}
 	},
 	upgradeRuleFormat(s) {
 		if (typeof(s.matchType) === "undefined") {