42CTF/bot
42CTF
/
bot
3
1
Fork 2
Code Issues Pull Requests Projects Releases Wiki Activity

Compare commits

base: 42CTF:bb25b8559d9b255d578cc0387852f88f03839566
Branches Tags
42CTF:main
42CTF:docker-compose
..
compare: 42CTF:f364e545cd508ed51144a016319507a2a768c436
Branches Tags
42CTF:main
42CTF:docker-compose

No commits in common. "bb25b8559d9b255d578cc0387852f88f03839566" and "f364e545cd508ed51144a016319507a2a768c436" have entirely different histories.
bb25b8559d ... f364e545cd

13 changed files with 4442 additions and 320 deletions
Show Stats Expand all files Collapse all files

12
Dockerfile
View File

@ -1,12 +0,0 @@
FROM golang:1.19
WORKDIR /usr/src/app/
COPY go.mod go.sum ./
RUN go mod download && go mod verify
COPY . .
RUN go build -v -o /usr/local/bin/bot ./...
CMD ["bot"]

19
README.md
View File

@ -11,22 +11,5 @@ AUTH_TOKEN=<api_42CTF_token>
And then run it with : And then run it with :
``` ```
docker-compose up --build -d docker-compose up -d
``` ```
By default it will run on 42CTF Guild and with RolesID from this server.
You can run :
```
docker-compose run discord_bot [--wait WAIT] [--force] [--dryrun]
```
## Arguments
| arg | Default | Usage |
|:------------:|:-------:|:---------------------------------------------:|
| --guild / -g | `nil` | Guild ID |
| --top1 | `nil` | TOP1 Role ID |
| --top10 | `nil` | TOP10 Role ID |
| --top50 | `nil` | TOP10 Role ID |
| --wait | `1` | Minutes to wait between runs |
| --force | `false` | Force update of all users |
| --dryrun | `false` | Display log but don't execute discord command |

33
api.go
View File

@ -1,33 +0,0 @@
package main
import (
"encoding/json"
"io"
"net/http"
"time"
)
func getUsers42CTF() (map[string]int, error) {
var jsonParse map[string]int
client := http.Client{ Timeout: time.Second * 2}
req, err := http.NewRequest(http.MethodGet, api, nil)
if err != nil {
return nil, err
}
res, err := client.Do(req)
if err != nil {
return nil, err
}
defer res.Body.Close()
body, err := io.ReadAll(res.Body)
if err != nil {
return nil, err
}
err = json.Unmarshal([]byte(body), &jsonParse)
if err != nil {
return nil, err
}
return jsonParse, nil
}

24
discord.go
View File

@ -1,24 +0,0 @@
package main
import (
"github.com/sirupsen/logrus"
"github.com/bwmarrin/discordgo"
)
func removeRole(discordClient *discordgo.Session, guildID string, userID string, roleID string, dryrun bool) {
if ! dryrun {
err := discordClient.GuildMemberRoleRemove(guild, userID, roleID)
if err != nil {
logrus.Error(err.Error())
}
}
}
func addRole(discordClient *discordgo.Session, guildID string, userID string, roleID string, dryrun bool) {
if ! dryrun {
err := discordClient.GuildMemberRoleAdd(guild, userID, roleID)
if err != nil {
logrus.Error(err.Error())
}
}
}

3
docker-compose.yaml
View File

@ -2,7 +2,6 @@ version: "2"
services: services:
bot_discord: bot_discord:
restart: always restart: always
build: . build: ./src
container_name: bot_discord container_name: bot_discord
env_file: ./env_file env_file: ./env_file
entrypoint: bot --guild=943459216901955604 --top1=798638767359524875 --top10=801787467064672286 --top50=803729539145924649

17
go.mod
View File

@ -1,17 +0,0 @@
module gitea.42ctf.org/42CTF/bot
go 1.20
require (
github.com/bwmarrin/discordgo v0.27.0
golang.org/x/exp v0.0.0-20230203172020-98cc5a0785f9
)
require (
github.com/alexflint/go-arg v1.4.3 // indirect
github.com/alexflint/go-scalar v1.1.0 // indirect
github.com/gorilla/websocket v1.4.2 // indirect
github.com/sirupsen/logrus v1.9.0 // indirect
golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b // indirect
golang.org/x/sys v0.1.0 // indirect
)

