Fix config

src/config.js

@@ -1,7 +0,0 @@
-module.exports = {
-    TOKEN: process.env.TOKEN,
-    COOKIE: process.env.COOKIE,
-    TEAM_ID: process.env.TEAM_ID,
-    CLAUDE: process.env.CLAUDE,
-    API_KEY: process.env.API_KEY,
-};  

src/slack.js

@@ -3,10 +3,11 @@ const { v4: uuidv4 } = require('uuid');
 const https = require('https');
 const WebSocket = require('ws');
 
-const { readBody, headers, createBaseForm, convertToUnixTime, currentTime, buildPrompt } = require('./utils');
+const { readBody, genHeaders, createBaseForm, convertToUnixTime, currentTime, buildPrompt } = require('./utils');
 
 async function sendPromptMessage(config, prompt) {
-    const form = createBaseForm();
+    const form = createBaseForm(config);
+    const headers = genHeaders(config);
 
     form.append('ts', convertToUnixTime(new Date()));
     form.append('type', 'message');
@@ -42,7 +43,8 @@ async function sendPromptMessage(config, prompt) {
 }
 
 async function sendChatReset(config) {
-    const form = createBaseForm();
+    const form = createBaseForm(config);
+    const headers = genHeaders(config);
 
     form.append('command', '/reset');
     form.append('disp', '/reset');
@@ -74,6 +76,8 @@ async function sendChatReset(config) {
 }
 
 async function waitForWebSocketResponse(config, messages, onData) {
+    const headers = genHeaders(config);
+
     return new Promise(async (resolve, reject) => {
         const websocketURL = `wss://wss-primary.slack.com/?token=${config.token}`;
 

src/utils.js

@@ -1,7 +1,5 @@
 const FormData = require('form-data');
 
-const { TOKEN, COOKIE, TEAM_ID, CLAUDE } = require('./config');
-
 const wait = (duration) => {
     return new Promise((resolve) => {
         setTimeout(() => {
@@ -63,9 +61,11 @@ const currentTime = () => {
     return `${year}-${month}-${day} ${hours}:${minutes}:${seconds}.${milliseconds}`;
 };
 
-const headers = {
-    'Cookie': `d=${COOKIE};`,
-    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:109.0) Gecko/20100101 Firefox/112.0',
+const genHeaders = (config) => {
+    return {
+        'Cookie': `d=${config.cookie};`,
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:109.0) Gecko/20100101 Firefox/112.0',
+    }
 }
 
 function splitJsonArray(jsonArray, maxLength) {
@@ -101,10 +101,10 @@ function convertToUnixTime(date) {
     return `${unixTime}.xxxxx${randomDigit}`;
 }
 
-function createBaseForm() {
+function createBaseForm(config) {
     const form = new FormData();
-    form.append('token', TOKEN);
-    form.append('channel', `${CLAUDE}`);
+    form.append('token', config.token);
+    form.append('channel', `${config.claudeId}`);
     form.append('_x_mode', 'online');
     form.append('_x_sonic', 'true');
     return form;
@@ -148,7 +148,7 @@ module.exports = {
     readBody,
     preparePrompt,
     currentTime,
-    headers,
+    genHeaders,
     convertToUnixTime,
     createBaseForm,
     splitJsonArray,