diff --git a/plugins/cookie.chrome.js b/plugins/cookie.chrome.js index e2d4278..a0a43f9 100644 --- a/plugins/cookie.chrome.js +++ b/plugins/cookie.chrome.js @@ -4,6 +4,7 @@ var log = require('../log'); var Plugin = require('../plugin'); var Queue = require('../queue'); var session = require('../session'); +var config = require('../config'); // [Usage] // @@ -134,9 +135,14 @@ Chrome.getCookies = function(cb) { db.on('error', cb); var KEYS = ['csrftoken', 'LEETCODE_SESSION']; + let host = config.sys.cookie_host; + if (!host) { + host = 'leetcode.com'; // default host key + } + db.serialize(function() { var cookies = {}; - var sql = 'select name, encrypted_value from cookies where host_key like "%leetcode.com"'; + var sql = 'select name, encrypted_value from cookies where host_key like "%' + host +'"'; db.each(sql, function(e, x) { if (e) return cb(e); if (KEYS.indexOf(x.name) < 0) return; diff --git a/plugins/cookie.firefox.js b/plugins/cookie.firefox.js index f69f869..ded46bc 100644 --- a/plugins/cookie.firefox.js +++ b/plugins/cookie.firefox.js @@ -1,6 +1,7 @@ var log = require('../log'); var Plugin = require('../plugin'); var session = require('../session'); +var config = require('../config'); // [Usage] // @@ -34,9 +35,14 @@ function getCookies(cb) { var db = new sqlite3.Database(files[0]); var KEYS = ['csrftoken', 'LEETCODE_SESSION']; + let host = config.sys.cookie_host; + if (!host) { + host = 'leetcode.com'; // default host key + } + db.serialize(function() { var cookies = {}; - var sql = 'select name, value from moz_cookies where host like "%leetcode.com"'; + var sql = 'select name, value from moz_cookies where host like "%' + host +'"'; db.each(sql, function(e, x) { if (e) return cb(e); if (KEYS.indexOf(x.name) < 0) return; diff --git a/plugins/leetcode.cn.js b/plugins/leetcode.cn.js index 07d5d2d..8094ba6 100644 --- a/plugins/leetcode.cn.js +++ b/plugins/leetcode.cn.js @@ -31,6 +31,7 @@ plugin.init = function() { config.sys.urls.verify = 'https://leetcode-cn.com/submissions/detail/$id/check/'; config.sys.urls.favorites = 'https://leetcode-cn.com/list/api/questions'; config.sys.urls.favorite_delete = 'https://leetcode-cn.com/list/api/questions/$hash/$id'; + config.sys.cookie_host = 'leetcode-cn.com'; }; // FIXME: refactor those