30
go.sum
View File

@ -1,30 +0,0 @@
github.com/alexflint/go-arg v1.4.3 h1:9rwwEBpMXfKQKceuZfYcwuc/7YY7tWJbFsgG5cAU/uo=
github.com/alexflint/go-arg v1.4.3/go.mod h1:3PZ/wp/8HuqRZMUUgu7I+e1qcpUbvmS258mRXkFH4IA=
github.com/alexflint/go-scalar v1.1.0 h1:aaAouLLzI9TChcPXotr6gUhq+Scr8rl0P9P4PnltbhM=
github.com/alexflint/go-scalar v1.1.0/go.mod h1:LoFvNMqS1CPrMVltza4LvnGKhaSpc3oyLEBUZVhhS2o=
github.com/bwmarrin/discordgo v0.27.0 h1:4ZK9KN+rGIxZ0fdGTmgdCcliQeW8Zhu6MnlFI92nf0Q=
github.com/bwmarrin/discordgo v0.27.0/go.mod h1:NJZpH+1AfhIcyQsPeuBKsUtYrRnjkyu0kIVMCHkZtRY=
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/gorilla/websocket v1.4.2 h1:+/TMaTYc4QFitKJxsQ7Yye35DkWvkdLcvGKqM+x0Ufc=
github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/sirupsen/logrus v1.9.0 h1:trlNQbNUG3OdDrDil03MCb1H2o9nJ1x4/5LYw7byDE0=
github.com/sirupsen/logrus v1.9.0/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b h1:7mWr3k41Qtv8XlltBkDkl8LoP3mpSgBW8BUoxtEdbXg=
golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b/go.mod h1:T9bdIzuCu7OtxOm1hfPfRQxPLYneinmdGuTeoZ9dtd4=
golang.org/x/exp v0.0.0-20230203172020-98cc5a0785f9 h1:frX3nT9RkKybPnjyI+yvZh6ZucTZatCCEm9D47sZ2zo=
golang.org/x/exp v0.0.0-20230203172020-98cc5a0785f9/go.mod h1:CxIveKay+FTh1D0yPZemJVgC/95VzuuOLq5Qi4xnoYc=
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.1.0 h1:kunALQeHf1/185U1i0GOB/fy1IPRDDpuoOOqRReG57U=
golang.org/x/sys v0.1.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

65
main.go
View File

@ -1,65 +0,0 @@
package main
import (
"fmt"
"os"
"time"
"github.com/alexflint/go-arg"
"github.com/bwmarrin/discordgo"
"github.com/sirupsen/logrus"
)
type UserRole struct {
Rank int
CurrentRole string
}
// Global var (I know, and i don't care)
var api string = "https://preprod.42ctf.org/api/bot/discord?token=" + os.Getenv("AUTH_TOKEN")
var guild string
var rolesID map[string]string
var Users map[string]*UserRole
var args struct {
Guild string `arg:"-g,required,env:GUILD_ID"`
Top1 string `arg:"env:TOP_1"`
Top10 string `arg:"env:TOP_10"`
Top50 string `arg:"env:TOP_50"`
Wait int `default:"2" help:"Minutes between run"`
Force bool
DryRun bool
}
func main() {
fmt.Println(len(os.Args), os.Args)
arg.MustParse(&args)
guild = args.Guild
rolesID = make(map[string]string)
rolesID["top1"] = args.Top1
rolesID["top10"] = args.Top10
rolesID["top50"] = args.Top50
discord, err := discordgo.New("Bot " + os.Getenv("DISCORD_TOKEN"))
if err != nil {
logrus.Fatal(err.Error())
}
if err := populateUserRole(discord); err != nil {
logrus.Fatal(err.Error())
}
if args.DryRun {
logrus.Warn("Running in Dry Run mode")
}
if args.Force {
logrus.Warn("Force update of all users")
forceUpdate(discord, args.DryRun)
}
for {
logrus.Info("Start update")
updateUserRole(discord, args.DryRun)
logrus.Info("End update, sleep for 2 minutes")
time.Sleep(time.Duration(args.Wait) * time.Minute)
}
}

119
rank.go
View File

