bot.py

@@ -0,0 +1,104 @@
+import os
+import discord
+import discord.utils
+import urllib.request, json
+import asyncio
+import json
+import logging
+
+TOKEN = os.getenv('DISCORD_TOKEN')
+GUILD = '42ctf'
+
+intents = discord.Intents.all()
+client = discord.Client(intents=intents)
+
+db_file = open('members.json', 'r')
+users = json.load(db_file)
+db_file.close()
+
+logging.basicConfig(filename='bot.log', format='%(asctime)s %(message)s', level=logging.INFO)
+
+guild = ''
+roles = {}
+
+def get_rank(token):
+    url = urllib.request.urlopen("https://www.42ctf.org/accounts/rank/" + token)
+    data = json.loads(url.read().decode())
+    rank = data['rank']
+    return rank
+
+async def watch_roles():
+    global users
+    await client.wait_until_ready() # ensures cache is loaded
+    while not client.is_closed():
+        for member_id, token in users.items():
+            if (token == "0000"):
+                continue
+            member = discord.utils.get(guild.members, id=int(member_id))
+            rank = get_rank(token)
+            if rank == 1 and roles['top1'] not in member.roles:
+                await member.add_roles(roles['top1'])
+                await member.remove_roles(roles['top10'])
+                await member.remove_roles(roles['top50'])
+            elif rank > 1 and rank <= 10 and roles['top10'] not in member.roles:
+                await member.add_roles(roles['top10'])
+                await member.remove_roles(roles['top1'])
+                await member.remove_roles(roles['top50'])
+            elif rank > 10 and rank <= 50 and roles['top50'] not in member.roles:
+                await member.add_roles(roles['top50'])
+                await member.remove_roles(roles['top10'])
+                await member.remove_roles(roles['top1'])
+            elif rank > 50:
+                await member.remove_roles(roles['top1'])
+                await member.remove_roles(roles['top10'])
+                await member.remove_roles(roles['top50'])
+        await asyncio.sleep(60)
+
+@client.event
+async def on_ready():
+    global guild, roles
+    guild = discord.utils.get(client.guilds, name=GUILD)
+    roles['top10'] = discord.utils.get(guild.roles, id=801787467064672286)
+    roles['top1'] = discord.utils.get(guild.roles, id=798638767359524875)
+    roles['top50'] = discord.utils.get(guild.roles, id=803729539145924649)
+
+    logging.info('%s is connected to the following guild: %s(id: %d)', client.user, guild.name, guild.id)
+    client.loop.create_task(watch_roles())
+
+@client.event
+async def on_message(message):
+    global guild, roles
+
+    if message.author == client.user:
+        return
+
+    if '!connect' in message.content:
+        try:
+            user_token = message.content.split(' ')[1]
+            member = discord.utils.get(guild.members, name=message.author.name)
+            rank = get_rank(user_token)
+            users[str(member.id)] = user_token
+            logging.info("MESSAGE: from %s with token %s", message.author.name, user_token)
+            with open('members.json', 'w') as json_file:
+                json.dump(users, json_file)
+            if rank == 1:
+                await member.add_roles(roles['top1'])
+                response = "Congratulations, you're now Top 1. But for how long ?"
+            
+            elif (rank <= 10):
+                await member.add_roles(roles['top10'])
+                response = "You've been granted the Top 10 role. Now, go away and flag !"
+            
+            elif rank <= 50:
+                await member.add_roles(roles['top50'])
+                response = "You've been granted the Top 50 role. Now, go away and flag !"
+
+            else:
+                response = "No role for you now, but I'll keep watching you."
+        except IndexError:
+            response = 'usage: !connect 42ctf_token'
+        await message.author.create_dm()
+        await message.author.dm_channel.send(response)
+
+
+client.run(TOKEN)

src/accounts/templates/accounts/edit.html

@@ -40,30 +40,7 @@
 		</div>
 	</div>
 
-		<div class="ctf-block">
-			<div class="ctf-head">
-				<h3>{% trans "Connected accounts" %}</h3>
   </div>
-			<div class="bloc-body">
-				<div class="d-flex">
-					{% if user.userprofileinfo.discord_id|length > 0 %}
-					<form action="{% url 'accounts:connections-disconnect-discord' %}" method='POST'
-						class="form-inline p-2">
-						{%csrf_token%}
-						<button class="btn btn-dark" type="submit">{% trans "Disconnect Discord" %}</button>
-					</form>
-					{% else %}
-					<form action="{% url 'accounts:connections-connect-discord' %}" method='POST'
-						class="form-inline p-2">
-						{%csrf_token%}
-						<button class="btn btn-dark" type="submit">{% trans "Connect Discord" %}</button>
-					</form>
-					{% endif %}
-				</div>
-			</div>
-		</div>
-	</div>
-
   <div class="d-none d-md-block col-10 col-md-3 right-sidebar">
     <ul class="list-group">
 		<li class="list-group-item">{{ user.username }}</li>
