claudecodeui/server/routes/gemini.js

import express from 'express';
import sessionManager from '../sessionManager.js';
import { sessionNamesDb } from '../database/db.js';
import { getGeminiCliSessionMessages } from '../projects.js';

const router = express.Router();

router.get('/sessions/:sessionId/messages', async (req, res) => {
    try {
        const { sessionId } = req.params;

        if (!sessionId || typeof sessionId !== 'string' || !/^[a-zA-Z0-9_.-]{1,100}$/.test(sessionId)) {
            return res.status(400).json({ success: false, error: 'Invalid session ID format' });
        }

        let messages = sessionManager.getSessionMessages(sessionId);

        // Fallback to Gemini CLI sessions on disk
        if (messages.length === 0) {
            messages = await getGeminiCliSessionMessages(sessionId);
        }

        res.json({
            success: true,
            messages: messages,
            total: messages.length,
            hasMore: false,
            offset: 0,
            limit: messages.length
        });
    } catch (error) {
        console.error('Error fetching Gemini session messages:', error);
        res.status(500).json({ success: false, error: error.message });
    }
});

router.delete('/sessions/:sessionId', async (req, res) => {
    try {
        const { sessionId } = req.params;

        if (!sessionId || typeof sessionId !== 'string' || !/^[a-zA-Z0-9_.-]{1,100}$/.test(sessionId)) {
            return res.status(400).json({ success: false, error: 'Invalid session ID format' });
        }

        await sessionManager.deleteSession(sessionId);
        sessionNamesDb.deleteName(sessionId, 'gemini');
        res.json({ success: true });
    } catch (error) {
        console.error(`Error deleting Gemini session ${req.params.sessionId}:`, error);
        res.status(500).json({ success: false, error: error.message });
    }
});

export default router;