add google Bard

src/services/apis/bard-web.mjs

@@ -0,0 +1,26 @@
+import { pushRecord } from './shared.mjs'
+import Bard from '../clients/bard'
+
+/**
+ * @param {Runtime.Port} port
+ * @param {string} question
+ * @param {Session} session
+ * @param {string} cookies
+ */
+export async function generateAnswersWithBardWebApi(port, question, session, cookies) {
+  // const { controller, messageListener } = setAbortController(port)
+  const bot = new Bard(cookies)
+
+  // eslint-disable-next-line
+  try {
+    const { answer, conversationObj } = await bot.ask(question, session.bard_conversationObj || {})
+    session.bard_conversationObj = conversationObj
+    pushRecord(session, question, answer)
+    console.debug('conversation history', { content: session.conversationRecords })
+    // port.onMessage.removeListener(messageListener)
+    port.postMessage({ answer: answer, done: true, session: session })
+  } catch (err) {
+    // port.onMessage.removeListener(messageListener)
+    throw err
+  }
+}

src/services/clients/bard/index.mjs

@@ -0,0 +1,137 @@
+// https://github.com/PawanOsman/GoogleBard
+
+export default class Bard {
+  cookies = ''
+
+  constructor(cookies) {
+    this.cookies = cookies
+  }
+
+  ParseResponse(text) {
+    let resData = {
+      r: '',
+      c: '',
+      rc: '',
+      responses: [],
+    }
+    try {
+      let parseData = (data) => {
+        if (typeof data === 'string') {
+          if (data?.startsWith('c_')) {
+            resData.c = data
+            return
+          }
+          if (data?.startsWith('r_')) {
+            resData.r = data
+            return
+          }
+          if (data?.startsWith('rc_')) {
+            resData.rc = data
+            return
+          }
+          resData.responses.push(data)
+        }
+        if (Array.isArray(data)) {
+          data.forEach((item) => {
+            parseData(item)
+          })
+        }
+      }
+      try {
+        const lines = text.split('\n')
+        for (let i in lines) {
+          const line = lines[i]
+          if (line.includes('wrb.fr')) {
+            let data = JSON.parse(line)
+            let responsesData = JSON.parse(data[0][2])
+            responsesData.forEach((response) => {
+              parseData(response)
+            })
+          }
+        }
+      } catch (e) {
+        throw new Error(
+          `Error parsing response: make sure you are using the correct cookie, copy the value of "__Secure-1PSID" cookie and set it like this: \n\nnew Bard("__Secure-1PSID=<COOKIE_VALUE>")\n\nAlso using a US proxy is recommended.\n\nIf this error persists, please open an issue on github.\nhttps://github.com/PawanOsman/GoogleBard`,
+        )
+      }
+    } catch (err) {
+      throw new Error(
+        `Error parsing response: make sure you are using the correct cookie, copy the value of "__Secure-1PSID" cookie and set it like this: \n\nnew Bard("__Secure-1PSID=<COOKIE_VALUE>")\n\nAlso using a US proxy is recommended.\n\nIf this error persists, please open an issue on github.\nhttps://github.com/PawanOsman/GoogleBard`,
+      )
+    }
+    return resData
+  }
+
+  async GetRequestParams() {
+    try {
+      const response = await fetch('https://bard.google.com', {
+        headers: {
+          Cookie: this.cookies,
+        },
+      })
+      const text = await response.text()
+      const cfb2h = text.match(/"cfb2h":\s*"([^"]+)"/)?.[1]
+      const SNlM0e = text.match(/"SNlM0e":\s*"([^"]+)"/)?.[1]
+      const context = { googleData: { cfb2h, SNlM0e } }
+      const at = context.googleData.SNlM0e
+      const bl = context.googleData.cfb2h
+      return { at, bl }
+    } catch (e) {
+      throw new Error(
+        `Error parsing response: make sure you are using the correct cookie, copy the value of "__Secure-1PSID" cookie and set it like this: \n\nnew Bard("__Secure-1PSID=<COOKIE_VALUE>")\n\nAlso using a US proxy is recommended.\n\nIf this error persists, please open an issue on github.\nhttps://github.com/PawanOsman/GoogleBard`,
+      )
+    }
+  }
+
+  async ask(prompt, conversationObj) {
+    return await this.send(prompt, conversationObj)
+  }
+
+  async send(prompt, conversationObj) {
+    let conversation = {
+      id: conversationObj.id || '',
+      c: conversationObj.c || '',
+      r: conversationObj.r || '',
+      rc: conversationObj.rc || '',
+      lastActive: Date.now(),
+    }
+    // eslint-disable-next-line
+    try {
+      let { at, bl } = await this.GetRequestParams()
+      const response = await fetch(
+        'https://bard.google.com/_/BardChatUi/data/assistant.lamda.BardFrontendService/StreamGenerate?' +
+          new URLSearchParams({
+            bl: bl,
+            rt: 'c',
+            _reqid: 0,
+          }),
+        {
+          method: 'POST',
+          body: new URLSearchParams({
+            at: at,
+            'f.req': JSON.stringify([
+              null,
+              `[[${JSON.stringify(prompt)}],null,${JSON.stringify([
+                conversation.c,
+                conversation.r,
+                conversation.rc,
+              ])}]`,
+            ]),
+          }),
+          headers: {
+            Cookie: this.cookies,
+          },
+        },
+      )
+      const data = await response.text()
+      let parsedResponse = this.ParseResponse(data)
+      conversation.c = parsedResponse.c
+      conversation.r = parsedResponse.r
+      conversation.rc = parsedResponse.rc
+      const conversationObj = { c: conversation.c, r: conversation.r, rc: conversation.rc }
+      return { answer: parsedResponse.responses[0], conversationObj: conversationObj }
+    } catch (e) {
+      throw e
+    }
+  }
+}

src/services/init-session.mjs

@@ -24,6 +24,7 @@ import { v4 as uuidv4 } from 'uuid'
  * @property {string|null} bingWeb_parentMessageId
  * @property {Object|null} bingWeb_jailbreakConversationCache
  * @property {number|null} poe_chatId
+ * @property {object|null} bard_conversationObj
  */
 /**
  * @param {string|null} question
@@ -74,5 +75,8 @@ export function initSession({
 
     // poe
     poe_chatId: null,
+
+    // bard
+    bard_conversationObj: null,
   }
 }

src/services/wrappers.mjs

@@ -34,6 +34,12 @@ export async function getBingAccessToken() {
   return (await Browser.cookies.get({ url: 'https://bing.com/', name: '_U' }))?.value
 }
 
+export async function getBardCookies() {
+  const token = (await Browser.cookies.get({ url: 'https://google.com/', name: '__Secure-1PSID' }))
+    ?.value
+  return '__Secure-1PSID=' + token
+}
+
 export function registerPortListener(executor) {
   Browser.runtime.onConnect.addListener((port) => {
     console.debug('connected')