feat: Support http_proxy for chatGPT (#87)

Signed-off-by: Lancelot <[email protected]>

Author: Lan-ce-lot

README.md

@@ -4,8 +4,7 @@
 
 A 🇨🇳 and 🇺🇸🇬🇧 translate tool in your command line, powered by youdao/iciba/ChatGPT.
 
-[![NPM version](https://img.shields.io/npm/v/fanyi.svg?style=flat-square)](https://npmjs.org/package/fanyi)
-[![NPM downloads](http://img.shields.io/npm/dm/fanyi.svg?style=flat-square)](https://npmjs.org/package/fanyi)
+[![NPM version](https://img.shields.io/npm/v/fanyi.svg?style=flat-square)](https://npmjs.org/package/fanyi) [![NPM downloads](http://img.shields.io/npm/dm/fanyi.svg?style=flat-square)](https://npmjs.org/package/fanyi)
 
 ![](https://gw.alipayobjects.com/zos/rmsportal/EAuwmtfxDGueGOdUgVKc.png)
 
@@ -150,6 +149,18 @@ Turn off ChatGPT translation.
 $ fanyi config --openai-api-key false
 ```
 
+If we get this error:
+
+> Cannot reach OpenAI api, please check network
+
+We need to ensure that the proxy and $http_proxy is correctly set.
+
+For example(for clash X):
+
+```bash
+$ export http_proxy=http://127.0.0.1:7890
+```
+
 ## Error: spawn festival ENOENT
 
 Try this workaround from [say.js](https://github.com/Marak/say.js#linux-notes) in Linux.

lib/print.js

@@ -2,6 +2,7 @@ const chalk = require('chalk');
 const ora = require('ora');
 const gradient = require('gradient-string');
 const { saveHistory } = require('./searchHistory');
+const { HttpsProxyAgent } = require('https-proxy-agent');
 require('isomorphic-unfetch');
 
 exports.iciba = function (data, options = {}) {
@@ -165,29 +166,45 @@ exports.chatgpt = async (word, options) => {
     chalk = initChalkWithNoColor();
   }
   const { ChatGPTAPI } = await import('chatgpt');
-  const api = new ChatGPTAPI({
-    apiKey: options.openaiApiKey,
-  });
+
+  const api = process.env.http_proxy
+    ? new ChatGPTAPI({
+        apiKey: options.openaiApiKey,
+        fetch: (url, options) => {
+          options.agent = new HttpsProxyAgent(process.env.http_proxy);
+          return fetch(url, options);
+        },
+      })
+    : new ChatGPTAPI({ apiKey: options.openaiApiKey });
+
   const spinner = ora().start();
-  return api.sendMessage(`
+  return api
+    .sendMessage(
+      `
     你是一本中英文字典，请给出这个词的翻译、音标或拼音、同义词和近义词、并提供相关例句：${word}
-  `).then(res => {
-    spinner.stop();
-    spinner.clear();
-    log(word + chalk.gray('  ~  ChatGPT'));
-    log();
-    res.text.trim().split('\n').forEach(line => {
-      log(options.color === false ? line : gradient.atlas(line));
+  `,
+    )
+    .then((res) => {
+      spinner.stop();
+      spinner.clear();
+      log(word + chalk.gray('  ~  ChatGPT'));
+      log();
+      res.text
+        .trim()
+        .split('\n')
+        .forEach((line) => {
+          log(options.color === false ? line : gradient.atlas(line));
+        });
+      log();
+      log(chalk.gray('-----'));
+      log();
+    })
+    .catch((e) => {
+      spinner.stop();
+      spinner.clear();
+      log(chalk.yellow('Cannot reach OpenAI api, please check network.'));
+      log();
     });
-    log();
-    log(chalk.gray('-----'));
-    log();
-  }).catch(e => {
-    spinner.stop();
-    spinner.clear();
-    log(chalk.yellow('Cannot reach OpenAI api, please check network.'));
-    log();
-  });
 };
 
 function log(message, indentNum) {

package.json

@@ -44,6 +44,7 @@
     "fs-extra": "^10.1.0",
     "gradient-string": "^2.0.2",
     "html-entities": "^1.1.2",
+    "https-proxy-agent": "^5.0.1",
     "is-chinese": "^1.2.5",
     "isomorphic-unfetch": "^4.0.2",
     "needle": "^3.1.0",