@@ -88,3 +65,4 @@
   </div>
 </div>
 {% endblock %}
+

src/accounts/views/connection.py

@@ -4,7 +4,6 @@ from django.views.decorators.http import require_POST
 from django.views.defaults import bad_request
 from django.urls import reverse
 from django.shortcuts import redirect
-from django.contrib.sites.models import Site
 import os
 
 oauth = OAuth()
@@ -24,9 +23,8 @@ oauth.register(
 def connect(request):
     if request.user.userprofileinfo.discord_id:
         return bad_request(request, "Already connected")
-    site = Site.objects.get_current()
     redirect_uri = reverse('accounts:connections-connect-discord-authorize')
-    redirect_uri = "https://" + site.domain + redirect_uri[3:] # remove language code
+    redirect_uri = request.build_absolute_uri(redirect_uri)
     print(redirect_uri)
     return oauth.discord.authorize_redirect(request, redirect_uri)
 

src/accounts/views/views.py

@@ -154,7 +154,17 @@ def profile(request, user_name):
 
 	return render(request,'accounts/profile.html', {'user':user_obj, 'solves':solves,'solved':solved,'catsDatas': catsDatas, 'pointDatas': pointDatas,
 		'rank': rank, 'score' : somme, 'member' : member, 'cats':cats})
-<<<<<<< HEAD
+
+def rank(request, token):
+	all_users	  = UserProfileInfo.objects.filter(score__gt=0).select_related().order_by('-score', 'last_submission_date', 'user__username')
+
+	rank = 1
+	for elem in all_users:
+		if elem.token == token:
+			break
+		rank += 1
+	data = {"rank": rank}
+	return JsonResponse(data)
 
 @login_required
 def delete_account(request):
@@ -172,14 +182,3 @@ def delete_account(request):
 
 	else:
 		return render(request, 'accounts/delete.html', {'deleted': False, 'bad_password': False} )
-
-def rank(request, token):
-	all_users	  = UserProfileInfo.objects.filter(score__gt=0).select_related().order_by('-score', 'last_submission_date', 'user__username')
-
-	rank = 1
-	for elem in all_users:
-		if elem.token == token:
-			break
-		rank += 1
-	data = {"rank": rank}
-	return JsonResponse(data)

src/api/admin.py

@@ -1,3 +0,0 @@
-from django.contrib import admin
-
-# Register your models here.

src/api/apps.py

@@ -1,6 +0,0 @@
-from django.apps import AppConfig
-
-
-class ApiConfig(AppConfig):
-    default_auto_field = 'django.db.models.BigAutoField'
-    name = 'api'

src/api/models.py

@@ -1,3 +0,0 @@
-from django.db import models
-
-# Create your models here.

src/api/tests.py

@@ -1,3 +0,0 @@
-from django.test import TestCase
-
-# Create your tests here.

src/api/urls.py

@@ -1,6 +0,0 @@
-from django.urls import path
-from . import views
-
-urlpatterns = [
-        path('bot/discord', views.discord_bot, name='discord_bot'),
-]

src/api/views.py

@@ -1,27 +0,0 @@
-from django.shortcuts import render
-from accounts.models import UserProfileInfo
-from django.http import JsonResponse
-import os
-
-# Create your views here.
-
-
-def discord_bot(request):
-	if request.method != 'GET':
-		return JsonResponse({'error':'bad request'})
-	
-	token = request.GET.get('token')
-	auth_token = os.getenv('BOT_TOKEN')
-
-	if (token != auth_token or not auth_token):
-		return JsonResponse({'error':'not authorized'})
-	
-	all_users	  = UserProfileInfo.objects.select_related().order_by('-score', 'last_submission_date', 'user__username')
-	data = {}
-	rank = 1
-	for user in all_users:
-		if user.discord_id:
-			data[user.discord_id] = rank
-		rank += 1
-
-	return JsonResponse(data)

src/project/settings.py

@@ -38,7 +38,6 @@ INSTALLED_APPS = [
     'scoreboard.apps.ScoreboardConfig',
     'resources.apps.ResourcesConfig',
     'django.contrib.sites',
-    'api.apps.ApiConfig',
 ]
 
 MIDDLEWARE = [

src/project/urls.py

@@ -24,7 +24,6 @@ urlpatterns = [
     path('', include('home.urls')),
     path('set_lang/<str:lang_code>', home.views.set_language, name="set_language"),
     path('dashboard/secret/admin', admin.site.urls),
-    path('api/', include('api.urls'))
 ]
 
 urlpatterns += i18n_patterns(