Update index.js

index.js

@@ -1,29 +1,18 @@
 const os = require('os')
-const uuid = require('uuid')
 const bytes = require('bytes')
 const logger = require('morgan')
 const express = require('express')
+const puppeteer = require('puppeteer')
 const CharacterAI = require('node_characterai')
-const { executablePath } = require('puppeteer')
+const Parser = require('node_characterai/parser')
 
-const sessions = new Map()
+const characterAI = new CharacterAI()
+characterAI.puppeteerPath = puppeteer.executablePath()
 
 const app = express()
-app.set('trust proxy', 1)
 app.set('json spaces', 4)
 app.use(logger('dev'))
 
-app.use(async (req, res, next) => {
-	for (let [sessionId, characterAI] of sessions) {
-		const timeReset = 10 * 60 * 1000 // 10 min
-		if (Date.now() - characterAI.lastFetch > timeReset) {
-			await characterAI.unauthenticate()
-			sessions.delete(sessionId)
-		}
-	}
-	next()
-})
-
 app.all('/', (req, res) => {
 	const obj = {}
 	const used = process.memoryUsage()
@@ -41,56 +30,36 @@ app.all('/', (req, res) => {
 	})
 })
 
-app.get('/ip', (req, res) => res.send(req.ip))
-
-app.get('/sessions', (_, res) => res.send(eval(Array.from(sessions))))
-
 app.get('/api', async (req, res) => {
 	try {
-		const { characterId, text, sessionId } = req.query
+		let { characterId, text, sessionId } = req.query
 		if (!characterId) return res.json({ success: false, message: 'Input parameter characterId' })
 		if (!text) return res.json({ success: false, message: 'Input parameter text' })
 		
-		if (sessionId) {
-			if (!sessions.has(sessionId)) return res.json({ success: false, message: 'sessionId not found' })
-			const characterAI = sessions.get(sessionId)
-			sessions.set(sessionId, Object.assign(characterAI, { lastFetch: Date.now() }))
-			
-			const chat = await characterAI.createOrContinueChat(characterId)
-			const response = await chat.sendAndAwaitResponse(text, true)
-			const urlAvatar = `https://characterai.io/i/80/static/avatars/${response.srcAvatarFileName}`
-			
-			delete response.chat
-			res.json({
-				success: true,
-				message: 'sessionId will be deleted in 10 minutes if not used',
-				result: { ...response, urlAvatar, sessionId }
+		if (!sessionId) {
+			const request = await characterAI.requester.request('https://beta.character.ai/chat/history/create/', {
+				headers: characterAI.getHeaders(), method: 'POST',
+				body: Parser.stringify({ character_external_id: characterId })
 			})
-			return
+			if (!request.ok()) return res.json({ success: false, message: 'Couldn\'t create a new chat' })
+			
+			const json = await Parser.parseJSON(request)
+			sessionId = json.external_id
 		}
 		
-		const id = uuid.v4()
-		const characterAI = new CharacterAI()
-		characterAI.requester.puppeteerPath = executablePath()
-		await characterAI.authenticateWithToken(process.env.cToken)
-		sessions.set(id, Object.assign(characterAI, { lastFetch: Date.now() }))
-		
-		const chat = await characterAI.createOrContinueChat(characterId)
+		const chat = await characterAI.createOrContinueChat(characterId, sessionId)
 		const response = await chat.sendAndAwaitResponse(text, true)
 		const urlAvatar = `https://characterai.io/i/80/static/avatars/${response.srcAvatarFileName}`
-		
+			
 		delete response.chat
 		res.json({
 			success: true,
-			message: 'sessionId will be deleted in 10 minutes if not used',
-			result: { ...response, urlAvatar, sessionId: id }
+			message: '',
+			result: { ...response, urlAvatar, sessionId }
 		})
 	} catch (e) {
 		console.log(e)
-		res.json({
-			error: true,
-			message: String(e) === '[object Object]' ? 'Internal Server Error' : String(e)
-		})
+		res.json({ error: true, message: String(e) === '[object Object]' ? 'Internal Server Error' : String(e) })
 	}
 })
 
@@ -99,19 +68,13 @@ app.get('/api/chara/info', async (req, res) => {
 		const { characterId } = req.query
 		if (!characterId) return res.json({ success: false, message: 'Input parameter characterId' })
 		
-		const characterAI = new CharacterAI()
-		characterAI.requester.puppeteerPath = executablePath()
-		await characterAI.authenticateWithToken(process.env.cToken)
 		const result = await characterAI.fetchCharacterInfo(characterId)
 		if (!result) return res.json({ success: false, message: 'Invalid characterId' })
 		
 		res.json({ success: true, message: '', result })
 	} catch (e) {
 		console.log(e)
-		res.json({
-			error: true,
-			message: String(e) === '[object Object]' ? 'Internal Server Error' : String(e)
-		})
+		res.json({ error: true, message: String(e) === '[object Object]' ? 'Internal Server Error' : String(e) })
 	}
 })
 
@@ -120,25 +83,22 @@ app.get('/api/chara/search', async (req, res) => {
 		const { name } = req.query
 		if (!name) return res.json({ success: false, message: 'Input parameter name' })
 		
-		const characterAI = new CharacterAI()
-		characterAI.requester.puppeteerPath = executablePath()
-		await characterAI.authenticateWithToken(process.env.cToken)
-		const result = await characterAI.searchCharacters(name)
-		if (!result.characters.length) return res.json({ success: false, message: 'Character not found' })
+		const { characters } = await characterAI.searchCharacters(name)
+		if (!characters.length) return res.json({ success: false, message: 'Character not found' })
 		
-		res.json({ success: true, message: '', result: result.characters })
+		res.json({ success: true, message: '', result: characters })
 	} catch (e) {
 		console.log(e)
-		res.json({
-			error: true,
-			message: String(e) === '[object Object]' ? 'Internal Server Error' : String(e)
-		})
+		res.json({ error: true, message: String(e) === '[object Object]' ? 'Internal Server Error' : String(e) })
 	}
 })
 
 const PORT = process.env.PORT || 7860
-app.listen(PORT, () => console.log('App running on port', PORT))
+app.listen(PORT, async () => {
+	console.log('App running on port', PORT)
+	await characterAI.authenticateWithToken(process.env.cToken)
+})
 
 function formatSize(num) {
 	return bytes(+num, { unitSeparator: ' ' })
-}
+}