@ -1,119 +0,0 @@
package main
import (
"github.com/bwmarrin/discordgo"
"github.com/sirupsen/logrus"
"golang.org/x/exp/slices"
)
// Populate Users map for the first launch
// This variable will be keep as sort of cache for all
// operation
func populateUserRole(discordClient *discordgo.Session) error {
Users = make(map[string]*UserRole)
APIResponse, err := getUsers42CTF()
if err != nil {
return err
}
for id, rank := range APIResponse {
tmp, err := discordClient.GuildMember(guild, id)
if err == nil {
if slices.Contains(tmp.Roles, rolesID["top1"]) {
Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top1"]}
} else if slices.Contains(tmp.Roles, rolesID["top10"]) {
Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top10"]}
} else if slices.Contains(tmp.Roles, rolesID["top50"]) {
Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top50"]}
} else {
Users[id] = &UserRole{Rank: rank, CurrentRole: ""}
}
} else {
logrus.Error("ID not found", id)
}
}
return nil
}
func updateUserRole(discordClient *discordgo.Session, dryrun bool) error {
APIResponse, err := getUsers42CTF()
if err != nil {
return err
}
for id, rank := range APIResponse {
tmp, err := discordClient.GuildMember(guild, id)
if err == nil {
if slices.Contains(tmp.Roles, rolesID["top1"]) {
Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top1"]}
} else if slices.Contains(tmp.Roles, rolesID["top10"]) {
Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top10"]}
} else if slices.Contains(tmp.Roles, rolesID["top50"]) {
Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top50"]}
} else {
Users[id] = &UserRole{Rank: rank, CurrentRole: ""}
}
} else {
logrus.Error("ID not found", id)
}
if Users[id].Rank != rank {
logrus.Info("New rank (", rank, ") for user ", id)
if Users[id].CurrentRole != "" {
logrus.Info("Remove role for ", id)
removeRole(discordClient, guild, id, Users[id].CurrentRole, dryrun)
Users[id] = &UserRole{Rank: rank, CurrentRole: ""}
}
if rank == 1 {
logrus.Info("Add role `top1` for ", id)
addRole(discordClient, guild, id, rolesID["top1"], dryrun)
Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top1"]}
} else if rank <= 10 {
logrus.Info("Add role `top10` for ", id)
addRole(discordClient, guild, id, rolesID["top10"], dryrun)
Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top10"]}
} else if rank <= 50 {
logrus.Info("Add role `top50` for ", id)
addRole(discordClient, guild, id, rolesID["top50"], dryrun)
Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top50"]}
}
}
}
return nil
}
func forceUpdate(discordClient *discordgo.Session, dryrun bool) (error) {
APIResponse, err := getUsers42CTF()
if err != nil {
return err
}
for id, rank := range APIResponse {
tmp, err := discordClient.GuildMember(guild, id)
if err == nil {
if slices.Contains(tmp.Roles, rolesID["top1"]) {
removeRole(discordClient, guild, id, rolesID["top1"], dryrun)
}
if slices.Contains(tmp.Roles, rolesID["top10"]) {
removeRole(discordClient, guild, id, rolesID["top10"], dryrun)
}
if slices.Contains(tmp.Roles, rolesID["top50"]) {
removeRole(discordClient, guild, id, rolesID["top50"], dryrun)
}
Users[id] = &UserRole{Rank: rank, CurrentRole: ""}
} else {
logrus.Error("ID not found", id)
}
if rank == 1 {
logrus.Info("Add role `top1` for ", id)
addRole(discordClient, guild, id, rolesID["top1"], dryrun)
Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top1"]}
} else if rank <= 10 {
logrus.Info("Add role `top10` for ", id)
addRole(discordClient, guild, id, rolesID["top10"], dryrun)
Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top10"]}
} else if rank <= 50 {
logrus.Info("Add role `top50` for ", id)
addRole(discordClient, guild, id, rolesID["top50"], dryrun)
Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top50"]}
}
}
return nil
}

15
src/Dockerfile Normal file
View File

@ -0,0 +1,15 @@
FROM debian:9
ADD . /app/
WORKDIR /app
RUN apt-get update -yq \
&& apt-get install curl gnupg -yq \
&& curl -sL https://deb.nodesource.com/setup_16.x | bash - \
&& apt-get install nodejs -yq \
&& apt-get clean -y \
&& npm i
CMD node bot.js

147
src/bot.js Normal file
View File

