ai-web-search 1.0.1 → 1.0.3

package/README.md

@@ -102,6 +102,7 @@ PORT=3000
 HOST=localhost
 RATE_LIMIT_MAX=100
 SEARCH_FUNCTION_PATH=./examples/custom-search.js
+CORS_ORIGIN=http://localhost,http://127.0.0.1
 ```
 
 ### 自定义搜索引擎 / Custom Search Engine
@@ -291,9 +292,7 @@ ai-web-search -p 8080
 - 检查防火墙设置 / Check firewall settings
 - 验证端口是否正确 / Verify port is correct
 
-
-
-### 5. npm install失败 / npm install failed
+### 4. npm install失败 / npm install failed
 
 ```bash
 # 清除npm缓存 / Clear npm cache
@@ -306,6 +305,15 @@ npm install --registry=https://registry.npmmirror.com
 node --version  # 需要 >= 14.0.0 / Requires >= 14.0.0
 ```
 
+### 5. 百度搜索结果是跳转链接 / Baidu results are redirect links
+
+当使用百度搜索回退时，返回的URL是百度跳转链接（如 `https://www.baidu.com/link?url=...`）。
+When using Baidu fallback, returned URLs are Baidu redirect links.
+
+- 点击链接后会自动跳转到目标网站 / Clicking the link will redirect to the destination
+- 这是百度搜索的正常行为 / This is normal Baidu search behavior
+- 如需直接获取真实URL，建议使用自定义搜索引擎实现 / For direct URLs, consider implementing a custom search engine
+
 ## 开发 / Development
 
 ### 本地开发 / Local Development
@@ -335,23 +343,13 @@ ai-web-search/
 │   ├── custom-search.js # 自定义搜索引擎示例 / Custom search engine example
 │   └── baidu-search.js  # 百度搜索引擎实现 / Baidu search engine implementation
 ├── test/
-│   ├── test.js          # 测试文件 / Test file
-│   └── test-custom.js   # 自定义搜索测试 / Custom search test
+│   └── test.js          # 测试文件 / Test file
 ├── package.json
 ├── README.md
 ├── .gitignore
