interviews-agent/vite-plugin-api.js at main · runtimerevolution/interviews-agent · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
// Simple in-memory API for local development
const sessions = new Map();

export function apiPlugin() {
  return {
    name: 'vite-plugin-api',
    configureServer(server) {
      server.middlewares.use('/api/code-session', async (req, res, next) => {
        console.log(`[API] ${req.method} /api/code-session`);

        // Parse URL
        const url = new URL(req.url, `http://${req.headers.host}`);
        const sessionId = url.searchParams.get('sessionId');

        console.log('[API] Session ID:', sessionId);

        // Set CORS headers
        res.setHeader('Access-Control-Allow-Origin', '*');
        res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS');
        res.setHeader('Access-Control-Allow-Headers', 'Content-Type');

        if (req.method === 'OPTIONS') {
          res.statusCode = 200;
          res.end();
          return;
        }

        if (!sessionId) {
          res.statusCode = 400;
          res.setHeader('Content-Type', 'application/json');
          res.end(JSON.stringify({ error: 'Session ID is required' }));
          return;
        }

        try {
          if (req.method === 'GET') {
            const session = sessions.get(sessionId);

            res.statusCode = 200;
            res.setHeader('Content-Type', 'application/json');

            if (!session) {
              console.log('[API] Session not found');
              res.end(JSON.stringify({
                exists: false,
                code: '',
                questionInfo: null,
                timestamp: null,
                lastModified: null,
              }));
            } else {
              console.log('[API] Returning session:', { codeLength: session.code?.length, lastModified: session.lastModified });
              res.end(JSON.stringify({
                exists: true,
                code: session.code || '',
                questionInfo: session.questionInfo || null,
                timestamp: session.timestamp,
                lastModified: session.lastModified,
              }));
            }
            return;
          }

          if (req.method === 'POST') {
            let body = '';
            req.on('data', chunk => {
              body += chunk.toString();
            });

            req.on('end', () => {
              try {
                const { code, questionInfo } = JSON.parse(body);

                if (code === undefined) {
                  res.statusCode = 400;
                  res.setHeader('Content-Type', 'application/json');
                  res.end(JSON.stringify({ error: 'Code is required' }));
                  return;
                }

                const now = Date.now();
                const sessionData = {
                  code,
                  questionInfo: questionInfo || null,
                  timestamp: new Date().toISOString(),
                  lastModified: now,
                };

                sessions.set(sessionId, sessionData);
                console.log('[API] Session saved:', { sessionId, codeLength: code?.length, sessionsCount: sessions.size });

                res.statusCode = 200;
                res.setHeader('Content-Type', 'application/json');
                res.end(JSON.stringify({
                  success: true,
                  message: 'Code session saved',
                  timestamp: sessionData.timestamp,
                }));
              } catch (error) {
                res.statusCode = 400;
                res.setHeader('Content-Type', 'application/json');
                res.end(JSON.stringify({ error: 'Invalid JSON' }));
              }
            });
            return;
          }

          res.statusCode = 405;
          res.setHeader('Content-Type', 'application/json');
          res.end(JSON.stringify({ error: 'Method not allowed' }));
        } catch (error) {
          console.error('Error handling code session:', error);
          res.statusCode = 500;
          res.setHeader('Content-Type', 'application/json');
          res.end(JSON.stringify({ error: 'Internal server error' }));
        }
      });
    },
  };
}