Compare commits
15 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
36a0cd309b | ||
|
|
b49b464afb | ||
|
|
2caf54346a | ||
|
|
6e6467c2a5 | ||
|
|
b463b0a8cb | ||
|
|
42ef38db14 | ||
|
|
af23db20bb | ||
|
|
117b195095 | ||
|
|
b361636a93 | ||
|
|
4dbd45bccd | ||
|
|
02ffb6a196 | ||
|
|
060494e883 | ||
|
|
352d88ee9d | ||
|
|
e60c2f88b8 | ||
|
|
b498276978 |
@@ -1,9 +1,6 @@
|
|||||||
# Discord token for the bot
|
# Discord token for the bot
|
||||||
CLIENT_TOKEN = BOT_TOKEN
|
CLIENT_TOKEN = BOT_TOKEN
|
||||||
|
|
||||||
# id token of a discord server
|
|
||||||
GUILD_ID = GUILD_ID
|
|
||||||
|
|
||||||
# model for the bot to query from (i.e. llama2 [llama2:13b], mistral, codellama, etc... )
|
# model for the bot to query from (i.e. llama2 [llama2:13b], mistral, codellama, etc... )
|
||||||
MODEL = MODEL_NAME
|
MODEL = MODEL_NAME
|
||||||
|
|
||||||
|
|||||||
19
.github/workflows/build.yml
vendored
19
.github/workflows/build.yml
vendored
@@ -1,20 +1,9 @@
|
|||||||
name: Builds
|
name: Builds
|
||||||
run-name: Validate Node and Docker Builds
|
run-name: Validate Node and Docker Builds
|
||||||
on:
|
on:
|
||||||
pull_request:
|
push:
|
||||||
branches:
|
branches:
|
||||||
- master
|
- master # runs after Pull Request is merged
|
||||||
paths:
|
|
||||||
- '/'
|
|
||||||
- 'src/**'
|
|
||||||
- 'tests/**'
|
|
||||||
- '!docs/**'
|
|
||||||
- '!imgs/**'
|
|
||||||
- '!.github/**'
|
|
||||||
- '.github/workflows/**'
|
|
||||||
- '!.gitignore'
|
|
||||||
- '!LICENSE'
|
|
||||||
- '!README'
|
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
Discord-Node-Build: # test if the node install and run
|
Discord-Node-Build: # test if the node install and run
|
||||||
@@ -42,7 +31,6 @@ jobs:
|
|||||||
run: |
|
run: |
|
||||||
touch .env
|
touch .env
|
||||||
echo CLIENT_TOKEN = ${{ secrets.BOT_TOKEN }} >> .env
|
echo CLIENT_TOKEN = ${{ secrets.BOT_TOKEN }} >> .env
|
||||||
echo GUILD_ID = ${{ secrets.GUILD_ID }} >> .env
|
|
||||||
echo MODEL = ${{ secrets.MODEL }} >> .env
|
echo MODEL = ${{ secrets.MODEL }} >> .env
|
||||||
echo CLIENT_UID = ${{ secrets.CLIENT_UID }} >> .env
|
echo CLIENT_UID = ${{ secrets.CLIENT_UID }} >> .env
|
||||||
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
||||||
@@ -71,7 +59,6 @@ jobs:
|
|||||||
run: |
|
run: |
|
||||||
touch .env
|
touch .env
|
||||||
echo CLIENT_TOKEN = ${{ secrets.BOT_TOKEN }} >> .env
|
echo CLIENT_TOKEN = ${{ secrets.BOT_TOKEN }} >> .env
|
||||||
echo GUILD_ID = ${{ secrets.GUILD_ID }} >> .env
|
|
||||||
echo MODEL = ${{ secrets.MODEL }} >> .env
|
echo MODEL = ${{ secrets.MODEL }} >> .env
|
||||||
echo CLIENT_UID = ${{ secrets.CLIENT_UID }} >> .env
|
echo CLIENT_UID = ${{ secrets.CLIENT_UID }} >> .env
|
||||||
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
||||||
@@ -83,7 +70,7 @@ jobs:
|
|||||||
|
|
||||||
- name: Check Images Exist
|
- name: Check Images Exist
|
||||||
run: |
|
run: |
|
||||||
(docker images | grep -q 'discord/bot' && docker images | grep -qE 'ollama/ollama') || exit 1
|
(docker images | grep -q 'kevinthedang/discord-ollama' && docker images | grep -qE 'ollama/ollama') || exit 1
|
||||||
|
|
||||||
- name: Check Containers Exist
|
- name: Check Containers Exist
|
||||||
run: |
|
run: |
|
||||||
|
|||||||
50
.github/workflows/release.yml
vendored
Normal file
50
.github/workflows/release.yml
vendored
Normal file
@@ -0,0 +1,50 @@
|
|||||||
|
name: Deploy
|
||||||
|
run-name: Release Docker Image
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
tags:
|
||||||
|
- 'v*'
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
Release-Docker-Image:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
environment: release
|
||||||
|
timeout-minutes: 3
|
||||||
|
steps:
|
||||||
|
- name: Checkout Repository
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
|
||||||
|
- name: Set up Node Environment lts/hydrogen
|
||||||
|
uses: actions/setup-node@v4
|
||||||
|
with:
|
||||||
|
node-version: lts/hydrogen
|
||||||
|
cache: "npm"
|
||||||
|
|
||||||
|
- name: Create Environment Variables
|
||||||
|
run: |
|
||||||
|
touch .env
|
||||||
|
echo CLIENT_TOKEN = NOT_REAL_TOKEN >> .env
|
||||||
|
echo MODEL = ${{ secrets.MODEL }} >> .env
|
||||||
|
echo CLIENT_UID = ${{ secrets.CLIENT_UID }} >> .env
|
||||||
|
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
||||||
|
echo OLLAMA_PORT = ${{ secrets.OLLAMA_PORT }} >> .env
|
||||||
|
|
||||||
|
- name: Get Version from package.json
|
||||||
|
run: echo "VERSION=$(jq -r '.version' package.json)" >> $GITHUB_ENV
|
||||||
|
|
||||||
|
- name: Build Image
|
||||||
|
run: |
|
||||||
|
npm run docker:build
|
||||||
|
|
||||||
|
- name: Build Image as Latest
|
||||||
|
run: |
|
||||||
|
npm run docker:build-latest
|
||||||
|
|
||||||
|
- name: Log into Docker
|
||||||
|
run: |
|
||||||
|
docker login --username ${{ vars.DOCKER_USER }} --password ${{ secrets.DOCKER_PASS }}
|
||||||
|
|
||||||
|
- name: Release Docker Image
|
||||||
|
run: |
|
||||||
|
docker push ${{ vars.DOCKER_USER }}/discord-ollama:${{ env.VERSION }}
|
||||||
|
docker push ${{ vars.DOCKER_USER }}/discord-ollama:latest
|
||||||
35
.github/workflows/test.yml
vendored
35
.github/workflows/test.yml
vendored
@@ -5,7 +5,8 @@ on:
|
|||||||
branches:
|
branches:
|
||||||
- master
|
- master
|
||||||
paths:
|
paths:
|
||||||
- '/'
|
- '*'
|
||||||
|
- 'package*.json'
|
||||||
- 'src/**'
|
- 'src/**'
|
||||||
- 'tests/**'
|
- 'tests/**'
|
||||||
- '!docs/**'
|
- '!docs/**'
|
||||||
@@ -38,7 +39,6 @@ jobs:
|
|||||||
run: |
|
run: |
|
||||||
touch .env
|
touch .env
|
||||||
echo CLIENT_TOKEN = ${{ secrets.BOT_TOKEN }} >> .env
|
echo CLIENT_TOKEN = ${{ secrets.BOT_TOKEN }} >> .env
|
||||||
echo GUILD_ID = ${{ secrets.GUILD_ID }} >> .env
|
|
||||||
echo MODEL = ${{ secrets.MODEL }} >> .env
|
echo MODEL = ${{ secrets.MODEL }} >> .env
|
||||||
echo CLIENT_UID = ${{ secrets.CLIENT_UID }} >> .env
|
echo CLIENT_UID = ${{ secrets.CLIENT_UID }} >> .env
|
||||||
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
||||||
@@ -47,34 +47,3 @@ jobs:
|
|||||||
- name: Test Application
|
- name: Test Application
|
||||||
run: |
|
run: |
|
||||||
npm run test:run
|
npm run test:run
|
||||||
|
|
||||||
Discord-Ollama-Container-Test:
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
timeout-minutes: 2
|
|
||||||
steps:
|
|
||||||
- name: Checkout Repository
|
|
||||||
uses: actions/checkout@v4
|
|
||||||
|
|
||||||
- name: Set up Node Environment lts/hydrogen
|
|
||||||
uses: actions/setup-node@v4
|
|
||||||
with:
|
|
||||||
node-version: lts/hydrogen
|
|
||||||
cache: "npm"
|
|
||||||
|
|
||||||
- name: Create Environment Variables
|
|
||||||
run: |
|
|
||||||
touch .env
|
|
||||||
echo CLIENT_TOKEN = ${{ secrets.BOT_TOKEN }} >> .env
|
|
||||||
echo GUILD_ID = ${{ secrets.GUILD_ID }} >> .env
|
|
||||||
echo MODEL = ${{ secrets.MODEL }} >> .env
|
|
||||||
echo CLIENT_UID = ${{ secrets.CLIENT_UID }} >> .env
|
|
||||||
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
|
||||||
echo OLLAMA_PORT = ${{ secrets.OLLAMA_PORT }} >> .env
|
|
||||||
|
|
||||||
- name: Setup Docker Network and Images
|
|
||||||
run: |
|
|
||||||
npm run docker:start-cpu
|
|
||||||
|
|
||||||
- name: Test Docker Container
|
|
||||||
run: |
|
|
||||||
npm run docker:test
|
|
||||||
13
README.md
13
README.md
@@ -5,6 +5,7 @@
|
|||||||
<p><a href="#"></a><a href="https://creativecommons.org/licenses/by/4.0/"><img alt="License" src="https://img.shields.io/badge/License-CC_BY_4.0-darkgreen.svg" /></a>
|
<p><a href="#"></a><a href="https://creativecommons.org/licenses/by/4.0/"><img alt="License" src="https://img.shields.io/badge/License-CC_BY_4.0-darkgreen.svg" /></a>
|
||||||
<a href="#"></a><a href="https://github.com/kevinthedang/discord-ollama/releases/latest"><img alt="Release" src="https://img.shields.io/github/v/release/kevinthedang/discord-ollama?logo=github" /></a>
|
<a href="#"></a><a href="https://github.com/kevinthedang/discord-ollama/releases/latest"><img alt="Release" src="https://img.shields.io/github/v/release/kevinthedang/discord-ollama?logo=github" /></a>
|
||||||
<a href="#"></a><a href="https://github.com/kevinthedang/discord-ollama/actions/workflows/build.yml"><img alt="Build Status" src="https://github.com/kevinthedang/discord-ollama/actions/workflows/build.yml/badge.svg" /></a>
|
<a href="#"></a><a href="https://github.com/kevinthedang/discord-ollama/actions/workflows/build.yml"><img alt="Build Status" src="https://github.com/kevinthedang/discord-ollama/actions/workflows/build.yml/badge.svg" /></a>
|
||||||
|
<a href="#"></a><a href="https://github.com/kevinthedang/discord-ollama/actions/workflows/release.yml"><img alt="Release Status" src="https://github.com/kevinthedang/discord-ollama/actions/workflows/release.yml/badge.svg" /></a>
|
||||||
<a href="#"></a><a href="https://github.com/kevinthedang/discord-ollama/actions/workflows/test.yml"><img alt="Testing Status" src="https://github.com/kevinthedang/discord-ollama/actions/workflows/test.yml/badge.svg" /></a>
|
<a href="#"></a><a href="https://github.com/kevinthedang/discord-ollama/actions/workflows/test.yml"><img alt="Testing Status" src="https://github.com/kevinthedang/discord-ollama/actions/workflows/test.yml/badge.svg" /></a>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
@@ -34,25 +35,17 @@ The project aims to:
|
|||||||
* You will need a `.env` file in the root of the project directory with the bot's token. There is a `.env.sample` is provided for you as a reference for what environment variables.
|
* You will need a `.env` file in the root of the project directory with the bot's token. There is a `.env.sample` is provided for you as a reference for what environment variables.
|
||||||
* For example, `CLIENT_TOKEN = [Bot Token]`
|
* For example, `CLIENT_TOKEN = [Bot Token]`
|
||||||
* Please refer to the docs for bot setup.
|
* Please refer to the docs for bot setup.
|
||||||
|
* [Creating a Discord App](./docs/setup-discord-app.md)
|
||||||
* [Local Machine Setup](./docs/setup-local.md)
|
* [Local Machine Setup](./docs/setup-local.md)
|
||||||
* [Docker Setup for Servers and Local Machines](./docs/setup-docker.md)
|
* [Docker Setup for Servers and Local Machines](./docs/setup-docker.md)
|
||||||
* Nvidia is recommended for now, but support for other GPUs should be development.
|
* Nvidia is recommended for now, but support for other GPUs should be development.
|
||||||
* Local use is not recommended.
|
* Local use is not recommended.
|
||||||
* [Creating a Discord App](./docs/setup-discord-app.md)
|
|
||||||
|
|
||||||
## Resources
|
## Resources
|
||||||
* [NodeJS](https://nodejs.org/en)
|
* [NodeJS](https://nodejs.org/en)
|
||||||
* This project runs on `lts\hydrogen`.
|
* This project runs on `lts\hydrogen`.
|
||||||
* To run dev in `ts-node`/`nodemon`, using `v18.18.2` is recommended.
|
|
||||||
* To run dev with `tsx`, you can use `v20.10.0` or earlier.
|
|
||||||
* This project supports any NodeJS version above `16.x.x` to only allow ESModules.
|
* This project supports any NodeJS version above `16.x.x` to only allow ESModules.
|
||||||
* [Ollama](https://ollama.com/)
|
* [Ollama](https://ollama.com/)
|
||||||
* [Ollama Docker Image](https://hub.docker.com/r/ollama/ollama)
|
* [Ollama Docker Image](https://hub.docker.com/r/ollama/ollama)
|
||||||
|
|
||||||
> [!CAUTION]
|
|
||||||
> `v18.X.X` or `lts/hydrogen` will not run properly for `npm run dev-mon`. It is recommended to just use `npm run dev-tsx` for development. The nodemon version will likely be removed in a future update.
|
|
||||||
|
|
||||||
* [Discord Developer Portal](https://discord.com/developers/docs/intro)
|
|
||||||
* [Discord.js Docs](https://discord.js.org/docs/packages/discord.js/main)
|
* [Discord.js Docs](https://discord.js.org/docs/packages/discord.js/main)
|
||||||
* [Setting up Docker (Ubuntu 20.04)](https://www.digitalocean.com/community/tutorials/how-to-install-and-use-docker-on-ubuntu-20-04)
|
* [Setting up Docker (Ubuntu 20.04)](https://www.digitalocean.com/community/tutorials/how-to-install-and-use-docker-on-ubuntu-20-04)
|
||||||
* [Setting up Nvidia Container Toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html)
|
* [Setting up Nvidia Container Toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html)
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# creates the docker compose
|
# creates the docker compose
|
||||||
version: '3.7'
|
|
||||||
|
|
||||||
# build individual services
|
# build individual services
|
||||||
services:
|
services:
|
||||||
@@ -8,10 +7,9 @@ services:
|
|||||||
build: ./ # find docker file in designated path
|
build: ./ # find docker file in designated path
|
||||||
container_name: discord
|
container_name: discord
|
||||||
restart: always # rebuild container always
|
restart: always # rebuild container always
|
||||||
image: discord/bot:0.5.5
|
image: kevinthedang/discord-ollama:0.5.11
|
||||||
environment:
|
environment:
|
||||||
CLIENT_TOKEN: ${CLIENT_TOKEN}
|
CLIENT_TOKEN: ${CLIENT_TOKEN}
|
||||||
GUILD_ID: ${GUILD_ID}
|
|
||||||
MODEL: ${MODEL}
|
MODEL: ${MODEL}
|
||||||
CLIENT_UID: ${CLIENT_UID}
|
CLIENT_UID: ${CLIENT_UID}
|
||||||
OLLAMA_IP: ${OLLAMA_IP}
|
OLLAMA_IP: ${OLLAMA_IP}
|
||||||
|
|||||||
@@ -56,7 +56,7 @@ sudo systemctl restart docker
|
|||||||
* `docker compose rm`
|
* `docker compose rm`
|
||||||
* `docker ps` to check if containers have been removed.
|
* `docker ps` to check if containers have been removed.
|
||||||
* This may not work if the nvidia installation was done incorrectly. If this is the case, please utilize the [Manual "Clean-up"](#manual-run-with-docker) shown below.
|
* This may not work if the nvidia installation was done incorrectly. If this is the case, please utilize the [Manual "Clean-up"](#manual-run-with-docker) shown below.
|
||||||
* You can also use `npm run clean` to clean up the containers and remove the network to address a possible `Address already in use` problem.
|
* You can also use `npm run clean` to clean up the containers and remove the network to address a possible `Address already in use` problem. This script does not have to end without error to work.
|
||||||
|
|
||||||
## Manual Run (with Docker)
|
## Manual Run (with Docker)
|
||||||
* Run the following commands:
|
* Run the following commands:
|
||||||
|
|||||||
@@ -14,7 +14,7 @@
|
|||||||
## To Run Locally (without Docker)
|
## To Run Locally (without Docker)
|
||||||
* Run `npm install` to install the npm packages.
|
* Run `npm install` to install the npm packages.
|
||||||
* Ensure that your [.env](../.env.sample) file's `OLLAMA_IP` is `127.0.0.1` to work properly.
|
* Ensure that your [.env](../.env.sample) file's `OLLAMA_IP` is `127.0.0.1` to work properly.
|
||||||
* You only need your `CLIENT_TOKEN`, `GUILD_ID`, `MODEL`, `CLIENT_UID`, `OLLAMA_IP`, `OLLAMA_PORT`.
|
* You only need your `CLIENT_TOKEN`, `MODEL`, `CLIENT_UID`, `OLLAMA_IP`, `OLLAMA_PORT`.
|
||||||
* The ollama ip and port should just use it's defaults by nature. If not, utilize `OLLAMA_IP = 127.0.0.1` and `OLLAMA_PORT = 11434`.
|
* The ollama ip and port should just use it's defaults by nature. If not, utilize `OLLAMA_IP = 127.0.0.1` and `OLLAMA_PORT = 11434`.
|
||||||
* Now, you can run the bot by running `npm run client` which will build and run the decompiled typescript and run the setup for ollama.
|
* Now, you can run the bot by running `npm run client` which will build and run the decompiled typescript and run the setup for ollama.
|
||||||
* **IMPORTANT**: This must be ran in the wsl/Linux instance to work properly! Using Command Prompt/Powershell/Git Bash/etc. will not work on Windows (at least in my experience).
|
* **IMPORTANT**: This must be ran in the wsl/Linux instance to work properly! Using Command Prompt/Powershell/Git Bash/etc. will not work on Windows (at least in my experience).
|
||||||
|
|||||||
12
nodemon.json
12
nodemon.json
@@ -1,12 +0,0 @@
|
|||||||
{
|
|
||||||
"restartable": "rs",
|
|
||||||
"ignore": ["node_modules/"],
|
|
||||||
"watch": ["src/"],
|
|
||||||
"execMap": {
|
|
||||||
"ts": "ts-node --esm"
|
|
||||||
},
|
|
||||||
"env": {
|
|
||||||
"NODE_ENV": "development"
|
|
||||||
},
|
|
||||||
"ext": "js,json,ts"
|
|
||||||
}
|
|
||||||
2061
package-lock.json
generated
2061
package-lock.json
generated
File diff suppressed because it is too large
Load Diff
36
package.json
36
package.json
@@ -1,44 +1,42 @@
|
|||||||
{
|
{
|
||||||
"name": "discord-ollama",
|
"name": "discord-ollama",
|
||||||
"version": "0.5.5",
|
"version": "0.5.11",
|
||||||
"description": "Ollama Integration into discord",
|
"description": "Ollama Integration into discord",
|
||||||
"main": "build/index.js",
|
"main": "build/index.js",
|
||||||
"exports": "./build/index.js",
|
"exports": "./build/index.js",
|
||||||
"scripts": {
|
"scripts": {
|
||||||
"dev-tsx": "tsx watch src/index.ts",
|
|
||||||
"dev-mon": "nodemon --config nodemon.json src/index.ts",
|
|
||||||
"build": "tsc",
|
|
||||||
"test:run": "vitest run",
|
"test:run": "vitest run",
|
||||||
"test:coverage": "vitest run --coverage",
|
"test:coverage": "vitest run --coverage",
|
||||||
|
"watch": "tsx watch src",
|
||||||
|
"build": "tsc",
|
||||||
"prod": "node .",
|
"prod": "node .",
|
||||||
"client": "npm run build && npm run prod",
|
"client": "npm run build && npm run prod",
|
||||||
"clean": "docker compose down && docker rmi $(docker images | grep $(node -p \"require('./package.json').version\") | tr -s ' ' | cut -d ' ' -f 3) && docker rmi $(docker images --filter \"dangling=true\" -q --no-trunc)",
|
"clean": "docker compose down && docker rmi $(docker images | grep kevinthedang | tr -s ' ' | cut -d ' ' -f 3) && docker rmi $(docker images --filter \"dangling=true\" -q --no-trunc)",
|
||||||
"start": "docker compose build --no-cache && docker compose up -d",
|
"start": "docker compose build --no-cache && docker compose up -d",
|
||||||
"docker:start": "npm run docker:network && npm run docker:build && npm run docker:client && npm run docker:ollama",
|
"docker:clean": "docker rm -f discord && docker rm -f ollama && docker network prune -f && docker rmi $(docker images | grep kevinthedang | tr -s ' ' | cut -d ' ' -f 3) && docker rmi $(docker images --filter \"dangling=true\" -q --no-trunc)",
|
||||||
"docker:start-cpu": "npm run docker:network && npm run docker:build && npm run docker:client && npm run docker:ollama-cpu",
|
|
||||||
"docker:clean": "docker rm -f discord && docker rm -f ollama && docker rmi $(docker images --filter \"dangling=true\" -q --no-trunc)",
|
|
||||||
"docker:network": "docker network create --subnet=172.18.0.0/16 ollama-net",
|
"docker:network": "docker network create --subnet=172.18.0.0/16 ollama-net",
|
||||||
"docker:build": "docker build --no-cache -t discord/bot:$(node -p \"require('./package.json').version\") .",
|
"docker:build": "docker build --no-cache -t kevinthedang/discord-ollama:$(node -p \"require('./package.json').version\") .",
|
||||||
"docker:test": "docker run -d --rm -v discord:/src/app --name test discord/bot:$(node -p \"require('./package.json').version\") npm run test:run",
|
"docker:build-latest": "docker build --no-cache -t kevinthedang/discord-ollama:latest .",
|
||||||
"docker:client": "docker run -d -v discord:/src/app --name discord --network ollama-net --ip 172.18.0.3 discord/bot:$(node -p \"require('./package.json').version\")",
|
"docker:client": "docker run -d -v discord:/src/app --name discord --network ollama-net --ip 172.18.0.3 kevinthedang/discord-ollama:$(node -p \"require('./package.json').version\")",
|
||||||
"docker:ollama": "docker run -d --gpus=all -v ollama:/root/.ollama -p 11434:11434 --name ollama --network ollama-net --ip 172.18.0.2 ollama/ollama:latest",
|
"docker:ollama": "docker run -d --gpus=all -v ollama:/root/.ollama -p 11434:11434 --name ollama --network ollama-net --ip 172.18.0.2 ollama/ollama:latest",
|
||||||
"docker:ollama-cpu": "docker run -d -v ollama:/root/.ollama -p 11434:11434 --name ollama --network ollama-net --ip 172.18.0.2 ollama/ollama:latest"
|
"docker:ollama-cpu": "docker run -d -v ollama:/root/.ollama -p 11434:11434 --name ollama --network ollama-net --ip 172.18.0.2 ollama/ollama:latest",
|
||||||
|
"docker:start": "docker network prune -f && npm run docker:network && npm run docker:build && npm run docker:client && npm run docker:ollama",
|
||||||
|
"docker:start-cpu": "docker network prune -f && npm run docker:network && npm run docker:build && npm run docker:client && npm run docker:ollama-cpu"
|
||||||
},
|
},
|
||||||
"author": "Kevin Dang",
|
"author": "Kevin Dang",
|
||||||
"license": "ISC",
|
"license": "ISC",
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"discord.js": "^14.15.3",
|
"discord.js": "^14.15.3",
|
||||||
"dotenv": "^16.4.5",
|
"dotenv": "^16.4.5",
|
||||||
"ollama": "^0.5.2"
|
"ollama": "^0.5.8"
|
||||||
},
|
},
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
"@types/node": "^20.14.2",
|
"@types/node": "^22.5.1",
|
||||||
"@vitest/coverage-v8": "^1.6.0",
|
"@vitest/coverage-v8": "^2.0.5",
|
||||||
"nodemon": "^3.1.3",
|
|
||||||
"ts-node": "^10.9.2",
|
"ts-node": "^10.9.2",
|
||||||
"tsx": "^4.15.5",
|
"tsx": "^4.19.0",
|
||||||
"typescript": "^5.4.5",
|
"typescript": "^5.5.4",
|
||||||
"vitest": "^1.6.0"
|
"vitest": "^2.0.5"
|
||||||
},
|
},
|
||||||
"type": "module",
|
"type": "module",
|
||||||
"engines": {
|
"engines": {
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
import { ChannelType, Client, CommandInteraction, ApplicationCommandOptionType } from 'discord.js'
|
import { ChannelType, Client, CommandInteraction, ApplicationCommandOptionType } from 'discord.js'
|
||||||
import { SlashCommand } from '../utils/commands.js'
|
import { SlashCommand } from '../utils/commands.js'
|
||||||
import { openConfig } from '../utils/jsonHandler.js'
|
import { openConfig } from '../utils/index.js'
|
||||||
|
|
||||||
export const Capacity: SlashCommand = {
|
export const Capacity: SlashCommand = {
|
||||||
name: 'modify-capacity',
|
name: 'modify-capacity',
|
||||||
@@ -20,7 +20,7 @@ export const Capacity: SlashCommand = {
|
|||||||
run: async (client: Client, interaction: CommandInteraction) => {
|
run: async (client: Client, interaction: CommandInteraction) => {
|
||||||
// fetch channel and message
|
// fetch channel and message
|
||||||
const channel = await client.channels.fetch(interaction.channelId)
|
const channel = await client.channels.fetch(interaction.channelId)
|
||||||
if (!channel || channel.type !== (ChannelType.PublicThread && ChannelType.GuildText)) return
|
if (!channel || channel.type !== (ChannelType.PrivateThread && ChannelType.PublicThread && ChannelType.GuildText)) return
|
||||||
|
|
||||||
// set state of bot chat features
|
// set state of bot chat features
|
||||||
openConfig(`${interaction.user.username}-config.json`, interaction.commandName, interaction.options.get('context-capacity')?.value)
|
openConfig(`${interaction.user.username}-config.json`, interaction.commandName, interaction.options.get('context-capacity')?.value)
|
||||||
|
|||||||
@@ -1,34 +0,0 @@
|
|||||||
import { ApplicationCommandOptionType, ChannelType, Client, CommandInteraction } from 'discord.js'
|
|
||||||
import { SlashCommand } from '../utils/commands.js'
|
|
||||||
import { openConfig } from '../utils/jsonHandler.js'
|
|
||||||
|
|
||||||
export const ChannelToggle: SlashCommand = {
|
|
||||||
name: 'channel-toggle',
|
|
||||||
description: 'toggles channel or thread usage.',
|
|
||||||
|
|
||||||
// set user option for toggling
|
|
||||||
options: [
|
|
||||||
{
|
|
||||||
name: 'toggle-channel',
|
|
||||||
description: 'toggle channel usage, otherwise threads',
|
|
||||||
type: ApplicationCommandOptionType.Boolean,
|
|
||||||
required: true
|
|
||||||
}
|
|
||||||
],
|
|
||||||
|
|
||||||
// Query for chatting preference
|
|
||||||
run: async (client: Client, interaction: CommandInteraction) => {
|
|
||||||
// fetch channel location
|
|
||||||
const channel = await client.channels.fetch(interaction.channelId)
|
|
||||||
if (!channel || channel.type !== (ChannelType.PublicThread && ChannelType.GuildText)) return
|
|
||||||
|
|
||||||
|
|
||||||
// set state of bot channel preferences
|
|
||||||
openConfig(`${interaction.guildId}-config.json`, interaction.commandName, interaction.options.get('toggle-channel')?.value)
|
|
||||||
|
|
||||||
interaction.reply({
|
|
||||||
content: `Channel Preferences have for Regular Channels set to \`${interaction.options.get('toggle-channel')?.value}\``,
|
|
||||||
ephemeral: true
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
34
src/commands/cleanUserChannelHistory.ts
Normal file
34
src/commands/cleanUserChannelHistory.ts
Normal file
@@ -0,0 +1,34 @@
|
|||||||
|
import { ChannelType, Client, CommandInteraction, TextChannel } from 'discord.js'
|
||||||
|
import { SlashCommand } from '../utils/commands.js'
|
||||||
|
import { clearChannelInfo } from '../utils/index.js'
|
||||||
|
|
||||||
|
export const ClearUserChannelHistory: SlashCommand = {
|
||||||
|
name: 'clear-user-channel-history',
|
||||||
|
description: 'clears history for user running this command in current channel',
|
||||||
|
|
||||||
|
// Clear channel history for intended user
|
||||||
|
run: async (client: Client, interaction: CommandInteraction) => {
|
||||||
|
// fetch current channel
|
||||||
|
const channel = await client.channels.fetch(interaction.channelId)
|
||||||
|
|
||||||
|
// if not an existing channel or a GuildText, fail command
|
||||||
|
if (!channel || channel.type !== ChannelType.GuildText) return
|
||||||
|
|
||||||
|
// clear channel info for user
|
||||||
|
const successfulWipe = await clearChannelInfo(interaction.channelId,
|
||||||
|
interaction.channel as TextChannel,
|
||||||
|
interaction.user.username)
|
||||||
|
|
||||||
|
// check result of clearing history
|
||||||
|
if (successfulWipe)
|
||||||
|
interaction.reply({
|
||||||
|
content: `Channel history in **${channel.name}** cleared for **${interaction.user.username}**.`,
|
||||||
|
ephemeral: true
|
||||||
|
})
|
||||||
|
else
|
||||||
|
interaction.reply({
|
||||||
|
content: `Channel history could not be found for **${interaction.user.username}** in **${channel.name}**.\n\nPlease chat with **${client.user?.username}** to start a chat history.`,
|
||||||
|
ephemeral: true
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,6 +1,6 @@
|
|||||||
import { ChannelType, Client, CommandInteraction, ApplicationCommandOptionType } from 'discord.js'
|
import { ChannelType, Client, CommandInteraction, ApplicationCommandOptionType } from 'discord.js'
|
||||||
import { SlashCommand } from '../utils/commands.js'
|
import { SlashCommand } from '../utils/commands.js'
|
||||||
import { openConfig } from '../utils/jsonHandler.js'
|
import { openConfig } from '../utils/index.js'
|
||||||
|
|
||||||
export const Disable: SlashCommand = {
|
export const Disable: SlashCommand = {
|
||||||
name: 'toggle-chat',
|
name: 'toggle-chat',
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ import { Disable } from './disable.js'
|
|||||||
import { Shutoff } from './shutoff.js'
|
import { Shutoff } from './shutoff.js'
|
||||||
import { Capacity } from './capacity.js'
|
import { Capacity } from './capacity.js'
|
||||||
import { PrivateThreadCreate } from './threadPrivateCreate.js'
|
import { PrivateThreadCreate } from './threadPrivateCreate.js'
|
||||||
import { ChannelToggle } from './channelToggle.js'
|
import { ClearUserChannelHistory } from './cleanUserChannelHistory.js'
|
||||||
|
|
||||||
export default [
|
export default [
|
||||||
ThreadCreate,
|
ThreadCreate,
|
||||||
@@ -16,5 +16,5 @@ export default [
|
|||||||
Disable,
|
Disable,
|
||||||
Shutoff,
|
Shutoff,
|
||||||
Capacity,
|
Capacity,
|
||||||
ChannelToggle
|
ClearUserChannelHistory
|
||||||
] as SlashCommand[]
|
] as SlashCommand[]
|
||||||
@@ -1,6 +1,6 @@
|
|||||||
import { ApplicationCommandOptionType, ChannelType, Client, CommandInteraction } from 'discord.js'
|
import { ApplicationCommandOptionType, ChannelType, Client, CommandInteraction } from 'discord.js'
|
||||||
import { SlashCommand } from '../utils/commands.js'
|
import { SlashCommand } from '../utils/commands.js'
|
||||||
import { openConfig } from '../utils/jsonHandler.js'
|
import { openConfig } from '../utils/index.js'
|
||||||
|
|
||||||
export const MessageStream: SlashCommand = {
|
export const MessageStream: SlashCommand = {
|
||||||
name: 'message-stream',
|
name: 'message-stream',
|
||||||
@@ -20,7 +20,7 @@ export const MessageStream: SlashCommand = {
|
|||||||
run: async (client: Client, interaction: CommandInteraction) => {
|
run: async (client: Client, interaction: CommandInteraction) => {
|
||||||
// verify channel
|
// verify channel
|
||||||
const channel = await client.channels.fetch(interaction.channelId)
|
const channel = await client.channels.fetch(interaction.channelId)
|
||||||
if (!channel || channel.type !== (ChannelType.PublicThread && ChannelType.GuildText)) return
|
if (!channel || channel.type !== (ChannelType.PrivateThread && ChannelType.PublicThread && ChannelType.GuildText)) return
|
||||||
|
|
||||||
// save value to json and write to it
|
// save value to json and write to it
|
||||||
openConfig(`${interaction.user.username}-config.json`, interaction.commandName, interaction.options.get('stream')?.value)
|
openConfig(`${interaction.user.username}-config.json`, interaction.commandName, interaction.options.get('stream')?.value)
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
import { ChannelType, Client, CommandInteraction, ApplicationCommandOptionType } from 'discord.js'
|
import { ChannelType, Client, CommandInteraction, ApplicationCommandOptionType } from 'discord.js'
|
||||||
import { SlashCommand } from '../utils/commands.js'
|
import { SlashCommand } from '../utils/commands.js'
|
||||||
import { openConfig } from '../utils/jsonHandler.js'
|
import { openConfig } from '../utils/index.js'
|
||||||
|
|
||||||
export const MessageStyle: SlashCommand = {
|
export const MessageStyle: SlashCommand = {
|
||||||
name: 'message-style',
|
name: 'message-style',
|
||||||
@@ -20,7 +20,7 @@ export const MessageStyle: SlashCommand = {
|
|||||||
run: async (client: Client, interaction: CommandInteraction) => {
|
run: async (client: Client, interaction: CommandInteraction) => {
|
||||||
// fetch channel and message
|
// fetch channel and message
|
||||||
const channel = await client.channels.fetch(interaction.channelId)
|
const channel = await client.channels.fetch(interaction.channelId)
|
||||||
if (!channel || channel.type !== (ChannelType.PublicThread && ChannelType.GuildText)) return
|
if (!channel || channel.type !== (ChannelType.PrivateThread && ChannelType.PublicThread && ChannelType.GuildText)) return
|
||||||
|
|
||||||
// set the message style
|
// set the message style
|
||||||
openConfig(`${interaction.user.username}-config.json`, interaction.commandName, interaction.options.get('embed')?.value)
|
openConfig(`${interaction.user.username}-config.json`, interaction.commandName, interaction.options.get('embed')?.value)
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
import { ChannelType, Client, CommandInteraction, TextChannel } from 'discord.js'
|
import { ChannelType, Client, CommandInteraction, TextChannel, ThreadChannel } from 'discord.js'
|
||||||
import { SlashCommand } from '../utils/commands.js'
|
import { SlashCommand } from '../utils/commands.js'
|
||||||
import { openThreadInfo } from '../utils/jsonHandler.js'
|
import { openChannelInfo } from '../utils/index.js'
|
||||||
|
|
||||||
export const ThreadCreate: SlashCommand = {
|
export const ThreadCreate: SlashCommand = {
|
||||||
name: 'thread',
|
name: 'thread',
|
||||||
@@ -22,8 +22,9 @@ export const ThreadCreate: SlashCommand = {
|
|||||||
thread.send(`Hello ${interaction.user} and others! \n\nIt's nice to meet you. Please talk to me by typing **@${client.user?.username}** with your prompt.`)
|
thread.send(`Hello ${interaction.user} and others! \n\nIt's nice to meet you. Please talk to me by typing **@${client.user?.username}** with your prompt.`)
|
||||||
|
|
||||||
// handle storing this chat channel
|
// handle storing this chat channel
|
||||||
// store: thread.id, thread.name
|
openChannelInfo(thread.id,
|
||||||
openThreadInfo(`${thread.id}.json`, thread)
|
thread as ThreadChannel,
|
||||||
|
interaction.user.tag)
|
||||||
|
|
||||||
// user only reply
|
// user only reply
|
||||||
return interaction.reply({
|
return interaction.reply({
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
import { ChannelType, Client, CommandInteraction, TextChannel } from 'discord.js'
|
import { ChannelType, Client, CommandInteraction, TextChannel, ThreadChannel } from 'discord.js'
|
||||||
import { SlashCommand } from '../utils/commands.js'
|
import { SlashCommand } from '../utils/commands.js'
|
||||||
import { openThreadInfo } from '../utils/jsonHandler.js'
|
import { openChannelInfo } from '../utils/index.js'
|
||||||
|
|
||||||
export const PrivateThreadCreate: SlashCommand = {
|
export const PrivateThreadCreate: SlashCommand = {
|
||||||
name: 'private-thread',
|
name: 'private-thread',
|
||||||
@@ -23,7 +23,10 @@ export const PrivateThreadCreate: SlashCommand = {
|
|||||||
|
|
||||||
// handle storing this chat channel
|
// handle storing this chat channel
|
||||||
// store: thread.id, thread.name
|
// store: thread.id, thread.name
|
||||||
openThreadInfo(`${thread.id}.json`, thread)
|
openChannelInfo(thread.id,
|
||||||
|
thread as ThreadChannel,
|
||||||
|
interaction.user.tag
|
||||||
|
)
|
||||||
|
|
||||||
// user only reply
|
// user only reply
|
||||||
return interaction.reply({
|
return interaction.reply({
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
import { embedMessage, event, Events, normalMessage, UserMessage } from '../utils/index.js'
|
import { embedMessage, event, Events, normalMessage, UserMessage } from '../utils/index.js'
|
||||||
import { getChannelInfo, getServerConfig, getThread, getUserConfig, openChannelInfo, openConfig, openThreadInfo, ServerConfig, UserConfig } from '../utils/jsonHandler.js'
|
import { getChannelInfo, getServerConfig, getUserConfig, openChannelInfo, openConfig, ServerConfig, UserConfig } from '../utils/index.js'
|
||||||
import { clean } from '../utils/mentionClean.js'
|
import { clean } from '../utils/mentionClean.js'
|
||||||
import { TextChannel, ThreadChannel } from 'discord.js'
|
import { TextChannel } from 'discord.js'
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Max Message length for free users is 2000 characters (bot or not).
|
* Max Message length for free users is 2000 characters (bot or not).
|
||||||
@@ -9,7 +9,7 @@ import { TextChannel, ThreadChannel } from 'discord.js'
|
|||||||
*
|
*
|
||||||
* @param message the message received from the channel
|
* @param message the message received from the channel
|
||||||
*/
|
*/
|
||||||
export default event(Events.MessageCreate, async ({ log, msgHist, tokens, ollama, client }, message) => {
|
export default event(Events.MessageCreate, async ({ log, msgHist, tokens, ollama }, message) => {
|
||||||
log(`Message \"${clean(message.content)}\" from ${message.author.tag} in channel/thread ${message.channelId}.`)
|
log(`Message \"${clean(message.content)}\" from ${message.author.tag} in channel/thread ${message.channelId}.`)
|
||||||
|
|
||||||
// Do not respond if bot talks in the chat
|
// Do not respond if bot talks in the chat
|
||||||
@@ -23,15 +23,12 @@ export default event(Events.MessageCreate, async ({ log, msgHist, tokens, ollama
|
|||||||
|
|
||||||
try {
|
try {
|
||||||
// Retrieve Server/Guild Preferences
|
// Retrieve Server/Guild Preferences
|
||||||
const serverConfig: ServerConfig = await new Promise((resolve, reject) => {
|
await new Promise((resolve, reject) => {
|
||||||
getServerConfig(`${message.guildId}-config.json`, (config) => {
|
getServerConfig(`${message.guildId}-config.json`, (config) => {
|
||||||
// check if config.json exists
|
// check if config.json exists
|
||||||
if (config === undefined) {
|
if (config === undefined) {
|
||||||
// Allowing chat options to be available
|
// Allowing chat options to be available
|
||||||
openConfig(`${message.guildId}-config.json`, 'toggle-chat', true)
|
openConfig(`${message.guildId}-config.json`, 'toggle-chat', true)
|
||||||
|
|
||||||
// default to channel scope chats
|
|
||||||
openConfig(`${message.guildId}-config.json`, 'channel-toggle', true)
|
|
||||||
reject(new Error('No Server Preferences is set up.\n\nCreating default server preferences file...\nPlease try chatting again.'))
|
reject(new Error('No Server Preferences is set up.\n\nCreating default server preferences file...\nPlease try chatting again.'))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -42,14 +39,6 @@ export default event(Events.MessageCreate, async ({ log, msgHist, tokens, ollama
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// ensure channel json exists, if not create it
|
|
||||||
if (config.options['channel-toggle']) {
|
|
||||||
openChannelInfo(message.channelId,
|
|
||||||
message.channel as TextChannel,
|
|
||||||
message.author.username
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
resolve(config)
|
resolve(config)
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
@@ -80,27 +69,35 @@ export default event(Events.MessageCreate, async ({ log, msgHist, tokens, ollama
|
|||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
// need new check for "open/active" threads/channels here!
|
// need new check for "open/active" threads/channels here!
|
||||||
const chatMessages: UserMessage[] = await new Promise((resolve) => {
|
let chatMessages: UserMessage[] = await new Promise((resolve) => {
|
||||||
// set new queue to modify
|
// set new queue to modify
|
||||||
if (serverConfig.options['channel-toggle']) {
|
getChannelInfo(`${message.channelId}-${message.author.username}.json`, (channelInfo) => {
|
||||||
|
if (channelInfo?.messages)
|
||||||
|
resolve(channelInfo.messages)
|
||||||
|
else {
|
||||||
|
log(`Channel/Thread ${message.channel}-${message.author.username} does not exist. File will be created shortly...`)
|
||||||
|
resolve([])
|
||||||
|
}
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
if (chatMessages.length === 0) {
|
||||||
|
chatMessages = await new Promise((resolve, reject) => {
|
||||||
|
openChannelInfo(message.channelId,
|
||||||
|
message.channel as TextChannel,
|
||||||
|
message.author.tag
|
||||||
|
)
|
||||||
getChannelInfo(`${message.channelId}-${message.author.username}.json`, (channelInfo) => {
|
getChannelInfo(`${message.channelId}-${message.author.username}.json`, (channelInfo) => {
|
||||||
if (channelInfo?.messages)
|
if (channelInfo?.messages)
|
||||||
resolve(channelInfo.messages)
|
resolve(channelInfo.messages)
|
||||||
else
|
else {
|
||||||
log(`Channel ${message.channel}-${message.author.username} does not exist.`)
|
log(`Channel/Thread ${message.channel}-${message.author.username} does not exist. File will be created shortly...`)
|
||||||
|
reject(new Error(`Failed to find ${message.author.username}'s history. Try chatting again.`))
|
||||||
|
}
|
||||||
})
|
})
|
||||||
} else {
|
})
|
||||||
getThread(`${message.channelId}.json`, (threadInfo) => {
|
}
|
||||||
if (threadInfo?.messages)
|
|
||||||
resolve(threadInfo.messages)
|
|
||||||
else
|
|
||||||
log(`Thread ${message.channelId} does not exist.`)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
// response string for ollama to put its response
|
// response string for ollama to put its response
|
||||||
let response: string
|
let response: string
|
||||||
@@ -136,18 +133,11 @@ export default event(Events.MessageCreate, async ({ log, msgHist, tokens, ollama
|
|||||||
})
|
})
|
||||||
|
|
||||||
// only update the json on success
|
// only update the json on success
|
||||||
if (serverConfig.options['channel-toggle']) {
|
openChannelInfo(message.channelId,
|
||||||
openChannelInfo(message.channelId,
|
message.channel as TextChannel,
|
||||||
message.channel as TextChannel,
|
message.author.tag,
|
||||||
message.author.tag,
|
msgHist.getItems()
|
||||||
msgHist.getItems()
|
)
|
||||||
)
|
|
||||||
} else {
|
|
||||||
openThreadInfo(`${message.channelId}.json`,
|
|
||||||
client.channels.fetch(message.channelId) as unknown as ThreadChannel,
|
|
||||||
msgHist.getItems()
|
|
||||||
)
|
|
||||||
}
|
|
||||||
} catch (error: any) {
|
} catch (error: any) {
|
||||||
msgHist.pop() // remove message because of failure
|
msgHist.pop() // remove message because of failure
|
||||||
message.reply(`**Error Occurred:**\n\n**Reason:** *${error.message}*`)
|
message.reply(`**Error Occurred:**\n\n**Reason:** *${error.message}*`)
|
||||||
|
|||||||
@@ -5,16 +5,36 @@ import fs from 'fs'
|
|||||||
/**
|
/**
|
||||||
* Event to remove the associated .json file for a thread once deleted
|
* Event to remove the associated .json file for a thread once deleted
|
||||||
*/
|
*/
|
||||||
export default event(Events.ThreadDelete, ({ log }, thread: ThreadChannel) => {
|
export default event(Events.ThreadDelete, async ({ log }, thread: ThreadChannel) => {
|
||||||
const filePath = `data/${thread.id}.json`
|
// iterate through every guild member in the thread and delete their history, except the bot
|
||||||
if (fs.existsSync(filePath)) {
|
try {
|
||||||
fs.unlink(filePath, (error) => {
|
log(`Number of User Guild Members in Thread being deleted: ${thread.memberCount!! - 1}`)
|
||||||
if (error)
|
const dirPath = 'data/'
|
||||||
log(`Error deleting file ${filePath}`, error)
|
|
||||||
else
|
// read all files in data/
|
||||||
log(`Successfully deleted ${filePath} thread info`)
|
fs.readdir(dirPath, (error, files) => {
|
||||||
|
if (error) {
|
||||||
|
log(`Error reading directory ${dirPath}`, error)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// filter files by thread id being deleted
|
||||||
|
const filesToDiscard = files.filter(
|
||||||
|
file => file.startsWith(`${thread.id}-`) &&
|
||||||
|
file.endsWith('.json'))
|
||||||
|
|
||||||
|
// remove files by unlinking
|
||||||
|
filesToDiscard.forEach(file => {
|
||||||
|
const filePath = dirPath + file
|
||||||
|
fs.unlink(filePath, error => {
|
||||||
|
if (error)
|
||||||
|
log(`Error deleting file ${filePath}`, error)
|
||||||
|
else
|
||||||
|
log(`Successfully deleted ${filePath} thread information`)
|
||||||
|
})
|
||||||
|
})
|
||||||
})
|
})
|
||||||
} else {
|
} catch (error) {
|
||||||
log(`File ${filePath} does not exist.`)
|
log(`Issue deleting user history files from ${thread.id}`)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
@@ -4,7 +4,6 @@ export const Keys = {
|
|||||||
clientToken: getEnvVar('CLIENT_TOKEN'),
|
clientToken: getEnvVar('CLIENT_TOKEN'),
|
||||||
model: getEnvVar('MODEL'),
|
model: getEnvVar('MODEL'),
|
||||||
clientUid: getEnvVar('CLIENT_UID'),
|
clientUid: getEnvVar('CLIENT_UID'),
|
||||||
guildId: getEnvVar('GUILD_ID'),
|
|
||||||
ipAddress: getEnvVar('OLLAMA_IP'),
|
ipAddress: getEnvVar('OLLAMA_IP'),
|
||||||
portAddress: getEnvVar('OLLAMA_PORT'),
|
portAddress: getEnvVar('OLLAMA_PORT'),
|
||||||
} as const // readonly keys
|
} as const // readonly keys
|
||||||
|
|||||||
51
src/utils/configInterfaces.ts
Normal file
51
src/utils/configInterfaces.ts
Normal file
@@ -0,0 +1,51 @@
|
|||||||
|
import { UserMessage } from './events.js'
|
||||||
|
|
||||||
|
export interface UserConfiguration {
|
||||||
|
'message-stream'?: boolean,
|
||||||
|
'message-style'?: boolean,
|
||||||
|
'modify-capacity': number
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface ServerConfiguration {
|
||||||
|
'toggle-chat'?: boolean,
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Parent Configuration interface
|
||||||
|
*
|
||||||
|
* @see ServerConfiguration server settings per guild
|
||||||
|
* @see UserConfiguration user configurations (only for the user for any server)
|
||||||
|
*/
|
||||||
|
export interface Configuration {
|
||||||
|
readonly name: string
|
||||||
|
options: UserConfiguration | ServerConfiguration
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* User config to use outside of this file
|
||||||
|
*/
|
||||||
|
export interface UserConfig {
|
||||||
|
readonly name: string
|
||||||
|
options: UserConfiguration
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface ServerConfig {
|
||||||
|
readonly name: string
|
||||||
|
options: ServerConfiguration
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface Channel {
|
||||||
|
readonly id: string
|
||||||
|
readonly name: string
|
||||||
|
readonly user: string
|
||||||
|
messages: UserMessage[]
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Check if the configuration we are editing/taking from is a Server Config
|
||||||
|
* @param key name of command we ran
|
||||||
|
* @returns true if command is from Server Config, false otherwise
|
||||||
|
*/
|
||||||
|
export function isServerConfigurationKey(key: string): key is keyof ServerConfiguration {
|
||||||
|
return ['toggle-chat'].includes(key);
|
||||||
|
}
|
||||||
115
src/utils/handlers/chatHistoryHandler.ts
Normal file
115
src/utils/handlers/chatHistoryHandler.ts
Normal file
@@ -0,0 +1,115 @@
|
|||||||
|
import { TextChannel, ThreadChannel } from 'discord.js'
|
||||||
|
import { Configuration, Channel, UserMessage } from '../index.js'
|
||||||
|
import fs from 'fs'
|
||||||
|
import path from 'path'
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Method to check if a thread history file exists
|
||||||
|
*
|
||||||
|
* @param channel parent thread of the requested thread (can be GuildText)
|
||||||
|
* @returns true if channel does not exist, false otherwise
|
||||||
|
*/
|
||||||
|
async function checkChannelInfoExists(channel: TextChannel, user: string) {
|
||||||
|
const doesExists: boolean = await new Promise((resolve) => {
|
||||||
|
getChannelInfo(`${channel.id}-${user}.json`, (channelInfo) => {
|
||||||
|
if (channelInfo?.messages)
|
||||||
|
resolve(true)
|
||||||
|
else
|
||||||
|
resolve(false)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
return doesExists
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Method to clear channel history for requesting user
|
||||||
|
*
|
||||||
|
* @param filename guild id string
|
||||||
|
* @param channel the TextChannel in the Guild
|
||||||
|
* @param user username of user
|
||||||
|
* @returns nothing
|
||||||
|
*/
|
||||||
|
export async function clearChannelInfo(filename: string, channel: TextChannel, user: string): Promise<boolean> {
|
||||||
|
const channelInfoExists: boolean = await checkChannelInfoExists(channel, user)
|
||||||
|
|
||||||
|
// If thread does not exist, file can't be found
|
||||||
|
if (!channelInfoExists) return false
|
||||||
|
|
||||||
|
// Attempt to clear user channel history
|
||||||
|
const fullFileName = `data/${filename}-${user}.json`
|
||||||
|
const cleanedHistory: boolean = await new Promise((resolve) => {
|
||||||
|
fs.readFile(fullFileName, 'utf8', (error, data) => {
|
||||||
|
if (error)
|
||||||
|
console.log(`[Error: openChannelInfo] Incorrect file format`)
|
||||||
|
else {
|
||||||
|
const object = JSON.parse(data)
|
||||||
|
if (object['messages'].length === 0) // already empty, let user know
|
||||||
|
resolve(false)
|
||||||
|
else {
|
||||||
|
object['messages'] = [] // cleared history
|
||||||
|
fs.writeFileSync(fullFileName, JSON.stringify(object, null, 2))
|
||||||
|
resolve(true)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
})
|
||||||
|
console.log(cleanedHistory)
|
||||||
|
return cleanedHistory
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Method to open the channel history
|
||||||
|
*
|
||||||
|
* @param filename name of the json file for the channel by user
|
||||||
|
* @param channel the text channel info
|
||||||
|
* @param user the user's name
|
||||||
|
* @param messages their messages
|
||||||
|
*/
|
||||||
|
export async function openChannelInfo(filename: string, channel: TextChannel | ThreadChannel, user: string, messages: UserMessage[] = []): Promise<void> {
|
||||||
|
const fullFileName = `data/${filename}-${user}.json`
|
||||||
|
if (fs.existsSync(fullFileName)) {
|
||||||
|
fs.readFile(fullFileName, 'utf8', (error, data) => {
|
||||||
|
if (error)
|
||||||
|
console.log(`[Error: openChannelInfo] Incorrect file format`)
|
||||||
|
else {
|
||||||
|
const object = JSON.parse(data)
|
||||||
|
if (object['messages'].length === 0)
|
||||||
|
object['messages'] = messages as []
|
||||||
|
else if (object['messages'].length !== 0 && messages.length !== 0)
|
||||||
|
object['messages'] = messages as []
|
||||||
|
fs.writeFileSync(fullFileName, JSON.stringify(object, null, 2))
|
||||||
|
}
|
||||||
|
})
|
||||||
|
} else { // file doesn't exist, create it
|
||||||
|
const object: Configuration = JSON.parse(`{ \"id\": \"${channel?.id}\", \"name\": \"${channel?.name}\", \"user\": \"${user}\", \"messages\": []}`)
|
||||||
|
|
||||||
|
const directory = path.dirname(fullFileName)
|
||||||
|
if (!fs.existsSync(directory))
|
||||||
|
fs.mkdirSync(directory, { recursive: true })
|
||||||
|
|
||||||
|
// only creating it, no need to add anything
|
||||||
|
fs.writeFileSync(fullFileName, JSON.stringify(object, null, 2))
|
||||||
|
console.log(`[Util: openChannelInfo] Created '${fullFileName}' in working directory`)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Method to get the channel information/history
|
||||||
|
*
|
||||||
|
* @param filename name of the json file for the channel by user
|
||||||
|
* @param callback function to handle resolving message history
|
||||||
|
*/
|
||||||
|
export async function getChannelInfo(filename: string, callback: (config: Channel | undefined) => void): Promise<void> {
|
||||||
|
const fullFileName = `data/${filename}`
|
||||||
|
if (fs.existsSync(fullFileName)) {
|
||||||
|
fs.readFile(fullFileName, 'utf8', (error, data) => {
|
||||||
|
if (error) {
|
||||||
|
callback(undefined)
|
||||||
|
return // something went wrong... stop
|
||||||
|
}
|
||||||
|
callback(JSON.parse(data))
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
callback(undefined) // file not found
|
||||||
|
}
|
||||||
|
}
|
||||||
92
src/utils/handlers/configHandler.ts
Normal file
92
src/utils/handlers/configHandler.ts
Normal file
@@ -0,0 +1,92 @@
|
|||||||
|
import { Configuration, ServerConfig, UserConfig, isServerConfigurationKey } from '../index.js'
|
||||||
|
import fs from 'fs'
|
||||||
|
import path from 'path'
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Method to open a file in the working directory and modify/create it
|
||||||
|
*
|
||||||
|
* @param filename name of the file
|
||||||
|
* @param key key value to access
|
||||||
|
* @param value new value to assign
|
||||||
|
*/
|
||||||
|
// add type of change (server, user)
|
||||||
|
export function openConfig(filename: string, key: string, value: any) {
|
||||||
|
const fullFileName = `data/${filename}`
|
||||||
|
|
||||||
|
// check if the file exists, if not then make the config file
|
||||||
|
if (fs.existsSync(fullFileName)) {
|
||||||
|
fs.readFile(fullFileName, 'utf8', (error, data) => {
|
||||||
|
if (error)
|
||||||
|
console.log(`[Error: openConfig] Incorrect file format`)
|
||||||
|
else {
|
||||||
|
const object = JSON.parse(data)
|
||||||
|
object['options'][key] = value
|
||||||
|
fs.writeFileSync(fullFileName, JSON.stringify(object, null, 2))
|
||||||
|
}
|
||||||
|
})
|
||||||
|
} else { // work on dynamic file creation
|
||||||
|
let object: Configuration
|
||||||
|
if (isServerConfigurationKey(key))
|
||||||
|
object = JSON.parse('{ \"name\": \"Server Confirgurations\" }')
|
||||||
|
else
|
||||||
|
object = JSON.parse('{ \"name\": \"User Confirgurations\" }')
|
||||||
|
|
||||||
|
// set standard information for config file and options
|
||||||
|
object['options'] = {
|
||||||
|
[key]: value
|
||||||
|
}
|
||||||
|
|
||||||
|
const directory = path.dirname(fullFileName)
|
||||||
|
if (!fs.existsSync(directory))
|
||||||
|
fs.mkdirSync(directory, { recursive: true })
|
||||||
|
|
||||||
|
fs.writeFileSync(`data/${filename}`, JSON.stringify(object, null, 2))
|
||||||
|
console.log(`[Util: openConfig] Created '${filename}' in working directory`)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Method to obtain the configurations of the message chat/thread
|
||||||
|
*
|
||||||
|
* @param filename name of the configuration file to get
|
||||||
|
* @param callback function to allow a promise from getting the config
|
||||||
|
*/
|
||||||
|
export async function getServerConfig(filename: string, callback: (config: ServerConfig | undefined) => void): Promise<void> {
|
||||||
|
const fullFileName = `data/${filename}`
|
||||||
|
|
||||||
|
// attempt to read the file and get the configuration
|
||||||
|
if (fs.existsSync(fullFileName)) {
|
||||||
|
fs.readFile(fullFileName, 'utf8', (error, data) => {
|
||||||
|
if (error) {
|
||||||
|
callback(undefined)
|
||||||
|
return // something went wrong... stop
|
||||||
|
}
|
||||||
|
callback(JSON.parse(data))
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
callback(undefined) // file not found
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Method to obtain the configurations of the message chat/thread
|
||||||
|
*
|
||||||
|
* @param filename name of the configuration file to get
|
||||||
|
* @param callback function to allow a promise from getting the config
|
||||||
|
*/
|
||||||
|
export async function getUserConfig(filename: string, callback: (config: UserConfig | undefined) => void): Promise<void> {
|
||||||
|
const fullFileName = `data/${filename}`
|
||||||
|
|
||||||
|
// attempt to read the file and get the configuration
|
||||||
|
if (fs.existsSync(fullFileName)) {
|
||||||
|
fs.readFile(fullFileName, 'utf8', (error, data) => {
|
||||||
|
if (error) {
|
||||||
|
callback(undefined)
|
||||||
|
return // something went wrong... stop
|
||||||
|
}
|
||||||
|
callback(JSON.parse(data))
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
callback(undefined) // file not found
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,5 +1,5 @@
|
|||||||
import { ChatResponse } from "ollama"
|
import { ChatResponse } from "ollama"
|
||||||
import { ChatParams } from "./index.js"
|
import { ChatParams } from "../index.js"
|
||||||
import { AbortableAsyncIterator } from "ollama/src/utils.js"
|
import { AbortableAsyncIterator } from "ollama/src/utils.js"
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -4,4 +4,9 @@ export * from './events.js'
|
|||||||
export * from './messageEmbed.js'
|
export * from './messageEmbed.js'
|
||||||
export * from './messageNormal.js'
|
export * from './messageNormal.js'
|
||||||
export * from './commands.js'
|
export * from './commands.js'
|
||||||
export * from './streamHandler.js'
|
export * from './configInterfaces.js'
|
||||||
|
|
||||||
|
// handler imports
|
||||||
|
export * from './handlers/chatHistoryHandler.js'
|
||||||
|
export * from './handlers/configHandler.js'
|
||||||
|
export * from './handlers/streamHandler.js'
|
||||||
|
|||||||
@@ -1,270 +0,0 @@
|
|||||||
import { TextChannel, ThreadChannel } from 'discord.js'
|
|
||||||
import { UserMessage } from './events.js'
|
|
||||||
import fs from 'fs'
|
|
||||||
import path from 'path'
|
|
||||||
|
|
||||||
export interface UserConfiguration {
|
|
||||||
'message-stream'?: boolean,
|
|
||||||
'message-style'?: boolean,
|
|
||||||
'modify-capacity': number
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface ServerConfiguration {
|
|
||||||
'toggle-chat'?: boolean,
|
|
||||||
'channel-toggle'?: boolean
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Parent Configuration interface
|
|
||||||
*
|
|
||||||
* @see ServerConfiguration server settings per guild
|
|
||||||
* @see UserConfiguration user configurations (only for the user for any server)
|
|
||||||
*/
|
|
||||||
export interface Configuration {
|
|
||||||
readonly name: string
|
|
||||||
options: UserConfiguration | ServerConfiguration
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* User config to use outside of this file
|
|
||||||
*/
|
|
||||||
export interface UserConfig {
|
|
||||||
readonly name: string
|
|
||||||
options: UserConfiguration
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface ServerConfig {
|
|
||||||
readonly name: string
|
|
||||||
options: ServerConfiguration
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface Thread {
|
|
||||||
readonly id: string
|
|
||||||
readonly name: string
|
|
||||||
messages: UserMessage[]
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface Channel {
|
|
||||||
readonly id: string
|
|
||||||
readonly name: string
|
|
||||||
readonly user: string
|
|
||||||
messages: UserMessage[]
|
|
||||||
}
|
|
||||||
|
|
||||||
function isUserConfigurationKey(key: string): key is keyof UserConfiguration {
|
|
||||||
return ['message-stream', 'message-style', 'modify-capacity'].includes(key);
|
|
||||||
}
|
|
||||||
|
|
||||||
function isServerConfigurationKey(key: string): key is keyof ServerConfiguration {
|
|
||||||
return ['toggle-chat', 'channel-toggle'].includes(key);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Method to open a file in the working directory and modify/create it
|
|
||||||
*
|
|
||||||
* @param filename name of the file
|
|
||||||
* @param key key value to access
|
|
||||||
* @param value new value to assign
|
|
||||||
*/
|
|
||||||
// add type of change (server, user)
|
|
||||||
export function openConfig(filename: string, key: string, value: any) {
|
|
||||||
const fullFileName = `data/${filename}`
|
|
||||||
|
|
||||||
// check if the file exists, if not then make the config file
|
|
||||||
if (fs.existsSync(fullFileName)) {
|
|
||||||
fs.readFile(fullFileName, 'utf8', (error, data) => {
|
|
||||||
if (error)
|
|
||||||
console.log(`[Error: openConfig] Incorrect file format`)
|
|
||||||
else {
|
|
||||||
const object = JSON.parse(data)
|
|
||||||
object['options'][key] = value
|
|
||||||
fs.writeFileSync(fullFileName, JSON.stringify(object, null, 2))
|
|
||||||
}
|
|
||||||
})
|
|
||||||
} else { // work on dynamic file creation
|
|
||||||
let object: Configuration
|
|
||||||
if (isServerConfigurationKey(key))
|
|
||||||
object = JSON.parse('{ \"name\": \"Server Confirgurations\" }')
|
|
||||||
else
|
|
||||||
object = JSON.parse('{ \"name\": \"User Confirgurations\" }')
|
|
||||||
|
|
||||||
// set standard information for config file and options
|
|
||||||
object['options'] = {
|
|
||||||
[key]: value
|
|
||||||
}
|
|
||||||
|
|
||||||
fs.writeFileSync(`data/${filename}`, JSON.stringify(object, null, 2))
|
|
||||||
console.log(`[Util: openConfig] Created '${filename}' in working directory`)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Method to obtain the configurations of the message chat/thread
|
|
||||||
*
|
|
||||||
* @param filename name of the configuration file to get
|
|
||||||
* @param callback function to allow a promise from getting the config
|
|
||||||
*/
|
|
||||||
export async function getServerConfig(filename: string, callback: (config: ServerConfig | undefined) => void): Promise<void> {
|
|
||||||
const fullFileName = `data/${filename}`
|
|
||||||
|
|
||||||
// attempt to read the file and get the configuration
|
|
||||||
if (fs.existsSync(fullFileName)) {
|
|
||||||
fs.readFile(fullFileName, 'utf8', (error, data) => {
|
|
||||||
if (error) {
|
|
||||||
callback(undefined)
|
|
||||||
return // something went wrong... stop
|
|
||||||
}
|
|
||||||
callback(JSON.parse(data))
|
|
||||||
})
|
|
||||||
} else {
|
|
||||||
callback(undefined) // file not found
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Method to obtain the configurations of the message chat/thread
|
|
||||||
*
|
|
||||||
* @param filename name of the configuration file to get
|
|
||||||
* @param callback function to allow a promise from getting the config
|
|
||||||
*/
|
|
||||||
export async function getUserConfig(filename: string, callback: (config: UserConfig | undefined) => void): Promise<void> {
|
|
||||||
const fullFileName = `data/${filename}`
|
|
||||||
|
|
||||||
// attempt to read the file and get the configuration
|
|
||||||
if (fs.existsSync(fullFileName)) {
|
|
||||||
fs.readFile(fullFileName, 'utf8', (error, data) => {
|
|
||||||
if (error) {
|
|
||||||
callback(undefined)
|
|
||||||
return // something went wrong... stop
|
|
||||||
}
|
|
||||||
callback(JSON.parse(data))
|
|
||||||
})
|
|
||||||
} else {
|
|
||||||
callback(undefined) // file not found
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Method to open/create and modify a json file containing thread information
|
|
||||||
*
|
|
||||||
* @param filename name of the thread file
|
|
||||||
* @param thread the thread with all of the interactions
|
|
||||||
* @param message message contents and from who
|
|
||||||
*/
|
|
||||||
export function openThreadInfo(filename: string, thread: ThreadChannel, messages: UserMessage[] = []) {
|
|
||||||
// check if the file exists, if not then make the config file
|
|
||||||
const fullFileName = `data/${filename}`
|
|
||||||
if (fs.existsSync(fullFileName)) {
|
|
||||||
fs.readFile(fullFileName, 'utf8', (error, data) => {
|
|
||||||
if (error)
|
|
||||||
console.log(`[Error: openThreadInfo] Incorrect file format`)
|
|
||||||
else {
|
|
||||||
const object = JSON.parse(data)
|
|
||||||
object['messages'] = messages as []
|
|
||||||
fs.writeFileSync(fullFileName, JSON.stringify(object, null, 2))
|
|
||||||
}
|
|
||||||
})
|
|
||||||
} else { // file doesn't exist, create it
|
|
||||||
const object: Configuration = JSON.parse(`{ \"id\": \"${thread?.id}\", \"name\": \"${thread?.name}\", \"messages\": []}`)
|
|
||||||
|
|
||||||
const directory = path.dirname(fullFileName)
|
|
||||||
if (!fs.existsSync(directory))
|
|
||||||
fs.mkdirSync(directory, { recursive: true })
|
|
||||||
|
|
||||||
// only creating it, no need to add anything
|
|
||||||
fs.writeFileSync(fullFileName, JSON.stringify(object, null, 2))
|
|
||||||
console.log(`[Util: openThreadInfo] Created '${fullFileName}' in working directory`)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Method to obtain the configurations of the message chat/thread
|
|
||||||
*
|
|
||||||
* @param filename name of the configuration file to get
|
|
||||||
* @param callback function to allow a promise from getting the config
|
|
||||||
*/
|
|
||||||
export async function getThread(filename: string, callback: (config: Thread | undefined) => void): Promise<void> {
|
|
||||||
// attempt to read the file and get the configuration
|
|
||||||
const fullFileName = `data/${filename}`
|
|
||||||
if (fs.existsSync(fullFileName)) {
|
|
||||||
fs.readFile(fullFileName, 'utf8', (error, data) => {
|
|
||||||
if (error) {
|
|
||||||
callback(undefined)
|
|
||||||
return // something went wrong... stop
|
|
||||||
}
|
|
||||||
callback(JSON.parse(data))
|
|
||||||
})
|
|
||||||
} else {
|
|
||||||
callback(undefined) // file not found
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Method to open the channel history
|
|
||||||
*
|
|
||||||
* @param filename name of the json file for the channel by user
|
|
||||||
* @param channel the text channel info
|
|
||||||
* @param user the user's name
|
|
||||||
* @param messages their messages
|
|
||||||
*/
|
|
||||||
export async function openChannelInfo(filename: string, channel: TextChannel, user: string, messages: UserMessage[] = []): Promise<void> {
|
|
||||||
// thread exist handler
|
|
||||||
const isThread: boolean = await new Promise((resolve) => {
|
|
||||||
getThread(`${channel.id}.json`, (threadInfo) => {
|
|
||||||
if (threadInfo?.messages)
|
|
||||||
resolve(true)
|
|
||||||
else
|
|
||||||
resolve(false)
|
|
||||||
})
|
|
||||||
})
|
|
||||||
|
|
||||||
// This is an existing thread, don't create another json
|
|
||||||
if (isThread) return
|
|
||||||
|
|
||||||
const fullFileName = `data/${filename}-${user}.json`
|
|
||||||
if (fs.existsSync(fullFileName)) {
|
|
||||||
fs.readFile(fullFileName, 'utf8', (error, data) => {
|
|
||||||
if (error)
|
|
||||||
console.log(`[Error: openChannelInfo] Incorrect file format`)
|
|
||||||
else {
|
|
||||||
const object = JSON.parse(data)
|
|
||||||
if (object['messages'].length === 0)
|
|
||||||
object['messages'] = messages as []
|
|
||||||
else if (object['messages'].length !== 0 && messages.length !== 0)
|
|
||||||
object['messages'] = messages as []
|
|
||||||
fs.writeFileSync(fullFileName, JSON.stringify(object, null, 2))
|
|
||||||
}
|
|
||||||
})
|
|
||||||
} else { // file doesn't exist, create it
|
|
||||||
const object: Configuration = JSON.parse(`{ \"id\": \"${channel?.id}\", \"name\": \"${channel?.name}\", \"user\": \"${user}\", \"messages\": []}`)
|
|
||||||
|
|
||||||
const directory = path.dirname(fullFileName)
|
|
||||||
if (!fs.existsSync(directory))
|
|
||||||
fs.mkdirSync(directory, { recursive: true })
|
|
||||||
|
|
||||||
// only creating it, no need to add anything
|
|
||||||
fs.writeFileSync(fullFileName, JSON.stringify(object, null, 2))
|
|
||||||
console.log(`[Util: openChannelInfo] Created '${fullFileName}' in working directory`)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Method to get the channel information/history
|
|
||||||
*
|
|
||||||
* @param filename name of the json file for the channel by user
|
|
||||||
* @param callback function to handle resolving message history
|
|
||||||
*/
|
|
||||||
export async function getChannelInfo(filename: string, callback: (config: Channel | undefined) => void): Promise<void> {
|
|
||||||
const fullFileName = `data/${filename}`
|
|
||||||
if (fs.existsSync(fullFileName)) {
|
|
||||||
fs.readFile(fullFileName, 'utf8', (error, data) => {
|
|
||||||
if (error) {
|
|
||||||
callback(undefined)
|
|
||||||
return // something went wrong... stop
|
|
||||||
}
|
|
||||||
callback(JSON.parse(data))
|
|
||||||
})
|
|
||||||
} else {
|
|
||||||
callback(undefined) // file not found
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -22,6 +22,6 @@ describe('#commands', () => {
|
|||||||
// test specific commands in the object
|
// test specific commands in the object
|
||||||
it('references specific commands', () => {
|
it('references specific commands', () => {
|
||||||
const commandsString = commands.map(e => e.name).join(', ')
|
const commandsString = commands.map(e => e.name).join(', ')
|
||||||
expect(commandsString).toBe('thread, private-thread, message-style, message-stream, toggle-chat, shutoff, modify-capacity, channel-toggle')
|
expect(commandsString).toBe('thread, private-thread, message-style, message-stream, toggle-chat, shutoff, modify-capacity, clear-user-channel-history')
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
@@ -5,7 +5,7 @@ export default defineConfig({
|
|||||||
test: {
|
test: {
|
||||||
globals: true, // <-- reduces test file imports
|
globals: true, // <-- reduces test file imports
|
||||||
coverage: {
|
coverage: {
|
||||||
exclude: [...configDefaults.exclude, 'build/*'], // <-- exclude JS build
|
exclude: [...configDefaults.exclude, 'build/*', 'tests/*'], // <-- exclude JS build
|
||||||
reporter: ['text', 'html'] // <-- reports in text, html
|
reporter: ['text', 'html'] // <-- reports in text, html
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user