Enhance Slack-Integrated Task Funding Service with Robust Security and Flexible Funding Mechanisms

.babelrc

@@ -0,0 +1 @@
+{}

.gitignore

@@ -1,4 +1,8 @@
-node_modules
+node_modules/
+dist/
 .env
-.env.funder
-*.pem
+__pycache__/
+.DS_Store
+*.log
+.jest-cache/
+coverage/

index.js

@@ -1,16 +1,23 @@
-const express = require('express');
-const { FundTask, KPLEstablishConnection, KPLFundTask, getTaskStateInfo, KPLCheckProgram } = require('@_koii/create-task-cli');
-const { establishConnection, checkProgram } = require('@_koii/create-task-cli');
-const {PublicKey, Connection,Keypair} = require('@_koii/web3.js');
-const crypto = require('crypto');
-const { parse } = require('path');
-const axios = require('axios');
+import express from 'express';
+import { FundTask, KPLEstablishConnection, KPLFundTask, getTaskStateInfo, KPLCheckProgram } from '@_koii/create-task-cli';
+import { establishConnection, checkProgram } from '@_koii/create-task-cli';
+import { PublicKey, Connection, Keypair } from '@_koii/web3.js';
+import crypto from 'crypto';
+import { parse } from 'path';
+import axios from 'axios';
+import dotenv from 'dotenv';
+
+dotenv.config();
+
 const app = express();
-const port = 3000;
-const SIGNING_SECRET = process.env.SIGNING_SECRET
-const funder_keypair = process.env.funder_keypair
-const user_id_list = ['U06NM9A2VC1', 'U02QTSK9R3N', 'U02QNL3PPFF']
+const port = process.env.PORT || 3000;
+
+const SIGNING_SECRET = process.env.SIGNING_SECRET;
+const funder_keypair = process.env.funder_keypair;
+const user_id_list = ['U06NM9A2VC1', 'U02QTSK9R3N', 'U02QNL3PPFF'];
+
 app.use(express.raw({ type: 'application/x-www-form-urlencoded' }));
+
 function verifySlackRequest(req) {
     const slackSignature = req.headers['x-slack-signature'];
     const timestamp = req.headers['x-slack-request-timestamp'];
@@ -31,7 +38,6 @@ function verifySlackRequest(req) {
 
 // Route to handle funding task
 app.post('/fundtask', async (req, res) => {
-
     if (!verifySlackRequest(req)) {
         return res.status(400).send('Invalid request signature');
     }
@@ -48,64 +54,58 @@ app.post('/fundtask', async (req, res) => {
     const text = parsedBody.text;
     const response_url = parsedBody.response_url;
     const user_id = parsedBody.user_id; 
+
     if (!user_id || !user_id_list.includes(user_id)) {
         await axios.post(response_url, {
             response_type: "in_channel",
             text: 'Sorry, please tag <@U06NM9A2VC1> to add you to the list! '
-        })
+        });
+        return;
     }
 
     let parts = text.split(' ').filter(part => part.trim() !== '');
     let TASK_ID = parts[0].trim();
     let AMOUNT = parts[1].trim();
-    try{
-        await generic_fund_task(TASK_ID, AMOUNT)
+
+    try {
+        await generic_fund_task(TASK_ID, AMOUNT);
         await axios.post(response_url, {
             response_type: "in_channel",
             text: `Congrats! <@${user_id}> You funded ${AMOUNT} to task ${TASK_ID} successfully. `
-        })
-    }catch(e){
+        });
+    } catch(e) {
         await axios.post(response_url, {
             response_type: "in_channel",
             text: `Failed to fund ${AMOUNT} to ${TASK_ID}. ${e}`
-        })
+        });
     }
 });
 
-app.listen(port, () => {
-    console.log(`App running on port ${port}`);
-});
-
-
-async function generic_fund_task(TASK_ID, AMOUNT){
+async function generic_fund_task(TASK_ID, AMOUNT) {
     const connection = new Connection("https://testnet.koii.network", "confirmed");
 
     const taskStateJSON = await getTaskStateInfo(
         connection,
         TASK_ID,
-      );
+    );
     const stakePotAccount = new PublicKey(taskStateJSON.stake_pot_account, connection);
+
     if (taskStateJSON.token_type) {
         const mint_uint8 = Uint8Array.from(taskStateJSON.token_type);
-
-        // Create the PublicKey
         const mint_publicKey = new PublicKey(mint_uint8);
-        await fund_a_KPL_task(TASK_ID, AMOUNT, stakePotAccount, connection, mint_publicKey)
-
-    }else{
-
-        await fund_a_task(TASK_ID, AMOUNT, stakePotAccount, connection)
-
+        await fund_a_KPL_task(TASK_ID, AMOUNT, stakePotAccount, connection, mint_publicKey);
+    } else {
+        await fund_a_task(TASK_ID, AMOUNT, stakePotAccount, connection);
     }
 }
-async function fund_a_task(TASK_ID, AMOUNT, stakePotAccount,connection){
+
+async function fund_a_task(TASK_ID, AMOUNT, stakePotAccount, connection) {
     console.log("Start Funding:");
     console.log("Funding task with Id: ", TASK_ID);
     console.log("Funding amount: ", AMOUNT);
+
     const payerKeypairString = process.env.funder_keypair;
-    // Parse the JSON string into an array
     const payerKeypairArray = JSON.parse(payerKeypairString);
-    // Convert the array to a Uint8Array
     const payerWallet = Uint8Array.from(payerKeypairArray);
     const payerKeypair = Keypair.fromSecretKey(payerWallet);
     const taskStateInfoAddress = new PublicKey(TASK_ID);
@@ -115,23 +115,29 @@ async function fund_a_task(TASK_ID, AMOUNT, stakePotAccount,connection){
     // Create-task-cli package setup
     await establishConnection(connection);
     await checkProgram();
-    await FundTask(payerKeypair,taskStateInfoAddress,stakePotAccount, amount);
+    await FundTask(payerKeypair, taskStateInfoAddress, stakePotAccount, amount);
 }
 
-async function fund_a_KPL_task(TASK_ID, AMOUNT, stakePotAccount,connection, mint_publicKey){
+async function fund_a_KPL_task(TASK_ID, AMOUNT, stakePotAccount, connection, mint_publicKey) {
     console.log("Start Funding:");
     console.log("Funding task with Id: ", TASK_ID);
     console.log("Funding amount: ", AMOUNT);
-    const payerKeypairString = funder_keypair
-    // Parse the JSON string into an array
+
+    const payerKeypairString = funder_keypair;
     const payerKeypairArray = JSON.parse(payerKeypairString);
-    // Convert the array to a Uint8Array
     const payerWallet = Uint8Array.from(payerKeypairArray);
     const payerKeypair = Keypair.fromSecretKey(payerWallet);
     const taskStateInfoAddress = new PublicKey(TASK_ID);
     const amount = parseInt(AMOUNT);
+
     // Create-task-cli package setup
     await KPLEstablishConnection(connection);
     await KPLCheckProgram(); 
-    await KPLFundTask(payerKeypair,taskStateInfoAddress, stakePotAccount, amount, mint_publicKey);
+    await KPLFundTask(payerKeypair, taskStateInfoAddress, stakePotAccount, amount, mint_publicKey);
 }
+
+const server = app.listen(port, () => {
+    console.log(`App running on port ${port}`);
+});
+
+export { app, server };