From e21a690e9eb31dff526b1b8b435d8077c9500349 Mon Sep 17 00:00:00 2001
From: Lawrence Onah <paplow01@gmail.com>
Date: Thu, 19 Jan 2023 14:31:08 +0100
Subject: [PATCH] chore: add support for userDataDir

---
 package.json               |  1 +
 src/chatgpt-api-browser.ts | 13 +++++++++++--
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/package.json b/package.json
index bb33cfb..e775416 100644
--- a/package.json
+++ b/package.json
@@ -47,6 +47,7 @@
     "random": "^4.1.0",
     "remark": "^14.0.2",
     "strip-markdown": "^5.0.0",
+    "tempy": "^3.0.0",
     "uuid": "^9.0.0"
   },
   "devDependencies": {
diff --git a/src/chatgpt-api-browser.ts b/src/chatgpt-api-browser.ts
index 866316a..7644fdb 100644
--- a/src/chatgpt-api-browser.ts
+++ b/src/chatgpt-api-browser.ts
@@ -1,5 +1,6 @@
 import delay from 'delay'
 import type { Browser, HTTPRequest, HTTPResponse, Page } from 'puppeteer'
+import { temporaryDirectory } from 'tempy'
 import { v4 as uuidv4 } from 'uuid'
 
 import * as types from './types'
@@ -37,6 +38,7 @@ export class ChatGPTAPIBrowser extends AChatGPTAPI {
     string,
     (partialResponse: types.ChatResponse) => void
   >
+  protected _userDataDir: string
 
   /**
    * Creates a new client for automating the ChatGPT webapp.
@@ -71,6 +73,9 @@ export class ChatGPTAPIBrowser extends AChatGPTAPI {
 
     /** @defaultValue `undefined` **/
     proxyServer?: string
+
+    /** @defaultValue `random directory with email as prefix` **/
+    userDataDir?: string
   }) {
     super()
 
@@ -85,7 +90,8 @@ export class ChatGPTAPIBrowser extends AChatGPTAPI {
       captchaToken,
       nopechaKey,
       executablePath,
-      proxyServer
+      proxyServer,
+      userDataDir
     } = opts
 
     this._email = email
@@ -102,6 +108,8 @@ export class ChatGPTAPIBrowser extends AChatGPTAPI {
     this._proxyServer = proxyServer
     this._isRefreshing = false
     this._messageOnProgressHandlers = {}
+    this._userDataDir =
+      userDataDir ?? temporaryDirectory({ prefix: this._email })
 
     if (!this._email) {
       const error = new types.ChatGPTError('ChatGPT invalid email')
@@ -127,7 +135,8 @@ export class ChatGPTAPIBrowser extends AChatGPTAPI {
         nopechaKey: this._nopechaKey,
         executablePath: this._executablePath,
         proxyServer: this._proxyServer,
-        minimize: this._minimize
+        minimize: this._minimize,
+        userDataDir: this._userDataDir
       })
 
       this._page = await getPage(this._browser, {