Create index.js

index.js

@@ -0,0 +1,144 @@
+const os = require('os')
+const uuid = require('uuid')
+const bytes = require('bytes')
+const logger = require('morgan')
+const express = require('express')
+const CharacterAI = require('node_characterai')
+const { executablePath } = require('puppeteer')
+
+const sessions = new Map()
+
+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()
+	for (let key in used) obj[key] = formatSize(used[key])
+	
+	const totalmem = os.totalmem()
+	const freemem = os.freemem()
+	obj.memoryUsage = `${formatSize(totalmem - freemem)} / ${formatSize(totalmem)}`
+	
+	res.json({
+		success: true,
+		message: 'Hello World!',
+		uptime: new Date(process.uptime() * 1000).toUTCString().split(' ')[4],
+		status: obj
+	})
+})
+
+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
+		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 }
+			})
+			return
+		}
+		
+		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 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 }
+		})
+	} catch (e) {
+		console.log(e)
+		res.json({
+			error: true,
+			message: String(e) === '[object Object]' ? 'Internal Server Error' : String(e)
+		})
+	}
+})
+
+app.get('/api/chara/info', async (req, res) => {
+	try {
+		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)
+		})
+	}
+})
+
+app.get('/api/chara/search', async (req, res) => {
+	try {
+		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' })
+		
+		res.json({ success: true, message: '', result: result.characters })
+	} catch (e) {
+		console.log(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))
+
+function formatSize(num) {
+	return bytes(+num, { unitSeparator: ' ' })
+}