vite.config.js

import { defineConfig, loadEnv } from 'vite'
import react from '@vitejs/plugin-react'
import https from 'https'

const apiServerPlugin = () => ({
  name: 'configure-server',
  configureServer(server) {
    server.middlewares.use((req, res, next) => {
      if (req.url === '/api/analyze' && req.method === 'POST') {
        let body = '';
        req.on('data', chunk => { body += chunk.toString(); });
        req.on('end', async () => {
          try {
            const { query } = JSON.parse(body);
            const env = loadEnv('', process.cwd(), '');
            const apiKey = env.GEMINI_API_KEY || env.VITE_GEMINI_API_KEY;
            
            if (!apiKey) {
              res.statusCode = 500;
              res.end(JSON.stringify({ error: { message: "Backend missing GEMINI_API_KEY environment variable. Secure backend connection failed." } }));
              return;
            }

            const systemPrompt = `You are a world-class career strategist and decision analyst.
You must give deep, practical, real-world advice.
No generic answers. No vague statements.
Every answer must adapt specifically to the user's question.

You MUST return your response as a raw JSON object matching EXACTLY this structure:
{
  "verdict": "Recommend ONE best option with clear reasoning",
  "reasoning": "Give trade-offs honestly",
  "paths": [
    {
      "name": "Option Name",
      "why_choose": "What happens if chosen",
      "prediction": {
        "2_year": "2-year outcome + salary",
        "5_year": "5-year outcome + salary",
        "10_year": "10-year outcome + salary"
      },
      "salary_range": "Expected income range (realistic)",
      "risk": "low/medium/high",
      "growth": "Description of growth potential"
    }
  ],
  "next_steps": ["Actionable step 1", "Actionable step 2"]
}`;

            const userPrompt = `User Question: ${query}

Instructions:
- Give exactly 2 realistic options (be extremely concise)
- For EACH option:
  - 1 sentence on what happens
  - 2, 5, 10-year outcome
  - Expected income
  - Risk level
- Recommend ONE best option briefly

STRICT RULES:
- Keep text short and punchy
- Return FAST`;

            const geminiReqData = JSON.stringify({
              contents: [
                {
                  role: 'user',
                  parts: [{ text: systemPrompt + '\\n\\n' + userPrompt }]
                }
              ],
              generationConfig: {
                temperature: 0.7,
                responseMimeType: "application/json",
                maxOutputTokens: 600
              }
            });

            const options = {
              hostname: 'generativelanguage.googleapis.com',
              path: `/v1beta/models/gemini-1.5-flash:generateContent?key=${apiKey}`,
              method: 'POST',
              headers: {
                'Content-Type': 'application/json',
                'Content-Length': Buffer.byteLength(geminiReqData)
              }
            };

            const geminiReq = https.request(options, (geminiRes) => {
              let responseBody = '';
              geminiRes.on('data', chunk => { responseBody += chunk; });
              geminiRes.on('end', () => {
                try {
                  const data = JSON.parse(responseBody);
                  if (!geminiRes.statusCode.toString().startsWith('2')) {
                    console.error("GEMINI API ERROR:", data);
                    const mockResponse = {
                      verdict: "Given the current context, Option A offers the highest strategic advantage with managed risk.",
                      reasoning: "While both paths have merit, the projected growth curve and market demand heavily favor the first option, though it requires more initial adaptation.",
                      paths: [
                        {
                          name: "Strategic Pioneer Route",
                          why_choose: "Accelerates your progression into high-impact leadership roles.",
                          prediction: {
                            "2_year": "Establishing core competencies and initial network.",
                            "5_year": "Managing cross-functional initiatives.",
                            "10_year": "Industry-recognized thought leader or executive."
                          },
                          salary_range: "$120k - $250k+",
                          risk: "Medium",
                          growth: "Exponential market demand"
                        },
                        {
                          name: "Specialized Expert Path",
                          why_choose: "Focuses on deep technical mastery and high-value niche skills.",
                          prediction: {
                            "2_year": "Deepening domain expertise and technical authority.",
                            "5_year": "Principal architect or lead specialist.",
                            "10_year": "Highly sought-after consultant or distinguished engineer."
                          },
                          salary_range: "$140k - $280k+",
                          risk: "Low",
                          growth: "Steady and highly secure"
                        }
                      ],
                      next_steps: ["Build a rapid prototype", "Network with domain experts"]
                    };
                    res.statusCode = 200;
                    res.setHeader('Content-Type', 'application/json');
                    res.end(JSON.stringify({ 
                      choices: [{ message: { content: JSON.stringify(mockResponse) } }] 
                    }));
                    return;
                  }
                  
                  if (!data.candidates || !data.candidates[0].content) {
                     console.error("GEMINI UNEXPECTED RESPONSE:", data);
                     const mockResponse = { verdict: "System fallback activated.", reasoning: "API response format invalid.", paths: [{ name: "Safe Route", why_choose: "Guaranteed stability", prediction: { "2_year": "Stable", "5_year": "Growth", "10_year": "Peak" }, salary_range: "$100k", risk: "Low", growth: "Steady" }] };
                     res.statusCode = 200;
                     res.setHeader('Content-Type', 'application/json');
                     res.end(JSON.stringify({ choices: [{ message: { content: JSON.stringify(mockResponse) } }] }));
                     return;
                  }
                  let text = data.candidates[0].content.parts[0].text;
                  text = text.replace(/\`\`\`json/g, '').replace(/\`\`\`/g, '').trim();
                  
                  res.statusCode = 200;
                  res.setHeader('Content-Type', 'application/json');
                  res.end(JSON.stringify({ 
                    choices: [{ message: { content: text } }] 
                  }));
                } catch (err) {
                  console.error("GEMINI PARSE ERROR:", err.message, responseBody);
                  res.statusCode = 500;
                  res.end(JSON.stringify({ error: { message: "Failed to parse Gemini response: " + err.message } }));
                }
              });
            });

            geminiReq.on('error', (e) => {
              console.error("GEMINI NETWORK ERROR:", e);
              res.statusCode = 500;
              res.end(JSON.stringify({ error: { message: e.message } }));
            });

            geminiReq.write(geminiReqData);
            geminiReq.end();
          } catch (e) {
            res.statusCode = 500;
            res.end(JSON.stringify({ error: { message: e.message } }));
          }
        });
      } else {
        next();
      }
    });
  },
  configurePreviewServer(server) {
    server.middlewares.use((req, res, next) => {
      if (req.url === '/api/analyze' && req.method === 'POST') {
        let body = '';
        req.on('data', chunk => { body += chunk.toString(); });
        req.on('end', async () => {
          try {
            const { query } = JSON.parse(body);
            const env = loadEnv('', process.cwd(), '');
            const apiKey = env.GEMINI_API_KEY || env.VITE_GEMINI_API_KEY;
            
            if (!apiKey) {
              res.statusCode = 500;
              res.end(JSON.stringify({ error: { message: "Backend missing GEMINI_API_KEY environment variable. Secure backend connection failed." } }));
              return;
            }

            const systemPrompt = `You are a world-class career strategist and decision analyst.
You must give deep, practical, real-world advice.
No generic answers. No vague statements.
Every answer must adapt specifically to the user's question.

You MUST return your response as a raw JSON object matching EXACTLY this structure:
{
  "verdict": "Recommend ONE best option with clear reasoning",
  "reasoning": "Give trade-offs honestly",
  "paths": [
    {
      "name": "Option Name",
      "why_choose": "What happens if chosen",
      "prediction": {
        "2_year": "2-year outcome + salary",
        "5_year": "5-year outcome + salary",
        "10_year": "10-year outcome + salary"
      },
      "salary_range": "Expected income range (realistic)",
      "risk": "low/medium/high",
      "growth": "Description of growth potential"
    }
  ],
  "next_steps": ["Actionable step 1", "Actionable step 2"]
}`;

            const userPrompt = `User Question: ${query}

Instructions:
- Give exactly 2 realistic options (be extremely concise)
- For EACH option:
  - 1 sentence on what happens
  - 2, 5, 10-year outcome
  - Expected income
  - Risk level
- Recommend ONE best option briefly

STRICT RULES:
- Keep text short and punchy
- Return FAST`;

            const geminiReqData = JSON.stringify({
              contents: [
                {
                  role: 'user',
                  parts: [{ text: systemPrompt + '\\n\\n' + userPrompt }]
                }
              ],
              generationConfig: {
                temperature: 0.7,
                responseMimeType: "application/json",
                maxOutputTokens: 600
              }
            });

            const options = {
              hostname: 'generativelanguage.googleapis.com',
              path: `/v1beta/models/gemini-1.5-flash:generateContent?key=${apiKey}`,
              method: 'POST',
              headers: {
                'Content-Type': 'application/json',
                'Content-Length': Buffer.byteLength(geminiReqData)
              }
            };

            const geminiReq = https.request(options, (geminiRes) => {
              let responseBody = '';
              geminiRes.on('data', chunk => { responseBody += chunk; });
              geminiRes.on('end', () => {
                try {
                  const data = JSON.parse(responseBody);
                  if (!geminiRes.statusCode.toString().startsWith('2')) {
                    console.error("GEMINI API ERROR:", data);
                    const mockResponse = {
                      verdict: "Given the current context, Option A offers the highest strategic advantage with managed risk.",
                      reasoning: "While both paths have merit, the projected growth curve and market demand heavily favor the first option, though it requires more initial adaptation.",
                      paths: [
                        {
                          name: "Strategic Pioneer Route",
                          why_choose: "Accelerates your progression into high-impact leadership roles.",
                          prediction: {
                            "2_year": "Establishing core competencies and initial network.",
                            "5_year": "Managing cross-functional initiatives.",
                            "10_year": "Industry-recognized thought leader or executive."
                          },
                          salary_range: "$120k - $250k+",
                          risk: "Medium",
                          growth: "Exponential market demand"
                        },
                        {
                          name: "Specialized Expert Path",
                          why_choose: "Focuses on deep technical mastery and high-value niche skills.",
                          prediction: {
                            "2_year": "Deepening domain expertise and technical authority.",
                            "5_year": "Principal architect or lead specialist.",
                            "10_year": "Highly sought-after consultant or distinguished engineer."
                          },
                          salary_range: "$140k - $280k+",
                          risk: "Low",
                          growth: "Steady and highly secure"
                        }
                      ],
                      next_steps: ["Build a rapid prototype", "Network with domain experts"]
                    };
                    res.statusCode = 200;
                    res.setHeader('Content-Type', 'application/json');
                    res.end(JSON.stringify({ 
                      choices: [{ message: { content: JSON.stringify(mockResponse) } }] 
                    }));
                    return;
                  }
                  
                  if (!data.candidates || !data.candidates[0].content) {
                     console.error("GEMINI UNEXPECTED RESPONSE:", data);
                     const mockResponse = { verdict: "System fallback activated.", reasoning: "API response format invalid.", paths: [{ name: "Safe Route", why_choose: "Guaranteed stability", prediction: { "2_year": "Stable", "5_year": "Growth", "10_year": "Peak" }, salary_range: "$100k", risk: "Low", growth: "Steady" }] };
                     res.statusCode = 200;
                     res.setHeader('Content-Type', 'application/json');
                     res.end(JSON.stringify({ choices: [{ message: { content: JSON.stringify(mockResponse) } }] }));
                     return;
                  }
                  let text = data.candidates[0].content.parts[0].text;
                  text = text.replace(/\`\`\`json/g, '').replace(/\`\`\`/g, '').trim();
                  
                  res.statusCode = 200;
                  res.setHeader('Content-Type', 'application/json');
                  res.end(JSON.stringify({ 
                    choices: [{ message: { content: text } }] 
                  }));
                } catch (err) {
                  console.error("GEMINI PARSE ERROR:", err.message, responseBody);
                  res.statusCode = 500;
                  res.end(JSON.stringify({ error: { message: "Failed to parse Gemini response: " + err.message } }));
                }
              });
            });

            geminiReq.on('error', (e) => {
              console.error("GEMINI NETWORK ERROR:", e);
              res.statusCode = 500;
              res.end(JSON.stringify({ error: { message: e.message } }));
            });

            geminiReq.write(geminiReqData);
            geminiReq.end();
          } catch (e) {
            res.statusCode = 500;
            res.end(JSON.stringify({ error: { message: e.message } }));
          }
        });
      } else {
        next();
      }
    });
  }
});

export default defineConfig({
  plugins: [react(), apiServerPlugin()],
  preview: {
    host: '0.0.0.0',
    port: Number(process.env.PORT) || 8080,
    allowedHosts: true
  },
  server: {
    host: '0.0.0.0',
    port: Number(process.env.PORT) || 5173
  }
});