forked from Skynet/discord-bot
Compare commits
178 commits
#6-incrime
...
main
Author | SHA1 | Date | |
---|---|---|---|
058f8a7a7d | |||
b43f760fb1 | |||
a11ba15f4a | |||
7d7afcd00c | |||
934842cbc9 | |||
143483d3b3 | |||
0bedf96da5 | |||
555e38ee26 | |||
6a5f651ba2 | |||
f046410cdc | |||
44bb40d96d | |||
7406f0e620 | |||
09ce45f70f | |||
b67894fc6e | |||
6481fcb89f | |||
9ce5b8136b | |||
348020ecfe | |||
8645a9b3ce | |||
0eba54b6f2 | |||
25fcc04287 | |||
c79944bd6e | |||
262eb0c991 | |||
30287466cb | |||
86f71f0fa4 | |||
6b84f33d2e | |||
a8c1cc9cf1 | |||
8c81fb864a | |||
b6cffd8a22 | |||
1aef86ad01 | |||
4eeb7f2135 | |||
cab04a068f | |||
c79113921d | |||
5fcc24a867 | |||
0a4f5281a5 | |||
a9d3af024e | |||
b7d36de976 | |||
68ffa55dc5 | |||
9b3c71e321 | |||
0478f634fa | |||
ee0c8f0987 | |||
b55650b221 | |||
4691869ae9 | |||
68d7b53905 | |||
bf55dfe31e | |||
ad94b197ae | |||
1f3c33458e | |||
bab6e4fdec | |||
f00db7ef5d | |||
37ea38f516 | |||
ca55a78447 | |||
93359698f0 | |||
dda05d7ca1 | |||
5dee9acbaa | |||
96a61e6fc8 | |||
94292fa388 | |||
77a7b7b81d | |||
2f75dc41c8 | |||
c98baa9d72 | |||
e4a8cce725 | |||
5b22f699d6 | |||
6739c7e068 | |||
d673dce6fa | |||
015f23b922 | |||
7a6421469c | |||
733827c3e6 | |||
2daa010d25 | |||
da4d006bc0 | |||
344d6d3585 | |||
b7161e2614 | |||
32249364ff | |||
f1138a3c81 | |||
61e76db8dd | |||
3e6dc9d560 | |||
aff6299ac7 | |||
bd80bda22f | |||
fe5aa5b252 | |||
273c58d035 | |||
3927734083 | |||
41407ecefb | |||
79f880daea | |||
ceade9b972 | |||
b2d8238c17 | |||
a7e8f5698e | |||
80c9191eee | |||
1c3ccbecf5 | |||
d1af8a7c1f | |||
0d9ce2de7f | |||
5e7964ae26 | |||
32292a3c0b | |||
ffce78a10d | |||
7980739627 | |||
2aad895bb3 | |||
ec74dc0aa7 | |||
42f301455a | |||
5b21bc47bd | |||
2db136a736 | |||
72c4b1e794 | |||
5e17a98bff | |||
0ab290a876 | |||
a62f893a98 | |||
43c5cd2eff | |||
d9211dca9a | |||
c71dbe7214 | |||
bf08aa650c | |||
f3ef03a418 | |||
11240914ac | |||
e9aed40f41 | |||
0df7c8a29f | |||
4998dba225 | |||
8d1c6b1bd1 | |||
d3a975a1d1 | |||
0e1a7d56b6 | |||
fd32adb138 | |||
04a487cd8f | |||
|
5f4e46bb51 | ||
28b911c468 | |||
e7caf148dd | |||
9452c0ac2e | |||
befced76f8 | |||
421864e151 | |||
557dcb9f8c | |||
b75fa39bcf | |||
986d2f19c9 | |||
cfb5d95be4 | |||
7ed20108fb | |||
c63945bb86 | |||
8ba92cc47e | |||
50d2923425 | |||
5c2502f726 | |||
bda3fbe2ad | |||
439d49db43 | |||
7e40b862d3 | |||
905aaa9620 | |||
c447577eee | |||
7ac8b90f27 | |||
48b52f3c09 | |||
394d6b4545 | |||
bee41c192f | |||
1d14499400 | |||
d8b232b546 | |||
7c2d392e35 | |||
9654963198 | |||
0541a70714 | |||
15720a1f13 | |||
86bb566e5e | |||
c90186295c | |||
acb6432129 | |||
55b2e534d4 | |||
9481358068 | |||
33cebe7782 | |||
c2a6407ef0 | |||
2970549eb0 | |||
d549627714 | |||
87e836619f | |||
0f774258a1 | |||
c446c10f2d | |||
9c284f2a5c | |||
d58d837940 | |||
ed4c46e81d | |||
982b9defd4 | |||
6cbbab80bd | |||
d0b63190b3 | |||
3d925fcfff | |||
cf2c7683d2 | |||
7e6d892b67 | |||
bbd55202bd | |||
bd74cdd09b | |||
2c28f3edcc | |||
f417b9993a | |||
0d0a50c84b | |||
4a1b1cc7f9 | |||
d5877e99e6 | |||
2761098c8d | |||
a9f55da04d | |||
480fc9b1a0 | |||
4bd23e7638 | |||
9dafba03b5 | |||
c6eaa8ad9a |
44 changed files with 4749 additions and 1937 deletions
67
.forgejo/workflows/push.yaml
Normal file
67
.forgejo/workflows/push.yaml
Normal file
|
@ -0,0 +1,67 @@
|
||||||
|
name: On_Push
|
||||||
|
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
branches:
|
||||||
|
- 'main'
|
||||||
|
paths:
|
||||||
|
- flake.*
|
||||||
|
- src/**/*
|
||||||
|
- Cargo.*
|
||||||
|
- .forgejo/**/*
|
||||||
|
- rust-toolchain.toml
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
# rust code must be formatted for standardisation
|
||||||
|
lint_fmt:
|
||||||
|
# build it using teh base nixos system, helps with caching
|
||||||
|
runs-on: nix
|
||||||
|
steps:
|
||||||
|
# get the repo first
|
||||||
|
- uses: https://code.forgejo.org/actions/checkout@v4
|
||||||
|
- uses: https://forgejo.skynet.ie/Skynet/actions/get_lfs@v3
|
||||||
|
with:
|
||||||
|
repository: ${{ gitea.repository }}
|
||||||
|
ref_name: ${{ gitea.ref_name }}
|
||||||
|
- run: nix build .#fmt --verbose
|
||||||
|
|
||||||
|
# clippy is incredibly useful for making yer code better
|
||||||
|
lint_clippy:
|
||||||
|
# build it using teh base nixos system, helps with caching
|
||||||
|
runs-on: nix
|
||||||
|
permissions:
|
||||||
|
checks: write
|
||||||
|
steps:
|
||||||
|
# get the repo first
|
||||||
|
- uses: https://code.forgejo.org/actions/checkout@v4
|
||||||
|
- uses: https://forgejo.skynet.ie/Skynet/actions/get_lfs@v3
|
||||||
|
with:
|
||||||
|
repository: ${{ gitea.repository }}
|
||||||
|
ref_name: ${{ gitea.ref_name }}
|
||||||
|
- run: nix build .#clippy --verbose
|
||||||
|
|
||||||
|
build:
|
||||||
|
# build it using teh base nixos system, helps with caching
|
||||||
|
runs-on: nix
|
||||||
|
needs: [ lint_fmt, lint_clippy ]
|
||||||
|
steps:
|
||||||
|
# get the repo first
|
||||||
|
- uses: https://code.forgejo.org/actions/checkout@v4
|
||||||
|
- uses: https://forgejo.skynet.ie/Skynet/actions/get_lfs@v3
|
||||||
|
with:
|
||||||
|
repository: ${{ gitea.repository }}
|
||||||
|
ref_name: ${{ gitea.ref_name }}
|
||||||
|
- name: "Build it locally"
|
||||||
|
run: nix build --verbose
|
||||||
|
|
||||||
|
# deploy it upstream
|
||||||
|
deploy:
|
||||||
|
# runs on teh default docker container
|
||||||
|
runs-on: docker
|
||||||
|
needs: [ build ]
|
||||||
|
steps:
|
||||||
|
- name: "Deploy to Skynet"
|
||||||
|
uses: https://forgejo.skynet.ie/Skynet/actions/deploy@v3
|
||||||
|
with:
|
||||||
|
input: 'skynet_discord_bot'
|
||||||
|
token: ${{ secrets.API_TOKEN_FORGEJO }}
|
37
.gitattributes
vendored
Normal file
37
.gitattributes
vendored
Normal file
|
@ -0,0 +1,37 @@
|
||||||
|
# Git config here
|
||||||
|
* text eol=lf
|
||||||
|
|
||||||
|
#############################################
|
||||||
|
# Git lfs stuff
|
||||||
|
# Documents
|
||||||
|
*.pdf filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.doc filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.docx filter=lfs diff=lfs merge=lfs -text
|
||||||
|
# Excel
|
||||||
|
*.xls filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.xlsx filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.xlsm filter=lfs diff=lfs merge=lfs -text
|
||||||
|
# Powerpoints
|
||||||
|
*.ppt filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.pptx filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.ppsx filter=lfs diff=lfs merge=lfs -text
|
||||||
|
# Images
|
||||||
|
*.png filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.jpg filter=lfs diff=lfs merge=lfs -text
|
||||||
|
# Video
|
||||||
|
*.mkv filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.mp4 filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.wmv filter=lfs diff=lfs merge=lfs -text
|
||||||
|
# Misc
|
||||||
|
*.zip filter=lfs diff=lfs merge=lfs -text
|
||||||
|
# ET4011
|
||||||
|
*.cbe filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.pbs filter=lfs diff=lfs merge=lfs -text
|
||||||
|
# Open/Libre office
|
||||||
|
# from https://www.libreoffice.org/discover/what-is-opendocument/
|
||||||
|
*.odt filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.ods filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.odp filter=lfs diff=lfs merge=lfs -text
|
||||||
|
*.odg filter=lfs diff=lfs merge=lfs -text
|
||||||
|
# QT
|
||||||
|
*.ui filter=lfs diff=lfs merge=lfs -text
|
2
.gitignore
vendored
2
.gitignore
vendored
|
@ -2,11 +2,13 @@
|
||||||
/.idea
|
/.idea
|
||||||
|
|
||||||
.env
|
.env
|
||||||
|
*.env
|
||||||
|
|
||||||
result
|
result
|
||||||
/result
|
/result
|
||||||
|
|
||||||
*.db
|
*.db
|
||||||
|
*.db.*
|
||||||
|
|
||||||
tmp/
|
tmp/
|
||||||
tmp.*
|
tmp.*
|
||||||
|
|
|
@ -1,86 +0,0 @@
|
||||||
# copied a good chunk from my bfom config
|
|
||||||
image: rust:latest
|
|
||||||
|
|
||||||
stages:
|
|
||||||
- lint
|
|
||||||
- build
|
|
||||||
- deploy
|
|
||||||
|
|
||||||
cache:
|
|
||||||
key: "$CI_JOB_NAME"
|
|
||||||
paths:
|
|
||||||
- target/
|
|
||||||
|
|
||||||
# Set any required environment variables here
|
|
||||||
variables:
|
|
||||||
RUST_BACKTRACE: FULL
|
|
||||||
|
|
||||||
|
|
||||||
# clippy and fmt are magic
|
|
||||||
# runs on all commits/branches
|
|
||||||
lint-clippy:
|
|
||||||
stage: lint
|
|
||||||
script:
|
|
||||||
- rustup component add clippy
|
|
||||||
- rustc --version
|
|
||||||
- cargo version
|
|
||||||
- cargo clippy
|
|
||||||
rules:
|
|
||||||
- if: $CI_COMMIT_TAG
|
|
||||||
when: never
|
|
||||||
- changes:
|
|
||||||
- src/**/*
|
|
||||||
- cargo.*
|
|
||||||
when: always
|
|
||||||
|
|
||||||
lint-fmt:
|
|
||||||
stage: lint
|
|
||||||
script:
|
|
||||||
- rustup component add rustfmt
|
|
||||||
- rustc --version
|
|
||||||
- cargo version
|
|
||||||
- cargo fmt -- --check
|
|
||||||
rules:
|
|
||||||
- if: $CI_COMMIT_TAG
|
|
||||||
when: never
|
|
||||||
- changes:
|
|
||||||
- src/**/*
|
|
||||||
- cargo.*
|
|
||||||
when: always
|
|
||||||
|
|
||||||
|
|
||||||
# has to actually compile
|
|
||||||
build:
|
|
||||||
stage: build
|
|
||||||
script:
|
|
||||||
- rustc --version
|
|
||||||
- cargo version
|
|
||||||
- cargo build --verbose
|
|
||||||
- RUST_BACKTRACE=1 cargo test --verbose
|
|
||||||
rules:
|
|
||||||
- if: $CI_COMMIT_TAG
|
|
||||||
when: never
|
|
||||||
- changes:
|
|
||||||
- src/**/*
|
|
||||||
- cargo.*
|
|
||||||
when: on_success
|
|
||||||
|
|
||||||
|
|
||||||
# from https://docs.gitlab.com/ee/ci/pipelines/multi_project_pipelines.html
|
|
||||||
# so simple to deploy now
|
|
||||||
nixos:
|
|
||||||
stage: deploy
|
|
||||||
variables:
|
|
||||||
PACKAGE_NAME: "skynet_discord_bot"
|
|
||||||
UPDATE_FLAKE: "yes"
|
|
||||||
trigger: compsoc1/skynet/nixos
|
|
||||||
rules:
|
|
||||||
- if: $CI_COMMIT_TAG
|
|
||||||
when: never
|
|
||||||
- if: $CI_COMMIT_BRANCH == $CI_DEFAULT_BRANCH
|
|
||||||
when: on_success
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
2503
Cargo.lock
generated
2503
Cargo.lock
generated
File diff suppressed because it is too large
Load diff
32
Cargo.toml
32
Cargo.toml
|
@ -11,26 +11,38 @@ name = "update_data"
|
||||||
[[bin]]
|
[[bin]]
|
||||||
name = "update_users"
|
name = "update_users"
|
||||||
|
|
||||||
|
[[bin]]
|
||||||
|
name = "update_committee"
|
||||||
|
|
||||||
|
[[bin]]
|
||||||
|
name = "update_minecraft"
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
serenity = { version = "0.11.6", default-features = false, features = ["client", "gateway", "rustls_backend", "model", "cache"] }
|
# discord library
|
||||||
tokio = { version = "1.0", features = ["macros", "rt-multi-thread"] }
|
serenity = { version = "0.12", default-features = false, features = ["client", "gateway", "rustls_backend", "model", "cache"] }
|
||||||
|
tokio = { version = "1", features = ["macros", "rt-multi-thread", "full"] }
|
||||||
|
|
||||||
|
# wolves api
|
||||||
|
wolves_oxidised = { git = "https://forgejo.skynet.ie/Skynet/wolves-oxidised.git", features = ["unstable"] }
|
||||||
|
# wolves_oxidised = { path = "../wolves-oxidised", features = ["unstable"] }
|
||||||
|
|
||||||
# to make the http requests
|
# to make the http requests
|
||||||
surf = "2.3.2"
|
surf = "2.3"
|
||||||
|
|
||||||
dotenvy = "0.15.7"
|
dotenvy = "0.15"
|
||||||
|
|
||||||
# For sqlite
|
# For sqlite
|
||||||
sqlx = { version = "0.7.1", features = [ "runtime-tokio", "sqlite" ] }
|
sqlx = { version = "0.8", features = [ "runtime-tokio", "sqlite", "migrate" ] }
|
||||||
|
serde_json = { version = "1.0", features = ["raw_value"] }
|
||||||
|
|
||||||
# create random strings
|
# create random strings
|
||||||
rand = "0.8.5"
|
rand = "0.9"
|
||||||
|
|
||||||
# fancy time stuff
|
# fancy time stuff
|
||||||
chrono = "0.4.26"
|
chrono = "0.4"
|
||||||
|
|
||||||
# for email
|
# for email
|
||||||
lettre = "0.10.4"
|
lettre = "0.11"
|
||||||
maud = "0.25.0"
|
maud = "0.27"
|
||||||
|
|
||||||
serde = "1.0.188"
|
serde = "1.0"
|
9
LICENSE
Normal file
9
LICENSE
Normal file
|
@ -0,0 +1,9 @@
|
||||||
|
MIT License
|
||||||
|
|
||||||
|
Copyright (c) 2024 Skynet
|
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
|
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
|
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
10
README.md
Normal file
10
README.md
Normal file
|
@ -0,0 +1,10 @@
|
||||||
|
# Skynet Discord Bot
|
||||||
|
The Skynet bot is designed to manage users on Discord.
|
||||||
|
It allows users to link their UL Wolves account with Wolves in a GDPR compliant manner.
|
||||||
|
Skynet (bot) is hosted is hosted by the Computer Society on Skynet (computer cluster).
|
||||||
|
|
||||||
|
## Documentation
|
||||||
|
We have split up the documentation into different segments depending on who the user is.
|
||||||
|
|
||||||
|
* [Committees](./doc/Committee.md)
|
||||||
|
* [Member](./doc/User.md)
|
10
db/migrations/10_member_committee-roles.sql
Normal file
10
db/migrations/10_member_committee-roles.sql
Normal file
|
@ -0,0 +1,10 @@
|
||||||
|
|
||||||
|
CREATE TABLE IF NOT EXISTS committee_roles (
|
||||||
|
id_wolves integer PRIMARY KEY,
|
||||||
|
id_role integer DEFAULT 1,
|
||||||
|
id_channel integer DEFAULT 1,
|
||||||
|
-- not strictly required but for readability and debugging
|
||||||
|
name_role text NOT NULL DEFAULT '',
|
||||||
|
name_channel text NOT NULL DEFAULT '',
|
||||||
|
count integer DEFAULT 0
|
||||||
|
);
|
38
db/migrations/1_setup.sql
Normal file
38
db/migrations/1_setup.sql
Normal file
|
@ -0,0 +1,38 @@
|
||||||
|
-- setup initial tables
|
||||||
|
|
||||||
|
-- this handles the users "floating" account
|
||||||
|
CREATE TABLE IF NOT EXISTS wolves (
|
||||||
|
id_wolves integer PRIMARY KEY,
|
||||||
|
email text not null,
|
||||||
|
discord integer,
|
||||||
|
minecraft text
|
||||||
|
);
|
||||||
|
CREATE INDEX IF NOT EXISTS index_discord ON wolves (discord);
|
||||||
|
|
||||||
|
-- used to verify the users email address
|
||||||
|
CREATE TABLE IF NOT EXISTS wolves_verify (
|
||||||
|
discord integer PRIMARY KEY,
|
||||||
|
email text not null,
|
||||||
|
auth_code text not null,
|
||||||
|
date_expiry text not null
|
||||||
|
);
|
||||||
|
CREATE INDEX IF NOT EXISTS index_date_expiry ON wolves_verify (date_expiry);
|
||||||
|
|
||||||
|
-- information on teh server the bot is registered on
|
||||||
|
CREATE TABLE IF NOT EXISTS servers (
|
||||||
|
server integer PRIMARY KEY,
|
||||||
|
wolves_api text not null,
|
||||||
|
role_past integer,
|
||||||
|
role_current integer,
|
||||||
|
member_past integer DEFAULT 0,
|
||||||
|
member_current integer DEFAULT 0
|
||||||
|
);
|
||||||
|
|
||||||
|
-- keep track of the members on the server
|
||||||
|
CREATE TABLE IF NOT EXISTS server_members (
|
||||||
|
server integer not null,
|
||||||
|
id_wolves integer not null,
|
||||||
|
expiry text not null,
|
||||||
|
PRIMARY KEY(server,id_wolves),
|
||||||
|
FOREIGN KEY (id_wolves) REFERENCES wolves (id_wolves)
|
||||||
|
);
|
3
db/migrations/2_minecraft-server.sql
Normal file
3
db/migrations/2_minecraft-server.sql
Normal file
|
@ -0,0 +1,3 @@
|
||||||
|
-- add teh option to associate each discord server with a minecraft one managed by skynet
|
||||||
|
ALTER TABLE servers
|
||||||
|
ADD server_minecraft text;
|
18
db/migrations/3_minecraft-server-migrate.sql
Normal file
18
db/migrations/3_minecraft-server-migrate.sql
Normal file
|
@ -0,0 +1,18 @@
|
||||||
|
-- Create the new table
|
||||||
|
CREATE TABLE IF NOT EXISTS minecraft
|
||||||
|
(
|
||||||
|
server_discord integer not null,
|
||||||
|
server_minecraft text not null,
|
||||||
|
PRIMARY KEY (server_discord, server_minecraft),
|
||||||
|
FOREIGN KEY (server_discord) REFERENCES servers (server)
|
||||||
|
);
|
||||||
|
|
||||||
|
-- Import in the data
|
||||||
|
INSERT INTO minecraft (server_discord, server_minecraft)
|
||||||
|
SELECT server, server_minecraft
|
||||||
|
FROM servers
|
||||||
|
where server_minecraft is not null;
|
||||||
|
|
||||||
|
-- delete the col in teh server table
|
||||||
|
ALTER TABLE servers
|
||||||
|
DROP COLUMN server_minecraft;
|
7
db/migrations/4_committee-mk-i.sql
Normal file
7
db/migrations/4_committee-mk-i.sql
Normal file
|
@ -0,0 +1,7 @@
|
||||||
|
-- temp table to allow folks to verify by committee email.
|
||||||
|
CREATE TABLE IF NOT EXISTS committee (
|
||||||
|
discord integer PRIMARY KEY,
|
||||||
|
email text not null,
|
||||||
|
auth_code text not null,
|
||||||
|
committee integer DEFAULT 0
|
||||||
|
);
|
5
db/migrations/5_welcome-message.sql
Normal file
5
db/migrations/5_welcome-message.sql
Normal file
|
@ -0,0 +1,5 @@
|
||||||
|
-- temp table to allow folks to verify by committee email.
|
||||||
|
-- delete the col in teh server table
|
||||||
|
ALTER TABLE servers ADD COLUMN bot_channel_id integer DEFAULT 0;
|
||||||
|
ALTER TABLE servers ADD COLUMN server_name text NOT NULL DEFAULT "";
|
||||||
|
ALTER TABLE servers ADD COLUMN wolves_link text NOT NULL DEFAULT "";
|
11
db/migrations/6_role-adder.sql
Normal file
11
db/migrations/6_role-adder.sql
Normal file
|
@ -0,0 +1,11 @@
|
||||||
|
CREATE TABLE IF NOT EXISTS roles_adder (
|
||||||
|
server integer not null,
|
||||||
|
role_a integer not null,
|
||||||
|
role_b integer not null,
|
||||||
|
role_c integer not null,
|
||||||
|
PRIMARY KEY(server,role_a,role_b,role_c)
|
||||||
|
);
|
||||||
|
CREATE INDEX IF NOT EXISTS index_roles_adder_server ON roles_adder (server);
|
||||||
|
CREATE INDEX IF NOT EXISTS index_roles_adder_from ON roles_adder (role_a,role_b);
|
||||||
|
CREATE INDEX IF NOT EXISTS index_roles_adder_to ON roles_adder (role_c);
|
||||||
|
CREATE INDEX IF NOT EXISTS index_roles_adder_search ON roles_adder (server,role_a,role_b);
|
3
db/migrations/7_minecraft-bedrock.sql
Normal file
3
db/migrations/7_minecraft-bedrock.sql
Normal file
|
@ -0,0 +1,3 @@
|
||||||
|
|
||||||
|
-- Using this col we will be able to see if someone has a bedrock account (as well as a java one)
|
||||||
|
ALTER TABLE wolves ADD COLUMN minecraft_uid TEXT;
|
14
db/migrations/8_committee-mk-ii.sql
Normal file
14
db/migrations/8_committee-mk-ii.sql
Normal file
|
@ -0,0 +1,14 @@
|
||||||
|
-- No longer using the previous committee table
|
||||||
|
DROP TABLE committee;
|
||||||
|
|
||||||
|
-- new table pulling from teh api
|
||||||
|
CREATE TABLE IF NOT EXISTS committees (
|
||||||
|
id integer PRIMARY KEY,
|
||||||
|
name_profile text not null,
|
||||||
|
name_plain text not null,
|
||||||
|
name_full text not null,
|
||||||
|
link text not null,
|
||||||
|
committee text not null
|
||||||
|
);
|
||||||
|
|
||||||
|
ALTER TABLE servers DROP COLUMN wolves_link;
|
6
db/migrations/9_member_committee-id.sql
Normal file
6
db/migrations/9_member_committee-id.sql
Normal file
|
@ -0,0 +1,6 @@
|
||||||
|
-- No need for this col since it is goign to be in "committees" anyways
|
||||||
|
ALTER TABLE servers DROP COLUMN server_name;
|
||||||
|
|
||||||
|
-- we do care about teh ID of the club/soc though
|
||||||
|
ALTER TABLE servers ADD COLUMN wolves_id integer DEFAULT 0;
|
||||||
|
|
71
doc/Committee.md
Normal file
71
doc/Committee.md
Normal file
|
@ -0,0 +1,71 @@
|
||||||
|
# Skynet Discord Bot
|
||||||
|
This bots core purpose is to give members roles based on their status on <https://ulwolves.ie>.
|
||||||
|
It uses an api key provided by wolves to get member lists.
|
||||||
|
|
||||||
|
Users are able to link their wolves account to the bot and that works across discord servers.
|
||||||
|
For example is a user links on the CompSoc Discord then they will also get their roles (automagically) on Games Dev if they are a member there.
|
||||||
|
|
||||||
|
## Setup - Committee
|
||||||
|
You need admin access to run any of the commands in this section.
|
||||||
|
Either the server owner or a user with the ``Manage Server`` permission.
|
||||||
|
|
||||||
|
### Get the API Key
|
||||||
|
The ``api_key`` is used by the Bot in order to request information, it will be used later in the process.
|
||||||
|
|
||||||
|
1. Email ``keith@assurememberships.com`` from committee email and say you want an ``api_key`` for ``193.1.99.74``
|
||||||
|
* The committee email is the one here: <https://cp.ulwolves.ie/mailbox/>
|
||||||
|
* This may take up to a week to get the key.
|
||||||
|
|
||||||
|
### Setup Server
|
||||||
|
The Bot reason for existing is being able to give members Roles.
|
||||||
|
So we have to create those.
|
||||||
|
|
||||||
|
1. Create a role for Current Members.
|
||||||
|
* You can call it whatever you want.
|
||||||
|
* ``member-current`` is a good choice.
|
||||||
|
* This should be a new role
|
||||||
|
2. **Optional**: you can create a role that is given to folks who were but no longer a member.
|
||||||
|
* ``member`` would be a good choice for this
|
||||||
|
* If you have an existing member role this is also a good fit.
|
||||||
|
|
||||||
|
The reason for both roles is ye have one for active members while the second is for all current and past members.
|
||||||
|
|
||||||
|
### Invite Bot
|
||||||
|
1. Invite the bot https://discord.com/api/oauth2/authorize?client_id=1145761669256069270&permissions=139855185984&scope=bot
|
||||||
|
2. Make sure the bot role ``@skynet`` is above these two roles created in the previous step
|
||||||
|
* This is so it can manage the roles (give and remove them from users)
|
||||||
|
|
||||||
|
### Setup Bot
|
||||||
|
This is where the bot is configured.
|
||||||
|
You will need the ``api_key`` from the start of the process.
|
||||||
|
You (personally) will need a role with ``Manage Server`` permission to be able to do this.
|
||||||
|
|
||||||
|
1. Use the command ``/add`` and a list of options will pop up.
|
||||||
|
2. ``api_key`` is the key you got from Keith earlier.
|
||||||
|
3. ``role_current`` is the ``member-current`` that you created earlier.
|
||||||
|
4. ``role_past`` (optional) is the role for all current and past members.
|
||||||
|
5. ``bot_channel`` is a channel that folks are recommended to use the bot.
|
||||||
|
* You can have it so folks cannot see message history
|
||||||
|
|
||||||
|
At this point the bot is set up and no further action is required.
|
||||||
|
|
||||||
|
### Minecraft
|
||||||
|
The bot is able to manage the whitelist of a Minecraft server managed by the Computer Society.
|
||||||
|
Talk to us to get a server.
|
||||||
|
|
||||||
|
#### Add
|
||||||
|
This links a minecraft server with your club/society.
|
||||||
|
|
||||||
|
``/minecraft_add SERVER_ID``
|
||||||
|
|
||||||
|
|
||||||
|
#### List
|
||||||
|
List the servers linked to your club/society.
|
||||||
|
It is possible to have more than one minecraft server
|
||||||
|
|
||||||
|
``/minecraft_list``
|
||||||
|
|
||||||
|
#### Delete
|
||||||
|
This unlinks a minecraft server from your club/society.
|
||||||
|
|
||||||
|
``/minecraft_delete SERVER_ID``
|
26
doc/User.md
Normal file
26
doc/User.md
Normal file
|
@ -0,0 +1,26 @@
|
||||||
|
# Skynet Discord Bot
|
||||||
|
The Skynet bot is designed to make it easy to verify that you are a member of a Club/Society.
|
||||||
|
The bot will be able to give you member roles for any partnered servers.
|
||||||
|
It also provides secondary manifests such as granting access to minecraft servers managed by teh Computer Society.
|
||||||
|
|
||||||
|
## Setup
|
||||||
|
This is to link your Discord account with your UL Wolves account.
|
||||||
|
**You will only need to do this once**.
|
||||||
|
|
||||||
|
### Setup
|
||||||
|
1. In a Discord server with the Skynet Bot enter ``/link_wolves YOUR_WOLVES_CONTACT_EMAIL``
|
||||||
|
<img src="../media/setup_user_01.png" alt="link process start" width="50%" height="50%">
|
||||||
|
* Your ``YOUR_WOLVES_CONTACT_EMAIL`` is the email in the Contact Email here: <https://ulwolves.ie/memberships/profile>
|
||||||
|
* This is most likely your student mail
|
||||||
|
2. An email will be sent to you with a verification code.
|
||||||
|
<img src="../media/setup_user_02.png" alt="signup email" width="50%" height="50%">
|
||||||
|
3. Verify the code using ``/verify CODE_FROM_EMAIL`` in Discord.
|
||||||
|
<img src="../media/setup_user_03.png" alt="verify in discord" width="50%" height="50%">
|
||||||
|
4. Once complete your Wolves and Discord accounts will be linked.
|
||||||
|
|
||||||
|
You will get member roles on any Discord that is using the bot that you are a member of.
|
||||||
|
|
||||||
|
### Minecraft
|
||||||
|
You can link your Minecraft username to grant you access to any Minecraft server run by UL Computer Society.
|
||||||
|
|
||||||
|
``/link_minecraft MINECRAFT_USERNAME``
|
8
example.env
Normal file
8
example.env
Normal file
|
@ -0,0 +1,8 @@
|
||||||
|
HOME="."
|
||||||
|
DATABASE="database.db"
|
||||||
|
DISCORD_TOKEN=""
|
||||||
|
DISCORD_TOKEN_MINECRAFT=""
|
||||||
|
EMAIL_SMTP=""
|
||||||
|
EMAIL_USER=""
|
||||||
|
EMAIL_PASS=""
|
||||||
|
WOLVES_URL=""
|
26
flake.lock
generated
26
flake.lock
generated
|
@ -5,11 +5,11 @@
|
||||||
"nixpkgs": "nixpkgs"
|
"nixpkgs": "nixpkgs"
|
||||||
},
|
},
|
||||||
"locked": {
|
"locked": {
|
||||||
"lastModified": 1692351612,
|
"lastModified": 1721727458,
|
||||||
"narHash": "sha256-KTGonidcdaLadRnv9KFgwSMh1ZbXoR/OBmPjeNMhFwU=",
|
"narHash": "sha256-r/xppY958gmZ4oTfLiHN0ZGuQ+RSTijDblVgVLFi1mw=",
|
||||||
"owner": "nix-community",
|
"owner": "nix-community",
|
||||||
"repo": "naersk",
|
"repo": "naersk",
|
||||||
"rev": "78789c30d64dea2396c9da516bbcc8db3a475207",
|
"rev": "3fb418eaf352498f6b6c30592e3beb63df42ef11",
|
||||||
"type": "github"
|
"type": "github"
|
||||||
},
|
},
|
||||||
"original": {
|
"original": {
|
||||||
|
@ -20,11 +20,11 @@
|
||||||
},
|
},
|
||||||
"nixpkgs": {
|
"nixpkgs": {
|
||||||
"locked": {
|
"locked": {
|
||||||
"lastModified": 1693060755,
|
"lastModified": 1723151389,
|
||||||
"narHash": "sha256-KNsbfqewEziFJEpPR0qvVz4rx0x6QXxw1CcunRhlFdk=",
|
"narHash": "sha256-9AVY0ReCmSGXHrlx78+1RrqcDgVSRhHUKDVV1LLBy28=",
|
||||||
"owner": "NixOS",
|
"owner": "NixOS",
|
||||||
"repo": "nixpkgs",
|
"repo": "nixpkgs",
|
||||||
"rev": "c66ccfa00c643751da2fd9290e096ceaa30493fc",
|
"rev": "13fe00cb6c75461901f072ae62b5805baef9f8b2",
|
||||||
"type": "github"
|
"type": "github"
|
||||||
},
|
},
|
||||||
"original": {
|
"original": {
|
||||||
|
@ -34,16 +34,16 @@
|
||||||
},
|
},
|
||||||
"nixpkgs_2": {
|
"nixpkgs_2": {
|
||||||
"locked": {
|
"locked": {
|
||||||
"lastModified": 1693087214,
|
"lastModified": 1722995383,
|
||||||
"narHash": "sha256-Kn1SSqRfPpqcI1MDy82JXrPT1WI8c03TA2F0xu6kS+4=",
|
"narHash": "sha256-UzuXo7ZM8ZK0SkWFhHocKkLSGQPHS4JxaE1jvVR4fUo=",
|
||||||
"owner": "NixOS",
|
"owner": "NixOS",
|
||||||
"repo": "nixpkgs",
|
"repo": "nixpkgs",
|
||||||
"rev": "f155f0cf4ea43c4e3c8918d2d327d44777b6cad4",
|
"rev": "957d95fc8b9bf1eb60d43f8d2eba352b71bbf2be",
|
||||||
"type": "github"
|
"type": "github"
|
||||||
},
|
},
|
||||||
"original": {
|
"original": {
|
||||||
"id": "nixpkgs",
|
"id": "nixpkgs",
|
||||||
"ref": "nixos-23.05",
|
"ref": "nixos-unstable",
|
||||||
"type": "indirect"
|
"type": "indirect"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
@ -74,11 +74,11 @@
|
||||||
"systems": "systems"
|
"systems": "systems"
|
||||||
},
|
},
|
||||||
"locked": {
|
"locked": {
|
||||||
"lastModified": 1692799911,
|
"lastModified": 1710146030,
|
||||||
"narHash": "sha256-3eihraek4qL744EvQXsK1Ha6C3CR7nnT8X2qWap4RNk=",
|
"narHash": "sha256-SZ5L6eA7HJ/nmkzGG7/ISclqe6oZdOZTNoesiInkXPQ=",
|
||||||
"owner": "numtide",
|
"owner": "numtide",
|
||||||
"repo": "flake-utils",
|
"repo": "flake-utils",
|
||||||
"rev": "f9e7cf818399d17d347f847525c5a5a8032e4e44",
|
"rev": "b1d9ab70662946ef0850d488da1c9019f3a9752a",
|
||||||
"type": "github"
|
"type": "github"
|
||||||
},
|
},
|
||||||
"original": {
|
"original": {
|
||||||
|
|
142
flake.nix
142
flake.nix
|
@ -2,31 +2,56 @@
|
||||||
description = "Skynet Discord Bot";
|
description = "Skynet Discord Bot";
|
||||||
|
|
||||||
inputs = {
|
inputs = {
|
||||||
nixpkgs.url = "nixpkgs/nixos-23.05";
|
nixpkgs.url = "nixpkgs/nixos-unstable";
|
||||||
naersk.url = "github:nix-community/naersk";
|
naersk.url = "github:nix-community/naersk";
|
||||||
utils.url = "github:numtide/flake-utils";
|
utils.url = "github:numtide/flake-utils";
|
||||||
};
|
};
|
||||||
|
|
||||||
outputs = { self, nixpkgs, utils, naersk }: utils.lib.eachDefaultSystem (system:
|
nixConfig = {
|
||||||
let
|
extra-substituters = "https://nix-cache.skynet.ie/skynet-cache";
|
||||||
pkgs = nixpkgs.legacyPackages."${system}";
|
extra-trusted-public-keys = "skynet-cache:zMFLzcRZPhUpjXUy8SF8Cf7KGAZwo98SKrzeXvdWABo=";
|
||||||
naersk-lib = naersk.lib."${system}";
|
|
||||||
package_name = "skynet_discord_bot";
|
|
||||||
desc = "Skynet Discord Bot";
|
|
||||||
in rec {
|
|
||||||
|
|
||||||
# `nix build`
|
|
||||||
packages."${package_name}" = naersk-lib.buildPackage {
|
|
||||||
pname = "${package_name}";
|
|
||||||
root = ./.;
|
|
||||||
|
|
||||||
buildInputs = [
|
|
||||||
pkgs.openssl
|
|
||||||
pkgs.pkg-config
|
|
||||||
];
|
|
||||||
};
|
};
|
||||||
|
|
||||||
defaultPackage = packages."${package_name}";
|
outputs = {
|
||||||
|
self,
|
||||||
|
nixpkgs,
|
||||||
|
utils,
|
||||||
|
naersk,
|
||||||
|
}:
|
||||||
|
utils.lib.eachDefaultSystem (
|
||||||
|
system: let
|
||||||
|
pkgs = (import nixpkgs) {inherit system;};
|
||||||
|
naersk' = pkgs.callPackage naersk {};
|
||||||
|
package_name = "skynet_discord_bot";
|
||||||
|
desc = "Skynet Discord Bot";
|
||||||
|
buildInputs = with pkgs; [
|
||||||
|
openssl
|
||||||
|
pkg-config
|
||||||
|
rustfmt
|
||||||
|
];
|
||||||
|
in rec {
|
||||||
|
packages = {
|
||||||
|
# For `nix build` & `nix run`:
|
||||||
|
default = naersk'.buildPackage {
|
||||||
|
pname = "${package_name}";
|
||||||
|
src = ./.;
|
||||||
|
buildInputs = buildInputs;
|
||||||
|
};
|
||||||
|
# Run `nix build .#fmt` to run tests
|
||||||
|
fmt = naersk'.buildPackage {
|
||||||
|
src = ./.;
|
||||||
|
mode = "fmt";
|
||||||
|
buildInputs = buildInputs;
|
||||||
|
};
|
||||||
|
# Run `nix build .#clippy` to lint code
|
||||||
|
clippy = naersk'.buildPackage {
|
||||||
|
src = ./.;
|
||||||
|
mode = "clippy";
|
||||||
|
buildInputs = buildInputs;
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
|
defaultPackage = packages.default;
|
||||||
|
|
||||||
# `nix run`
|
# `nix run`
|
||||||
apps."${package_name}" = utils.lib.mkApp {
|
apps."${package_name}" = utils.lib.mkApp {
|
||||||
|
@ -37,28 +62,31 @@
|
||||||
|
|
||||||
# `nix develop`
|
# `nix develop`
|
||||||
devShell = pkgs.mkShell {
|
devShell = pkgs.mkShell {
|
||||||
nativeBuildInputs = with pkgs; [ rustc cargo pkg-config openssl];
|
nativeBuildInputs = with pkgs; [rustc cargo pkg-config openssl rustfmt];
|
||||||
};
|
};
|
||||||
|
|
||||||
nixosModule = { lib, pkgs, config, ... }:
|
nixosModule = {
|
||||||
with lib;
|
lib,
|
||||||
let
|
pkgs,
|
||||||
|
config,
|
||||||
|
...
|
||||||
|
}:
|
||||||
|
with lib; let
|
||||||
cfg = config.services."${package_name}";
|
cfg = config.services."${package_name}";
|
||||||
# secret options are in the env file(s) loaded separately
|
# secret options are in the env file(s) loaded separately
|
||||||
environment_config = {
|
environment_config = {
|
||||||
LDAP_API = cfg.ldap;
|
DATABASE_HOME = cfg.home;
|
||||||
SKYNET_SERVER = cfg.discord.server;
|
|
||||||
HOME = cfg.home;
|
|
||||||
DATABASE = "database.db";
|
DATABASE = "database.db";
|
||||||
};
|
};
|
||||||
|
|
||||||
service_name = script: lib.strings.sanitizeDerivationName("${cfg.user}@${script}");
|
service_name = script: lib.strings.sanitizeDerivationName "${cfg.user}@${script}";
|
||||||
|
|
||||||
# oneshot scripts to run
|
# oneshot scripts to run
|
||||||
serviceGenerator = mapAttrs' (script: time: nameValuePair (service_name script) {
|
serviceGenerator = mapAttrs' (script: time:
|
||||||
|
nameValuePair (service_name script) {
|
||||||
description = "Service for ${desc} ${script}";
|
description = "Service for ${desc} ${script}";
|
||||||
wantedBy = [ ];
|
wantedBy = [];
|
||||||
after = [ "network-online.target" ];
|
after = ["network-online.target"];
|
||||||
environment = environment_config;
|
environment = environment_config;
|
||||||
|
|
||||||
serviceConfig = {
|
serviceConfig = {
|
||||||
|
@ -67,7 +95,6 @@
|
||||||
Group = "${cfg.user}";
|
Group = "${cfg.user}";
|
||||||
ExecStart = "${self.defaultPackage."${system}"}/bin/${script}";
|
ExecStart = "${self.defaultPackage."${system}"}/bin/${script}";
|
||||||
EnvironmentFile = [
|
EnvironmentFile = [
|
||||||
"${cfg.env.ldap}"
|
|
||||||
"${cfg.env.discord}"
|
"${cfg.env.discord}"
|
||||||
"${cfg.env.mail}"
|
"${cfg.env.mail}"
|
||||||
"${cfg.env.wolves}"
|
"${cfg.env.wolves}"
|
||||||
|
@ -76,11 +103,12 @@
|
||||||
});
|
});
|
||||||
|
|
||||||
# each timer will run the above service
|
# each timer will run the above service
|
||||||
timerGenerator = mapAttrs' (script: time: nameValuePair (service_name script) {
|
timerGenerator = mapAttrs' (script: time:
|
||||||
|
nameValuePair (service_name script) {
|
||||||
description = "Timer for ${desc} ${script}";
|
description = "Timer for ${desc} ${script}";
|
||||||
|
|
||||||
wantedBy = [ "timers.target" ];
|
wantedBy = ["timers.target"];
|
||||||
partOf = [ "${service_name script}.service" ];
|
partOf = ["${service_name script}.service"];
|
||||||
timerConfig = {
|
timerConfig = {
|
||||||
OnCalendar = time;
|
OnCalendar = time;
|
||||||
Unit = "${service_name script}.service";
|
Unit = "${service_name script}.service";
|
||||||
|
@ -91,23 +119,22 @@
|
||||||
# modify these
|
# modify these
|
||||||
scripts = {
|
scripts = {
|
||||||
# every 20 min
|
# every 20 min
|
||||||
"update_data" = "*:0,20,40";
|
"update_data" = "*:0,10,20,30,40,50";
|
||||||
# groups are updated every hour, offset from teh ldap
|
# groups are updated every hour, offset from teh ldap
|
||||||
"update_users" = "*:05:00";
|
"update_users" = "*:05:00";
|
||||||
|
# Committee server has its own timer
|
||||||
|
"update_committee" = "*:15:00";
|
||||||
|
# minecraft stuff is updated at 5am
|
||||||
|
"update_minecraft" = "5:10:00";
|
||||||
};
|
};
|
||||||
|
|
||||||
in {
|
in {
|
||||||
options.services."${package_name}" = {
|
options.services."${package_name}" = {
|
||||||
enable = mkEnableOption "enable ${package_name}";
|
enable = mkEnableOption "enable ${package_name}";
|
||||||
|
|
||||||
env = {
|
env = {
|
||||||
ldap = mkOption rec {
|
|
||||||
type = types.str;
|
|
||||||
description = "ENV file with LDAP_DISCORD_AUTH";
|
|
||||||
};
|
|
||||||
discord = mkOption rec {
|
discord = mkOption rec {
|
||||||
type = types.str;
|
type = types.str;
|
||||||
description = "ENV file with DISCORD_TOKEN";
|
description = "ENV file with DISCORD_TOKEN, DISCORD_TOKEN_MINECRAFT";
|
||||||
};
|
};
|
||||||
mail = mkOption rec {
|
mail = mkOption rec {
|
||||||
type = types.str;
|
type = types.str;
|
||||||
|
@ -115,23 +142,10 @@
|
||||||
};
|
};
|
||||||
wolves = mkOption rec {
|
wolves = mkOption rec {
|
||||||
type = types.str;
|
type = types.str;
|
||||||
description = "Mail details, has WOLVES_URL, WOLVES_KEY";
|
description = "Mail details, has WOLVES_URL";
|
||||||
};
|
};
|
||||||
};
|
};
|
||||||
|
|
||||||
discord = {
|
|
||||||
server = mkOption rec {
|
|
||||||
type = types.str;
|
|
||||||
description = "ID of the server the bot runs on";
|
|
||||||
};
|
|
||||||
};
|
|
||||||
|
|
||||||
ldap = mkOption rec {
|
|
||||||
type = types.str;
|
|
||||||
default = "https://api.account.skynet.ie";
|
|
||||||
description = "Location of the ldap api";
|
|
||||||
};
|
|
||||||
|
|
||||||
user = mkOption rec {
|
user = mkOption rec {
|
||||||
type = types.str;
|
type = types.str;
|
||||||
default = "${package_name}";
|
default = "${package_name}";
|
||||||
|
@ -153,8 +167,7 @@
|
||||||
};
|
};
|
||||||
|
|
||||||
config = mkIf cfg.enable {
|
config = mkIf cfg.enable {
|
||||||
|
users.groups."${cfg.user}" = {};
|
||||||
users.groups."${cfg.user}" = { };
|
|
||||||
|
|
||||||
users.users."${cfg.user}" = {
|
users.users."${cfg.user}" = {
|
||||||
createHome = true;
|
createHome = true;
|
||||||
|
@ -163,13 +176,14 @@
|
||||||
group = "${cfg.user}";
|
group = "${cfg.user}";
|
||||||
};
|
};
|
||||||
|
|
||||||
systemd.services = {
|
systemd.services =
|
||||||
|
{
|
||||||
# main service
|
# main service
|
||||||
"${package_name}" = {
|
"${package_name}" = {
|
||||||
description = desc;
|
description = desc;
|
||||||
wantedBy = [ "multi-user.target" ];
|
wantedBy = ["multi-user.target"];
|
||||||
after = [ "network-online.target" ];
|
after = ["network-online.target"];
|
||||||
wants = [ ];
|
wants = [];
|
||||||
environment = environment_config;
|
environment = environment_config;
|
||||||
|
|
||||||
serviceConfig = {
|
serviceConfig = {
|
||||||
|
@ -179,24 +193,22 @@
|
||||||
ExecStart = "${self.defaultPackage."${system}"}/bin/${package_name}";
|
ExecStart = "${self.defaultPackage."${system}"}/bin/${package_name}";
|
||||||
# can have multiple env files
|
# can have multiple env files
|
||||||
EnvironmentFile = [
|
EnvironmentFile = [
|
||||||
"${cfg.env.ldap}"
|
|
||||||
"${cfg.env.discord}"
|
"${cfg.env.discord}"
|
||||||
"${cfg.env.mail}"
|
"${cfg.env.mail}"
|
||||||
"${cfg.env.wolves}"
|
"${cfg.env.wolves}"
|
||||||
];
|
];
|
||||||
};
|
};
|
||||||
restartTriggers = [
|
restartTriggers = [
|
||||||
"${cfg.env.ldap}"
|
|
||||||
"${cfg.env.discord}"
|
"${cfg.env.discord}"
|
||||||
"${cfg.env.mail}"
|
"${cfg.env.mail}"
|
||||||
"${cfg.env.wolves}"
|
"${cfg.env.wolves}"
|
||||||
];
|
];
|
||||||
};
|
};
|
||||||
} // serviceGenerator scripts;
|
}
|
||||||
|
// serviceGenerator scripts;
|
||||||
|
|
||||||
# timers to run the above services
|
# timers to run the above services
|
||||||
systemd.timers = timerGenerator scripts;
|
systemd.timers = timerGenerator scripts;
|
||||||
|
|
||||||
};
|
};
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
BIN
media/setup_user_01.png
(Stored with Git LFS)
Normal file
BIN
media/setup_user_01.png
(Stored with Git LFS)
Normal file
Binary file not shown.
BIN
media/setup_user_02.png
(Stored with Git LFS)
Normal file
BIN
media/setup_user_02.png
(Stored with Git LFS)
Normal file
Binary file not shown.
BIN
media/setup_user_03.png
(Stored with Git LFS)
Normal file
BIN
media/setup_user_03.png
(Stored with Git LFS)
Normal file
Binary file not shown.
2
rust-toolchain.toml
Normal file
2
rust-toolchain.toml
Normal file
|
@ -0,0 +1,2 @@
|
||||||
|
[toolchain]
|
||||||
|
channel = "1.80"
|
54
src/bin/update_committee.rs
Normal file
54
src/bin/update_committee.rs
Normal file
|
@ -0,0 +1,54 @@
|
||||||
|
use serenity::{
|
||||||
|
async_trait,
|
||||||
|
client::{Context, EventHandler},
|
||||||
|
model::gateway::{GatewayIntents, Ready},
|
||||||
|
Client,
|
||||||
|
};
|
||||||
|
use skynet_discord_bot::common::database::{db_init, DataBase};
|
||||||
|
use skynet_discord_bot::common::set_roles::committee;
|
||||||
|
use skynet_discord_bot::{get_config, Config};
|
||||||
|
use std::{process, sync::Arc};
|
||||||
|
use tokio::sync::RwLock;
|
||||||
|
|
||||||
|
#[tokio::main]
|
||||||
|
async fn main() {
|
||||||
|
let config = get_config();
|
||||||
|
let db = match db_init(&config).await {
|
||||||
|
Ok(x) => x,
|
||||||
|
Err(_) => return,
|
||||||
|
};
|
||||||
|
|
||||||
|
// Intents are a bitflag, bitwise operations can be used to dictate which intents to use
|
||||||
|
let intents = GatewayIntents::GUILDS | GatewayIntents::GUILD_MESSAGES | GatewayIntents::MESSAGE_CONTENT | GatewayIntents::GUILD_MEMBERS;
|
||||||
|
// Build our client.
|
||||||
|
let mut client = Client::builder(&config.discord_token, intents)
|
||||||
|
.event_handler(Handler {})
|
||||||
|
.await
|
||||||
|
.expect("Error creating client");
|
||||||
|
|
||||||
|
{
|
||||||
|
let mut data = client.data.write().await;
|
||||||
|
|
||||||
|
data.insert::<Config>(Arc::new(RwLock::new(config)));
|
||||||
|
data.insert::<DataBase>(Arc::new(RwLock::new(db)));
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Err(why) = client.start().await {
|
||||||
|
println!("Client error: {:?}", why);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct Handler;
|
||||||
|
#[async_trait]
|
||||||
|
impl EventHandler for Handler {
|
||||||
|
async fn ready(&self, ctx: Context, ready: Ready) {
|
||||||
|
let ctx = Arc::new(ctx);
|
||||||
|
println!("{} is connected!", ready.user.name);
|
||||||
|
|
||||||
|
// u[date committee server
|
||||||
|
committee::check_committee(Arc::clone(&ctx)).await;
|
||||||
|
|
||||||
|
// finish up
|
||||||
|
process::exit(0);
|
||||||
|
}
|
||||||
|
}
|
|
@ -4,7 +4,10 @@ use serenity::{
|
||||||
model::gateway::{GatewayIntents, Ready},
|
model::gateway::{GatewayIntents, Ready},
|
||||||
Client,
|
Client,
|
||||||
};
|
};
|
||||||
use skynet_discord_bot::{db_init, get_config, get_data::get_wolves, Config, DataBase};
|
use skynet_discord_bot::common::database::{db_init, DataBase};
|
||||||
|
use skynet_discord_bot::common::wolves::cns::get_wolves;
|
||||||
|
use skynet_discord_bot::common::wolves::committees::get_cns;
|
||||||
|
use skynet_discord_bot::{get_config, Config};
|
||||||
use std::{process, sync::Arc};
|
use std::{process, sync::Arc};
|
||||||
use tokio::sync::RwLock;
|
use tokio::sync::RwLock;
|
||||||
|
|
||||||
|
@ -13,7 +16,10 @@ async fn main() {
|
||||||
let config = get_config();
|
let config = get_config();
|
||||||
let db = match db_init(&config).await {
|
let db = match db_init(&config).await {
|
||||||
Ok(x) => x,
|
Ok(x) => x,
|
||||||
Err(_) => return,
|
Err(e) => {
|
||||||
|
dbg!(e);
|
||||||
|
return;
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
// Intents are a bitflag, bitwise operations can be used to dictate which intents to use
|
// Intents are a bitflag, bitwise operations can be used to dictate which intents to use
|
||||||
|
@ -43,8 +49,12 @@ impl EventHandler for Handler {
|
||||||
let ctx = Arc::new(ctx);
|
let ctx = Arc::new(ctx);
|
||||||
println!("{} is connected!", ready.user.name);
|
println!("{} is connected!", ready.user.name);
|
||||||
|
|
||||||
|
// get the data for each individual club/soc
|
||||||
get_wolves(&ctx).await;
|
get_wolves(&ctx).await;
|
||||||
|
|
||||||
|
// get teh data for the clubs/socs committees
|
||||||
|
get_cns(&ctx).await;
|
||||||
|
|
||||||
// finish up
|
// finish up
|
||||||
process::exit(0);
|
process::exit(0);
|
||||||
}
|
}
|
||||||
|
|
27
src/bin/update_minecraft.rs
Normal file
27
src/bin/update_minecraft.rs
Normal file
|
@ -0,0 +1,27 @@
|
||||||
|
use skynet_discord_bot::common::database::db_init;
|
||||||
|
use skynet_discord_bot::common::minecraft::{get_minecraft_config, update_server, whitelist_wipe};
|
||||||
|
use skynet_discord_bot::get_config;
|
||||||
|
use std::collections::HashSet;
|
||||||
|
|
||||||
|
#[tokio::main]
|
||||||
|
async fn main() {
|
||||||
|
let config = get_config();
|
||||||
|
let db = match db_init(&config).await {
|
||||||
|
Ok(x) => x,
|
||||||
|
Err(_) => return,
|
||||||
|
};
|
||||||
|
|
||||||
|
let servers = get_minecraft_config(&db).await;
|
||||||
|
let mut wiped = HashSet::new();
|
||||||
|
|
||||||
|
for server in &servers {
|
||||||
|
// wipe whitelist first
|
||||||
|
if !wiped.contains(&server.minecraft) {
|
||||||
|
whitelist_wipe(&server.minecraft, &config.discord_token_minecraft).await;
|
||||||
|
// add it to teh done list so its not done again
|
||||||
|
wiped.insert(&server.minecraft);
|
||||||
|
}
|
||||||
|
|
||||||
|
update_server(&server.minecraft, &db, &server.discord, &config).await;
|
||||||
|
}
|
||||||
|
}
|
|
@ -4,7 +4,9 @@ use serenity::{
|
||||||
model::gateway::{GatewayIntents, Ready},
|
model::gateway::{GatewayIntents, Ready},
|
||||||
Client,
|
Client,
|
||||||
};
|
};
|
||||||
use skynet_discord_bot::{db_init, get_config, get_server_config_bulk, set_roles::update_server, Config, DataBase};
|
use skynet_discord_bot::common::database::{db_init, get_server_config_bulk, DataBase};
|
||||||
|
use skynet_discord_bot::common::set_roles::normal;
|
||||||
|
use skynet_discord_bot::{get_config, Config};
|
||||||
use std::{process, sync::Arc};
|
use std::{process, sync::Arc};
|
||||||
use tokio::sync::RwLock;
|
use tokio::sync::RwLock;
|
||||||
|
|
||||||
|
@ -43,14 +45,15 @@ impl EventHandler for Handler {
|
||||||
let ctx = Arc::new(ctx);
|
let ctx = Arc::new(ctx);
|
||||||
println!("{} is connected!", ready.user.name);
|
println!("{} is connected!", ready.user.name);
|
||||||
|
|
||||||
bulk_check(Arc::clone(&ctx)).await;
|
// this goes into each server and sets roles for each wolves member
|
||||||
|
check_bulk(Arc::clone(&ctx)).await;
|
||||||
|
|
||||||
// finish up
|
// finish up
|
||||||
process::exit(0);
|
process::exit(0);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn bulk_check(ctx: Arc<Context>) {
|
async fn check_bulk(ctx: Arc<Context>) {
|
||||||
let db_lock = {
|
let db_lock = {
|
||||||
let data_read = ctx.data.read().await;
|
let data_read = ctx.data.read().await;
|
||||||
data_read.get::<DataBase>().expect("Expected Config in TypeMap.").clone()
|
data_read.get::<DataBase>().expect("Expected Config in TypeMap.").clone()
|
||||||
|
@ -59,6 +62,6 @@ async fn bulk_check(ctx: Arc<Context>) {
|
||||||
let db = db_lock.read().await;
|
let db = db_lock.read().await;
|
||||||
|
|
||||||
for server_config in get_server_config_bulk(&db).await {
|
for server_config in get_server_config_bulk(&db).await {
|
||||||
update_server(&ctx, &server_config, &[], &vec![]).await;
|
normal::update_server(&ctx, &server_config, &[], &[]).await;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,81 +1,49 @@
|
||||||
use serenity::{
|
use serenity::all::{CommandDataOption, CommandDataOptionValue, CommandInteraction, CommandOptionType, CreateCommandOption};
|
||||||
builder::CreateApplicationCommand,
|
use serenity::{builder::CreateCommand, client::Context};
|
||||||
client::Context,
|
use skynet_discord_bot::common::database::{get_server_config, DataBase, Servers};
|
||||||
model::{
|
use skynet_discord_bot::common::set_roles::normal::update_server;
|
||||||
application::interaction::application_command::ApplicationCommandInteraction,
|
use skynet_discord_bot::common::wolves::cns::get_wolves;
|
||||||
prelude::{command::CommandOptionType, interaction::application_command::CommandDataOptionValue},
|
|
||||||
},
|
|
||||||
};
|
|
||||||
use skynet_discord_bot::get_data::get_wolves;
|
|
||||||
use skynet_discord_bot::{get_server_config, set_roles::update_server, DataBase, Servers};
|
|
||||||
use sqlx::{Error, Pool, Sqlite};
|
use sqlx::{Error, Pool, Sqlite};
|
||||||
|
|
||||||
pub async fn run(command: &ApplicationCommandInteraction, ctx: &Context) -> String {
|
pub async fn run(command: &CommandInteraction, ctx: &Context) -> String {
|
||||||
// check if user has high enough permisssions
|
let wolves_api = if let Some(CommandDataOption {
|
||||||
let mut admin = false;
|
value: CommandDataOptionValue::String(key),
|
||||||
|
..
|
||||||
let g_id = match command.guild_id {
|
}) = command.data.options.first()
|
||||||
None => return "Not in a server".to_string(),
|
|
||||||
Some(x) => x,
|
|
||||||
};
|
|
||||||
|
|
||||||
let roles_server = g_id.roles(&ctx.http).await.unwrap_or_default();
|
|
||||||
|
|
||||||
if let Ok(member) = g_id.member(&ctx.http, command.user.id).await {
|
|
||||||
if let Some(permissions) = member.permissions {
|
|
||||||
if permissions.administrator() {
|
|
||||||
admin = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for role_id in member.roles {
|
|
||||||
if admin {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
if let Some(role) = roles_server.get(&role_id) {
|
|
||||||
if role.permissions.administrator() {
|
|
||||||
admin = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if !admin {
|
|
||||||
return "Administrator permission required".to_string();
|
|
||||||
}
|
|
||||||
|
|
||||||
let api_key = if let CommandDataOptionValue::String(key) = command
|
|
||||||
.data
|
|
||||||
.options
|
|
||||||
.get(0)
|
|
||||||
.expect("Expected user option")
|
|
||||||
.resolved
|
|
||||||
.as_ref()
|
|
||||||
.expect("Expected user object")
|
|
||||||
{
|
{
|
||||||
key
|
key.to_string()
|
||||||
} else {
|
} else {
|
||||||
return "Please provide a wolves API key".to_string();
|
return "Please provide a wolves API key".to_string();
|
||||||
};
|
};
|
||||||
|
|
||||||
let role_current = if let CommandDataOptionValue::Role(role) = command
|
let role_current = if let Some(CommandDataOption {
|
||||||
.data
|
value: CommandDataOptionValue::Role(role),
|
||||||
.options
|
..
|
||||||
.get(1)
|
}) = command.data.options.get(1)
|
||||||
.expect("Expected role option")
|
|
||||||
.resolved
|
|
||||||
.as_ref()
|
|
||||||
.expect("Expected role object")
|
|
||||||
{
|
{
|
||||||
Some(role.id.to_owned())
|
role.to_owned()
|
||||||
} else {
|
} else {
|
||||||
return "Please provide a valid role for ``Role Current``".to_string();
|
return "Please provide a valid role for ``Role Current``".to_string();
|
||||||
};
|
};
|
||||||
|
|
||||||
let mut role_past = None;
|
let role_past = if let Some(CommandDataOption {
|
||||||
if let Some(x) = command.data.options.get(2) {
|
value: CommandDataOptionValue::Role(role),
|
||||||
if let Some(CommandDataOptionValue::Role(role)) = &x.resolved {
|
..
|
||||||
role_past = Some(role.id.to_owned());
|
}) = command.data.options.get(5)
|
||||||
}
|
{
|
||||||
|
Some(role.to_owned())
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
};
|
||||||
|
|
||||||
|
let bot_channel_id = if let Some(CommandDataOption {
|
||||||
|
value: CommandDataOptionValue::Channel(channel),
|
||||||
|
..
|
||||||
|
}) = command.data.options.get(2)
|
||||||
|
{
|
||||||
|
channel.to_owned()
|
||||||
|
} else {
|
||||||
|
return "Please provide a valid channel for ``Bot Channel``".to_string();
|
||||||
};
|
};
|
||||||
|
|
||||||
let db_lock = {
|
let db_lock = {
|
||||||
|
@ -86,11 +54,13 @@ pub async fn run(command: &ApplicationCommandInteraction, ctx: &Context) -> Stri
|
||||||
|
|
||||||
let server_data = Servers {
|
let server_data = Servers {
|
||||||
server: command.guild_id.unwrap_or_default(),
|
server: command.guild_id.unwrap_or_default(),
|
||||||
wolves_api: api_key.to_owned(),
|
wolves_api,
|
||||||
|
wolves_id: 0,
|
||||||
role_past,
|
role_past,
|
||||||
role_current,
|
role_current,
|
||||||
member_past: 0,
|
member_past: 0,
|
||||||
member_current: 0,
|
member_current: 0,
|
||||||
|
bot_channel_id,
|
||||||
};
|
};
|
||||||
|
|
||||||
match add_server(&db, ctx, &server_data).await {
|
match add_server(&db, ctx, &server_data).await {
|
||||||
|
@ -104,48 +74,31 @@ pub async fn run(command: &ApplicationCommandInteraction, ctx: &Context) -> Stri
|
||||||
"Added/Updated server info".to_string()
|
"Added/Updated server info".to_string()
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn register(command: &mut CreateApplicationCommand) -> &mut CreateApplicationCommand {
|
pub fn register() -> CreateCommand {
|
||||||
command
|
CreateCommand::new("add")
|
||||||
.name("add")
|
|
||||||
.description("Enable the bot for this discord")
|
.description("Enable the bot for this discord")
|
||||||
.create_option(|option| {
|
.default_member_permissions(serenity::model::Permissions::MANAGE_GUILD)
|
||||||
option
|
.add_option(CreateCommandOption::new(CommandOptionType::String, "api_key", "UL Wolves API Key").required(true))
|
||||||
.name("api_key")
|
.add_option(CreateCommandOption::new(CommandOptionType::Role, "role_current", "Role for Current members").required(true))
|
||||||
.description("UL Wolves API Key")
|
.add_option(CreateCommandOption::new(CommandOptionType::Channel, "bot_channel", "Safe space for folks to use the bot commands.").required(true))
|
||||||
.kind(CommandOptionType::String)
|
.add_option(CreateCommandOption::new(CommandOptionType::Role, "role_past", "Role for Past members").required(false))
|
||||||
.required(true)
|
|
||||||
})
|
|
||||||
.create_option(|option| {
|
|
||||||
option
|
|
||||||
.name("role_current")
|
|
||||||
.description("Role for Current members")
|
|
||||||
.kind(CommandOptionType::Role)
|
|
||||||
.required(true)
|
|
||||||
})
|
|
||||||
.create_option(|option| {
|
|
||||||
option
|
|
||||||
.name("role_past")
|
|
||||||
.description("Role for Past members")
|
|
||||||
.kind(CommandOptionType::Role)
|
|
||||||
.required(false)
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn add_server(db: &Pool<Sqlite>, ctx: &Context, server: &Servers) -> Result<Option<Servers>, Error> {
|
async fn add_server(db: &Pool<Sqlite>, ctx: &Context, server: &Servers) -> Result<Option<Servers>, Error> {
|
||||||
let existing = get_server_config(db, &server.server).await;
|
let existing = get_server_config(db, &server.server).await;
|
||||||
let role_past = server.role_past.map(|x| *x.as_u64() as i64);
|
let role_past = server.role_past.map(|x| x.get() as i64);
|
||||||
let role_current = server.role_current.map(|x| *x.as_u64() as i64);
|
|
||||||
|
|
||||||
let insert = sqlx::query_as::<_, Servers>(
|
let insert = sqlx::query_as::<_, Servers>(
|
||||||
"
|
"
|
||||||
INSERT OR REPLACE INTO servers (server, wolves_api, role_past, role_current)
|
INSERT OR REPLACE INTO servers (server, wolves_api, role_past, role_current, bot_channel_id)
|
||||||
VALUES (?1, ?2, ?3, ?4)
|
VALUES (?1, ?2, ?3, ?4, ?5)
|
||||||
",
|
",
|
||||||
)
|
)
|
||||||
.bind(*server.server.as_u64() as i64)
|
.bind(server.server.get() as i64)
|
||||||
.bind(&server.wolves_api)
|
.bind(&server.wolves_api)
|
||||||
.bind(role_past)
|
.bind(role_past)
|
||||||
.bind(role_current)
|
.bind(server.role_current.get() as i64)
|
||||||
|
.bind(server.bot_channel_id.get() as i64)
|
||||||
.fetch_optional(db)
|
.fetch_optional(db)
|
||||||
.await;
|
.await;
|
||||||
|
|
||||||
|
@ -160,7 +113,7 @@ async fn add_server(db: &Pool<Sqlite>, ctx: &Context, server: &Servers) -> Resul
|
||||||
if x.role_current != server.role_current {
|
if x.role_current != server.role_current {
|
||||||
result.0 = true;
|
result.0 = true;
|
||||||
result.1 = true;
|
result.1 = true;
|
||||||
result.2 = x.role_current;
|
result.2 = Some(x.role_current);
|
||||||
}
|
}
|
||||||
if x.role_past != server.role_past {
|
if x.role_past != server.role_past {
|
||||||
result.0 = true;
|
result.0 = true;
|
||||||
|
@ -183,7 +136,7 @@ async fn add_server(db: &Pool<Sqlite>, ctx: &Context, server: &Servers) -> Resul
|
||||||
if past_remove {
|
if past_remove {
|
||||||
roles_remove.push(past_role)
|
roles_remove.push(past_role)
|
||||||
}
|
}
|
||||||
update_server(ctx, server, &roles_remove, &vec![]).await;
|
update_server(ctx, server, &roles_remove, &[]).await;
|
||||||
}
|
}
|
||||||
|
|
||||||
insert
|
insert
|
||||||
|
|
157
src/commands/count.rs
Normal file
157
src/commands/count.rs
Normal file
|
@ -0,0 +1,157 @@
|
||||||
|
pub mod committee {
|
||||||
|
|
||||||
|
// get the list of all the current clubs/socs members
|
||||||
|
|
||||||
|
use serenity::all::{
|
||||||
|
CommandDataOption, CommandDataOptionValue, CommandInteraction, CommandOptionType, Context, CreateCommand, CreateCommandOption,
|
||||||
|
};
|
||||||
|
use skynet_discord_bot::common::database::DataBase;
|
||||||
|
use skynet_discord_bot::common::set_roles::committee::db_roles_get;
|
||||||
|
|
||||||
|
pub async fn run(command: &CommandInteraction, ctx: &Context) -> String {
|
||||||
|
let sub_options = if let Some(CommandDataOption {
|
||||||
|
value: CommandDataOptionValue::SubCommand(key),
|
||||||
|
..
|
||||||
|
}) = command.data.options.first()
|
||||||
|
{
|
||||||
|
key
|
||||||
|
} else {
|
||||||
|
return "Please provide a wolves API key".to_string();
|
||||||
|
};
|
||||||
|
|
||||||
|
let all = if let Some(x) = sub_options.first() {
|
||||||
|
match x.value {
|
||||||
|
CommandDataOptionValue::Boolean(y) => y,
|
||||||
|
_ => false,
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
false
|
||||||
|
};
|
||||||
|
|
||||||
|
let db_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<DataBase>().expect("Expected Databse in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
let db = db_lock.read().await;
|
||||||
|
|
||||||
|
let mut cs = vec![];
|
||||||
|
// pull it from a DB
|
||||||
|
for committee in db_roles_get(&db).await {
|
||||||
|
if !all && committee.count == 0 {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
cs.push((committee.count, committee.name_role.to_owned()));
|
||||||
|
}
|
||||||
|
|
||||||
|
cs.sort_by_key(|(count, _)| *count);
|
||||||
|
cs.reverse();
|
||||||
|
|
||||||
|
// msg can be a max 2000 chars long
|
||||||
|
let mut limit = 2000 - 3;
|
||||||
|
|
||||||
|
let mut response = vec!["```".to_string()];
|
||||||
|
for (count, name) in cs {
|
||||||
|
let leading = if count < 10 { " " } else { "" };
|
||||||
|
|
||||||
|
let line = format!("{}{} {}", leading, count, name);
|
||||||
|
|
||||||
|
let length = line.len() + 1;
|
||||||
|
|
||||||
|
if length < (limit + 3) {
|
||||||
|
response.push(line);
|
||||||
|
limit -= length;
|
||||||
|
} else {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
response.push("```".to_string());
|
||||||
|
|
||||||
|
response.join("\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn register() -> CreateCommand {
|
||||||
|
CreateCommand::new("count")
|
||||||
|
.description("Count Committee Members")
|
||||||
|
// All Committee members are able to add reactions to posts
|
||||||
|
.default_member_permissions(serenity::model::Permissions::ADD_REACTIONS)
|
||||||
|
.add_option(
|
||||||
|
CreateCommandOption::new(CommandOptionType::SubCommand, "committee", "List out the Committee Roles Numbers")
|
||||||
|
.add_sub_option(CreateCommandOption::new(CommandOptionType::Boolean, "all", "List out all the Committee Roles Numbers").required(false)),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub mod servers {
|
||||||
|
// get the list of all the current clubs/socs
|
||||||
|
use serenity::all::{CommandInteraction, CommandOptionType, Context, CreateCommand, CreateCommandOption};
|
||||||
|
use skynet_discord_bot::common::database::{get_server_config_bulk, DataBase};
|
||||||
|
use skynet_discord_bot::common::set_roles::committee::get_committees;
|
||||||
|
use std::collections::HashMap;
|
||||||
|
|
||||||
|
pub async fn run(_command: &CommandInteraction, ctx: &Context) -> String {
|
||||||
|
let db_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<DataBase>().expect("Expected Databse in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
let db = db_lock.read().await;
|
||||||
|
|
||||||
|
let mut committees = HashMap::new();
|
||||||
|
for committee in get_committees(&db).await {
|
||||||
|
committees.insert(committee.id, committee.to_owned());
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut cs = vec![];
|
||||||
|
// pull it from a DB
|
||||||
|
for server_config in get_server_config_bulk(&db).await {
|
||||||
|
if let Some(x) = committees.get(&server_config.wolves_id) {
|
||||||
|
cs.push((server_config.member_current, server_config.member_past, x.name_full.to_owned()));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
cs.sort_by_key(|(count, _, _)| *count);
|
||||||
|
cs.reverse();
|
||||||
|
|
||||||
|
// msg can be a max 2000 chars long
|
||||||
|
let mut limit = 2000 - 3;
|
||||||
|
|
||||||
|
let mut response = vec!["```".to_string()];
|
||||||
|
for (current, past, name) in cs {
|
||||||
|
let current_leading = if current < 10 {
|
||||||
|
" "
|
||||||
|
} else if current < 100 {
|
||||||
|
" "
|
||||||
|
} else {
|
||||||
|
""
|
||||||
|
};
|
||||||
|
let past_leading = if past < 10 {
|
||||||
|
" "
|
||||||
|
} else if past < 100 {
|
||||||
|
" "
|
||||||
|
} else {
|
||||||
|
""
|
||||||
|
};
|
||||||
|
|
||||||
|
let line = format!("{}{} {}{} {}", current_leading, current, past_leading, past, name);
|
||||||
|
|
||||||
|
let length = line.len() + 1;
|
||||||
|
|
||||||
|
// +3 is to account for the closing fense
|
||||||
|
if length < (limit + 3) {
|
||||||
|
response.push(line);
|
||||||
|
limit -= length;
|
||||||
|
} else {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
response.push("```".to_string());
|
||||||
|
|
||||||
|
response.join("\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn register() -> CreateCommand {
|
||||||
|
CreateCommand::new("count")
|
||||||
|
.description("Count the servers")
|
||||||
|
.default_member_permissions(serenity::model::Permissions::MANAGE_GUILD)
|
||||||
|
.add_option(CreateCommandOption::new(CommandOptionType::SubCommand, "servers", "List out all servers using the skynet bot"))
|
||||||
|
}
|
||||||
|
}
|
|
@ -4,21 +4,16 @@ use lettre::{
|
||||||
Message, SmtpTransport, Transport,
|
Message, SmtpTransport, Transport,
|
||||||
};
|
};
|
||||||
use maud::html;
|
use maud::html;
|
||||||
use serenity::{
|
use serenity::{builder::CreateCommand, client::Context, model::id::UserId};
|
||||||
builder::CreateApplicationCommand,
|
use skynet_discord_bot::common::database::{DataBase, Wolves, WolvesVerify};
|
||||||
client::Context,
|
use skynet_discord_bot::{get_now_iso, random_string, Config};
|
||||||
model::{
|
|
||||||
application::interaction::application_command::ApplicationCommandInteraction,
|
|
||||||
id::UserId,
|
|
||||||
prelude::{command::CommandOptionType, interaction::application_command::CommandDataOptionValue},
|
|
||||||
},
|
|
||||||
};
|
|
||||||
use skynet_discord_bot::{get_now_iso, random_string, Config, DataBase, Wolves, WolvesVerify};
|
|
||||||
use sqlx::{Pool, Sqlite};
|
use sqlx::{Pool, Sqlite};
|
||||||
pub(crate) mod link {
|
|
||||||
use super::*;
|
|
||||||
|
|
||||||
pub async fn run(command: &ApplicationCommandInteraction, ctx: &Context) -> String {
|
pub mod link {
|
||||||
|
use super::*;
|
||||||
|
use serenity::all::{CommandDataOption, CommandDataOptionValue, CommandInteraction, CommandOptionType, CreateCommand, CreateCommandOption};
|
||||||
|
|
||||||
|
pub async fn run(command: &CommandInteraction, ctx: &Context) -> String {
|
||||||
let db_lock = {
|
let db_lock = {
|
||||||
let data_read = ctx.data.read().await;
|
let data_read = ctx.data.read().await;
|
||||||
data_read.get::<DataBase>().expect("Expected Databse in TypeMap.").clone()
|
data_read.get::<DataBase>().expect("Expected Databse in TypeMap.").clone()
|
||||||
|
@ -41,16 +36,11 @@ pub(crate) mod link {
|
||||||
return "Linking already in process, please check email.".to_string();
|
return "Linking already in process, please check email.".to_string();
|
||||||
}
|
}
|
||||||
|
|
||||||
let option = command
|
let email = if let Some(CommandDataOption {
|
||||||
.data
|
value: CommandDataOptionValue::String(email),
|
||||||
.options
|
..
|
||||||
.get(0)
|
}) = command.data.options.first()
|
||||||
.expect("Expected email option")
|
{
|
||||||
.resolved
|
|
||||||
.as_ref()
|
|
||||||
.expect("Expected email object");
|
|
||||||
|
|
||||||
let email = if let CommandDataOptionValue::String(email) = option {
|
|
||||||
email.trim()
|
email.trim()
|
||||||
} else {
|
} else {
|
||||||
return "Please provide a valid user".to_string();
|
return "Please provide a valid user".to_string();
|
||||||
|
@ -59,7 +49,34 @@ pub(crate) mod link {
|
||||||
// check if email exists
|
// check if email exists
|
||||||
let details = match get_server_member_email(&db, email).await {
|
let details = match get_server_member_email(&db, email).await {
|
||||||
None => {
|
None => {
|
||||||
return "Please check it is your preferred contact on https://ulwolves.ie/memberships/profile and that you are fully paid up.".to_string()
|
let invalid_user = "Please check it matches (including case) your preferred contact on https://ulwolves.ie/memberships/profile and that you are fully paid up.".to_string();
|
||||||
|
|
||||||
|
let wolves = wolves_oxidised::Client::new(&config.wolves_url, Some(&config.wolves_api));
|
||||||
|
|
||||||
|
// see if the user actually exists
|
||||||
|
let id = match wolves.get_member(email).await {
|
||||||
|
None => {
|
||||||
|
return invalid_user;
|
||||||
|
}
|
||||||
|
Some(x) => x,
|
||||||
|
};
|
||||||
|
|
||||||
|
// save teh user id and email to teh db
|
||||||
|
match save_to_db_user(&db, id, email).await {
|
||||||
|
Ok(x) => x,
|
||||||
|
Err(x) => {
|
||||||
|
dbg!(x);
|
||||||
|
return "Error: unable to save user to teh database, contact Computer Society".to_string();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// pull it back out (technically could do it in previous step but more explicit)
|
||||||
|
match get_server_member_email(&db, email).await {
|
||||||
|
None => {
|
||||||
|
return "Error: failed to read user from database.".to_string();
|
||||||
|
}
|
||||||
|
Some(x) => x,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
Some(x) => x,
|
Some(x) => x,
|
||||||
};
|
};
|
||||||
|
@ -70,7 +87,7 @@ pub(crate) mod link {
|
||||||
|
|
||||||
// generate a auth key
|
// generate a auth key
|
||||||
let auth = random_string(20);
|
let auth = random_string(20);
|
||||||
match send_mail(&config, &details, &auth, &command.user.name) {
|
match send_mail(&config, &details.email, &auth, &command.user.name) {
|
||||||
Ok(_) => match save_to_db(&db, &details, &auth, &command.user.id).await {
|
Ok(_) => match save_to_db(&db, &details, &auth, &command.user.id).await {
|
||||||
Ok(_) => {}
|
Ok(_) => {}
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
|
@ -85,14 +102,13 @@ pub(crate) mod link {
|
||||||
format!("Verification email sent to {}, it may take up to 15 min for it to arrive. If it takes longer check the Junk folder.", email)
|
format!("Verification email sent to {}, it may take up to 15 min for it to arrive. If it takes longer check the Junk folder.", email)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn register(command: &mut CreateApplicationCommand) -> &mut CreateApplicationCommand {
|
pub fn register() -> CreateCommand {
|
||||||
command
|
CreateCommand::new("link_wolves")
|
||||||
.name("link")
|
|
||||||
.description("Set Wolves Email")
|
.description("Set Wolves Email")
|
||||||
.create_option(|option| option.name("email").description("UL Wolves Email").kind(CommandOptionType::String).required(true))
|
.add_option(CreateCommandOption::new(CommandOptionType::String, "email", "UL Wolves Email").required(true))
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn get_server_member_discord(db: &Pool<Sqlite>, user: &UserId) -> Option<Wolves> {
|
pub async fn get_server_member_discord(db: &Pool<Sqlite>, user: &UserId) -> Option<Wolves> {
|
||||||
sqlx::query_as::<_, Wolves>(
|
sqlx::query_as::<_, Wolves>(
|
||||||
r#"
|
r#"
|
||||||
SELECT *
|
SELECT *
|
||||||
|
@ -100,7 +116,7 @@ pub(crate) mod link {
|
||||||
WHERE discord = ?
|
WHERE discord = ?
|
||||||
"#,
|
"#,
|
||||||
)
|
)
|
||||||
.bind(*user.as_u64() as i64)
|
.bind(user.get() as i64)
|
||||||
.fetch_one(db)
|
.fetch_one(db)
|
||||||
.await
|
.await
|
||||||
.ok()
|
.ok()
|
||||||
|
@ -120,33 +136,35 @@ pub(crate) mod link {
|
||||||
.ok()
|
.ok()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn send_mail(config: &Config, email: &Wolves, auth: &str, user: &str) -> Result<smtp::response::Response, smtp::Error> {
|
fn send_mail(config: &Config, mail: &str, auth: &str, user: &str) -> Result<smtp::response::Response, smtp::Error> {
|
||||||
let mail = &email.email;
|
let discord = "https://computer.discord.skynet.ie";
|
||||||
let discord = "https://discord.skynet.ie";
|
|
||||||
let sender = format!("UL Computer Society <{}>", &config.mail_user);
|
let sender = format!("UL Computer Society <{}>", &config.mail_user);
|
||||||
|
|
||||||
// Create the html we want to send.
|
// Create the html we want to send.
|
||||||
let html = html! {
|
let html = html! {
|
||||||
head {
|
head {
|
||||||
title { "Hello from Skynet!" }
|
title { "UL Wolves Discord Linker" }
|
||||||
style type="text/css" {
|
style type="text/css" {
|
||||||
"h2, h4 { font-family: Arial, Helvetica, sans-serif; }"
|
"h2, h4 { font-family: Arial, Helvetica, sans-serif; }"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
div style="display: flex; flex-direction: column; align-items: center;" {
|
div {
|
||||||
h2 { "Hello from Skynet!" }
|
h2 { "UL Wolves Discord Linker" }
|
||||||
|
|
||||||
|
h3 { "Link your UL Wolves Account to Discord" }
|
||||||
// Substitute in the name of our recipient.
|
// Substitute in the name of our recipient.
|
||||||
p { "Hi " (user) "," }
|
p { "Hi " (user) "," }
|
||||||
p {
|
p {
|
||||||
"Please use " pre { "/verify code: " (auth)} " to verify your discord account."
|
"Please paste this line into Discord (and press enter) to verify your discord account:"
|
||||||
|
br;
|
||||||
|
pre { "/verify code: " (auth)}
|
||||||
}
|
}
|
||||||
|
hr;
|
||||||
|
h3 { "Help & Support" }
|
||||||
p {
|
p {
|
||||||
"If you have issues please refer to our Discord server:"
|
"If you have issues please refer to our Computer Society Discord Server:"
|
||||||
br;
|
br;
|
||||||
a href=(discord) { (discord) }
|
a href=(discord) { (discord) }
|
||||||
}
|
|
||||||
p {
|
|
||||||
"Skynet Team"
|
|
||||||
br;
|
br;
|
||||||
"UL Computer Society"
|
"UL Computer Society"
|
||||||
}
|
}
|
||||||
|
@ -155,14 +173,22 @@ pub(crate) mod link {
|
||||||
|
|
||||||
let body_text = format!(
|
let body_text = format!(
|
||||||
r#"
|
r#"
|
||||||
Hi {user}
|
UL Wolves Discord Linker
|
||||||
|
Link your UL Wolves Account to Discord
|
||||||
|
|
||||||
Please use "/verify code: {auth}" to verify your discord account.
|
Link your Account
|
||||||
|
|
||||||
If you have issues please refer to our Discord server:
|
Hi {user},
|
||||||
|
|
||||||
|
Please paste this line into Discord (and press enter) to verify your Discord account:
|
||||||
|
/verify code: {auth}
|
||||||
|
|
||||||
|
-------------------------------------------------------------------------
|
||||||
|
|
||||||
|
Help & Support
|
||||||
|
|
||||||
|
If you have issues please refer to our Computer Society Discord Server:
|
||||||
{discord}
|
{discord}
|
||||||
|
|
||||||
Skynet Team
|
|
||||||
UL Computer Society
|
UL Computer Society
|
||||||
"#
|
"#
|
||||||
);
|
);
|
||||||
|
@ -171,7 +197,7 @@ pub(crate) mod link {
|
||||||
let email = Message::builder()
|
let email = Message::builder()
|
||||||
.from(sender.parse().unwrap())
|
.from(sender.parse().unwrap())
|
||||||
.to(mail.parse().unwrap())
|
.to(mail.parse().unwrap())
|
||||||
.subject("Skynet-Discord: Link Wolves.")
|
.subject("Skynet: Link Discord to Wolves.")
|
||||||
.multipart(
|
.multipart(
|
||||||
// This is composed of two parts.
|
// This is composed of two parts.
|
||||||
// also helps not trip spam settings (uneven number of url's
|
// also helps not trip spam settings (uneven number of url's
|
||||||
|
@ -184,7 +210,7 @@ pub(crate) mod link {
|
||||||
let creds = Credentials::new(config.mail_user.clone(), config.mail_pass.clone());
|
let creds = Credentials::new(config.mail_user.clone(), config.mail_pass.clone());
|
||||||
|
|
||||||
// Open a remote connection to gmail using STARTTLS
|
// Open a remote connection to gmail using STARTTLS
|
||||||
let mailer = SmtpTransport::starttls_relay(&config.mail_smtp).unwrap().credentials(creds).build();
|
let mailer = SmtpTransport::starttls_relay(&config.mail_smtp)?.credentials(creds).build();
|
||||||
|
|
||||||
// Send the email
|
// Send the email
|
||||||
mailer.send(&email)
|
mailer.send(&email)
|
||||||
|
@ -212,7 +238,7 @@ pub(crate) mod link {
|
||||||
WHERE discord = ?
|
WHERE discord = ?
|
||||||
"#,
|
"#,
|
||||||
)
|
)
|
||||||
.bind(*user.as_u64() as i64)
|
.bind(user.get() as i64)
|
||||||
.fetch_one(db)
|
.fetch_one(db)
|
||||||
.await
|
.await
|
||||||
.ok()
|
.ok()
|
||||||
|
@ -226,45 +252,57 @@ pub(crate) mod link {
|
||||||
",
|
",
|
||||||
)
|
)
|
||||||
.bind(record.email.to_owned())
|
.bind(record.email.to_owned())
|
||||||
.bind(*user.as_u64() as i64)
|
.bind(user.get() as i64)
|
||||||
.bind(auth.to_owned())
|
.bind(auth.to_owned())
|
||||||
.bind(get_now_iso(false))
|
.bind(get_now_iso(false))
|
||||||
.fetch_optional(db)
|
.fetch_optional(db)
|
||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn save_to_db_user(db: &Pool<Sqlite>, id_wolves: i64, email: &str) -> Result<Option<Wolves>, sqlx::Error> {
|
||||||
|
sqlx::query_as::<_, Wolves>(
|
||||||
|
"
|
||||||
|
INSERT INTO wolves (id_wolves, email)
|
||||||
|
VALUES ($1, $2)
|
||||||
|
ON CONFLICT(id_wolves) DO UPDATE SET email = $2
|
||||||
|
",
|
||||||
|
)
|
||||||
|
.bind(id_wolves)
|
||||||
|
.bind(email)
|
||||||
|
.fetch_optional(db)
|
||||||
|
.await
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) mod verify {
|
pub mod verify {
|
||||||
use super::*;
|
use super::*;
|
||||||
use crate::commands::link_email::link::{db_pending_clear_expired, get_verify_from_db};
|
use crate::commands::link_email::link::{db_pending_clear_expired, get_server_member_discord, get_verify_from_db};
|
||||||
|
use serenity::all::{CommandDataOption, CommandDataOptionValue, CommandInteraction, CommandOptionType, CreateCommandOption, GuildId, RoleId};
|
||||||
use serenity::model::user::User;
|
use serenity::model::user::User;
|
||||||
use skynet_discord_bot::{get_server_config, ServerMembersWolves, Servers};
|
use skynet_discord_bot::common::database::get_server_config;
|
||||||
|
use skynet_discord_bot::common::database::{ServerMembersWolves, Servers};
|
||||||
|
use skynet_discord_bot::common::wolves::committees::Committees;
|
||||||
use sqlx::Error;
|
use sqlx::Error;
|
||||||
|
|
||||||
pub async fn run(command: &ApplicationCommandInteraction, ctx: &Context) -> String {
|
pub async fn run(command: &CommandInteraction, ctx: &Context) -> String {
|
||||||
let db_lock = {
|
let db_lock = {
|
||||||
let data_read = ctx.data.read().await;
|
let data_read = ctx.data.read().await;
|
||||||
data_read.get::<DataBase>().expect("Expected Databse in TypeMap.").clone()
|
data_read.get::<DataBase>().expect("Expected Databse in TypeMap.").clone()
|
||||||
};
|
};
|
||||||
let db = db_lock.read().await;
|
let db = db_lock.read().await;
|
||||||
|
|
||||||
// check if user has used /link
|
// check if user has used /link_wolves
|
||||||
let details = if let Some(x) = get_verify_from_db(&db, &command.user.id).await {
|
let details = if let Some(x) = get_verify_from_db(&db, &command.user.id).await {
|
||||||
x
|
x
|
||||||
} else {
|
} else {
|
||||||
return "Please use /link first".to_string();
|
return "Please use /link_wolves first".to_string();
|
||||||
};
|
};
|
||||||
|
|
||||||
let option = command
|
let code = if let Some(CommandDataOption {
|
||||||
.data
|
value: CommandDataOptionValue::String(code),
|
||||||
.options
|
..
|
||||||
.get(0)
|
}) = command.data.options.first()
|
||||||
.expect("Expected code option")
|
{
|
||||||
.resolved
|
|
||||||
.as_ref()
|
|
||||||
.expect("Expected code object");
|
|
||||||
|
|
||||||
let code = if let CommandDataOptionValue::String(code) = option {
|
|
||||||
code
|
code
|
||||||
} else {
|
} else {
|
||||||
return "Please provide a verification code".to_string();
|
return "Please provide a verification code".to_string();
|
||||||
|
@ -282,11 +320,15 @@ pub(crate) mod verify {
|
||||||
Ok(_) => {
|
Ok(_) => {
|
||||||
// get teh right roles for the user
|
// get teh right roles for the user
|
||||||
set_server_roles(&db, &command.user, ctx).await;
|
set_server_roles(&db, &command.user, ctx).await;
|
||||||
|
|
||||||
|
// check if they are a committee member, and on that server
|
||||||
|
set_server_roles_committee(&db, &command.user, ctx).await;
|
||||||
|
|
||||||
"Discord username linked to Wolves".to_string()
|
"Discord username linked to Wolves".to_string()
|
||||||
}
|
}
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
println!("{:?}", e);
|
println!("{:?}", e);
|
||||||
"Failed to save, please try /link again".to_string()
|
"Failed to save, please try /link_wolves again".to_string()
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
@ -296,14 +338,10 @@ pub(crate) mod verify {
|
||||||
"Failed to verify".to_string()
|
"Failed to verify".to_string()
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn register(command: &mut CreateApplicationCommand) -> &mut CreateApplicationCommand {
|
pub fn register() -> CreateCommand {
|
||||||
command.name("verify").description("Verify Wolves Email").create_option(|option| {
|
CreateCommand::new("verify")
|
||||||
option
|
.description("Verify Wolves Email")
|
||||||
.name("code")
|
.add_option(CreateCommandOption::new(CommandOptionType::String, "code", "Code from verification email").required(true))
|
||||||
.description("Code from verification email")
|
|
||||||
.kind(CommandOptionType::String)
|
|
||||||
.required(true)
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn db_pending_clear_successful(pool: &Pool<Sqlite>, user: &UserId) -> Result<Option<WolvesVerify>, Error> {
|
async fn db_pending_clear_successful(pool: &Pool<Sqlite>, user: &UserId) -> Result<Option<WolvesVerify>, Error> {
|
||||||
|
@ -314,7 +352,7 @@ pub(crate) mod verify {
|
||||||
WHERE discord = ?
|
WHERE discord = ?
|
||||||
"#,
|
"#,
|
||||||
)
|
)
|
||||||
.bind(*user.as_u64() as i64)
|
.bind(user.get() as i64)
|
||||||
.fetch_optional(pool)
|
.fetch_optional(pool)
|
||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
|
@ -327,7 +365,7 @@ pub(crate) mod verify {
|
||||||
WHERE email = ?
|
WHERE email = ?
|
||||||
",
|
",
|
||||||
)
|
)
|
||||||
.bind(*discord.as_u64() as i64)
|
.bind(discord.get() as i64)
|
||||||
.bind(email)
|
.bind(email)
|
||||||
.fetch_optional(db)
|
.fetch_optional(db)
|
||||||
.await
|
.await
|
||||||
|
@ -336,7 +374,7 @@ pub(crate) mod verify {
|
||||||
async fn set_server_roles(db: &Pool<Sqlite>, discord: &User, ctx: &Context) {
|
async fn set_server_roles(db: &Pool<Sqlite>, discord: &User, ctx: &Context) {
|
||||||
if let Ok(servers) = get_servers(db, &discord.id).await {
|
if let Ok(servers) = get_servers(db, &discord.id).await {
|
||||||
for server in servers {
|
for server in servers {
|
||||||
if let Ok(mut member) = server.server.member(&ctx.http, &discord.id).await {
|
if let Ok(member) = server.server.member(&ctx.http, &discord.id).await {
|
||||||
if let Some(config) = get_server_config(db, &server.server).await {
|
if let Some(config) = get_server_config(db, &server.server).await {
|
||||||
let Servers {
|
let Servers {
|
||||||
role_past,
|
role_past,
|
||||||
|
@ -352,10 +390,8 @@ pub(crate) mod verify {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Some(role) = &role_current {
|
if !member.roles.contains(&role_current) {
|
||||||
if !member.roles.contains(role) {
|
roles.push(role_current.to_owned());
|
||||||
roles.push(role.to_owned());
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Err(e) = member.add_roles(&ctx, &roles).await {
|
if let Err(e) = member.add_roles(&ctx, &roles).await {
|
||||||
|
@ -367,6 +403,38 @@ pub(crate) mod verify {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn get_committees_id(db: &Pool<Sqlite>, wolves_id: i64) -> Vec<Committees> {
|
||||||
|
sqlx::query_as::<_, Committees>(
|
||||||
|
r#"
|
||||||
|
SELECT *
|
||||||
|
FROM committees
|
||||||
|
WHERE committee LIKE ?1
|
||||||
|
"#,
|
||||||
|
)
|
||||||
|
.bind(format!("%{}%", wolves_id))
|
||||||
|
.fetch_all(db)
|
||||||
|
.await
|
||||||
|
.unwrap_or_else(|e| {
|
||||||
|
dbg!(e);
|
||||||
|
vec![]
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn set_server_roles_committee(db: &Pool<Sqlite>, discord: &User, ctx: &Context) {
|
||||||
|
if let Some(x) = get_server_member_discord(db, &discord.id).await {
|
||||||
|
// if they are a member of one or more committees, and in teh committee server then give the teh general committee role
|
||||||
|
// they will get teh more specific vanity role later
|
||||||
|
if !get_committees_id(db, x.id_wolves).await.is_empty() {
|
||||||
|
let server = GuildId::new(1220150752656363520);
|
||||||
|
let committee_member = RoleId::new(1226602779968274573);
|
||||||
|
|
||||||
|
if let Ok(member) = server.member(ctx, &discord.id).await {
|
||||||
|
member.add_roles(&ctx, &[committee_member]).await.unwrap_or_default();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
async fn get_servers(db: &Pool<Sqlite>, discord: &UserId) -> Result<Vec<ServerMembersWolves>, Error> {
|
async fn get_servers(db: &Pool<Sqlite>, discord: &UserId) -> Result<Vec<ServerMembersWolves>, Error> {
|
||||||
sqlx::query_as::<_, ServerMembersWolves>(
|
sqlx::query_as::<_, ServerMembersWolves>(
|
||||||
"
|
"
|
||||||
|
@ -376,7 +444,7 @@ pub(crate) mod verify {
|
||||||
WHERE discord = ?
|
WHERE discord = ?
|
||||||
",
|
",
|
||||||
)
|
)
|
||||||
.bind(*discord.as_u64() as i64)
|
.bind(discord.get() as i64)
|
||||||
.fetch_all(db)
|
.fetch_all(db)
|
||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
|
|
389
src/commands/minecraft.rs
Normal file
389
src/commands/minecraft.rs
Normal file
|
@ -0,0 +1,389 @@
|
||||||
|
use serenity::{builder::CreateCommand, client::Context};
|
||||||
|
|
||||||
|
use skynet_discord_bot::common::database::DataBase;
|
||||||
|
use sqlx::{Pool, Sqlite};
|
||||||
|
|
||||||
|
pub(crate) mod user {
|
||||||
|
use super::*;
|
||||||
|
pub(crate) mod add {
|
||||||
|
use super::*;
|
||||||
|
use crate::commands::link_email::link::get_server_member_discord;
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
use serenity::all::{CommandDataOption, CommandDataOptionValue, CommandInteraction, CommandOptionType, CreateCommandOption};
|
||||||
|
use serenity::model::id::UserId;
|
||||||
|
use skynet_discord_bot::common::database::Wolves;
|
||||||
|
use skynet_discord_bot::common::minecraft::{whitelist_update, Minecraft};
|
||||||
|
use skynet_discord_bot::Config;
|
||||||
|
use sqlx::Error;
|
||||||
|
|
||||||
|
pub fn register() -> CreateCommand {
|
||||||
|
CreateCommand::new("link_minecraft")
|
||||||
|
.description("Link your minecraft account")
|
||||||
|
.add_option(CreateCommandOption::new(CommandOptionType::String, "minecraft_username", "Your Minecraft username").required(true))
|
||||||
|
.add_option(CreateCommandOption::new(CommandOptionType::Boolean, "bedrock_account", "Is this a Bedrock account?").required(false))
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn run(command: &CommandInteraction, ctx: &Context) -> String {
|
||||||
|
let db_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<DataBase>().expect("Expected Databse in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
let db = db_lock.read().await;
|
||||||
|
|
||||||
|
let config_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<Config>().expect("Expected Config in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
let config = config_lock.read().await;
|
||||||
|
|
||||||
|
// user has to have previously linked with wolves
|
||||||
|
if get_server_member_discord(&db, &command.user.id).await.is_none() {
|
||||||
|
return "Not linked with wolves, please use ``/link_wolves`` with your wolves email.".to_string();
|
||||||
|
}
|
||||||
|
|
||||||
|
let username = if let Some(CommandDataOption {
|
||||||
|
value: CommandDataOptionValue::String(username),
|
||||||
|
..
|
||||||
|
}) = command.data.options.first()
|
||||||
|
{
|
||||||
|
username.trim()
|
||||||
|
} else {
|
||||||
|
return "Please provide a valid username".to_string();
|
||||||
|
};
|
||||||
|
|
||||||
|
// this is always true unless they state its not
|
||||||
|
let java = if let Some(CommandDataOption {
|
||||||
|
value: CommandDataOptionValue::Boolean(z),
|
||||||
|
..
|
||||||
|
}) = command.data.options.get(1)
|
||||||
|
{
|
||||||
|
!z
|
||||||
|
} else {
|
||||||
|
true
|
||||||
|
};
|
||||||
|
|
||||||
|
let username_mc;
|
||||||
|
if java {
|
||||||
|
// insert the username into the database
|
||||||
|
match add_minecraft(&db, &command.user.id, username).await {
|
||||||
|
Ok(_) => {}
|
||||||
|
Err(e) => {
|
||||||
|
dbg!("{:?}", e);
|
||||||
|
return format!("Failure to minecraft username {:?}", username);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
username_mc = username.to_string();
|
||||||
|
} else {
|
||||||
|
match get_minecraft_bedrock(username, &config.minecraft_mcprofile).await {
|
||||||
|
None => {
|
||||||
|
return format!("No UID found for {:?}", username);
|
||||||
|
}
|
||||||
|
Some(x) => {
|
||||||
|
match add_minecraft_bedrock(&db, &command.user.id, &x.floodgateuid).await {
|
||||||
|
Ok(_) => {}
|
||||||
|
Err(e) => {
|
||||||
|
dbg!("{:?}", e);
|
||||||
|
return format!("Failure to minecraft UID {:?}", &x.floodgateuid);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
username_mc = x.floodgateuid;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// get a list of servers that the user is a member of
|
||||||
|
if let Ok(servers) = get_servers(&db, &command.user.id).await {
|
||||||
|
for server in servers {
|
||||||
|
whitelist_update(&vec![(username_mc.to_owned(), java)], &server.minecraft, &config.discord_token_minecraft).await;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
"Added/Updated minecraft_user info".to_string()
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn add_minecraft(db: &Pool<Sqlite>, user: &UserId, minecraft: &str) -> Result<Option<Wolves>, Error> {
|
||||||
|
sqlx::query_as::<_, Wolves>(
|
||||||
|
"
|
||||||
|
UPDATE wolves
|
||||||
|
SET minecraft = ?2
|
||||||
|
WHERE discord = ?1;
|
||||||
|
",
|
||||||
|
)
|
||||||
|
.bind(user.get() as i64)
|
||||||
|
.bind(minecraft)
|
||||||
|
.fetch_optional(db)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, Debug)]
|
||||||
|
struct BedrockDetails {
|
||||||
|
pub gamertag: String,
|
||||||
|
pub xuid: String,
|
||||||
|
pub floodgateuid: String,
|
||||||
|
pub icon: String,
|
||||||
|
pub gamescore: String,
|
||||||
|
pub accounttier: String,
|
||||||
|
pub textureid: String,
|
||||||
|
pub skin: String,
|
||||||
|
pub linked: bool,
|
||||||
|
pub java_uuid: String,
|
||||||
|
pub java_name: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn get_minecraft_bedrock(username: &str, api_key: &str) -> Option<BedrockDetails> {
|
||||||
|
let url = format!("https://mcprofile.io/api/v1/bedrock/gamertag/{username}/");
|
||||||
|
match surf::get(url)
|
||||||
|
.header("x-api-key", api_key)
|
||||||
|
.header("User-Agent", "UL Computer Society")
|
||||||
|
.recv_json()
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(res) => Some(res),
|
||||||
|
Err(e) => {
|
||||||
|
dbg!(e);
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn add_minecraft_bedrock(db: &Pool<Sqlite>, user: &UserId, minecraft: &str) -> Result<Option<Wolves>, Error> {
|
||||||
|
sqlx::query_as::<_, Wolves>(
|
||||||
|
"
|
||||||
|
UPDATE wolves
|
||||||
|
SET minecraft_uid = ?2
|
||||||
|
WHERE discord = ?1;
|
||||||
|
",
|
||||||
|
)
|
||||||
|
.bind(user.get() as i64)
|
||||||
|
.bind(minecraft)
|
||||||
|
.fetch_optional(db)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn get_servers(db: &Pool<Sqlite>, discord: &UserId) -> Result<Vec<Minecraft>, Error> {
|
||||||
|
sqlx::query_as::<_, Minecraft>(
|
||||||
|
"
|
||||||
|
SELECT minecraft.*
|
||||||
|
FROM minecraft
|
||||||
|
JOIN (
|
||||||
|
SELECT server
|
||||||
|
FROM server_members
|
||||||
|
JOIN wolves USING (id_wolves)
|
||||||
|
WHERE discord = ?1
|
||||||
|
) sub on minecraft.server_discord = sub.server
|
||||||
|
",
|
||||||
|
)
|
||||||
|
.bind(discord.get() as i64)
|
||||||
|
.fetch_all(db)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) mod server {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
pub(crate) mod add {
|
||||||
|
use serenity::all::{CommandDataOption, CommandDataOptionValue, CommandInteraction, CommandOptionType, CreateCommand, CreateCommandOption};
|
||||||
|
use serenity::model::id::GuildId;
|
||||||
|
use sqlx::Error;
|
||||||
|
// this is to managfe the server side of commands related to minecraft
|
||||||
|
use super::*;
|
||||||
|
use skynet_discord_bot::common::minecraft::update_server;
|
||||||
|
use skynet_discord_bot::common::minecraft::Minecraft;
|
||||||
|
use skynet_discord_bot::Config;
|
||||||
|
|
||||||
|
pub fn register() -> CreateCommand {
|
||||||
|
CreateCommand::new("minecraft_add")
|
||||||
|
.description("Add a minecraft server")
|
||||||
|
.default_member_permissions(serenity::model::Permissions::MANAGE_GUILD)
|
||||||
|
.add_option(
|
||||||
|
CreateCommandOption::new(CommandOptionType::String, "server_id", "ID of the Minecraft server hosted by the Computer Society")
|
||||||
|
.required(true),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn run(command: &CommandInteraction, ctx: &Context) -> String {
|
||||||
|
let g_id = match command.guild_id {
|
||||||
|
None => return "Not in a server".to_string(),
|
||||||
|
Some(x) => x,
|
||||||
|
};
|
||||||
|
|
||||||
|
let server_minecraft = if let Some(CommandDataOption {
|
||||||
|
value: CommandDataOptionValue::String(id),
|
||||||
|
..
|
||||||
|
}) = command.data.options.first()
|
||||||
|
{
|
||||||
|
id.to_string()
|
||||||
|
} else {
|
||||||
|
return String::from("Expected Server ID");
|
||||||
|
};
|
||||||
|
|
||||||
|
let db_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<DataBase>().expect("Expected Databse in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
let db = db_lock.read().await;
|
||||||
|
|
||||||
|
match add_server(&db, &g_id, &server_minecraft).await {
|
||||||
|
Ok(_) => {}
|
||||||
|
Err(e) => {
|
||||||
|
println!("{:?}", e);
|
||||||
|
return format!("Failure to insert into Minecraft {} {}", &g_id, &server_minecraft);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let config_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<Config>().expect("Expected Config in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
let config = config_lock.read().await;
|
||||||
|
|
||||||
|
update_server(&server_minecraft, &db, &g_id, &config).await;
|
||||||
|
|
||||||
|
"Added/Updated minecraft_server info".to_string()
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn add_server(db: &Pool<Sqlite>, discord: &GuildId, minecraft: &str) -> Result<Option<Minecraft>, Error> {
|
||||||
|
sqlx::query_as::<_, Minecraft>(
|
||||||
|
"
|
||||||
|
INSERT OR REPLACE INTO minecraft (server_discord, server_minecraft)
|
||||||
|
VALUES (?1, ?2)
|
||||||
|
",
|
||||||
|
)
|
||||||
|
.bind(discord.get() as i64)
|
||||||
|
.bind(minecraft)
|
||||||
|
.fetch_optional(db)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) mod list {
|
||||||
|
use serenity::all::CommandInteraction;
|
||||||
|
use serenity::builder::CreateCommand;
|
||||||
|
use serenity::client::Context;
|
||||||
|
use skynet_discord_bot::common::database::DataBase;
|
||||||
|
use skynet_discord_bot::common::minecraft::{get_minecraft_config_server, server_information};
|
||||||
|
use skynet_discord_bot::Config;
|
||||||
|
|
||||||
|
pub fn register() -> CreateCommand {
|
||||||
|
CreateCommand::new("minecraft_list")
|
||||||
|
.default_member_permissions(serenity::model::Permissions::MANAGE_GUILD)
|
||||||
|
.description("List your minecraft servers")
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn run(command: &CommandInteraction, ctx: &Context) -> String {
|
||||||
|
let g_id = match command.guild_id {
|
||||||
|
None => return "Not in a server".to_string(),
|
||||||
|
Some(x) => x,
|
||||||
|
};
|
||||||
|
|
||||||
|
let db_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<DataBase>().expect("Expected Databse in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
let db = db_lock.read().await;
|
||||||
|
|
||||||
|
let servers = get_minecraft_config_server(&db, g_id).await;
|
||||||
|
|
||||||
|
if servers.is_empty() {
|
||||||
|
return "No minecraft servers, use /minecraft_add to add one".to_string();
|
||||||
|
}
|
||||||
|
|
||||||
|
let config_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<Config>().expect("Expected Config in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
let config = config_lock.read().await;
|
||||||
|
|
||||||
|
let mut result = "Server Information:\n".to_string();
|
||||||
|
for server in get_minecraft_config_server(&db, g_id).await {
|
||||||
|
if let Some(x) = server_information(&server.minecraft, &config.discord_token_minecraft).await {
|
||||||
|
result.push_str(&format!(
|
||||||
|
r#"
|
||||||
|
Name: {name}
|
||||||
|
ID: {id}
|
||||||
|
Online: {online}
|
||||||
|
Info: {description}
|
||||||
|
Link: <https://panel.games.skynet.ie/server/{id}>
|
||||||
|
"#,
|
||||||
|
name = &x.attributes.name,
|
||||||
|
online = !x.attributes.is_suspended,
|
||||||
|
description = &x.attributes.description,
|
||||||
|
id = &x.attributes.identifier
|
||||||
|
));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
result.to_string()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) mod delete {
|
||||||
|
use serenity::all::{CommandDataOption, CommandDataOptionValue, CommandInteraction, CommandOptionType, CreateCommandOption};
|
||||||
|
use serenity::builder::CreateCommand;
|
||||||
|
use serenity::client::Context;
|
||||||
|
use serenity::model::id::GuildId;
|
||||||
|
use skynet_discord_bot::common::database::DataBase;
|
||||||
|
use skynet_discord_bot::common::minecraft::Minecraft;
|
||||||
|
use sqlx::{Error, Pool, Sqlite};
|
||||||
|
|
||||||
|
pub fn register() -> CreateCommand {
|
||||||
|
CreateCommand::new("minecraft_delete")
|
||||||
|
.description("Delete a minecraft server")
|
||||||
|
.default_member_permissions(serenity::model::Permissions::MANAGE_GUILD)
|
||||||
|
.add_option(
|
||||||
|
CreateCommandOption::new(CommandOptionType::String, "server_id", "ID of the Minecraft server hosted by the Computer Society")
|
||||||
|
.required(true),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn run(command: &CommandInteraction, ctx: &Context) -> String {
|
||||||
|
let g_id = match command.guild_id {
|
||||||
|
None => return "Not in a server".to_string(),
|
||||||
|
Some(x) => x,
|
||||||
|
};
|
||||||
|
|
||||||
|
let server_minecraft = if let Some(CommandDataOption {
|
||||||
|
value: CommandDataOptionValue::String(id),
|
||||||
|
..
|
||||||
|
}) = command.data.options.first()
|
||||||
|
{
|
||||||
|
id.to_string()
|
||||||
|
} else {
|
||||||
|
return String::from("Expected Server ID");
|
||||||
|
};
|
||||||
|
|
||||||
|
let db_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<DataBase>().expect("Expected Databse in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
let db = db_lock.read().await;
|
||||||
|
|
||||||
|
match server_remove(&db, &g_id, &server_minecraft).await {
|
||||||
|
Ok(_) => {}
|
||||||
|
Err(e) => {
|
||||||
|
println!("{:?}", e);
|
||||||
|
return format!("Failure to insert into Minecraft {} {}", &g_id, &server_minecraft);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// no need to clear teh whitelist as it will be reset within 24hr anyways
|
||||||
|
|
||||||
|
"Removed minecraft_server info".to_string()
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn server_remove(db: &Pool<Sqlite>, discord: &GuildId, minecraft: &str) -> Result<Option<Minecraft>, Error> {
|
||||||
|
sqlx::query_as::<_, Minecraft>(
|
||||||
|
"
|
||||||
|
DELETE FROM minecraft
|
||||||
|
WHERE server_discord = ?1 AND server_minecraft = ?2
|
||||||
|
",
|
||||||
|
)
|
||||||
|
.bind(discord.get() as i64)
|
||||||
|
.bind(minecraft)
|
||||||
|
.fetch_optional(db)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -1,2 +1,5 @@
|
||||||
pub mod add_server;
|
pub mod add_server;
|
||||||
|
pub mod count;
|
||||||
pub mod link_email;
|
pub mod link_email;
|
||||||
|
pub mod minecraft;
|
||||||
|
pub mod role_adder;
|
||||||
|
|
198
src/commands/role_adder.rs
Normal file
198
src/commands/role_adder.rs
Normal file
|
@ -0,0 +1,198 @@
|
||||||
|
use serenity::client::Context;
|
||||||
|
|
||||||
|
use skynet_discord_bot::common::database::{DataBase, RoleAdder};
|
||||||
|
use sqlx::{Error, Pool, Sqlite};
|
||||||
|
|
||||||
|
pub mod edit {
|
||||||
|
use super::*;
|
||||||
|
use serenity::all::{CommandDataOption, CommandDataOptionValue, CommandInteraction, CommandOptionType, CreateCommand, CreateCommandOption};
|
||||||
|
|
||||||
|
pub async fn run(command: &CommandInteraction, ctx: &Context) -> String {
|
||||||
|
let role_a = if let Some(CommandDataOption {
|
||||||
|
value: CommandDataOptionValue::Role(role),
|
||||||
|
..
|
||||||
|
}) = command.data.options.first()
|
||||||
|
{
|
||||||
|
role.to_owned()
|
||||||
|
} else {
|
||||||
|
return "Please provide a valid role for ``Role Current``".to_string();
|
||||||
|
};
|
||||||
|
|
||||||
|
let role_b = if let Some(CommandDataOption {
|
||||||
|
value: CommandDataOptionValue::Role(role),
|
||||||
|
..
|
||||||
|
}) = command.data.options.get(1)
|
||||||
|
{
|
||||||
|
role.to_owned()
|
||||||
|
} else {
|
||||||
|
return "Please provide a valid role for ``Role Current``".to_string();
|
||||||
|
};
|
||||||
|
|
||||||
|
let role_c = if let Some(CommandDataOption {
|
||||||
|
value: CommandDataOptionValue::Role(role),
|
||||||
|
..
|
||||||
|
}) = command.data.options.get(2)
|
||||||
|
{
|
||||||
|
role.to_owned()
|
||||||
|
} else {
|
||||||
|
return "Please provide a valid role for ``Role Current``".to_string();
|
||||||
|
};
|
||||||
|
|
||||||
|
if role_a == role_b {
|
||||||
|
return "Roles A and B must be different".to_string();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (role_c == role_a) || (role_c == role_b) {
|
||||||
|
return "Role C cannot be same as A or B".to_string();
|
||||||
|
}
|
||||||
|
|
||||||
|
let delete = if let Some(CommandDataOption {
|
||||||
|
value: CommandDataOptionValue::Boolean(z),
|
||||||
|
..
|
||||||
|
}) = command.data.options.get(3)
|
||||||
|
{
|
||||||
|
*z
|
||||||
|
} else {
|
||||||
|
false
|
||||||
|
};
|
||||||
|
|
||||||
|
let db_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<DataBase>().expect("Expected Databse in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
let db = db_lock.read().await;
|
||||||
|
|
||||||
|
let server = command.guild_id.unwrap_or_default();
|
||||||
|
let server_data = RoleAdder {
|
||||||
|
server,
|
||||||
|
role_a,
|
||||||
|
role_b,
|
||||||
|
role_c,
|
||||||
|
};
|
||||||
|
|
||||||
|
match add_server(&db, &server_data, delete).await {
|
||||||
|
Ok(_) => {}
|
||||||
|
Err(e) => {
|
||||||
|
println!("{:?}", e);
|
||||||
|
return format!("Failure to insert into Servers {:?}", server_data);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut role_a_name = String::new();
|
||||||
|
let mut role_b_name = String::new();
|
||||||
|
let mut role_c_name = String::new();
|
||||||
|
|
||||||
|
if let Ok(x) = server.roles(&ctx).await {
|
||||||
|
if let Some(y) = x.get(&role_a) {
|
||||||
|
role_a_name = y.to_owned().name;
|
||||||
|
}
|
||||||
|
if let Some(y) = x.get(&role_b) {
|
||||||
|
role_b_name = y.to_owned().name;
|
||||||
|
}
|
||||||
|
if let Some(y) = x.get(&role_b) {
|
||||||
|
role_c_name = y.to_owned().name;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if delete {
|
||||||
|
format!("Removed {} + {} = {}", role_a_name, role_b_name, role_c_name)
|
||||||
|
} else {
|
||||||
|
format!("Added {} + {} = {}", role_a_name, role_b_name, role_c_name)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn register() -> CreateCommand {
|
||||||
|
CreateCommand::new("roles_adder")
|
||||||
|
.description("Combine roles together to an new one")
|
||||||
|
.default_member_permissions(serenity::model::Permissions::MANAGE_GUILD)
|
||||||
|
.add_option(CreateCommandOption::new(CommandOptionType::Role, "role_a", "A role you want to add to Role B").required(true))
|
||||||
|
.add_option(CreateCommandOption::new(CommandOptionType::Role, "role_b", "A role you want to add to Role A").required(true))
|
||||||
|
.add_option(CreateCommandOption::new(CommandOptionType::Role, "role_c", "Sum of A and B").required(true))
|
||||||
|
.add_option(CreateCommandOption::new(CommandOptionType::Boolean, "delete", "Delete this entry.").required(false))
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn add_server(db: &Pool<Sqlite>, server: &RoleAdder, delete: bool) -> Result<Option<RoleAdder>, Error> {
|
||||||
|
if delete {
|
||||||
|
sqlx::query_as::<_, RoleAdder>(
|
||||||
|
"
|
||||||
|
DELETE FROM roles_adder
|
||||||
|
WHERE server = ?1 AND role_a = ?2 AND role_b = ?3 AND role_c = ?4
|
||||||
|
",
|
||||||
|
)
|
||||||
|
.bind(server.server.get() as i64)
|
||||||
|
.bind(server.role_a.get() as i64)
|
||||||
|
.bind(server.role_b.get() as i64)
|
||||||
|
.bind(server.role_c.get() as i64)
|
||||||
|
.fetch_optional(db)
|
||||||
|
.await
|
||||||
|
} else {
|
||||||
|
sqlx::query_as::<_, RoleAdder>(
|
||||||
|
"
|
||||||
|
INSERT OR REPLACE INTO roles_adder (server, role_a, role_b, role_c)
|
||||||
|
VALUES (?1, ?2, ?3, ?4)
|
||||||
|
",
|
||||||
|
)
|
||||||
|
.bind(server.server.get() as i64)
|
||||||
|
.bind(server.role_a.get() as i64)
|
||||||
|
.bind(server.role_b.get() as i64)
|
||||||
|
.bind(server.role_c.get() as i64)
|
||||||
|
.fetch_optional(db)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO
|
||||||
|
pub mod list {}
|
||||||
|
|
||||||
|
pub mod tools {
|
||||||
|
use serenity::client::Context;
|
||||||
|
use serenity::model::guild::Member;
|
||||||
|
use skynet_discord_bot::common::database::RoleAdder;
|
||||||
|
use sqlx::{Pool, Sqlite};
|
||||||
|
|
||||||
|
pub async fn on_role_change(db: &Pool<Sqlite>, ctx: &Context, new_data: Member) {
|
||||||
|
// check if the role changed is part of the oens for this server
|
||||||
|
if let Ok(role_adders) = sqlx::query_as::<_, RoleAdder>(
|
||||||
|
r#"
|
||||||
|
SELECT *
|
||||||
|
FROM roles_adder
|
||||||
|
WHERE server = ?
|
||||||
|
"#,
|
||||||
|
)
|
||||||
|
.bind(new_data.guild_id.get() as i64)
|
||||||
|
.fetch_all(db)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
let mut roles_add = vec![];
|
||||||
|
let mut roles_remove = vec![];
|
||||||
|
|
||||||
|
for role_adder in role_adders {
|
||||||
|
// if the user has both A dnd B give them C
|
||||||
|
if new_data.roles.contains(&role_adder.role_a) && new_data.roles.contains(&role_adder.role_b) && !new_data.roles.contains(&role_adder.role_c)
|
||||||
|
{
|
||||||
|
roles_add.push(role_adder.role_c);
|
||||||
|
}
|
||||||
|
|
||||||
|
// If the suer has C but not A or B remove C
|
||||||
|
if new_data.roles.contains(&role_adder.role_c)
|
||||||
|
&& (!new_data.roles.contains(&role_adder.role_a) || !new_data.roles.contains(&role_adder.role_b))
|
||||||
|
{
|
||||||
|
roles_remove.push(role_adder.role_c);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !roles_add.is_empty() {
|
||||||
|
if let Err(e) = new_data.add_roles(&ctx, &roles_add).await {
|
||||||
|
println!("{:?}", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !roles_remove.is_empty() {
|
||||||
|
if let Err(e) = new_data.remove_roles(&ctx, &roles_remove).await {
|
||||||
|
println!("{:?}", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
274
src/common/database.rs
Normal file
274
src/common/database.rs
Normal file
|
@ -0,0 +1,274 @@
|
||||||
|
use crate::Config;
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
use serenity::model::guild;
|
||||||
|
use serenity::model::id::{ChannelId, GuildId, RoleId, UserId};
|
||||||
|
use serenity::prelude::TypeMapKey;
|
||||||
|
use sqlx::sqlite::{SqliteConnectOptions, SqlitePoolOptions, SqliteRow};
|
||||||
|
use sqlx::{Error, FromRow, Pool, Row, Sqlite};
|
||||||
|
use std::str::FromStr;
|
||||||
|
use std::sync::Arc;
|
||||||
|
use tokio::sync::RwLock;
|
||||||
|
|
||||||
|
pub struct DataBase;
|
||||||
|
impl TypeMapKey for DataBase {
|
||||||
|
type Value = Arc<RwLock<Pool<Sqlite>>>;
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Deserialize, Serialize)]
|
||||||
|
pub struct ServerMembers {
|
||||||
|
pub server: GuildId,
|
||||||
|
pub id_wolves: i64,
|
||||||
|
pub expiry: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'r> FromRow<'r, SqliteRow> for ServerMembers {
|
||||||
|
fn from_row(row: &'r SqliteRow) -> Result<Self, Error> {
|
||||||
|
let server_tmp: i64 = row.try_get("server")?;
|
||||||
|
let server = GuildId::from(server_tmp as u64);
|
||||||
|
|
||||||
|
Ok(Self {
|
||||||
|
server,
|
||||||
|
id_wolves: row.try_get("id_wolves")?,
|
||||||
|
expiry: row.try_get("expiry")?,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Deserialize, Serialize)]
|
||||||
|
pub struct ServerMembersWolves {
|
||||||
|
pub server: GuildId,
|
||||||
|
pub id_wolves: i64,
|
||||||
|
pub expiry: String,
|
||||||
|
pub email: String,
|
||||||
|
pub discord: Option<UserId>,
|
||||||
|
pub minecraft: Option<String>,
|
||||||
|
pub minecraft_uid: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'r> FromRow<'r, SqliteRow> for ServerMembersWolves {
|
||||||
|
fn from_row(row: &'r SqliteRow) -> Result<Self, Error> {
|
||||||
|
let server_tmp: i64 = row.try_get("server")?;
|
||||||
|
let server = GuildId::from(server_tmp as u64);
|
||||||
|
|
||||||
|
Ok(Self {
|
||||||
|
server,
|
||||||
|
id_wolves: row.try_get("id_wolves")?,
|
||||||
|
expiry: row.try_get("expiry")?,
|
||||||
|
email: row.try_get("email")?,
|
||||||
|
discord: get_discord_from_row(row),
|
||||||
|
minecraft: row.try_get("minecraft")?,
|
||||||
|
minecraft_uid: row.try_get("minecraft_uid")?,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_discord_from_row(row: &SqliteRow) -> Option<UserId> {
|
||||||
|
match row.try_get("discord") {
|
||||||
|
Ok(x) => {
|
||||||
|
let tmp: i64 = x;
|
||||||
|
if tmp == 0 {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(UserId::from(tmp as u64))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Deserialize, Serialize)]
|
||||||
|
pub struct Wolves {
|
||||||
|
pub id_wolves: i64,
|
||||||
|
pub email: String,
|
||||||
|
pub discord: Option<UserId>,
|
||||||
|
pub minecraft: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'r> FromRow<'r, SqliteRow> for Wolves {
|
||||||
|
fn from_row(row: &'r SqliteRow) -> Result<Self, Error> {
|
||||||
|
Ok(Self {
|
||||||
|
id_wolves: row.try_get("id_wolves")?,
|
||||||
|
email: row.try_get("email")?,
|
||||||
|
discord: get_discord_from_row(row),
|
||||||
|
minecraft: row.try_get("minecraft")?,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Deserialize, Serialize)]
|
||||||
|
pub struct WolvesVerify {
|
||||||
|
pub email: String,
|
||||||
|
pub discord: UserId,
|
||||||
|
pub auth_code: String,
|
||||||
|
pub date_expiry: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'r> FromRow<'r, SqliteRow> for WolvesVerify {
|
||||||
|
fn from_row(row: &'r SqliteRow) -> Result<Self, Error> {
|
||||||
|
let user_tmp: i64 = row.try_get("discord")?;
|
||||||
|
let discord = UserId::from(user_tmp as u64);
|
||||||
|
|
||||||
|
Ok(Self {
|
||||||
|
email: row.try_get("email")?,
|
||||||
|
discord,
|
||||||
|
auth_code: row.try_get("auth_code")?,
|
||||||
|
date_expiry: row.try_get("date_expiry")?,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Deserialize, Serialize)]
|
||||||
|
pub struct Servers {
|
||||||
|
pub server: GuildId,
|
||||||
|
pub wolves_api: String,
|
||||||
|
pub wolves_id: i64,
|
||||||
|
pub role_past: Option<RoleId>,
|
||||||
|
pub role_current: RoleId,
|
||||||
|
pub member_past: i64,
|
||||||
|
pub member_current: i64,
|
||||||
|
pub bot_channel_id: ChannelId,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'r> FromRow<'r, SqliteRow> for Servers {
|
||||||
|
fn from_row(row: &'r SqliteRow) -> Result<Self, Error> {
|
||||||
|
let server_tmp: i64 = row.try_get("server")?;
|
||||||
|
let server = GuildId::from(server_tmp as u64);
|
||||||
|
let role_past = match row.try_get("role_past") {
|
||||||
|
Ok(x) => {
|
||||||
|
let tmp: i64 = x;
|
||||||
|
if tmp == 0 {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(RoleId::from(tmp as u64))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => None,
|
||||||
|
};
|
||||||
|
let role_current = match row.try_get("role_current") {
|
||||||
|
Ok(x) => {
|
||||||
|
let tmp: i64 = x;
|
||||||
|
RoleId::from(tmp as u64)
|
||||||
|
}
|
||||||
|
_ => RoleId::from(0u64),
|
||||||
|
};
|
||||||
|
|
||||||
|
let bot_channel_tmp: i64 = row.try_get("bot_channel_id")?;
|
||||||
|
let bot_channel_id = ChannelId::from(bot_channel_tmp as u64);
|
||||||
|
|
||||||
|
Ok(Self {
|
||||||
|
server,
|
||||||
|
wolves_api: row.try_get("wolves_api")?,
|
||||||
|
wolves_id: row.try_get("wolves_id").unwrap_or(0),
|
||||||
|
role_past,
|
||||||
|
role_current,
|
||||||
|
member_past: row.try_get("member_past")?,
|
||||||
|
member_current: row.try_get("member_current")?,
|
||||||
|
bot_channel_id,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Deserialize, Serialize)]
|
||||||
|
pub struct RoleAdder {
|
||||||
|
pub server: GuildId,
|
||||||
|
pub role_a: RoleId,
|
||||||
|
pub role_b: RoleId,
|
||||||
|
pub role_c: RoleId,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'r> FromRow<'r, SqliteRow> for RoleAdder {
|
||||||
|
fn from_row(row: &'r SqliteRow) -> Result<Self, Error> {
|
||||||
|
let server_tmp: i64 = row.try_get("server")?;
|
||||||
|
let server = GuildId::from(server_tmp as u64);
|
||||||
|
|
||||||
|
Ok(Self {
|
||||||
|
server,
|
||||||
|
role_a: get_role_from_row(row, "role_a"),
|
||||||
|
role_b: get_role_from_row(row, "role_b"),
|
||||||
|
role_c: get_role_from_row(row, "role_c"),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn get_role_from_row(row: &SqliteRow, col: &str) -> RoleId {
|
||||||
|
let id = match row.try_get(col) {
|
||||||
|
Ok(x) => {
|
||||||
|
let tmp: i64 = x;
|
||||||
|
tmp as u64
|
||||||
|
}
|
||||||
|
_ => 0,
|
||||||
|
};
|
||||||
|
|
||||||
|
RoleId::from(id)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn get_channel_from_row(row: &SqliteRow, col: &str) -> ChannelId {
|
||||||
|
let id = match row.try_get(col) {
|
||||||
|
Ok(x) => {
|
||||||
|
let tmp: i64 = x;
|
||||||
|
tmp as u64
|
||||||
|
}
|
||||||
|
_ => 0,
|
||||||
|
};
|
||||||
|
|
||||||
|
ChannelId::from(id)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn db_init(config: &Config) -> Result<Pool<Sqlite>, Error> {
|
||||||
|
let database = format!("{}/{}", &config.home, &config.database);
|
||||||
|
|
||||||
|
let pool = SqlitePoolOptions::new()
|
||||||
|
.max_connections(5)
|
||||||
|
.connect_with(
|
||||||
|
SqliteConnectOptions::from_str(&format!("sqlite://{}", database))?
|
||||||
|
.foreign_keys(true)
|
||||||
|
.create_if_missing(true),
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
// migrations are amazing!
|
||||||
|
sqlx::migrate!("./db/migrations").run(&pool).await?;
|
||||||
|
|
||||||
|
Ok(pool)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn get_server_config(db: &Pool<Sqlite>, server: &GuildId) -> Option<Servers> {
|
||||||
|
sqlx::query_as::<_, Servers>(
|
||||||
|
r#"
|
||||||
|
SELECT *
|
||||||
|
FROM servers
|
||||||
|
WHERE server = ?
|
||||||
|
"#,
|
||||||
|
)
|
||||||
|
.bind(server.get() as i64)
|
||||||
|
.fetch_one(db)
|
||||||
|
.await
|
||||||
|
.ok()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn get_server_member(db: &Pool<Sqlite>, server: &GuildId, member: &guild::Member) -> Result<ServerMembersWolves, Error> {
|
||||||
|
sqlx::query_as::<_, ServerMembersWolves>(
|
||||||
|
r#"
|
||||||
|
SELECT *
|
||||||
|
FROM server_members
|
||||||
|
JOIN wolves USING (id_wolves)
|
||||||
|
WHERE server = ? AND discord = ?
|
||||||
|
"#,
|
||||||
|
)
|
||||||
|
.bind(server.get() as i64)
|
||||||
|
.bind(member.user.id.get() as i64)
|
||||||
|
.fetch_one(db)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn get_server_config_bulk(db: &Pool<Sqlite>) -> Vec<Servers> {
|
||||||
|
sqlx::query_as::<_, Servers>(
|
||||||
|
r#"
|
||||||
|
SELECT *
|
||||||
|
FROM servers
|
||||||
|
"#,
|
||||||
|
)
|
||||||
|
.fetch_all(db)
|
||||||
|
.await
|
||||||
|
.unwrap_or_default()
|
||||||
|
}
|
174
src/common/minecraft.rs
Normal file
174
src/common/minecraft.rs
Normal file
|
@ -0,0 +1,174 @@
|
||||||
|
use crate::common::set_roles::normal::get_server_member_bulk;
|
||||||
|
use crate::Config;
|
||||||
|
use serde::de::DeserializeOwned;
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
use serenity::model::id::GuildId;
|
||||||
|
use sqlx::sqlite::SqliteRow;
|
||||||
|
use sqlx::{Error, FromRow, Pool, Row, Sqlite};
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Deserialize, Serialize)]
|
||||||
|
pub struct Minecraft {
|
||||||
|
pub discord: GuildId,
|
||||||
|
pub minecraft: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'r> FromRow<'r, SqliteRow> for Minecraft {
|
||||||
|
fn from_row(row: &'r SqliteRow) -> Result<Self, Error> {
|
||||||
|
let server_tmp: i64 = row.try_get("server_discord")?;
|
||||||
|
let discord = GuildId::from(server_tmp as u64);
|
||||||
|
|
||||||
|
Ok(Self {
|
||||||
|
discord,
|
||||||
|
minecraft: row.try_get("server_minecraft")?,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
loop through all members of server
|
||||||
|
get a list of folks with mc accounts that are members
|
||||||
|
and a list that arent members
|
||||||
|
*/
|
||||||
|
pub async fn update_server(server_id: &str, db: &Pool<Sqlite>, g_id: &GuildId, config: &Config) {
|
||||||
|
let mut usernames = vec![];
|
||||||
|
for member in get_server_member_bulk(db, g_id).await {
|
||||||
|
if let Some(x) = member.minecraft {
|
||||||
|
usernames.push((x, true));
|
||||||
|
}
|
||||||
|
if let Some(x) = member.minecraft_uid {
|
||||||
|
usernames.push((x, false));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !usernames.is_empty() {
|
||||||
|
whitelist_update(&usernames, server_id, &config.discord_token_minecraft).await;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn post<T: Serialize>(url: &str, bearer: &str, data: &T) {
|
||||||
|
match surf::post(url)
|
||||||
|
.header("Authorization", bearer)
|
||||||
|
.header("Content-Type", "application/json")
|
||||||
|
.header("Accept", "Application/vnd.pterodactyl.v1+json")
|
||||||
|
.body_json(&data)
|
||||||
|
{
|
||||||
|
Ok(req) => {
|
||||||
|
req.await.ok();
|
||||||
|
}
|
||||||
|
Err(e) => {
|
||||||
|
dbg!(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Deserialize, Serialize, Debug)]
|
||||||
|
pub struct ServerDetailsResSub {
|
||||||
|
pub identifier: String,
|
||||||
|
pub name: String,
|
||||||
|
pub description: String,
|
||||||
|
pub is_suspended: bool,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Deserialize, Serialize, Debug)]
|
||||||
|
pub struct ServerDetailsRes {
|
||||||
|
pub attributes: ServerDetailsResSub,
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn get<T: Serialize + DeserializeOwned>(url: &str, bearer: &str) -> Option<T> {
|
||||||
|
match surf::get(url)
|
||||||
|
.header("Authorization", bearer)
|
||||||
|
.header("Content-Type", "application/json")
|
||||||
|
.header("Accept", "Application/vnd.pterodactyl.v1+json")
|
||||||
|
.recv_json()
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(res) => Some(res),
|
||||||
|
Err(e) => {
|
||||||
|
dbg!(e);
|
||||||
|
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Deserialize, Serialize, Debug)]
|
||||||
|
struct BodyCommand {
|
||||||
|
command: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Deserialize, Serialize, Debug)]
|
||||||
|
struct BodyDelete {
|
||||||
|
root: String,
|
||||||
|
files: Vec<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn whitelist_wipe(server: &str, token: &str) {
|
||||||
|
let url_base = format!("https://panel.games.skynet.ie/api/client/servers/{server}");
|
||||||
|
let bearer = format!("Bearer {token}");
|
||||||
|
|
||||||
|
// delete whitelist
|
||||||
|
let deletion = BodyDelete {
|
||||||
|
root: "/".to_string(),
|
||||||
|
files: vec!["whitelist.json".to_string()],
|
||||||
|
};
|
||||||
|
post(&format!("{url_base}/files/delete"), &bearer, &deletion).await;
|
||||||
|
|
||||||
|
// recreate teh file, passing in the type here so the compiler knows what type of vec it is
|
||||||
|
post::<Vec<&str>>(&format!("{url_base}/files/write?file=%2Fwhitelist.json"), &bearer, &vec![]).await;
|
||||||
|
|
||||||
|
// reload the whitelist
|
||||||
|
let data = BodyCommand {
|
||||||
|
command: "whitelist reload".to_string(),
|
||||||
|
};
|
||||||
|
post(&format!("{url_base}/command"), &bearer, &data).await;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn server_information(server: &str, token: &str) -> Option<ServerDetailsRes> {
|
||||||
|
let url_base = format!("https://panel.games.skynet.ie/api/client/servers/{server}");
|
||||||
|
let bearer = format!("Bearer {token}");
|
||||||
|
get::<ServerDetailsRes>(&format!("{url_base}/"), &bearer).await
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn get_minecraft_config(db: &Pool<Sqlite>) -> Vec<Minecraft> {
|
||||||
|
sqlx::query_as::<_, Minecraft>(
|
||||||
|
r#"
|
||||||
|
SELECT *
|
||||||
|
FROM minecraft
|
||||||
|
"#,
|
||||||
|
)
|
||||||
|
.fetch_all(db)
|
||||||
|
.await
|
||||||
|
.unwrap_or_default()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn get_minecraft_config_server(db: &Pool<Sqlite>, g_id: GuildId) -> Vec<Minecraft> {
|
||||||
|
sqlx::query_as::<_, Minecraft>(
|
||||||
|
r#"
|
||||||
|
SELECT *
|
||||||
|
FROM minecraft
|
||||||
|
WHERE server_discord = ?1
|
||||||
|
"#,
|
||||||
|
)
|
||||||
|
.bind(g_id.get() as i64)
|
||||||
|
.fetch_all(db)
|
||||||
|
.await
|
||||||
|
.unwrap_or_default()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn whitelist_update(add: &Vec<(String, bool)>, server: &str, token: &str) {
|
||||||
|
println!("Update whitelist for {}", server);
|
||||||
|
let url_base = format!("https://panel.games.skynet.ie/api/client/servers/{server}");
|
||||||
|
let bearer = format!("Bearer {token}");
|
||||||
|
|
||||||
|
for (name, java) in add {
|
||||||
|
let data = if *java {
|
||||||
|
BodyCommand {
|
||||||
|
command: format!("whitelist add {name}"),
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
BodyCommand {
|
||||||
|
command: format!("fwhitelist add {name}"),
|
||||||
|
}
|
||||||
|
};
|
||||||
|
post(&format!("{url_base}/command"), &bearer, &data).await;
|
||||||
|
}
|
||||||
|
}
|
4
src/common/mod.rs
Normal file
4
src/common/mod.rs
Normal file
|
@ -0,0 +1,4 @@
|
||||||
|
pub mod database;
|
||||||
|
pub mod minecraft;
|
||||||
|
pub mod set_roles;
|
||||||
|
pub mod wolves;
|
500
src/common/set_roles.rs
Normal file
500
src/common/set_roles.rs
Normal file
|
@ -0,0 +1,500 @@
|
||||||
|
pub mod normal {
|
||||||
|
use crate::common::database::{DataBase, ServerMembersWolves, Servers, Wolves};
|
||||||
|
use crate::get_now_iso;
|
||||||
|
use serenity::client::Context;
|
||||||
|
use serenity::model::id::{GuildId, RoleId, UserId};
|
||||||
|
use sqlx::{Pool, Sqlite};
|
||||||
|
|
||||||
|
pub async fn update_server(ctx: &Context, server: &Servers, remove_roles: &[Option<RoleId>], members_changed: &[UserId]) {
|
||||||
|
let db_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<DataBase>().expect("Expected Database in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
|
||||||
|
let db = db_lock.read().await;
|
||||||
|
|
||||||
|
let Servers {
|
||||||
|
server,
|
||||||
|
role_past,
|
||||||
|
role_current,
|
||||||
|
..
|
||||||
|
} = server;
|
||||||
|
|
||||||
|
let mut roles_set = [0, 0, 0];
|
||||||
|
let mut members = vec![];
|
||||||
|
|
||||||
|
for member in get_server_member_bulk(&db, server).await {
|
||||||
|
if let Some(x) = member.discord {
|
||||||
|
members.push(x);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let mut members_all = members.len();
|
||||||
|
|
||||||
|
if let Ok(x) = server.members(ctx, None, None).await {
|
||||||
|
for member in x {
|
||||||
|
// members_changed acts as an override to only deal with teh users in it
|
||||||
|
if !members_changed.is_empty() && !members_changed.contains(&member.user.id) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if members.contains(&member.user.id) {
|
||||||
|
let mut roles = vec![];
|
||||||
|
|
||||||
|
if let Some(role) = &role_past {
|
||||||
|
if !member.roles.contains(role) {
|
||||||
|
roles_set[0] += 1;
|
||||||
|
roles.push(role.to_owned());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !member.roles.contains(role_current) {
|
||||||
|
roles_set[1] += 1;
|
||||||
|
roles.push(role_current.to_owned());
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Err(e) = member.add_roles(ctx, &roles).await {
|
||||||
|
println!("{:?}", e);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// old and never
|
||||||
|
|
||||||
|
if let Some(role) = &role_past {
|
||||||
|
if member.roles.contains(role) {
|
||||||
|
members_all += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if member.roles.contains(role_current) {
|
||||||
|
roles_set[2] += 1;
|
||||||
|
// if theya re not a current member and have the role then remove it
|
||||||
|
if let Err(e) = member.remove_role(ctx, role_current).await {
|
||||||
|
println!("{:?}", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for role in remove_roles.iter().flatten() {
|
||||||
|
if let Err(e) = member.remove_role(ctx, role).await {
|
||||||
|
println!("{:?}", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
set_server_numbers(&db, server, members_all as i64, members.len() as i64).await;
|
||||||
|
|
||||||
|
// small bit of logging to note changes over time
|
||||||
|
println!("{:?} Changes: New: +{}, Current: +{}/-{}", server.get(), roles_set[0], roles_set[1], roles_set[2]);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn get_server_member_bulk(db: &Pool<Sqlite>, server: &GuildId) -> Vec<ServerMembersWolves> {
|
||||||
|
sqlx::query_as::<_, ServerMembersWolves>(
|
||||||
|
r#"
|
||||||
|
SELECT *
|
||||||
|
FROM server_members
|
||||||
|
JOIN wolves USING (id_wolves)
|
||||||
|
WHERE (
|
||||||
|
server = ?
|
||||||
|
AND discord IS NOT NULL
|
||||||
|
AND expiry > ?
|
||||||
|
)
|
||||||
|
"#,
|
||||||
|
)
|
||||||
|
.bind(server.get() as i64)
|
||||||
|
.bind(get_now_iso(true))
|
||||||
|
.fetch_all(db)
|
||||||
|
.await
|
||||||
|
.unwrap_or_default()
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn set_server_numbers(db: &Pool<Sqlite>, server: &GuildId, past: i64, current: i64) {
|
||||||
|
match sqlx::query_as::<_, Wolves>(
|
||||||
|
"
|
||||||
|
UPDATE servers
|
||||||
|
SET member_past = ?, member_current = ?
|
||||||
|
WHERE server = ?
|
||||||
|
",
|
||||||
|
)
|
||||||
|
.bind(past)
|
||||||
|
.bind(current)
|
||||||
|
.bind(server.get() as i64)
|
||||||
|
.fetch_optional(db)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(_) => {}
|
||||||
|
Err(e) => {
|
||||||
|
println!("Failure to insert into {}", server.get());
|
||||||
|
println!("{:?}", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// for updating committee members
|
||||||
|
pub mod committee {
|
||||||
|
use crate::common::database::{get_channel_from_row, get_role_from_row, DataBase, Wolves};
|
||||||
|
use crate::common::wolves::committees::Committees;
|
||||||
|
use crate::Config;
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
use serenity::all::EditRole;
|
||||||
|
use serenity::builder::CreateChannel;
|
||||||
|
use serenity::client::Context;
|
||||||
|
use serenity::model::channel::ChannelType;
|
||||||
|
use serenity::model::guild::Member;
|
||||||
|
use serenity::model::id::ChannelId;
|
||||||
|
use serenity::model::prelude::RoleId;
|
||||||
|
use sqlx::sqlite::SqliteRow;
|
||||||
|
use sqlx::{Error, FromRow, Pool, Row, Sqlite};
|
||||||
|
use std::collections::HashMap;
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
pub async fn check_committee(ctx: Arc<Context>) {
|
||||||
|
let db_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<DataBase>().expect("Expected Config in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
|
||||||
|
let db = db_lock.read().await;
|
||||||
|
|
||||||
|
let config_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<Config>().expect("Expected Config in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
let config_global = config_lock.read().await;
|
||||||
|
|
||||||
|
let server = config_global.committee_server;
|
||||||
|
|
||||||
|
// because to use it to update a single user we need to pre-get the members of teh server
|
||||||
|
let mut members = server.members(&ctx, None, None).await.unwrap_or_default();
|
||||||
|
|
||||||
|
update_committees(&db, &ctx, &config_global, &mut members).await;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
This function can take a vec of members (or just one) and gives tehm the appropiate roles on teh committee server
|
||||||
|
*/
|
||||||
|
pub async fn update_committees(db: &Pool<Sqlite>, ctx: &Context, config: &Config, members: &mut Vec<Member>) {
|
||||||
|
let server = config.committee_server;
|
||||||
|
let committee_member = RoleId::new(1226602779968274573);
|
||||||
|
let committees = get_committees(db).await;
|
||||||
|
let categories = [
|
||||||
|
ChannelId::new(1226606560973815839),
|
||||||
|
// C&S Chats 2
|
||||||
|
ChannelId::new(1341457244973305927),
|
||||||
|
// C&S Chats 3
|
||||||
|
ChannelId::new(1341457509717639279),
|
||||||
|
];
|
||||||
|
|
||||||
|
// information about the server
|
||||||
|
let mut roles_db = HashMap::new();
|
||||||
|
for role in db_roles_get(db).await {
|
||||||
|
roles_db.insert(
|
||||||
|
role.id_wolves,
|
||||||
|
CommitteeRoles {
|
||||||
|
id_wolves: role.id_wolves,
|
||||||
|
id_role: role.id_role,
|
||||||
|
id_channel: role.id_channel,
|
||||||
|
name_role: role.name_role,
|
||||||
|
name_channel: role.name_channel,
|
||||||
|
// always start at 0
|
||||||
|
count: 0,
|
||||||
|
},
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut channels = server.channels(&ctx).await.unwrap_or_default();
|
||||||
|
|
||||||
|
// a map of users and the roles they are goign to be getting
|
||||||
|
let mut users_roles = HashMap::new();
|
||||||
|
|
||||||
|
let mut re_order = false;
|
||||||
|
// we need to create roles and channels if tehy dont already exist
|
||||||
|
let mut category_index = 0;
|
||||||
|
let mut i = 0;
|
||||||
|
loop {
|
||||||
|
if i >= committees.len() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
let committee = &committees[i];
|
||||||
|
|
||||||
|
// if a club/soc ever changes their name
|
||||||
|
if let Some(x) = roles_db.get_mut(&committee.id) {
|
||||||
|
committee.name_full.clone_into(&mut x.name_role);
|
||||||
|
committee.name_profile.clone_into(&mut x.name_channel);
|
||||||
|
}
|
||||||
|
|
||||||
|
// handle new clubs/socs
|
||||||
|
if let std::collections::hash_map::Entry::Vacant(e) = roles_db.entry(committee.id) {
|
||||||
|
// create channel
|
||||||
|
// channel is first as the categories can only contain 50 channels
|
||||||
|
let channel = match server
|
||||||
|
.create_channel(
|
||||||
|
&ctx,
|
||||||
|
CreateChannel::new(&committee.name_profile)
|
||||||
|
.kind(ChannelType::Text)
|
||||||
|
.category(categories[category_index]),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(x) => {
|
||||||
|
println!("Created channel: {}", &committee.name_profile);
|
||||||
|
|
||||||
|
x.id
|
||||||
|
}
|
||||||
|
Err(x) => {
|
||||||
|
let tmp = x.to_string();
|
||||||
|
dbg!("Unable to create channel: ", &tmp, &tmp.contains("Maximum number of channels in category reached (50)"));
|
||||||
|
if x.to_string().contains("Maximum number of channels in category reached (50)") {
|
||||||
|
category_index += 1;
|
||||||
|
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
ChannelId::new(1)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// create role
|
||||||
|
let role = match server
|
||||||
|
.create_role(&ctx, EditRole::new().name(&committee.name_full).hoist(false).mentionable(true))
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(x) => x.id,
|
||||||
|
Err(_) => RoleId::new(1),
|
||||||
|
};
|
||||||
|
|
||||||
|
let tmp = CommitteeRoles {
|
||||||
|
id_wolves: committee.id,
|
||||||
|
id_role: role,
|
||||||
|
id_channel: channel,
|
||||||
|
name_role: committee.name_full.to_owned(),
|
||||||
|
name_channel: committee.name_profile.to_owned(),
|
||||||
|
count: 0,
|
||||||
|
};
|
||||||
|
|
||||||
|
// save it to the db in case of crash or error
|
||||||
|
db_role_set(db, &tmp).await;
|
||||||
|
|
||||||
|
// insert it into teh local cache
|
||||||
|
e.insert(tmp);
|
||||||
|
|
||||||
|
re_order = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
i += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
for committee in &committees {
|
||||||
|
let r = if let Some(x) = roles_db.get(&committee.id) {
|
||||||
|
x.id_role
|
||||||
|
} else {
|
||||||
|
continue;
|
||||||
|
};
|
||||||
|
|
||||||
|
for id_wolves in &committee.committee {
|
||||||
|
// ID in this is the wolves ID, so we need to get a matching discord ID (if one exists)
|
||||||
|
if let Some(x) = get_server_member_discord(db, id_wolves).await {
|
||||||
|
if let Some(member_tmp) = x.discord {
|
||||||
|
if server.member(ctx, &member_tmp).await.is_ok() {
|
||||||
|
let values = users_roles.entry(member_tmp).or_insert(vec![]);
|
||||||
|
values.push(r);
|
||||||
|
|
||||||
|
if let Some(x) = roles_db.get_mut(&committee.id) {
|
||||||
|
x.count += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// now we have a map of all users that should get roles time to go through all the folks on teh server
|
||||||
|
for member in members {
|
||||||
|
let roles_current = member.roles(ctx).unwrap_or_default();
|
||||||
|
|
||||||
|
let roles_required = match users_roles.get(&member.user.id) {
|
||||||
|
None => {
|
||||||
|
vec![]
|
||||||
|
}
|
||||||
|
Some(x) => {
|
||||||
|
let mut tmp = x.to_owned();
|
||||||
|
if !tmp.is_empty() {
|
||||||
|
tmp.push(committee_member);
|
||||||
|
}
|
||||||
|
tmp
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut roles_rem = vec![];
|
||||||
|
let mut roles_add = vec![];
|
||||||
|
// get a list of all the roles to remove from someone
|
||||||
|
|
||||||
|
let mut roles_current_id = vec![];
|
||||||
|
for role in &roles_current {
|
||||||
|
roles_current_id.push(role.id.to_owned());
|
||||||
|
if !roles_required.contains(&role.id) {
|
||||||
|
roles_rem.push(role.id.to_owned());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !roles_required.is_empty() {
|
||||||
|
// if there are committee roles then give the general purporse role
|
||||||
|
roles_add.push(committee_member);
|
||||||
|
|
||||||
|
if let Some(x) = roles_db.get_mut(&0) {
|
||||||
|
x.count += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for role in &roles_required {
|
||||||
|
if !roles_current_id.contains(role) {
|
||||||
|
roles_add.push(role.to_owned());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !roles_rem.is_empty() {
|
||||||
|
member.remove_roles(&ctx, &roles_rem).await.unwrap_or_default();
|
||||||
|
}
|
||||||
|
|
||||||
|
if !roles_add.is_empty() {
|
||||||
|
// these roles are flavor roles, only there to make folks mentionable
|
||||||
|
member.add_roles(&ctx, &roles_add).await.unwrap_or_default();
|
||||||
|
} else {
|
||||||
|
member.remove_roles(&ctx, &[committee_member]).await.unwrap_or_default();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut channel_names = vec![];
|
||||||
|
let mut positions = vec![];
|
||||||
|
for role in roles_db.values() {
|
||||||
|
// save these to db
|
||||||
|
db_role_set(db, role).await;
|
||||||
|
|
||||||
|
if re_order {
|
||||||
|
let channel_id = role.id_channel.to_owned();
|
||||||
|
if let Some(channel) = channels.get_mut(&channel_id) {
|
||||||
|
// record the position of each of teh C&S channels
|
||||||
|
positions.push(channel.position);
|
||||||
|
|
||||||
|
// pull out teh channel names
|
||||||
|
channel_names.push((role.name_channel.to_owned(), channel_id));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if re_order {
|
||||||
|
// sort by the position and name
|
||||||
|
positions.sort();
|
||||||
|
channel_names.sort_by_key(|(name, _)| name.to_owned());
|
||||||
|
|
||||||
|
let mut new_positions = vec![];
|
||||||
|
for (i, (_, id)) in channel_names.iter().enumerate() {
|
||||||
|
new_positions.push((id.to_owned(), positions[i] as u64));
|
||||||
|
}
|
||||||
|
|
||||||
|
if !new_positions.is_empty() {
|
||||||
|
match server.reorder_channels(&ctx, new_positions).await {
|
||||||
|
Ok(_) => {
|
||||||
|
println!("Successfully re-orderd the committee category");
|
||||||
|
}
|
||||||
|
Err(e) => {
|
||||||
|
dbg!("Failed to re-order ", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Deserialize, Serialize)]
|
||||||
|
pub struct CommitteeRoles {
|
||||||
|
id_wolves: i64,
|
||||||
|
id_role: RoleId,
|
||||||
|
id_channel: ChannelId,
|
||||||
|
pub name_role: String,
|
||||||
|
name_channel: String,
|
||||||
|
pub count: i64,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'r> FromRow<'r, SqliteRow> for CommitteeRoles {
|
||||||
|
fn from_row(row: &'r SqliteRow) -> Result<Self, Error> {
|
||||||
|
Ok(Self {
|
||||||
|
id_wolves: row.try_get("id_wolves")?,
|
||||||
|
id_role: get_role_from_row(row, "id_role"),
|
||||||
|
id_channel: get_channel_from_row(row, "id_channel"),
|
||||||
|
name_role: row.try_get("name_role")?,
|
||||||
|
name_channel: row.try_get("name_channel")?,
|
||||||
|
count: row.try_get("count")?,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn db_role_set(db: &Pool<Sqlite>, role: &CommitteeRoles) {
|
||||||
|
// expiry
|
||||||
|
match sqlx::query_as::<_, CommitteeRoles>(
|
||||||
|
"
|
||||||
|
INSERT INTO committee_roles (id_wolves, id_role, id_channel, name_role, name_channel, count)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, $6)
|
||||||
|
ON CONFLICT(id_wolves) DO UPDATE SET name_role = $4, name_channel = $5, count = $6
|
||||||
|
",
|
||||||
|
)
|
||||||
|
.bind(role.id_wolves)
|
||||||
|
.bind(role.id_role.get() as i64)
|
||||||
|
.bind(role.id_channel.get() as i64)
|
||||||
|
.bind(&role.name_role)
|
||||||
|
.bind(&role.name_channel)
|
||||||
|
.bind(role.count)
|
||||||
|
.fetch_optional(db)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(_) => {}
|
||||||
|
Err(e) => {
|
||||||
|
println!("Failure to insert into Wolves {:?}", role);
|
||||||
|
println!("{:?}", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn db_roles_get(db: &Pool<Sqlite>) -> Vec<CommitteeRoles> {
|
||||||
|
// expiry
|
||||||
|
sqlx::query_as::<_, CommitteeRoles>(
|
||||||
|
"
|
||||||
|
SELECT *
|
||||||
|
FROM committee_roles
|
||||||
|
",
|
||||||
|
)
|
||||||
|
.fetch_all(db)
|
||||||
|
.await
|
||||||
|
.unwrap_or_else(|e| {
|
||||||
|
println!("Failure to get Roles from committee_roles");
|
||||||
|
println!("{:?}", e);
|
||||||
|
vec![]
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn get_committees(db: &Pool<Sqlite>) -> Vec<Committees> {
|
||||||
|
sqlx::query_as::<_, Committees>(
|
||||||
|
r#"
|
||||||
|
SELECT *
|
||||||
|
FROM committees
|
||||||
|
"#,
|
||||||
|
)
|
||||||
|
.fetch_all(db)
|
||||||
|
.await
|
||||||
|
.unwrap_or_else(|e| {
|
||||||
|
dbg!(e);
|
||||||
|
vec![]
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn get_server_member_discord(db: &Pool<Sqlite>, user: &i64) -> Option<Wolves> {
|
||||||
|
sqlx::query_as::<_, Wolves>(
|
||||||
|
r#"
|
||||||
|
SELECT *
|
||||||
|
FROM wolves
|
||||||
|
WHERE id_wolves = ?
|
||||||
|
"#,
|
||||||
|
)
|
||||||
|
.bind(user)
|
||||||
|
.fetch_one(db)
|
||||||
|
.await
|
||||||
|
.ok()
|
||||||
|
}
|
||||||
|
}
|
278
src/common/wolves.rs
Normal file
278
src/common/wolves.rs
Normal file
|
@ -0,0 +1,278 @@
|
||||||
|
use crate::common::database::Wolves;
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
use sqlx::{Pool, Sqlite};
|
||||||
|
|
||||||
|
/**
|
||||||
|
This file relates to anything that directly interacts with teh wolves API
|
||||||
|
*/
|
||||||
|
|
||||||
|
#[derive(Deserialize, Serialize, Debug)]
|
||||||
|
struct WolvesResultUserMin {
|
||||||
|
// committee: String,
|
||||||
|
member_id: String,
|
||||||
|
// first_name: String,
|
||||||
|
// last_name: String,
|
||||||
|
contact_email: String,
|
||||||
|
// opt_in_email: String,
|
||||||
|
// student_id: Option<String>,
|
||||||
|
// note: Option<String>,
|
||||||
|
// expiry: String,
|
||||||
|
// requested: String,
|
||||||
|
// approved: String,
|
||||||
|
// sitename: String,
|
||||||
|
// domain: String,
|
||||||
|
}
|
||||||
|
async fn add_users_wolves(db: &Pool<Sqlite>, user: &WolvesResultUserMin) {
|
||||||
|
// expiry
|
||||||
|
match sqlx::query_as::<_, Wolves>(
|
||||||
|
"
|
||||||
|
INSERT INTO wolves (id_wolves, email)
|
||||||
|
VALUES ($1, $2)
|
||||||
|
ON CONFLICT(id_wolves) DO UPDATE SET email = $2
|
||||||
|
",
|
||||||
|
)
|
||||||
|
.bind(&user.member_id)
|
||||||
|
.bind(&user.contact_email)
|
||||||
|
.fetch_optional(db)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(_) => {}
|
||||||
|
Err(e) => {
|
||||||
|
println!("Failure to insert into Wolves {:?}", user);
|
||||||
|
println!("{:?}", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
This is getting data for Clubs and Socs
|
||||||
|
*/
|
||||||
|
pub mod cns {
|
||||||
|
use crate::common::database::{get_server_config_bulk, DataBase, ServerMembers, ServerMembersWolves, Servers};
|
||||||
|
use crate::common::set_roles::normal::update_server;
|
||||||
|
use crate::common::wolves::{add_users_wolves, WolvesResultUserMin};
|
||||||
|
use crate::Config;
|
||||||
|
use serenity::client::Context;
|
||||||
|
use serenity::model::id::GuildId;
|
||||||
|
use sqlx::{Pool, Sqlite};
|
||||||
|
use std::collections::BTreeMap;
|
||||||
|
|
||||||
|
impl From<&wolves_oxidised::WolvesUser> for WolvesResultUserMin {
|
||||||
|
fn from(value: &wolves_oxidised::WolvesUser) -> Self {
|
||||||
|
Self {
|
||||||
|
member_id: value.member_id.to_owned(),
|
||||||
|
contact_email: value.contact_email.to_owned(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn get_wolves(ctx: &Context) {
|
||||||
|
let db_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<DataBase>().expect("Expected Database in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
let db = db_lock.read().await;
|
||||||
|
|
||||||
|
let config_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<Config>().expect("Expected Config in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
let config = config_lock.read().await;
|
||||||
|
|
||||||
|
// set up teh client
|
||||||
|
let wolves = wolves_oxidised::Client::new(&config.wolves_url, Some(&config.wolves_api));
|
||||||
|
|
||||||
|
for server_config in get_server_config_bulk(&db).await {
|
||||||
|
let Servers {
|
||||||
|
server,
|
||||||
|
// this is the unique api key for each club/soc
|
||||||
|
wolves_api,
|
||||||
|
wolves_id,
|
||||||
|
..
|
||||||
|
} = &server_config;
|
||||||
|
// dbg!(&server_config);
|
||||||
|
|
||||||
|
let existing_tmp = get_server_member(&db, server).await;
|
||||||
|
let existing = existing_tmp.iter().map(|data| (data.id_wolves, data)).collect::<BTreeMap<_, _>>();
|
||||||
|
|
||||||
|
// list of users that need to be updated for this server
|
||||||
|
let mut user_to_update = vec![];
|
||||||
|
let mut server_name_tmp = None;
|
||||||
|
for user in wolves.get_members(wolves_api).await {
|
||||||
|
// dbg!(&user.committee);
|
||||||
|
if server_name_tmp.is_none() {
|
||||||
|
server_name_tmp = Some(user.committee_id);
|
||||||
|
}
|
||||||
|
let id = user.member_id.parse::<u64>().unwrap_or_default();
|
||||||
|
match existing.get(&(id as i64)) {
|
||||||
|
None => {
|
||||||
|
// user does not exist already, add everything
|
||||||
|
add_users_wolves(&db, &WolvesResultUserMin::from(&user)).await;
|
||||||
|
add_users_server_members(&db, server, &user).await;
|
||||||
|
}
|
||||||
|
Some(old) => {
|
||||||
|
// always update wolves table, in case data has changed
|
||||||
|
add_users_wolves(&db, &WolvesResultUserMin::from(&user)).await;
|
||||||
|
if old.expiry != user.expiry {
|
||||||
|
add_users_server_members(&db, server, &user).await;
|
||||||
|
|
||||||
|
if let Some(discord_id) = old.discord {
|
||||||
|
user_to_update.push(discord_id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(cs_id) = server_name_tmp {
|
||||||
|
if &cs_id != wolves_id {
|
||||||
|
set_server_member(&db, server, cs_id).await;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !user_to_update.is_empty() {
|
||||||
|
update_server(ctx, &server_config, &[], &user_to_update).await;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn set_server_member(db: &Pool<Sqlite>, server: &GuildId, wolves_id: i64) {
|
||||||
|
match sqlx::query_as::<_, Servers>(
|
||||||
|
"
|
||||||
|
UPDATE servers
|
||||||
|
SET wolves_id = ?
|
||||||
|
WHERE server = ?
|
||||||
|
",
|
||||||
|
)
|
||||||
|
.bind(wolves_id)
|
||||||
|
.bind(server.get() as i64)
|
||||||
|
.fetch_optional(db)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(_) => {}
|
||||||
|
Err(e) => {
|
||||||
|
println!("Failure to set server name {}", server.get());
|
||||||
|
println!("{:?}", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn get_server_member(db: &Pool<Sqlite>, server: &GuildId) -> Vec<ServerMembersWolves> {
|
||||||
|
sqlx::query_as::<_, ServerMembersWolves>(
|
||||||
|
r#"
|
||||||
|
SELECT *
|
||||||
|
FROM server_members
|
||||||
|
JOIN wolves USING (id_wolves)
|
||||||
|
WHERE (
|
||||||
|
server = ?
|
||||||
|
AND discord IS NOT NULL
|
||||||
|
)
|
||||||
|
"#,
|
||||||
|
)
|
||||||
|
.bind(server.get() as i64)
|
||||||
|
.fetch_all(db)
|
||||||
|
.await
|
||||||
|
.unwrap_or_default()
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn add_users_server_members(db: &Pool<Sqlite>, server: &GuildId, user: &wolves_oxidised::WolvesUser) {
|
||||||
|
match sqlx::query_as::<_, ServerMembers>(
|
||||||
|
"
|
||||||
|
INSERT OR REPLACE INTO server_members (server, id_wolves, expiry)
|
||||||
|
VALUES (?1, ?2, ?3)
|
||||||
|
",
|
||||||
|
)
|
||||||
|
.bind(server.get() as i64)
|
||||||
|
.bind(&user.member_id)
|
||||||
|
.bind(&user.expiry)
|
||||||
|
.fetch_optional(db)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(_) => {}
|
||||||
|
Err(e) => {
|
||||||
|
println!("Failure to insert into ServerMembers {} {:?}", server.get(), user);
|
||||||
|
println!("{:?}", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
Get and store the data on C&S committees
|
||||||
|
*/
|
||||||
|
pub mod committees {
|
||||||
|
use crate::common::database::DataBase;
|
||||||
|
use crate::Config;
|
||||||
|
use serenity::client::Context;
|
||||||
|
use sqlx::{Pool, Sqlite};
|
||||||
|
|
||||||
|
// Database entry for it
|
||||||
|
#[derive(Debug, Clone, sqlx::FromRow)]
|
||||||
|
pub struct Committees {
|
||||||
|
pub id: i64,
|
||||||
|
pub name_full: String,
|
||||||
|
pub name_profile: String,
|
||||||
|
pub name_plain: String,
|
||||||
|
pub link: String,
|
||||||
|
#[sqlx(json)]
|
||||||
|
pub committee: Vec<i64>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<wolves_oxidised::WolvesCNS> for Committees {
|
||||||
|
fn from(value: wolves_oxidised::WolvesCNS) -> Self {
|
||||||
|
Self {
|
||||||
|
id: value.id,
|
||||||
|
name_full: value.name_full,
|
||||||
|
name_profile: value.name_profile,
|
||||||
|
name_plain: value.name_plain,
|
||||||
|
link: value.link,
|
||||||
|
committee: value.committee,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn get_cns(ctx: &Context) {
|
||||||
|
let db_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<DataBase>().expect("Expected Database in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
let db = db_lock.read().await;
|
||||||
|
|
||||||
|
let config_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<Config>().expect("Expected Config in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
let config = config_lock.read().await;
|
||||||
|
|
||||||
|
let wolves = wolves_oxidised::Client::new(&config.wolves_url, Some(&config.wolves_api));
|
||||||
|
// request data from wolves
|
||||||
|
for committee_wolves in wolves.get_committees().await {
|
||||||
|
let committee = Committees::from(committee_wolves);
|
||||||
|
add_committee(&db, &committee).await;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn add_committee(db: &Pool<Sqlite>, committee: &Committees) {
|
||||||
|
match sqlx::query_as::<_, Committees>(
|
||||||
|
"
|
||||||
|
INSERT INTO committees (id, name_profile, name_full, name_plain, link, committee)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, $6)
|
||||||
|
ON CONFLICT(id) DO UPDATE SET committee = $6
|
||||||
|
",
|
||||||
|
)
|
||||||
|
.bind(committee.id)
|
||||||
|
.bind(&committee.name_profile)
|
||||||
|
.bind(&committee.name_full)
|
||||||
|
.bind(&committee.name_plain)
|
||||||
|
.bind(&committee.link)
|
||||||
|
.bind(serde_json::to_string(&committee.committee).unwrap_or_default())
|
||||||
|
.fetch_optional(db)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(_) => {}
|
||||||
|
Err(e) => {
|
||||||
|
println!("Failure to insert into Committees {:?}", committee);
|
||||||
|
println!("{:?}", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
661
src/lib.rs
661
src/lib.rs
|
@ -1,57 +1,51 @@
|
||||||
use dotenvy::dotenv;
|
pub mod common;
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
use serenity::{
|
|
||||||
model::{
|
|
||||||
guild,
|
|
||||||
id::{GuildId, RoleId},
|
|
||||||
},
|
|
||||||
prelude::TypeMapKey,
|
|
||||||
};
|
|
||||||
|
|
||||||
use chrono::{Datelike, SecondsFormat, Utc};
|
use chrono::{Datelike, SecondsFormat, Utc};
|
||||||
use rand::{distributions::Alphanumeric, thread_rng, Rng};
|
use dotenvy::dotenv;
|
||||||
use serenity::client::Context;
|
use rand::{distr::Alphanumeric, rng, Rng};
|
||||||
use serenity::model::id::UserId;
|
use serenity::model::id::{ChannelId, GuildId, RoleId};
|
||||||
use sqlx::{
|
use serenity::prelude::TypeMapKey;
|
||||||
sqlite::{SqliteConnectOptions, SqlitePoolOptions, SqliteRow},
|
use std::{env, sync::Arc};
|
||||||
Error, FromRow, Pool, Row, Sqlite,
|
|
||||||
};
|
|
||||||
use std::{env, str::FromStr, sync::Arc};
|
|
||||||
use tokio::sync::RwLock;
|
use tokio::sync::RwLock;
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct Config {
|
pub struct Config {
|
||||||
pub skynet_server: GuildId,
|
// manages where teh database is stored
|
||||||
pub ldap_api: String,
|
|
||||||
pub home: String,
|
pub home: String,
|
||||||
pub database: String,
|
pub database: String,
|
||||||
|
|
||||||
pub auth: String,
|
// tokens for discord and other API's
|
||||||
pub discord_token: String,
|
pub discord_token: String,
|
||||||
|
pub discord_token_minecraft: String,
|
||||||
|
pub minecraft_mcprofile: String,
|
||||||
|
|
||||||
|
// email settings
|
||||||
pub mail_smtp: String,
|
pub mail_smtp: String,
|
||||||
pub mail_user: String,
|
pub mail_user: String,
|
||||||
pub mail_pass: String,
|
pub mail_pass: String,
|
||||||
|
|
||||||
|
// wolves API base for clubs/socs
|
||||||
pub wolves_url: String,
|
pub wolves_url: String,
|
||||||
|
// API key for accessing more general resources
|
||||||
|
pub wolves_api: String,
|
||||||
|
|
||||||
|
// discord server for committee
|
||||||
|
pub committee_server: GuildId,
|
||||||
|
pub committee_role: RoleId,
|
||||||
|
pub committee_category: ChannelId,
|
||||||
}
|
}
|
||||||
impl TypeMapKey for Config {
|
impl TypeMapKey for Config {
|
||||||
type Value = Arc<RwLock<Config>>;
|
type Value = Arc<RwLock<Config>>;
|
||||||
}
|
}
|
||||||
|
|
||||||
pub struct DataBase;
|
|
||||||
impl TypeMapKey for DataBase {
|
|
||||||
type Value = Arc<RwLock<Pool<Sqlite>>>;
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn get_config() -> Config {
|
pub fn get_config() -> Config {
|
||||||
dotenv().ok();
|
dotenv().ok();
|
||||||
|
|
||||||
// reasonable defaults
|
// reasonable defaults
|
||||||
let mut config = Config {
|
let mut config = Config {
|
||||||
skynet_server: Default::default(),
|
|
||||||
ldap_api: "https://api.account.skynet.ie".to_string(),
|
|
||||||
auth: "".to_string(),
|
|
||||||
discord_token: "".to_string(),
|
discord_token: "".to_string(),
|
||||||
|
discord_token_minecraft: "".to_string(),
|
||||||
|
minecraft_mcprofile: "".to_string(),
|
||||||
|
|
||||||
home: ".".to_string(),
|
home: ".".to_string(),
|
||||||
database: "database.db".to_string(),
|
database: "database.db".to_string(),
|
||||||
|
@ -60,28 +54,28 @@ pub fn get_config() -> Config {
|
||||||
mail_user: "".to_string(),
|
mail_user: "".to_string(),
|
||||||
mail_pass: "".to_string(),
|
mail_pass: "".to_string(),
|
||||||
wolves_url: "".to_string(),
|
wolves_url: "".to_string(),
|
||||||
|
wolves_api: "".to_string(),
|
||||||
|
committee_server: GuildId::new(1),
|
||||||
|
committee_role: RoleId::new(1),
|
||||||
|
committee_category: ChannelId::new(1),
|
||||||
};
|
};
|
||||||
|
|
||||||
if let Ok(x) = env::var("LDAP_API") {
|
if let Ok(x) = env::var("DATABASE_HOME") {
|
||||||
config.ldap_api = x.trim().to_string();
|
|
||||||
}
|
|
||||||
if let Ok(x) = env::var("SKYNET_SERVER") {
|
|
||||||
config.skynet_server = GuildId::from(str_to_num::<u64>(&x));
|
|
||||||
}
|
|
||||||
if let Ok(x) = env::var("HOME") {
|
|
||||||
config.home = x.trim().to_string();
|
config.home = x.trim().to_string();
|
||||||
}
|
}
|
||||||
if let Ok(x) = env::var("DATABASE") {
|
if let Ok(x) = env::var("DATABASE") {
|
||||||
config.database = x.trim().to_string();
|
config.database = x.trim().to_string();
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Ok(x) = env::var("LDAP_DISCORD_AUTH") {
|
|
||||||
config.auth = x.trim().to_string();
|
|
||||||
}
|
|
||||||
|
|
||||||
if let Ok(x) = env::var("DISCORD_TOKEN") {
|
if let Ok(x) = env::var("DISCORD_TOKEN") {
|
||||||
config.discord_token = x.trim().to_string();
|
config.discord_token = x.trim().to_string();
|
||||||
}
|
}
|
||||||
|
if let Ok(x) = env::var("DISCORD_TOKEN_MINECRAFT") {
|
||||||
|
config.discord_token_minecraft = x.trim().to_string();
|
||||||
|
}
|
||||||
|
if let Ok(x) = env::var("MINECRAFT_MCPROFILE_KEY") {
|
||||||
|
config.minecraft_mcprofile = x.trim().to_string();
|
||||||
|
}
|
||||||
|
|
||||||
if let Ok(x) = env::var("EMAIL_SMTP") {
|
if let Ok(x) = env::var("EMAIL_SMTP") {
|
||||||
config.mail_smtp = x.trim().to_string();
|
config.mail_smtp = x.trim().to_string();
|
||||||
|
@ -93,279 +87,32 @@ pub fn get_config() -> Config {
|
||||||
config.mail_pass = x.trim().to_string();
|
config.mail_pass = x.trim().to_string();
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Ok(x) = env::var("WOLVES_URL") {
|
if let Ok(x) = env::var("WOLVES_URL_BASE") {
|
||||||
config.wolves_url = x.trim().to_string();
|
config.wolves_url = x.trim().to_string();
|
||||||
}
|
}
|
||||||
|
if let Ok(x) = env::var("WOLVES_API") {
|
||||||
|
config.wolves_api = x.trim().to_string();
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Ok(x) = env::var("COMMITTEE_DISCORD") {
|
||||||
|
if let Ok(x) = x.trim().parse::<u64>() {
|
||||||
|
config.committee_server = GuildId::new(x);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if let Ok(x) = env::var("COMMITTEE_DISCORD") {
|
||||||
|
if let Ok(x) = x.trim().parse::<u64>() {
|
||||||
|
config.committee_role = RoleId::new(x);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if let Ok(x) = env::var("COMMITTEE_CATEGORY") {
|
||||||
|
if let Ok(x) = x.trim().parse::<u64>() {
|
||||||
|
config.committee_category = ChannelId::new(x);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
config
|
config
|
||||||
}
|
}
|
||||||
|
|
||||||
fn str_to_num<T: FromStr + Default>(x: &str) -> T {
|
|
||||||
x.trim().parse::<T>().unwrap_or_default()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Clone, Deserialize, Serialize)]
|
|
||||||
pub struct ServerMembers {
|
|
||||||
pub server: GuildId,
|
|
||||||
pub id_wolves: i64,
|
|
||||||
pub expiry: String,
|
|
||||||
}
|
|
||||||
impl<'r> FromRow<'r, SqliteRow> for ServerMembers {
|
|
||||||
fn from_row(row: &'r SqliteRow) -> Result<Self, Error> {
|
|
||||||
let server_tmp: i64 = row.try_get("server")?;
|
|
||||||
let server = GuildId::from(server_tmp as u64);
|
|
||||||
|
|
||||||
Ok(Self {
|
|
||||||
server,
|
|
||||||
id_wolves: row.try_get("id_wolves")?,
|
|
||||||
expiry: row.try_get("expiry")?,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Clone, Deserialize, Serialize)]
|
|
||||||
pub struct ServerMembersWolves {
|
|
||||||
pub server: GuildId,
|
|
||||||
pub id_wolves: i64,
|
|
||||||
pub expiry: String,
|
|
||||||
pub email: String,
|
|
||||||
pub discord: Option<UserId>,
|
|
||||||
pub minecraft: Option<String>,
|
|
||||||
}
|
|
||||||
impl<'r> FromRow<'r, SqliteRow> for ServerMembersWolves {
|
|
||||||
fn from_row(row: &'r SqliteRow) -> Result<Self, Error> {
|
|
||||||
let server_tmp: i64 = row.try_get("server")?;
|
|
||||||
let server = GuildId::from(server_tmp as u64);
|
|
||||||
let discord = match row.try_get("discord") {
|
|
||||||
Ok(x) => {
|
|
||||||
let tmp: i64 = x;
|
|
||||||
if tmp == 0 {
|
|
||||||
None
|
|
||||||
} else {
|
|
||||||
Some(UserId::from(tmp as u64))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => None,
|
|
||||||
};
|
|
||||||
|
|
||||||
Ok(Self {
|
|
||||||
server,
|
|
||||||
id_wolves: row.try_get("id_wolves")?,
|
|
||||||
expiry: row.try_get("expiry")?,
|
|
||||||
email: row.try_get("email")?,
|
|
||||||
discord,
|
|
||||||
minecraft: row.try_get("minecraft")?,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Clone, Deserialize, Serialize)]
|
|
||||||
pub struct Wolves {
|
|
||||||
pub id_wolves: i64,
|
|
||||||
pub email: String,
|
|
||||||
pub discord: Option<UserId>,
|
|
||||||
pub minecraft: Option<String>,
|
|
||||||
}
|
|
||||||
impl<'r> FromRow<'r, SqliteRow> for Wolves {
|
|
||||||
fn from_row(row: &'r SqliteRow) -> Result<Self, Error> {
|
|
||||||
let discord = match row.try_get("discord") {
|
|
||||||
Ok(x) => {
|
|
||||||
let tmp: i64 = x;
|
|
||||||
if tmp == 0 {
|
|
||||||
None
|
|
||||||
} else {
|
|
||||||
Some(UserId::from(tmp as u64))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => None,
|
|
||||||
};
|
|
||||||
|
|
||||||
Ok(Self {
|
|
||||||
id_wolves: row.try_get("id_wolves")?,
|
|
||||||
email: row.try_get("email")?,
|
|
||||||
discord,
|
|
||||||
minecraft: row.try_get("minecraft")?,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Clone, Deserialize, Serialize)]
|
|
||||||
pub struct WolvesVerify {
|
|
||||||
pub email: String,
|
|
||||||
pub discord: UserId,
|
|
||||||
pub auth_code: String,
|
|
||||||
pub date_expiry: String,
|
|
||||||
}
|
|
||||||
impl<'r> FromRow<'r, SqliteRow> for WolvesVerify {
|
|
||||||
fn from_row(row: &'r SqliteRow) -> Result<Self, Error> {
|
|
||||||
let user_tmp: i64 = row.try_get("discord")?;
|
|
||||||
let discord = UserId::from(user_tmp as u64);
|
|
||||||
|
|
||||||
Ok(Self {
|
|
||||||
email: row.try_get("email")?,
|
|
||||||
discord,
|
|
||||||
auth_code: row.try_get("auth_code")?,
|
|
||||||
date_expiry: row.try_get("date_expiry")?,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Clone, Deserialize, Serialize)]
|
|
||||||
pub struct Servers {
|
|
||||||
pub server: GuildId,
|
|
||||||
pub wolves_api: String,
|
|
||||||
pub role_past: Option<RoleId>,
|
|
||||||
pub role_current: Option<RoleId>,
|
|
||||||
pub member_past: i64,
|
|
||||||
pub member_current: i64,
|
|
||||||
}
|
|
||||||
impl<'r> FromRow<'r, SqliteRow> for Servers {
|
|
||||||
fn from_row(row: &'r SqliteRow) -> Result<Self, Error> {
|
|
||||||
let server_tmp: i64 = row.try_get("server")?;
|
|
||||||
let server = GuildId::from(server_tmp as u64);
|
|
||||||
let role_past = match row.try_get("role_past") {
|
|
||||||
Ok(x) => {
|
|
||||||
let tmp: i64 = x;
|
|
||||||
if tmp == 0 {
|
|
||||||
None
|
|
||||||
} else {
|
|
||||||
Some(RoleId::from(tmp as u64))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => None,
|
|
||||||
};
|
|
||||||
let role_current = match row.try_get("role_current") {
|
|
||||||
Ok(x) => {
|
|
||||||
let tmp: i64 = x;
|
|
||||||
if tmp == 0 {
|
|
||||||
None
|
|
||||||
} else {
|
|
||||||
Some(RoleId::from(tmp as u64))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => None,
|
|
||||||
};
|
|
||||||
|
|
||||||
Ok(Self {
|
|
||||||
server,
|
|
||||||
wolves_api: row.try_get("wolves_api")?,
|
|
||||||
role_past,
|
|
||||||
role_current,
|
|
||||||
member_past: row.try_get("member_past")?,
|
|
||||||
member_current: row.try_get("member_current")?,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn db_init(config: &Config) -> Result<Pool<Sqlite>, Error> {
|
|
||||||
let database = format!("{}/{}", &config.home, &config.database);
|
|
||||||
|
|
||||||
let pool = SqlitePoolOptions::new()
|
|
||||||
.max_connections(5)
|
|
||||||
.connect_with(
|
|
||||||
SqliteConnectOptions::from_str(&format!("sqlite://{}", database))?
|
|
||||||
.foreign_keys(true)
|
|
||||||
.create_if_missing(true),
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
sqlx::query(
|
|
||||||
"CREATE TABLE IF NOT EXISTS wolves (
|
|
||||||
id_wolves integer PRIMARY KEY,
|
|
||||||
email text not null,
|
|
||||||
discord integer,
|
|
||||||
minecraft text
|
|
||||||
)",
|
|
||||||
)
|
|
||||||
.execute(&pool)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
sqlx::query("CREATE INDEX IF NOT EXISTS index_discord ON wolves (discord)").execute(&pool).await?;
|
|
||||||
|
|
||||||
sqlx::query(
|
|
||||||
"CREATE TABLE IF NOT EXISTS wolves_verify (
|
|
||||||
discord integer PRIMARY KEY,
|
|
||||||
email text not null,
|
|
||||||
auth_code text not null,
|
|
||||||
date_expiry text not null
|
|
||||||
)",
|
|
||||||
)
|
|
||||||
.execute(&pool)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
sqlx::query("CREATE INDEX IF NOT EXISTS index_date_expiry ON wolves_verify (date_expiry)")
|
|
||||||
.execute(&pool)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
sqlx::query(
|
|
||||||
"CREATE TABLE IF NOT EXISTS server_members (
|
|
||||||
server integer not null,
|
|
||||||
id_wolves integer not null,
|
|
||||||
expiry text not null,
|
|
||||||
PRIMARY KEY(server,id_wolves),
|
|
||||||
FOREIGN KEY (id_wolves) REFERENCES wolves (id_wolves)
|
|
||||||
)",
|
|
||||||
)
|
|
||||||
.execute(&pool)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
sqlx::query(
|
|
||||||
"CREATE TABLE IF NOT EXISTS servers (
|
|
||||||
server integer PRIMARY KEY,
|
|
||||||
wolves_api text not null,
|
|
||||||
role_past integer,
|
|
||||||
role_current integer,
|
|
||||||
member_past integer DEFAULT 0,
|
|
||||||
member_current integer DEFAULT 0
|
|
||||||
)",
|
|
||||||
)
|
|
||||||
.execute(&pool)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
Ok(pool)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn get_server_config(db: &Pool<Sqlite>, server: &GuildId) -> Option<Servers> {
|
|
||||||
sqlx::query_as::<_, Servers>(
|
|
||||||
r#"
|
|
||||||
SELECT *
|
|
||||||
FROM servers
|
|
||||||
WHERE server = ?
|
|
||||||
"#,
|
|
||||||
)
|
|
||||||
.bind(*server.as_u64() as i64)
|
|
||||||
.fetch_one(db)
|
|
||||||
.await
|
|
||||||
.ok()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn get_server_member(db: &Pool<Sqlite>, server: &GuildId, member: &guild::Member) -> Result<ServerMembersWolves, Error> {
|
|
||||||
sqlx::query_as::<_, ServerMembersWolves>(
|
|
||||||
r#"
|
|
||||||
SELECT *
|
|
||||||
FROM server_members
|
|
||||||
JOIN wolves USING (id_wolves)
|
|
||||||
WHERE server = ? AND discord = ?
|
|
||||||
"#,
|
|
||||||
)
|
|
||||||
.bind(*server.as_u64() as i64)
|
|
||||||
.bind(*member.user.id.as_u64() as i64)
|
|
||||||
.fetch_one(db)
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn get_server_config_bulk(db: &Pool<Sqlite>) -> Vec<Servers> {
|
|
||||||
sqlx::query_as::<_, Servers>(
|
|
||||||
r#"
|
|
||||||
SELECT *
|
|
||||||
FROM servers
|
|
||||||
"#,
|
|
||||||
)
|
|
||||||
.fetch_all(db)
|
|
||||||
.await
|
|
||||||
.unwrap_or_default()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn get_now_iso(short: bool) -> String {
|
pub fn get_now_iso(short: bool) -> String {
|
||||||
let now = Utc::now();
|
let now = Utc::now();
|
||||||
if short {
|
if short {
|
||||||
|
@ -376,305 +123,5 @@ pub fn get_now_iso(short: bool) -> String {
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn random_string(len: usize) -> String {
|
pub fn random_string(len: usize) -> String {
|
||||||
thread_rng().sample_iter(&Alphanumeric).take(len).map(char::from).collect()
|
rng().sample_iter(&Alphanumeric).take(len).map(char::from).collect()
|
||||||
}
|
|
||||||
|
|
||||||
pub mod set_roles {
|
|
||||||
use super::*;
|
|
||||||
pub async fn update_server(ctx: &Context, server: &Servers, remove_roles: &[Option<RoleId>], members_changed: &Vec<UserId>) {
|
|
||||||
let db_lock = {
|
|
||||||
let data_read = ctx.data.read().await;
|
|
||||||
data_read.get::<DataBase>().expect("Expected Database in TypeMap.").clone()
|
|
||||||
};
|
|
||||||
|
|
||||||
let db = db_lock.read().await;
|
|
||||||
|
|
||||||
let Servers {
|
|
||||||
server,
|
|
||||||
role_past,
|
|
||||||
role_current,
|
|
||||||
..
|
|
||||||
} = server;
|
|
||||||
|
|
||||||
let mut roles_set = [0, 0, 0];
|
|
||||||
let mut members = vec![];
|
|
||||||
|
|
||||||
for member in get_server_member_bulk(&db, server).await {
|
|
||||||
if let Some(x) = member.discord {
|
|
||||||
members.push(x);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
let mut members_all = members.len();
|
|
||||||
|
|
||||||
if let Ok(x) = server.members(ctx, None, None).await {
|
|
||||||
for mut member in x {
|
|
||||||
// members_changed acts as an override to only deal with teh users in it
|
|
||||||
if !members_changed.is_empty() && !members_changed.contains(&member.user.id) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
if members.contains(&member.user.id) {
|
|
||||||
let mut roles = vec![];
|
|
||||||
|
|
||||||
if let Some(role) = &role_past {
|
|
||||||
if !member.roles.contains(role) {
|
|
||||||
roles_set[0] += 1;
|
|
||||||
roles.push(role.to_owned());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if let Some(role) = &role_current {
|
|
||||||
if !member.roles.contains(role) {
|
|
||||||
roles_set[1] += 1;
|
|
||||||
roles.push(role.to_owned());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if let Err(e) = member.add_roles(ctx, &roles).await {
|
|
||||||
println!("{:?}", e);
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
// old and never
|
|
||||||
|
|
||||||
if let Some(role) = &role_past {
|
|
||||||
if member.roles.contains(role) {
|
|
||||||
members_all += 1;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if let Some(role) = &role_current {
|
|
||||||
if member.roles.contains(role) {
|
|
||||||
roles_set[2] += 1;
|
|
||||||
// if theya re not a current member and have the role then remove it
|
|
||||||
if let Err(e) = member.remove_role(ctx, role).await {
|
|
||||||
println!("{:?}", e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for role in remove_roles.iter().flatten() {
|
|
||||||
if let Err(e) = member.remove_role(ctx, role).await {
|
|
||||||
println!("{:?}", e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
set_server_numbers(&db, server, members_all as i64, members.len() as i64).await;
|
|
||||||
|
|
||||||
// small bit of logging to note changes over time
|
|
||||||
println!("{:?} Changes: New: +{}, Current: +{}/-{}", server.as_u64(), roles_set[0], roles_set[1], roles_set[2]);
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn get_server_member_bulk(db: &Pool<Sqlite>, server: &GuildId) -> Vec<ServerMembersWolves> {
|
|
||||||
sqlx::query_as::<_, ServerMembersWolves>(
|
|
||||||
r#"
|
|
||||||
SELECT *
|
|
||||||
FROM server_members
|
|
||||||
JOIN wolves USING (id_wolves)
|
|
||||||
WHERE (
|
|
||||||
server = ?
|
|
||||||
AND discord IS NOT NULL
|
|
||||||
AND expiry > ?
|
|
||||||
)
|
|
||||||
"#,
|
|
||||||
)
|
|
||||||
.bind(*server.as_u64() as i64)
|
|
||||||
.bind(get_now_iso(true))
|
|
||||||
.fetch_all(db)
|
|
||||||
.await
|
|
||||||
.unwrap_or_default()
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn set_server_numbers(db: &Pool<Sqlite>, server: &GuildId, past: i64, current: i64) {
|
|
||||||
match sqlx::query_as::<_, Wolves>(
|
|
||||||
"
|
|
||||||
UPDATE servers
|
|
||||||
SET member_past = ?, member_current = ?
|
|
||||||
WHERE server = ?
|
|
||||||
",
|
|
||||||
)
|
|
||||||
.bind(past)
|
|
||||||
.bind(current)
|
|
||||||
.bind(*server.as_u64() as i64)
|
|
||||||
.fetch_optional(db)
|
|
||||||
.await
|
|
||||||
{
|
|
||||||
Ok(_) => {}
|
|
||||||
Err(e) => {
|
|
||||||
println!("Failure to insert into {}", server.as_u64());
|
|
||||||
println!("{:?}", e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub mod get_data {
|
|
||||||
use super::*;
|
|
||||||
use crate::set_roles::update_server;
|
|
||||||
use std::collections::BTreeMap;
|
|
||||||
|
|
||||||
#[derive(Deserialize, Serialize, Debug)]
|
|
||||||
struct WolvesResultUser {
|
|
||||||
committee: String,
|
|
||||||
wolves_id: String,
|
|
||||||
first_name: String,
|
|
||||||
last_name: String,
|
|
||||||
contact_email: String,
|
|
||||||
student_id: Option<String>,
|
|
||||||
note: Option<String>,
|
|
||||||
expiry: String,
|
|
||||||
requested: String,
|
|
||||||
approved: String,
|
|
||||||
sitename: String,
|
|
||||||
domain: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Deserialize, Serialize, Debug)]
|
|
||||||
struct WolvesResult {
|
|
||||||
success: i8,
|
|
||||||
result: Vec<WolvesResultUser>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Deserialize, Serialize, Debug)]
|
|
||||||
struct WolvesResultLocal {
|
|
||||||
pub id_wolves: String,
|
|
||||||
pub email: String,
|
|
||||||
pub expiry: String,
|
|
||||||
}
|
|
||||||
pub async fn get_wolves(ctx: &Context) {
|
|
||||||
let db_lock = {
|
|
||||||
let data_read = ctx.data.read().await;
|
|
||||||
data_read.get::<DataBase>().expect("Expected Database in TypeMap.").clone()
|
|
||||||
};
|
|
||||||
let db = db_lock.read().await;
|
|
||||||
|
|
||||||
let config_lock = {
|
|
||||||
let data_read = ctx.data.read().await;
|
|
||||||
data_read.get::<Config>().expect("Expected Config in TypeMap.").clone()
|
|
||||||
};
|
|
||||||
let config = config_lock.read().await;
|
|
||||||
|
|
||||||
for server_config in get_server_config_bulk(&db).await {
|
|
||||||
let Servers {
|
|
||||||
server,
|
|
||||||
wolves_api,
|
|
||||||
..
|
|
||||||
} = &server_config;
|
|
||||||
|
|
||||||
let existing_tmp = get_server_member(&db, server).await;
|
|
||||||
let existing = existing_tmp.iter().map(|data| (data.id_wolves, data)).collect::<BTreeMap<_, _>>();
|
|
||||||
|
|
||||||
// list of users that need to be updated for this server
|
|
||||||
let mut user_to_update = vec![];
|
|
||||||
for user in get_wolves_sub(&config, wolves_api).await {
|
|
||||||
let id = user.wolves_id.parse::<u64>().unwrap_or_default();
|
|
||||||
match existing.get(&(id as i64)) {
|
|
||||||
None => {
|
|
||||||
// user does not exist already, add everything
|
|
||||||
add_users_wolves(&db, &user).await;
|
|
||||||
add_users_server_members(&db, server, &user).await;
|
|
||||||
}
|
|
||||||
Some(old) => {
|
|
||||||
// always update wolves table, in case data has changed
|
|
||||||
add_users_wolves(&db, &user).await;
|
|
||||||
if old.expiry != user.expiry {
|
|
||||||
add_users_server_members(&db, server, &user).await;
|
|
||||||
|
|
||||||
if let Some(discord_id) = old.discord {
|
|
||||||
user_to_update.push(discord_id);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if !user_to_update.is_empty() {
|
|
||||||
update_server(ctx, &server_config, &[], &user_to_update).await;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn get_server_member(db: &Pool<Sqlite>, server: &GuildId) -> Vec<ServerMembersWolves> {
|
|
||||||
sqlx::query_as::<_, ServerMembersWolves>(
|
|
||||||
r#"
|
|
||||||
SELECT *
|
|
||||||
FROM server_members
|
|
||||||
JOIN wolves USING (id_wolves)
|
|
||||||
WHERE (
|
|
||||||
server = ?
|
|
||||||
AND discord IS NOT NULL
|
|
||||||
)
|
|
||||||
"#,
|
|
||||||
)
|
|
||||||
.bind(*server.as_u64() as i64)
|
|
||||||
.fetch_all(db)
|
|
||||||
.await
|
|
||||||
.unwrap_or_default()
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn get_wolves_sub(config: &Config, wolves_api: &str) -> Vec<WolvesResultUser> {
|
|
||||||
if config.wolves_url.is_empty() {
|
|
||||||
return vec![];
|
|
||||||
}
|
|
||||||
|
|
||||||
// get wolves data
|
|
||||||
if let Ok(mut res) = surf::post(&config.wolves_url).header("X-AM-Identity", wolves_api).await {
|
|
||||||
if let Ok(WolvesResult {
|
|
||||||
success,
|
|
||||||
result,
|
|
||||||
}) = res.body_json().await
|
|
||||||
{
|
|
||||||
if success != 1 {
|
|
||||||
return vec![];
|
|
||||||
}
|
|
||||||
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
vec![]
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn add_users_wolves(db: &Pool<Sqlite>, user: &WolvesResultUser) {
|
|
||||||
// expiry
|
|
||||||
match sqlx::query_as::<_, Wolves>(
|
|
||||||
"
|
|
||||||
INSERT INTO wolves (id_wolves, email)
|
|
||||||
VALUES ($1, $2)
|
|
||||||
ON CONFLICT(id_wolves) DO UPDATE SET email = $2
|
|
||||||
",
|
|
||||||
)
|
|
||||||
.bind(&user.wolves_id)
|
|
||||||
.bind(&user.contact_email)
|
|
||||||
.fetch_optional(db)
|
|
||||||
.await
|
|
||||||
{
|
|
||||||
Ok(_) => {}
|
|
||||||
Err(e) => {
|
|
||||||
println!("Failure to insert into Wolves {:?}", user);
|
|
||||||
println!("{:?}", e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
async fn add_users_server_members(db: &Pool<Sqlite>, server: &GuildId, user: &WolvesResultUser) {
|
|
||||||
match sqlx::query_as::<_, ServerMembers>(
|
|
||||||
"
|
|
||||||
INSERT OR REPLACE INTO server_members (server, id_wolves, expiry)
|
|
||||||
VALUES (?1, ?2, ?3)
|
|
||||||
",
|
|
||||||
)
|
|
||||||
.bind(*server.as_u64() as i64)
|
|
||||||
.bind(&user.wolves_id)
|
|
||||||
.bind(&user.expiry)
|
|
||||||
.fetch_optional(db)
|
|
||||||
.await
|
|
||||||
{
|
|
||||||
Ok(_) => {}
|
|
||||||
Err(e) => {
|
|
||||||
println!("Failure to insert into ServerMembers {} {:?}", server.as_u64(), user);
|
|
||||||
println!("{:?}", e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
170
src/main.rs
170
src/main.rs
|
@ -1,32 +1,52 @@
|
||||||
mod commands;
|
pub mod commands;
|
||||||
|
|
||||||
|
use crate::commands::role_adder::tools::on_role_change;
|
||||||
|
use serenity::all::{ActivityData, Command, CreateMessage, EditInteractionResponse, GuildId, GuildMemberUpdateEvent, Interaction};
|
||||||
|
use serenity::model::guild::Member;
|
||||||
use serenity::{
|
use serenity::{
|
||||||
async_trait,
|
async_trait,
|
||||||
client::{Context, EventHandler},
|
client::{Context, EventHandler},
|
||||||
model::{
|
model::{
|
||||||
application::{command::Command, interaction::Interaction},
|
|
||||||
gateway::{GatewayIntents, Ready},
|
gateway::{GatewayIntents, Ready},
|
||||||
guild,
|
user::OnlineStatus,
|
||||||
},
|
},
|
||||||
Client,
|
Client,
|
||||||
};
|
};
|
||||||
|
use skynet_discord_bot::common::database::{db_init, get_server_config, get_server_member, DataBase};
|
||||||
|
use skynet_discord_bot::common::set_roles::committee::update_committees;
|
||||||
|
use skynet_discord_bot::common::wolves::committees::Committees;
|
||||||
|
use skynet_discord_bot::{get_config, Config};
|
||||||
|
use sqlx::{Pool, Sqlite};
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
use skynet_discord_bot::{db_init, get_config, get_server_config, get_server_member, Config, DataBase};
|
|
||||||
use tokio::sync::RwLock;
|
use tokio::sync::RwLock;
|
||||||
|
|
||||||
struct Handler;
|
struct Handler;
|
||||||
|
|
||||||
#[async_trait]
|
#[async_trait]
|
||||||
impl EventHandler for Handler {
|
impl EventHandler for Handler {
|
||||||
async fn guild_member_addition(&self, ctx: Context, mut new_member: guild::Member) {
|
// handles previously linked accounts joining the server
|
||||||
|
async fn guild_member_addition(&self, ctx: Context, new_member: Member) {
|
||||||
let db_lock = {
|
let db_lock = {
|
||||||
let data_read = ctx.data.read().await;
|
let data_read = ctx.data.read().await;
|
||||||
data_read.get::<DataBase>().expect("Expected Config in TypeMap.").clone()
|
data_read.get::<DataBase>().expect("Expected Config in TypeMap.").clone()
|
||||||
};
|
};
|
||||||
|
|
||||||
let db = db_lock.read().await;
|
let db = db_lock.read().await;
|
||||||
let config = match get_server_config(&db, &new_member.guild_id).await {
|
|
||||||
|
let config_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<Config>().expect("Expected Config in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
let config_global = config_lock.read().await;
|
||||||
|
|
||||||
|
// committee server takes priority
|
||||||
|
if new_member.guild_id.eq(&config_global.committee_server) {
|
||||||
|
let mut member = vec![new_member.clone()];
|
||||||
|
update_committees(&db, &ctx, &config_global, &mut member).await;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
let config_server = match get_server_config(&db, &new_member.guild_id).await {
|
||||||
None => return,
|
None => return,
|
||||||
Some(x) => x,
|
Some(x) => x,
|
||||||
};
|
};
|
||||||
|
@ -34,33 +54,102 @@ impl EventHandler for Handler {
|
||||||
if get_server_member(&db, &new_member.guild_id, &new_member).await.is_ok() {
|
if get_server_member(&db, &new_member.guild_id, &new_member).await.is_ok() {
|
||||||
let mut roles = vec![];
|
let mut roles = vec![];
|
||||||
|
|
||||||
if let Some(role) = &config.role_past {
|
if let Some(role) = &config_server.role_past {
|
||||||
if !new_member.roles.contains(role) {
|
if !new_member.roles.contains(role) {
|
||||||
roles.push(role.to_owned());
|
roles.push(role.to_owned());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Some(role) = &config.role_current {
|
if !new_member.roles.contains(&config_server.role_current) {
|
||||||
if !new_member.roles.contains(role) {
|
roles.push(config_server.role_current.to_owned());
|
||||||
roles.push(role.to_owned());
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Err(e) = new_member.add_roles(&ctx, &roles).await {
|
if let Err(e) = new_member.add_roles(&ctx, &roles).await {
|
||||||
println!("{:?}", e);
|
println!("{:?}", e);
|
||||||
}
|
}
|
||||||
|
} else {
|
||||||
|
let tmp = get_committee(&db, config_server.wolves_id).await;
|
||||||
|
if !tmp.is_empty() {
|
||||||
|
let committee = &tmp[0];
|
||||||
|
let msg = format!(
|
||||||
|
r#"
|
||||||
|
Welcome {} to the {} server!
|
||||||
|
Sign up on [UL Wolves]({}) and go to https://discord.com/channels/{}/{} and use ``/link_wolves`` to get full access.
|
||||||
|
"#,
|
||||||
|
new_member.display_name(),
|
||||||
|
committee.name_full,
|
||||||
|
committee.link,
|
||||||
|
&config_server.server,
|
||||||
|
&config_server.bot_channel_id
|
||||||
|
);
|
||||||
|
|
||||||
|
if let Err(err) = new_member.user.direct_message(&ctx, CreateMessage::new().content(&msg)).await {
|
||||||
|
dbg!(err);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// handles role updates
|
||||||
|
async fn guild_member_update(&self, ctx: Context, _old_data: Option<Member>, new_data: Option<Member>, _: GuildMemberUpdateEvent) {
|
||||||
|
// get config/db
|
||||||
|
let db_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<DataBase>().expect("Expected Config in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
|
||||||
|
let db = db_lock.read().await;
|
||||||
|
|
||||||
|
// check if the role changed is part of the oens for this server
|
||||||
|
if let Some(x) = new_data {
|
||||||
|
on_role_change(&db, &ctx, x).await;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn ready(&self, ctx: Context, ready: Ready) {
|
async fn ready(&self, ctx: Context, ready: Ready) {
|
||||||
println!("[Main] {} is connected!", ready.user.name);
|
println!("[Main] {} is connected!", ready.user.name);
|
||||||
|
ctx.set_presence(Some(ActivityData::playing("with humanity's fate")), OnlineStatus::Online);
|
||||||
|
|
||||||
match Command::set_global_application_commands(&ctx.http, |commands| {
|
match Command::set_global_commands(
|
||||||
commands
|
&ctx.http,
|
||||||
.create_application_command(|command| commands::add_server::register(command))
|
vec![
|
||||||
.create_application_command(|command| commands::link_email::link::register(command))
|
commands::add_server::register(),
|
||||||
.create_application_command(|command| commands::link_email::verify::register(command))
|
commands::role_adder::edit::register(),
|
||||||
})
|
commands::link_email::link::register(),
|
||||||
|
commands::link_email::verify::register(),
|
||||||
|
commands::minecraft::server::add::register(),
|
||||||
|
commands::minecraft::server::list::register(),
|
||||||
|
commands::minecraft::server::delete::register(),
|
||||||
|
commands::minecraft::user::add::register(),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(_) => {}
|
||||||
|
Err(e) => {
|
||||||
|
println!("{:?}", e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let config_lock = {
|
||||||
|
let data_read = ctx.data.read().await;
|
||||||
|
data_read.get::<Config>().expect("Expected Config in TypeMap.").clone()
|
||||||
|
};
|
||||||
|
let config_global = config_lock.read().await;
|
||||||
|
|
||||||
|
match &config_global
|
||||||
|
.committee_server
|
||||||
|
.set_commands(&ctx.http, vec![commands::count::committee::register()])
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(_) => {}
|
||||||
|
Err(e) => {
|
||||||
|
println!("{:?}", e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
match GuildId::new(689189992417067052)
|
||||||
|
.set_commands(&ctx.http, vec![commands::count::servers::register()])
|
||||||
.await
|
.await
|
||||||
{
|
{
|
||||||
Ok(_) => {}
|
Ok(_) => {}
|
||||||
|
@ -71,30 +160,63 @@ impl EventHandler for Handler {
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn interaction_create(&self, ctx: Context, interaction: Interaction) {
|
async fn interaction_create(&self, ctx: Context, interaction: Interaction) {
|
||||||
if let Interaction::ApplicationCommand(command) = interaction {
|
if let Interaction::Command(command) = interaction {
|
||||||
let _ = command.defer_ephemeral(&ctx.http).await;
|
let _ = command.defer_ephemeral(&ctx.http).await;
|
||||||
//println!("Received command interaction: {:#?}", command);
|
//println!("Received command interaction: {:#?}", command);
|
||||||
|
|
||||||
let content = match command.data.name.as_str() {
|
let content = match command.data.name.as_str() {
|
||||||
"add" => commands::add_server::run(&command, &ctx).await,
|
// user commands
|
||||||
"link" => commands::link_email::link::run(&command, &ctx).await,
|
"link_wolves" => commands::link_email::link::run(&command, &ctx).await,
|
||||||
"verify" => commands::link_email::verify::run(&command, &ctx).await,
|
"verify" => commands::link_email::verify::run(&command, &ctx).await,
|
||||||
_ => "not implemented :(".to_string(),
|
"link_minecraft" => commands::minecraft::user::add::run(&command, &ctx).await,
|
||||||
|
// admin commands
|
||||||
|
"add" => commands::add_server::run(&command, &ctx).await,
|
||||||
|
"roles_adder" => commands::role_adder::edit::run(&command, &ctx).await,
|
||||||
|
"minecraft_add" => commands::minecraft::server::add::run(&command, &ctx).await,
|
||||||
|
"minecraft_list" => commands::minecraft::server::list::run(&command, &ctx).await,
|
||||||
|
"minecraft_delete" => commands::minecraft::server::delete::run(&command, &ctx).await,
|
||||||
|
// sub command
|
||||||
|
"count" => match command.data.options.first() {
|
||||||
|
None => "Invalid Command".to_string(),
|
||||||
|
Some(x) => match x.name.as_str() {
|
||||||
|
"committee" => commands::count::committee::run(&command, &ctx).await,
|
||||||
|
"servers" => commands::count::servers::run(&command, &ctx).await,
|
||||||
|
&_ => format!("not implemented :( count {}", x.name.as_str()),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
_ => format!("not implemented :( {}", command.data.name.as_str()),
|
||||||
};
|
};
|
||||||
|
|
||||||
if let Err(why) = command.edit_original_interaction_response(&ctx.http, |response| response.content(content)).await {
|
if let Err(why) = command.edit_response(&ctx.http, EditInteractionResponse::new().content(content)).await {
|
||||||
println!("Cannot respond to slash command: {}", why);
|
println!("Cannot respond to slash command: {}", why);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn get_committee(db: &Pool<Sqlite>, wolves_id: i64) -> Vec<Committees> {
|
||||||
|
sqlx::query_as::<_, Committees>(
|
||||||
|
r#"
|
||||||
|
SELECT *
|
||||||
|
FROM committees
|
||||||
|
WHERE id = ?
|
||||||
|
"#,
|
||||||
|
)
|
||||||
|
.bind(wolves_id)
|
||||||
|
.fetch_all(db)
|
||||||
|
.await
|
||||||
|
.unwrap_or_default()
|
||||||
|
}
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() {
|
async fn main() {
|
||||||
let config = get_config();
|
let config = get_config();
|
||||||
let db = match db_init(&config).await {
|
let db = match db_init(&config).await {
|
||||||
Ok(x) => x,
|
Ok(x) => x,
|
||||||
Err(_) => return,
|
Err(err) => {
|
||||||
|
dbg!(err);
|
||||||
|
return;
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
// Intents are a bitflag, bitwise operations can be used to dictate which intents to use
|
// Intents are a bitflag, bitwise operations can be used to dictate which intents to use
|
||||||
|
|
Loading…
Add table
Reference in a new issue