@ -0,0 +1,147 @@
const { Client, Intents } = require('discord.js');
const CLIENT = new Client({ intents: [Intents.FLAGS.GUILDS, Intents.FLAGS.GUILD_MEMBERS] });
const https = require('https');
const D_TOKEN = process.env.DISCORD_TOKEN;
const O_TOKEN = process.env.AUTH_TOKEN;
async function get_guildID()
{
let Guilds = CLIENT.guilds;
let v = (await Guilds.fetch()).keyAt(0);
return v;
}
async function remove_notlogged(){
let guild = CLIENT.guilds.resolve(await get_guildID());
let token = O_TOKEN;
let url = `https://www.42ctf.local/api/bot/discord?token=${token}`;
await (guild.members.fetch()).then(() => {
console.log("Cache reloaded !");
});
const rolesCollection = new Map([
['top1', guild.roles.cache.get("798638767359524875")],
['top10', guild.roles.cache.get("801787467064672286")],
['top50', guild.roles.cache.get("803729539145924649")]
]);
const options = {
path: '/',
method: 'GET',
rejectUnauthorized: false,
requestCert: true,
agent: false
};
https.get(url, options, (res) => {
let body = "";
res.on("data", (chunk) => {
body += chunk;
});
res.on("end", () => {
try {
var json = JSON.parse(body);
let v = new Map(Object.entries(json))
let members = guild.members.cache;
members.forEach(async (key, value) => {
var r = false;
v.forEach((jkey, jvalue) => {
if (jvalue == key)
r = true;
})
if (r == false)
{
let user = await guild.members.fetch(key);
await user.roles.remove(rolesCollection.get('top1'));
await user.roles.remove(rolesCollection.get('top10'));
await user.roles.remove(rolesCollection.get('top50'));
}
})
}
catch (error) {
console.error(error.message);
};
});
}).on("error", (error) => {
console.log(error.message);
});
}
async function rank(token = O_TOKEN){
let url = `https://www.42ctf.local/api/bot/discord?token=${token}`;
let guild = CLIENT.guilds.resolve(await get_guildID());
const rolesCollection = new Map([
['top1', guild.roles.cache.get("798638767359524875")],
['top10', guild.roles.cache.get("801787467064672286")],
['top50', guild.roles.cache.get("803729539145924649")]
]);
const options = {
path: '/',
method: 'GET',
rejectUnauthorized: false,
requestCert: true,
agent: false
};
https.get(url,options,(res) => {
var body = "";
res.on("data", (chunk) => {
body += chunk;
});
res.on("end", () => {
try {
var json = JSON.parse(body);
let v = new Map(Object.entries(json))
v.forEach(async (key, value) => {
let user = guild.members.fetch(value);
let rank = parseInt(key);
if(rank == 1)
{
await (await user).roles.add(rolesCollection.get('top1'));
await (await user).roles.remove(rolesCollection.get('top10'));
await (await user).roles.remove(rolesCollection.get('top50'));
}
else if(rank <= 10)
{
await (await user).roles.remove(rolesCollection.get('top1'));
await (await user).roles.add(rolesCollection.get('top10'));
await (await user).roles.remove(rolesCollection.get('top50'));
}
else if(rank <= 50)
{
await (await user).roles.remove(rolesCollection.get('top1'));
await (await user).roles.remove(rolesCollection.get('top10'));
await (await user).roles.add(rolesCollection.get('top50'));
}
else
{
await (await user).roles.remove(rolesCollection.get('top1'));
await (await user).roles.remove(rolesCollection.get('top10'));
await (await user).roles.remove(rolesCollection.get('top50'));
}
})
} catch (error) {
console.error(error.message);
};
});
}).on("error", (error) => {
console.log(error.message);
});
}
CLIENT.on('ready', () => {
console.log(`Logged in as ${CLIENT.user.tag}!`);
setInterval(rank, 900000);
setInterval(remove_notlogged, 900000);
console.log('started rank() interval !');
});
CLIENT.login(D_TOKEN);

4270
src/package-lock.json generated Normal file
View File

File diff suppressed because it is too large Load Diff

8
src/package.json Normal file
View File

@ -0,0 +1,8 @@
{
"dependencies": {
"@discordjs/rest": "*",
"discord-api-types": "*",
"discord.js": "*",
"npm": "*"
}
}