From 1568a12a96e6a7500f9d4e257619254d01826aa4 Mon Sep 17 00:00:00 2001
From: donald <donald@theaken.com>
Date: Tue, 9 Dec 2025 18:28:29 +0800
Subject: [PATCH] fix: Add localhost host binding to Vite config and index fix
 script
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- Add host: 'localhost' to vite.config.js to ensure consistent IP
- Add scripts/fix-indexes.js for database index verification
- Add routes/llmTest.js for LLM testing endpoint

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
---
 routes/llmTest.js      | 219 +++++++++++++++++++++++++++++++++++++++++
 scripts/fix-indexes.js | 204 ++++++++++++++++++++++++++++++++++++++
 vite.config.js         |   1 +
 3 files changed, 424 insertions(+)
 create mode 100644 routes/llmTest.js
 create mode 100644 scripts/fix-indexes.js

diff --git a/routes/llmTest.js b/routes/llmTest.js
new file mode 100644
index 0000000..70186b6
--- /dev/null
+++ b/routes/llmTest.js
@@ -0,0 +1,219 @@
+import express from 'express';
+import { asyncHandler } from '../middleware/errorHandler.js';
+import { requireAuth, requireAdmin } from '../middleware/auth.js';
+
+const router = express.Router();
+
+/**
+ * GET /api/llm-test/models
+ * 列出可用的 LLM 模型（從 API 動態獲取）
+ */
+router.get('/models', requireAuth, asyncHandler(async (req, res) => {
+  const { api_url } = req.query;
+  const targetUrl = api_url || 'https://ollama_pjapi.theaken.com';
+
+  try {
+    const axios = (await import('axios')).default;
+
+    const response = await axios.get(`${targetUrl}/v1/models`, {
+      timeout: 10000
+    });
+
+    if (response.data && response.data.data) {
+      const models = response.data.data.map(model => ({
+        id: model.id,
+        name: model.info?.name || model.id,
+        description: model.info?.description || '',
+        best_for: model.info?.best_for || '',
+        owned_by: model.owned_by || 'unknown'
+      }));
+
+      res.json({
+        success: true,
+        data: models
+      });
+    } else {
+      throw new Error('Invalid response format');
+    }
+  } catch (error) {
+    res.status(500).json({
+      success: false,
+      error: '無法取得模型列表',
+      message: error.message
+    });
+  }
+}));
+
+/**
+ * POST /api/llm-test/quick
+ * 快速測試 LLM 連線（僅管理員）
+ */
+router.post('/quick', requireAdmin, asyncHandler(async (req, res) => {
+  const { api_url, api_key, model_name } = req.body;
+
+  if (!api_url || !model_name) {
+    return res.status(400).json({
+      success: false,
+      error: '請提供 API 端點和模型名稱'
+    });
+  }
+
+  try {
+    const axios = (await import('axios')).default;
+    const startTime = Date.now();
+
+    const response = await axios.post(
+      `${api_url}/v1/chat/completions`,
+      {
+        model: model_name,
+        messages: [
+          { role: 'user', content: 'Hello, please respond with just "OK"' }
+        ],
+        max_tokens: 10
+      },
+      {
+        timeout: 15000,
+        headers: {
+          'Content-Type': 'application/json',
+          ...(api_key && { 'Authorization': `Bearer ${api_key}` })
+        }
+      }
+    );
+
+    const responseTime = Date.now() - startTime;
+
+    if (response.data && response.data.choices) {
+      res.json({
+        success: true,
+        message: 'LLM API 連線測試成功',
+        responseTime: responseTime,
+        response: response.data.choices[0]?.message?.content || ''
+      });
+    } else {
+      throw new Error('Invalid API response format');
+    }
+  } catch (error) {
+    res.status(500).json({
+      success: false,
+      error: 'LLM API 連線測試失敗',
+      message: error.message
+    });
+  }
+}));
+
+/**
+ * POST /api/llm-test/chat
+ * 直接與 LLM 對話（測試用，僅管理員）
+ */
+router.post('/chat', requireAdmin, asyncHandler(async (req, res) => {
+  const { api_url, api_key, model_name, messages, temperature, max_tokens, stream } = req.body;
+
+  if (!api_url || !model_name || !messages) {
+    return res.status(400).json({
+      success: false,
+      error: '請提供必要參數'
+    });
+  }
+
+  try {
+    const axios = (await import('axios')).default;
+    const startTime = Date.now();
+
+    // 非串流模式
+    if (!stream) {
+      const response = await axios.post(
+        `${api_url}/v1/chat/completions`,
+        {
+          model: model_name,
+          messages: messages,
+          temperature: temperature || 0.7,
+          max_tokens: max_tokens || 2000,
+          stream: false
+        },
+        {
+          timeout: 120000,
+          headers: {
+            'Content-Type': 'application/json',
+            ...(api_key && { 'Authorization': `Bearer ${api_key}` })
+          }
+        }
+      );
+
+      const responseTime = Date.now() - startTime;
+
+      res.json({
+        success: true,
+        data: {
+          content: response.data.choices[0]?.message?.content || '',
+          usage: response.data.usage,
+          model: response.data.model,
+          responseTime: responseTime
+        }
+      });
+    } else {
+      // 串流模式 - 使用 Server-Sent Events
+      res.setHeader('Content-Type', 'text/event-stream');
+      res.setHeader('Cache-Control', 'no-cache');
+      res.setHeader('Connection', 'keep-alive');
+
+      const response = await axios.post(
+        `${api_url}/v1/chat/completions`,
+        {
+          model: model_name,
+          messages: messages,
+          temperature: temperature || 0.7,
+          max_tokens: max_tokens || 2000,
+          stream: true
+        },
+        {
+          timeout: 120000,
+          responseType: 'stream',
+          headers: {
+            'Content-Type': 'application/json',
+            ...(api_key && { 'Authorization': `Bearer ${api_key}` })
+          }
+        }
+      );
+
+      response.data.on('data', (chunk) => {
+        const lines = chunk.toString().split('\n');
+        for (const line of lines) {
+          if (line.startsWith('data: ')) {
+            const dataStr = line.slice(6).trim();
+            if (dataStr && dataStr !== '[DONE]') {
+              try {
+                const data = JSON.parse(dataStr);
+                const content = data.choices?.[0]?.delta?.content;
+                if (content) {
+                  res.write(`data: ${JSON.stringify({ content })}\n\n`);
+                }
+              } catch (e) {
+                // Skip invalid JSON
+              }
+            }
+          }
+        }
+      });
+
+      response.data.on('end', () => {
+        res.write('data: [DONE]\n\n');
+        res.end();
+      });
+
+      response.data.on('error', (error) => {
+        res.write(`data: ${JSON.stringify({ error: error.message })}\n\n`);
+        res.end();
+      });
+    }
+  } catch (error) {
+    if (!res.headersSent) {
+      res.status(500).json({
+        success: false,
+        error: 'LLM 對話失敗',
+        message: error.message
+      });
+    }
+  }
+}));
+
+export default router;
diff --git a/scripts/fix-indexes.js b/scripts/fix-indexes.js
new file mode 100644
index 0000000..5d4728b
--- /dev/null
+++ b/scripts/fix-indexes.js
@@ -0,0 +1,204 @@
+#!/usr/bin/env node
+/**
+ * Fix Database Indexes
+ * Ensures all indexes exist on renamed 5Why_ tables
+ *
+ * Run: node scripts/fix-indexes.js
+ */
+
+import mysql from 'mysql2/promise';
+import dotenv from 'dotenv';
+
+dotenv.config();
+
+const dbConfig = {
+  host: process.env.DB_HOST || 'mysql.theaken.com',
+  port: parseInt(process.env.DB_PORT) || 33306,
+  user: process.env.DB_USER || 'A102',
+  password: process.env.DB_PASSWORD || 'Bb123456',
+  database: process.env.DB_NAME || 'db_A102'
+};
+
+// Index definitions for each table
+const indexDefinitions = {
+  '5Why_users': [
+    { name: 'idx_employee_id', columns: 'employee_id' },
+    { name: 'idx_email', columns: 'email' },
+    { name: 'idx_role', columns: 'role' }
+  ],
+  '5Why_analyses': [
+    { name: 'idx_user_id', columns: 'user_id' },
+    { name: 'idx_status', columns: 'status' },
+    { name: 'idx_created_at', columns: 'created_at' }
+  ],
+  '5Why_analysis_perspectives': [
+    { name: 'idx_analysis_id', columns: 'analysis_id' }
+  ],
+  '5Why_analysis_whys': [
+    { name: 'idx_perspective_id', columns: 'perspective_id' },
+    { name: 'idx_level', columns: 'level' }
+  ],
+  '5Why_llm_configs': [
+    { name: 'idx_provider', columns: 'provider' },
+    { name: 'idx_is_active', columns: 'is_active' }
+  ],
+  '5Why_system_settings': [
+    { name: 'idx_setting_key', columns: 'setting_key' },
+    { name: 'idx_is_public', columns: 'is_public' }
+  ],
+  '5Why_audit_logs': [
+    { name: 'idx_user_id', columns: 'user_id' },
+    { name: 'idx_action', columns: 'action' },
+    { name: 'idx_created_at', columns: 'created_at' },
+    { name: 'idx_entity', columns: 'entity_type, entity_id' }
+  ],
+  '5Why_sessions': [
+    { name: 'idx_expires', columns: 'expires' }
+  ]
+};
+
+async function fixIndexes() {
+  let connection;
+
+  try {
+    console.log('\n╔════════════════════════════════════════════════════════╗');
+    console.log('║  5 Why Analyzer - Fix Database Indexes                 ║');
+    console.log('╚════════════════════════════════════════════════════════╝\n');
+
+    console.log('🔄 Connecting to database...');
+    connection = await mysql.createConnection(dbConfig);
+    console.log('✅ Connected successfully\n');
+
+    // Process each table
+    for (const [tableName, indexes] of Object.entries(indexDefinitions)) {
+      console.log(`📋 Processing table: ${tableName}`);
+
+      // Check if table exists
+      const [tableExists] = await connection.execute(
+        `SELECT COUNT(*) as count FROM information_schema.tables
+         WHERE table_schema = ? AND table_name = ?`,
+        [dbConfig.database, tableName]
+      );
+
+      if (tableExists[0].count === 0) {
+        console.log(`   ⚠ Table ${tableName} not found, skipping...`);
+        continue;
+      }
+
+      // Get existing indexes
+      const [existingIndexes] = await connection.execute(
+        `SELECT DISTINCT INDEX_NAME FROM information_schema.statistics
+         WHERE table_schema = ? AND table_name = ?`,
+        [dbConfig.database, tableName]
+      );
+      const existingIndexNames = existingIndexes.map(idx => idx.INDEX_NAME);
+
+      // Create missing indexes
+      for (const index of indexes) {
+        if (existingIndexNames.includes(index.name)) {
+          console.log(`   ✓ Index ${index.name} already exists`);
+        } else {
+          try {
+            await connection.execute(
+              `CREATE INDEX ${index.name} ON \`${tableName}\` (${index.columns})`
+            );
+            console.log(`   ✓ Created index: ${index.name} on (${index.columns})`);
+          } catch (err) {
+            if (err.code === 'ER_DUP_KEYNAME') {
+              console.log(`   ⚠ Index ${index.name} already exists (duplicate)`);
+            } else {
+              console.error(`   ✗ Error creating index ${index.name}: ${err.message}`);
+            }
+          }
+        }
+      }
+    }
+
+    // Verify unique constraints on 5Why_users
+    console.log('\n📋 Verifying unique constraints on 5Why_users...');
+
+    // Check employee_id unique constraint
+    const [empUnique] = await connection.execute(
+      `SELECT COUNT(*) as count FROM information_schema.statistics
+       WHERE table_schema = ? AND table_name = '5Why_users'
+       AND column_name = 'employee_id' AND non_unique = 0`,
+      [dbConfig.database]
+    );
+
+    if (empUnique[0].count > 0) {
+      console.log('   ✓ employee_id has unique constraint');
+    } else {
+      console.log('   ⚠ employee_id missing unique constraint, adding...');
+      try {
+        await connection.execute(
+          `ALTER TABLE 5Why_users ADD UNIQUE INDEX idx_employee_id_unique (employee_id)`
+        );
+        console.log('   ✓ Added unique constraint on employee_id');
+      } catch (err) {
+        if (err.code === 'ER_DUP_KEYNAME' || err.message.includes('Duplicate')) {
+          console.log('   ✓ Unique constraint already exists');
+        } else {
+          console.error(`   ✗ Error: ${err.message}`);
+        }
+      }
+    }
+
+    // Check email unique constraint
+    const [emailUnique] = await connection.execute(
+      `SELECT COUNT(*) as count FROM information_schema.statistics
+       WHERE table_schema = ? AND table_name = '5Why_users'
+       AND column_name = 'email' AND non_unique = 0`,
+      [dbConfig.database]
+    );
+
+    if (emailUnique[0].count > 0) {
+      console.log('   ✓ email has unique constraint');
+    } else {
+      console.log('   ⚠ email missing unique constraint, adding...');
+      try {
+        await connection.execute(
+          `ALTER TABLE 5Why_users ADD UNIQUE INDEX idx_email_unique (email)`
+        );
+        console.log('   ✓ Added unique constraint on email');
+      } catch (err) {
+        if (err.code === 'ER_DUP_KEYNAME' || err.message.includes('Duplicate')) {
+          console.log('   ✓ Unique constraint already exists');
+        } else {
+          console.error(`   ✗ Error: ${err.message}`);
+        }
+      }
+    }
+
+    console.log('\n╔════════════════════════════════════════════════════════╗');
+    console.log('║  Index Fix Complete!                                   ║');
+    console.log('╚════════════════════════════════════════════════════════╝\n');
+
+    // Show current indexes for 5Why_users
+    console.log('📊 Current indexes on 5Why_users:');
+    const [userIndexes] = await connection.execute(
+      `SELECT INDEX_NAME, COLUMN_NAME, NON_UNIQUE
+       FROM information_schema.statistics
+       WHERE table_schema = ? AND table_name = '5Why_users'
+       ORDER BY INDEX_NAME, SEQ_IN_INDEX`,
+      [dbConfig.database]
+    );
+
+    userIndexes.forEach(idx => {
+      const unique = idx.NON_UNIQUE === 0 ? ' (UNIQUE)' : '';
+      console.log(`   - ${idx.INDEX_NAME}: ${idx.COLUMN_NAME}${unique}`);
+    });
+
+  } catch (error) {
+    console.error('\n❌ Fix indexes failed:');
+    console.error('   Error:', error.message);
+    process.exit(1);
+  } finally {
+    if (connection) {
+      await connection.end();
+      console.log('\n🔌 Database connection closed\n');
+    }
+  }
+}
+
+// Execute
+fixIndexes();
diff --git a/vite.config.js b/vite.config.js
index 3922762..a8d2087 100644
--- a/vite.config.js
+++ b/vite.config.js
@@ -4,6 +4,7 @@ import react from '@vitejs/plugin-react'
 export default defineConfig({
   plugins: [react()],
   server: {
+    host: 'localhost',
     port: 5173,
     open: true
   }