-└── .env.example
+└── LICENSE
 ```
 
-## 许可证 / License
-
-MIT License
-
-## 贡献 / Contributing
-
-欢迎提交 Issue 和 Pull Request！
-Issues and Pull Requests are welcome!
-
 ## 相关项目 / Related Projects
 
 - [duck-duck-scrape](https://github.com/Snazzah/duck-duck-scrape) - DuckDuckGo 搜索库 / DuckDuckGo search library

package/bin/cli.js

@@ -1,4 +1,4 @@
-#!/usr/bin/env node
+#!/usr/bin/env node
 
 const { createServer } = require('../src/server');
 const dotenv = require('dotenv');
@@ -38,6 +38,13 @@ for (let i = 0; i < args.length; i++) {
       showVersion();
       process.exit(0);
       break;
+    default:
+      if (args[i].startsWith('-')) {
+        console.error(`❌ Unknown option: ${args[i]}`);
+        console.error('Run with -h or --help for usage information.');
+        process.exit(1);
+      }
+      break;
   }
 }
 
@@ -112,4 +119,16 @@ process.on('SIGTERM', () => {
     console.log('✅ Server stopped');
     process.exit(0);
   });
+});
+
+// Global error handlers to prevent process crashes
+process.on('uncaughtException', (err) => {
+  console.error('💥 Uncaught Exception:', err.message);
+  if (config.dev) console.error(err.stack);
+  // Allow the process to continue running in production
+});
+
+process.on('unhandledRejection', (reason, promise) => {
+  console.error('💥 Unhandled Rejection at:', promise, 'reason:', reason);
+  // Allow the process to continue running in production
 });

package/examples/baidu-search.js

@@ -98,18 +98,8 @@ async function baiduSearch(query, options = {}) {
     
   } catch (error) {
     console.error('Baidu search error:', error.message);
-    
-    // 返回一个友好的错误信息，而不是抛出异常
-    return {
-      results: [
-        {
-          title: '百度搜索暂时不可用',
-          url: 'https://www.baidu.com',
-          snippet: `搜索出错: ${error.message}。请稍后重试或检查网络连接。`,
-          source: 'baidu-error'
-        }
-      ]
-    };
+    // Throw so callers (e.g. fallback logic in search.js) know the search actually failed
+    throw new Error(`Baidu search failed: ${error.message}`);
   }
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-web-search",
-  "version": "1.0.1",
+  "version": "1.0.3",
   "description": "A lightweight tool that enables AI assistants to search the internet via local HTTP server",
   "main": "src/index.js",
   "type": "commonjs",

package/src/index.js

@@ -43,4 +43,14 @@ process.on('SIGTERM', () => {
     console.log('✅ Server stopped');
     process.exit(0);
   });
+});
+
+// Global error handlers to prevent process crashes
+process.on('uncaughtException', (err) => {
+  console.error('💥 Uncaught Exception:', err.message);
+  if (config.dev) console.error(err.stack);
+});
+
+process.on('unhandledRejection', (reason, promise) => {
+  console.error('💥 Unhandled Rejection at:', promise, 'reason:', reason);
 });

package/src/search.js

@@ -1,4 +1,6 @@
 const DuckDuckGo = require('duck-duck-scrape');
+const axios = require('axios');
+
 let baiduSearchFn = null;
 try {
   baiduSearchFn = require('../examples/baidu-search');
@@ -8,11 +10,11 @@ try {
 
 class SearchEngine {
   constructor(options = {}) {
+    // Merge options carefully: explicit values take precedence, then defaults
     this.options = {
-      safeSearch: options.safeSearch || DuckDuckGo.SafeSearchType.MODERATE,
-      timeout: options.timeout || 10000,
-      searchFn: options.searchFn || null,
-      ...options
+      safeSearch: options.safeSearch !== undefined ? options.safeSearch : DuckDuckGo.SafeSearchType.MODERATE,
+      timeout: options.timeout !== undefined ? options.timeout : 10000,
+      searchFn: options.searchFn !== undefined ? options.searchFn : null,
     };
   }
 
@@ -20,11 +22,11 @@ class SearchEngine {
    * Perform web search
    * @param {string} query - Search query
    * @param {Object} options - Search options
-   * @returns {Promise<Array>} Search results
+   * @returns {Promise<Object>} Search results
    */
   async webSearch(query, options = {}) {
     const { limit = 5, timeRange = 'month' } = options;
-    
+
     try {
       // Use custom search function if provided
       if (this.options.searchFn) {
@@ -34,7 +36,7 @@ class SearchEngine {
           type: 'web',
           ...options
         });
-        
+
         // Format results according to our standard
         const results = (searchResults.results || []).slice(0, limit).map(result => ({
           title: result.title || '',
@@ -50,14 +52,19 @@ class SearchEngine {
           timestamp: new Date().toISOString()
         };
       }
-      
+
       // Default: Use DuckDuckGo
       const searchResults = await DuckDuckGo.search(query, {
         safeSearch: this.options.safeSearch,
         timeRange,
-        ...this.options
+        timeout: this.options.timeout
       });
 
+      // Guard against unexpected response structure
+      if (!searchResults || !Array.isArray(searchResults.results)) {
+        throw new Error('Invalid response structure from DuckDuckGo');
+      }
+
       // Limit and format results
       const results = searchResults.results.slice(0, limit).map(result => ({
         title: result.title,
@@ -74,7 +81,7 @@ class SearchEngine {
       };
     } catch (error) {
       console.log(`⚠️ DuckDuckGo search failed: ${error.message}`);
-      
+
       // Try Baidu as fallback
       if (baiduSearchFn) {
         console.log('🔄 Trying Baidu as fallback...');
@@ -85,7 +92,7 @@ class SearchEngine {
             type: 'web',
             ...options
           });
-          
+
           const results = (searchResults.results || []).slice(0, limit).map(result => ({
             title: result.title || '',
             url: result.url || '',
@@ -104,7 +111,7 @@ class SearchEngine {
           console.log(`⚠️ Baidu fallback also failed: ${fallbackError.message}`);
         }
       }
-      
+
       throw new Error(`Search failed: ${error.message}`);
     }
   }
@@ -113,11 +120,11 @@ class SearchEngine {
    * Perform news search
    * @param {string} query - Search query
    * @param {Object} options - Search options
-   * @returns {Promise<Array>} News results
+   * @returns {Promise<Object>} News results
    */
   async newsSearch(query, options = {}) {
     const { limit = 5, timeRange = 'week' } = options;
-    
+
     try {
       // Use custom search function if provided
       if (this.options.searchFn) {
@@ -127,7 +134,7 @@ class SearchEngine {
           type: 'news',
           ...options
         });
-        
+
         // Format results according to our standard
         const results = (searchResults.results || []).slice(0, limit).map(result => ({
           title: result.title || '',
@@ -144,15 +151,20 @@ class SearchEngine {
           timestamp: new Date().toISOString()
         };
       }
-      
+
       // Default: Use DuckDuckGo
       const searchResults = await DuckDuckGo.search(query, {
         safeSearch: this.options.safeSearch,
         news: true,
         timeRange,
-        ...this.options
+        timeout: this.options.timeout
       });
 
+      // Guard against unexpected response structure
+      if (!searchResults || !Array.isArray(searchResults.results)) {
+        throw new Error('Invalid response structure from DuckDuckGo');
+      }
+
       const results = searchResults.results.slice(0, limit).map(result => ({
         title: result.title,
         url: result.url,
@@ -169,7 +181,7 @@ class SearchEngine {
       };
     } catch (error) {
       console.log(`⚠️ DuckDuckGo news search failed: ${error.message}`);
-      
+
       // Try Baidu as fallback for news
       if (baiduSearchFn) {
         console.log('🔄 Trying Baidu news as fallback...');
@@ -180,7 +192,7 @@ class SearchEngine {
             type: 'news',
             ...options
           });
-          
+
           const results = (searchResults.results || []).slice(0, limit).map(result => ({
             title: result.title || '',
             url: result.url || '',
@@ -200,25 +212,50 @@ class SearchEngine {
           console.log(`⚠️ Baidu news fallback also failed: ${fallbackError.message}`);
         }
       }
-      
+
       throw new Error(`News search failed: ${error.message}`);
     }
   }
 
   /**
-   * Get search suggestions (placeholder - not implemented in duck-duck-scrape)
+   * Get search suggestions
    * @param {string} query - Partial query
-   * @returns {Promise<Array>} Search suggestions
+   * @returns {Promise<Object>} Search suggestions
    */
   async getSuggestions(query) {
-    // duck-duck-scrape doesn't provide getSuggestions method
-    // Return empty suggestions for now
-    return {
-      query,
-      suggestions: [],
-      timestamp: new Date().toISOString()
-    };
+    try {
+      // Try DuckDuckGo suggestions API
+      const response = await axios.get('https://duckduckgo.com/ac/', {
+        params: { q: query, type: 'list' },
+        timeout: 5000,
+        headers: {
+          'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
+        }
+      });
+
+      let suggestions = [];
+      if (Array.isArray(response.data)) {
+        suggestions = response.data.map(item => {
+          if (typeof item === 'string') return item;
+          if (item && item.phrase) return item.phrase;
+          return null;
+        }).filter(Boolean);
+      }
+
+      return {
+        query,
+        suggestions,
+        timestamp: new Date().toISOString()
+      };
+    } catch (error) {
+      // If DuckDuckGo suggestions fail, return empty array gracefully
+      return {
+        query,
+        suggestions: [],
+        timestamp: new Date().toISOString()
+      };
+    }
   }
 }
 
-module.exports = SearchEngine;
+module.exports = SearchEngine;

package/src/server.js

@@ -1,29 +1,36 @@
 const express = require('express');
 const cors = require('cors');
 const rateLimit = require('express-rate-limit');
+const path = require('path');
 const SearchEngine = require('./search');
 
 function createServer(config) {
   const app = express();
   // Load custom search function if provided
+  // Resolve path relative to the current working directory (user's project root),
+  // not relative to this module's location inside node_modules
   let searchFn = null;
   if (config.searchFnPath) {
+    const resolvedPath = path.isAbsolute(config.searchFnPath)
+      ? config.searchFnPath
+      : path.resolve(process.cwd(), config.searchFnPath);
     try {
-      searchFn = require(config.searchFnPath);
-      console.log(`🔌 Loaded custom search function from ${config.searchFnPath}`);
+      searchFn = require(resolvedPath);
+      console.log(`🔌 Loaded custom search function from ${resolvedPath}`);
     } catch (error) {
-      console.error(`❌ Failed to load custom search function from ${config.searchFnPath}:`, error.message);
+      console.error(`❌ Failed to load custom search function from ${resolvedPath}:`, error.message);
     }
   }
-  
+
   const searchEngine = new SearchEngine({ searchFn });
 
-  // Middleware
-  app.use(cors());
+  // CORS: restrict origins in production unless explicitly configured
+  const corsOrigin = process.env.CORS_ORIGIN || (config.dev ? true : ['http://localhost', 'http://127.0.0.1']);
+  app.use(cors({ origin: corsOrigin }));
   app.use(express.json());
   app.use(express.urlencoded({ extended: true }));
 
-  // Rate limiting
+  // Rate limiting applied to all API endpoints
   const limiter = rateLimit({
     windowMs: config.rateLimit.windowMs,
     max: config.rateLimit.max,
@@ -32,7 +39,7 @@ function createServer(config) {
       message: 'Please try again later'
     }
   });
-  app.use('/search', limiter);
+  app.use(limiter);
 
   // Health check endpoint
   app.get('/health', (req, res) => {

package/test/test.js

@@ -4,9 +4,10 @@ const SearchEngine = require('../src/search');
 
 async function runTests() {
   console.log('🧪 Running AI Web Search Tests\n');
-  
+
   const searchEngine = new SearchEngine();
-  
+  let failed = 0;
+
   // Test 1: Basic web search
   console.log('1. Testing web search...');
   try {
@@ -19,10 +20,11 @@ async function runTests() {
     }
     console.log('');
   } catch (error) {
+    failed++;
     console.log('❌ Web search failed:', error.message);
     console.log('');
   }
-  
+
   // Test 2: News search
   console.log('2. Testing news search...');
   try {
@@ -34,14 +36,15 @@ async function runTests() {
     }
     console.log('');
   } catch (error) {
+    failed++;
     console.log('❌ News search failed:', error.message);
     console.log('');
   }
-  
+
   // Test 3: Search suggestions
   console.log('3. Testing search suggestions...');
   try {
-    const suggestions = await searchEngine.getSuggestions('机器学');
+    const suggestions = await searchEngine.getSuggestions('machine learn');
     console.log('✅ Suggestions successful');
     console.log(`   Got ${suggestions.suggestions.length} suggestions`);
     if (suggestions.suggestions.length > 0) {
@@ -49,13 +52,18 @@ async function runTests() {
     }
     console.log('');
   } catch (error) {
+    failed++;
     console.log('❌ Suggestions failed:', error.message);
     console.log('');
   }
-  
+
   console.log('🎉 All tests completed!');
-  console.log('\nNote: Some tests may fail due to network issues or DuckDuckGo rate limiting.');
-  console.log('If tests fail, try again in a few minutes.');
+  if (failed > 0) {
+    console.log(`\n⚠️  ${failed} test(s) failed.`);
+    console.log('Note: Some tests may fail due to network issues or DuckDuckGo rate limiting.');
+    console.log('If tests fail, try again in a few minutes.');
+    process.exit(1);
+  }
 }
 
 // Run tests