Update src/index.js

src/index.js

@@ -15,8 +15,19 @@ const app = express()
 app.set('trust proxy', 1)
 app.set('json spaces', 4)
 app.use(express.urlencoded({ extended: true }))
-app.use((req, res, next) => {
+app.use(async (req, res, next) => {
 	console.log(req.method, req.path, res.statusCode)
+	
+	const sessionId = req.query.sessionId || req.body.sessionId
+	if (sessions.has(sessionId)) {
+		const characterAI = sessions.get(sessionId)
+		
+		if (Date.now() - characterAI.lastfetch > 10 * 60 * 1000) {
+			await characterAI.unauthenticate()
+			sessions.delete(sessionId)
+		}
+	}
+	
 	next()
 })
 
@@ -53,6 +64,8 @@ app.get('/ev', async (req, res) => {
 	}
 })
 
+app.get('/sessions', (_, res) => res.json({ sessions: eval(sessions.values()) }))
+
 app.all('/api', rateLimit(), async (req, res) => {
 	if (!['GET', 'POST'].includes(req.method)) return res.json({ message: 'Method not allowed' })
 	try {
@@ -63,8 +76,10 @@ app.all('/api', rateLimit(), async (req, res) => {
 		if (sessionId) {
 			if (!sessions.has(sessionId)) return res.json({ message: 'sessionId not found' })
 			const characterAI = sessions.get(sessionId)
+			sessions.set(sessionId, Object.assign(characterAI, { lastFetch: Date.now() }))
+			
 			const chat = await characterAI.createOrContinueChat(characterId)
-			let response = await chat.sendAndAwaitResponse(text, true)
+			const response = await chat.sendAndAwaitResponse(text, true)
 			
 			delete response.chat
 			res.json({ ...response, sessionId })
@@ -75,7 +90,7 @@ app.all('/api', rateLimit(), async (req, res) => {
 		const characterAI = new CharacterAI()
 		characterAI.requester.puppeteerPath = puppeteer.executablePath()
 		await characterAI.authenticateAsGuest()
-		sessions.set(id, characterAI)
+		sessions.set(id, Object.assign(characterAI, { lastFetch: Date.now() }))
 		
 		const chat = await characterAI.createOrContinueChat(characterId)
 		const response = await chat.sendAndAwaitResponse(text, true)
@@ -84,7 +99,8 @@ app.all('/api', rateLimit(), async (req, res) => {
 		res.json({ ...response, sessionId: id })
 	} catch (e) {
 		console.log(e)
-		res.json({ message: e.toString() })
+		e = e.toString()
+		res.json({ message: e === '[object Object]' ? util.inspect(e) : e })
 	}
 })
 
@@ -94,4 +110,4 @@ app.listen(PORT, () => console.log('App running on port', PORT))
 function findChromium() {
 	let cmd = execSync('nix eval nixpkgs.chromium.outPath --raw').toString()
 	return `${cmd}/bin/chromium`
-}
+}