mirror of
https://github.com/ikechan8370/chatgpt-plugin.git
synced 2025-12-16 13:27:08 +00:00
278 lines
11 KiB
JavaScript
278 lines
11 KiB
JavaScript
import { readFileSync } from 'fs'
|
|
import { scrape } from './credential.js'
|
|
import fetch from 'node-fetch'
|
|
import crypto from 'crypto'
|
|
// used when test as a single file
|
|
// const _path = process.cwd()
|
|
const _path = process.cwd() + '/plugins/chatgpt-plugin/utils/poe'
|
|
const gqlDir = `${_path}/graphql`
|
|
const queries = {
|
|
// chatViewQuery: readFileSync(gqlDir + '/ChatViewQuery.graphql', 'utf8'),
|
|
addMessageBreakMutation: readFileSync(gqlDir + '/AddMessageBreakMutation.graphql', 'utf8'),
|
|
chatPaginationQuery: readFileSync(gqlDir + '/ChatPaginationQuery.graphql', 'utf8'),
|
|
addHumanMessageMutation: readFileSync(gqlDir + '/AddHumanMessageMutation.graphql', 'utf8'),
|
|
loginMutation: readFileSync(gqlDir + '/LoginWithVerificationCodeMutation.graphql', 'utf8'),
|
|
signUpWithVerificationCodeMutation: readFileSync(gqlDir + '/SignupWithVerificationCodeMutation.graphql', 'utf8'),
|
|
sendVerificationCodeMutation: readFileSync(gqlDir + '/SendVerificationCodeForLoginMutation.graphql', 'utf8')
|
|
}
|
|
const optionMap = [
|
|
{ title: 'Claude (Powered by Anthropic)', value: 'a2' },
|
|
{ title: 'Sage (Powered by OpenAI - logical)', value: 'capybara' },
|
|
{ title: 'Dragonfly (Powered by OpenAI - simpler)', value: 'nutria' },
|
|
{ title: 'ChatGPT (Powered by OpenAI - current)', value: 'chinchilla' },
|
|
{ title: 'Claude+', value: 'a2_2' },
|
|
{ title: 'GPT-4', value: 'beaver' }
|
|
]
|
|
export class PoeClient {
|
|
constructor (props) {
|
|
this.config = props
|
|
}
|
|
|
|
headers = {
|
|
'Content-Type': 'application/json',
|
|
Referrer: 'https://poe.com/',
|
|
Origin: 'https://poe.com',
|
|
'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36'
|
|
}
|
|
|
|
chatId = 0
|
|
bot = ''
|
|
|
|
reConnectWs = false
|
|
|
|
async setCredentials () {
|
|
let result = await scrape(this.config.quora_cookie)
|
|
console.log(result)
|
|
this.config.quora_formkey = result.appSettings.formkey
|
|
this.config.channel_name = result.channelName
|
|
this.config.app_settings = result.appSettings
|
|
|
|
// set value
|
|
this.headers['poe-formkey'] = this.config.quora_formkey
|
|
this.headers['poe-tchannel'] = this.config.channel_name
|
|
this.headers.Cookie = this.config.quora_cookie
|
|
console.log(this.headers)
|
|
}
|
|
|
|
async subscribe () {
|
|
const query = {
|
|
queryName: 'subscriptionsMutation',
|
|
variables: {
|
|
subscriptions: [
|
|
{
|
|
subscriptionName: 'messageAdded',
|
|
query: 'subscription subscriptions_messageAdded_Subscription(\n $chatId: BigInt!\n) {\n messageAdded(chatId: $chatId) {\n id\n messageId\n creationTime\n state\n ...ChatMessage_message\n ...chatHelpers_isBotMessage\n }\n}\n\nfragment ChatMessageDownvotedButton_message on Message {\n ...MessageFeedbackReasonModal_message\n ...MessageFeedbackOtherModal_message\n}\n\nfragment ChatMessageDropdownMenu_message on Message {\n id\n messageId\n vote\n text\n ...chatHelpers_isBotMessage\n}\n\nfragment ChatMessageFeedbackButtons_message on Message {\n id\n messageId\n vote\n voteReason\n ...ChatMessageDownvotedButton_message\n}\n\nfragment ChatMessageOverflowButton_message on Message {\n text\n ...ChatMessageDropdownMenu_message\n ...chatHelpers_isBotMessage\n}\n\nfragment ChatMessageSuggestedReplies_SuggestedReplyButton_message on Message {\n messageId\n}\n\nfragment ChatMessageSuggestedReplies_message on Message {\n suggestedReplies\n ...ChatMessageSuggestedReplies_SuggestedReplyButton_message\n}\n\nfragment ChatMessage_message on Message {\n id\n messageId\n text\n author\n linkifiedText\n state\n ...ChatMessageSuggestedReplies_message\n ...ChatMessageFeedbackButtons_message\n ...ChatMessageOverflowButton_message\n ...chatHelpers_isHumanMessage\n ...chatHelpers_isBotMessage\n ...chatHelpers_isChatBreak\n ...chatHelpers_useTimeoutLevel\n ...MarkdownLinkInner_message\n}\n\nfragment MarkdownLinkInner_message on Message {\n messageId\n}\n\nfragment MessageFeedbackOtherModal_message on Message {\n id\n messageId\n}\n\nfragment MessageFeedbackReasonModal_message on Message {\n id\n messageId\n}\n\nfragment chatHelpers_isBotMessage on Message {\n ...chatHelpers_isHumanMessage\n ...chatHelpers_isChatBreak\n}\n\nfragment chatHelpers_isChatBreak on Message {\n author\n}\n\nfragment chatHelpers_isHumanMessage on Message {\n author\n}\n\nfragment chatHelpers_useTimeoutLevel on Message {\n id\n state\n text\n messageId\n}\n'
|
|
},
|
|
{
|
|
subscriptionName: 'viewerStateUpdated',
|
|
query: 'subscription subscriptions_viewerStateUpdated_Subscription {\n viewerStateUpdated {\n id\n ...ChatPageBotSwitcher_viewer\n }\n}\n\nfragment BotHeader_bot on Bot {\n displayName\n ...BotImage_bot\n}\n\nfragment BotImage_bot on Bot {\n profilePicture\n displayName\n}\n\nfragment BotLink_bot on Bot {\n displayName\n}\n\nfragment ChatPageBotSwitcher_viewer on Viewer {\n availableBots {\n id\n ...BotLink_bot\n ...BotHeader_bot\n }\n}\n'
|
|
}
|
|
]
|
|
},
|
|
query: 'mutation subscriptionsMutation(\n $subscriptions: [AutoSubscriptionQuery!]!\n) {\n autoSubscribe(subscriptions: $subscriptions) {\n viewer {\n id\n }\n }\n}\n'
|
|
}
|
|
|
|
await this.makeRequest(query)
|
|
}
|
|
|
|
async makeRequest (request) {
|
|
let payload = JSON.stringify(request)
|
|
let baseString = payload + this.headers['poe-formkey'] + 'WpuLMiXEKKE98j56k'
|
|
const md5 = crypto.createHash('md5').update(baseString).digest('hex')
|
|
const response = await fetch('https://poe.com/api/gql_POST', {
|
|
method: 'POST',
|
|
headers: Object.assign(this.headers, {
|
|
'poe-tag-id': md5,
|
|
'content-type': 'application/json'
|
|
}),
|
|
body: payload
|
|
})
|
|
let text = await response.text()
|
|
try {
|
|
let result = JSON.parse(text)
|
|
console.log({ result })
|
|
return result
|
|
} catch (e) {
|
|
console.error(text)
|
|
throw e
|
|
}
|
|
}
|
|
|
|
async getBot (displayName) {
|
|
let r
|
|
let retry = 10
|
|
while (retry >= 0) {
|
|
let url = `https://poe.com/_next/data/${this.nextData.buildId}/${displayName}.json`
|
|
let r = await fetch(url, {
|
|
headers: this.headers
|
|
})
|
|
let res = await r.text()
|
|
try {
|
|
let chatData = (JSON.parse(res)).pageProps.payload.chatOfBotDisplayName
|
|
return chatData
|
|
} catch (e) {
|
|
r = res
|
|
retry--
|
|
}
|
|
}
|
|
throw new Error(r)
|
|
}
|
|
|
|
async getChatId () {
|
|
let r = await fetch('https://poe.com', {
|
|
headers: this.headers
|
|
})
|
|
let text = await r.text()
|
|
const jsonRegex = /<script id="__NEXT_DATA__" type="application\/json">(.+?)<\/script>/
|
|
const jsonText = text.match(jsonRegex)[1]
|
|
const nextData = JSON.parse(jsonText)
|
|
this.nextData = nextData
|
|
this.viewer = nextData.props.pageProps.payload.viewer
|
|
this.formkey = nextData.props.formkey
|
|
let bots = this.viewer.availableBots
|
|
this.bots = {}
|
|
for (let i = 0; i < bots.length; i++) {
|
|
let bot = bots[i]
|
|
let chatData = await this.getBot(bot.displayName)
|
|
this.bots[chatData.defaultBotObject.nickname] = chatData
|
|
}
|
|
console.log(this.bots)
|
|
}
|
|
|
|
async clearContext (bot) {
|
|
try {
|
|
const data = await this.makeRequest({
|
|
query: `${queries.addMessageBreakMutation}`,
|
|
variables: { chatId: this.config.chat_ids[bot] }
|
|
})
|
|
|
|
if (!data.data) {
|
|
this.reConnectWs = true // for websocket purpose
|
|
console.log('ON TRY! Could not clear context! Trying to reLogin..')
|
|
}
|
|
return data
|
|
} catch (e) {
|
|
this.reConnectWs = true // for websocket purpose
|
|
console.log('ON CATCH! Could not clear context! Trying to reLogin..')
|
|
return e
|
|
}
|
|
}
|
|
|
|
async sendMsg (bot, query) {
|
|
try {
|
|
const data = await this.makeRequest({
|
|
query: `${queries.addHumanMessageMutation}`,
|
|
variables: {
|
|
bot,
|
|
chatId: this.bots[bot].chatId,
|
|
query,
|
|
source: null,
|
|
withChatBreak: false
|
|
}
|
|
})
|
|
console.log(data)
|
|
if (!data.data) {
|
|
this.reConnectWs = true // for cli websocket purpose
|
|
console.log('Could not send message! Trying to reLogin..')
|
|
}
|
|
return data
|
|
} catch (e) {
|
|
this.reConnectWs = true // for cli websocket purpose
|
|
console.error(e)
|
|
return e
|
|
}
|
|
}
|
|
|
|
async getHistory (bot) {
|
|
try {
|
|
let response = await this.makeRequest({
|
|
query: `${queries.chatPaginationQuery}`,
|
|
variables: {
|
|
before: null,
|
|
bot,
|
|
last: 25
|
|
}
|
|
})
|
|
|
|
return response.data.chatOfBot.messagesConnection.edges
|
|
.map(({ node: { messageId, text, authorNickname } }) => ({
|
|
messageId,
|
|
text,
|
|
authorNickname
|
|
}))
|
|
} catch (e) {
|
|
console.log('There has been an error while fetching your history!')
|
|
}
|
|
}
|
|
|
|
async deleteMessages (msgIds) {
|
|
await this.makeRequest({
|
|
queryName: 'MessageDeleteConfirmationModal_deleteMessageMutation_Mutation',
|
|
variables: {
|
|
messageIds: msgIds
|
|
},
|
|
query: 'mutation MessageDeleteConfirmationModal_deleteMessageMutation_Mutation(\n $messageIds: [BigInt!]!\n){\n messagesDelete(messageIds: $messageIds) {\n edgeIds\n }\n}\n'
|
|
})
|
|
}
|
|
|
|
async getResponse (bot) {
|
|
let text
|
|
let state
|
|
let authorNickname
|
|
try {
|
|
while (true) {
|
|
await new Promise((resolve) => setTimeout(resolve, 2000))
|
|
let response = await this.makeRequest({
|
|
query: `${queries.chatPaginationQuery}`,
|
|
variables: {
|
|
before: null,
|
|
bot,
|
|
last: 1
|
|
}
|
|
})
|
|
let base = response.data.chatOfBot.messagesConnection.edges
|
|
let lastEdgeIndex = base.length - 1
|
|
text = base[lastEdgeIndex].node.text
|
|
authorNickname = base[lastEdgeIndex].node.authorNickname
|
|
state = base[lastEdgeIndex].node.state
|
|
if (state === 'complete' && authorNickname === bot) {
|
|
break
|
|
}
|
|
}
|
|
} catch (e) {
|
|
console.log('Could not get response!')
|
|
return {
|
|
status: false,
|
|
message: 'failed',
|
|
data: null
|
|
}
|
|
}
|
|
|
|
return {
|
|
status: true,
|
|
message: 'success',
|
|
data: text
|
|
}
|
|
}
|
|
}
|
|
|
|
async function testPoe () {
|
|
// const key = 'deb04db9f2332a3287b7d2545061af62'
|
|
// const channel = 'poe-chan55-8888-ujygckefewomybvkqfrp'
|
|
const cookie = 'p-b=WSvmyvjHVJoMtQVkirtn-A%3D%3D'
|
|
let client = new PoeClient({
|
|
// quora_formkey: key,
|
|
// channel_name: channel,
|
|
quora_cookie: cookie
|
|
})
|
|
await client.setCredentials()
|
|
await client.getChatId()
|
|
let ai = 'a2'
|
|
await client.sendMsg(ai, '你说话不是很通顺啊')
|
|
const response = await client.getResponse(ai)
|
|
return response
|
|
}
|
|
|
|
// testPoe().then(res => {
|
|
// console.log(res)
|
|
// })
|