From e075f11bdb1b194062d153cad181a6b04ad8fcfc Mon Sep 17 00:00:00 2001
From: Evgeny Zinoviev <me@ch1p.io>
Date: Tue, 15 Mar 2022 03:56:59 +0300
Subject: much fixes so improvements

---
 lib/cookies-storage.js | 80 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 80 insertions(+)
 create mode 100644 lib/cookies-storage.js

(limited to 'lib/cookies-storage.js')

diff --git a/lib/cookies-storage.js b/lib/cookies-storage.js
new file mode 100644
index 0000000..15714ca
--- /dev/null
+++ b/lib/cookies-storage.js
@@ -0,0 +1,80 @@
+const fs = require('fs').promises
+const {Mutex} = require('async-mutex')
+const {getLogger} = require('./logging')
+
+const logger = getLogger('CookiesStorage')
+
+let storageFileName = null
+const mutex = new Mutex()
+
+async function exists(path) {
+    try {
+        await fs.access(path)
+        return true
+    } catch {
+        return false
+    }
+}
+
+function cookiesAsHashed(cookies) {
+    if (!cookies.length)
+        return {}
+
+    const map = {}
+    const sep = '|;|;'
+    for (const c of cookies) {
+        const k = `${c.domain}${sep}${c.path}${sep}${c.name}`
+        map[k] = c
+    }
+    return map
+}
+
+module.exports = {
+    async save(newCookies) {
+        let currentCookies = await this.get()
+
+        await mutex.runExclusive(async () => {
+            if (currentCookies.length) {
+                for (let newCookie of newCookies) {
+                    if (!currentCookies.length)
+                        break
+                    let i = currentCookies.findIndex((oldCookie) => {
+                        return oldCookie.domain === newCookie.domain
+                            && oldCookie.path === newCookie.path
+                            && oldCookie.name === newCookie.name
+                            && (
+                                oldCookie.value !== newCookie.value
+                                || oldCookie.expires !== newCookie.expires
+                            )
+                    })
+                    if (i !== -1) {
+                        let removed = currentCookies.splice(i, 1)
+                        // logger.debug('removed cookie:', removed, 'instead got new one:', newCookie)
+                    }
+                }
+            }
+
+            const cookiesMap = Object.assign({}, cookiesAsHashed(currentCookies), cookiesAsHashed(newCookies))
+            await fs.writeFile(storageFileName, JSON.stringify(Object.values(cookiesMap), null, 2), 'utf-8')
+        })
+    },
+
+    async get() {
+        if (!(await exists(storageFileName)))
+            return []
+
+        try {
+            const raw = await mutex.runExclusive(async () => {
+                return await fs.readFile(storageFileName, 'utf-8')
+            })
+            return JSON.parse(raw)
+        } catch (e) {
+            logger.error('Failed to parse storage:', e)
+            return []
+        }
+    },
+
+    setFileName(name) {
+        storageFileName = name
+    }
+};
\ No newline at end of file
-- 
cgit v1.2.3