97 lines
2.8 KiB
Python
97 lines
2.8 KiB
Python
|
import json
|
||
|
|
|
||
|
|
from django.shortcuts import get_object_or_404, redirect
|
||
|
|
from django.utils.translation import gettext_lazy as _
|
||
|
|
from django.views.decorators.csrf import csrf_exempt
|
||
|
|
from django.views.generic.edit import DeleteView
|
||
|
|
from django.views.generic.base import View
|
||
|
|
from django.http import JsonResponse
|
||
|
|
from django_tables2 import SingleTableView
|
||
|
|
from pyvckit.verify import verify_vp, verify_vc
|
||
|
|
from uuid import uuid4
|
||
|
|
|
||
|
|
from idhub.mixins import AdminView
|
||
|
|
from webhook.models import Token
|
||
|
|
from webhook.tables import TokensTable
|
||
|
|
|
||
|
|
|
||
|
|
@csrf_exempt
|
||
|
|
def webhook_verify(request):
|
||
|
|
if request.method == 'POST':
|
||
|
|
auth_header = request.headers.get('Authorization')
|
||
|
|
if not auth_header or not auth_header.startswith('Bearer '):
|
||
|
|
return JsonResponse({'error': 'Invalid or missing token'}, status=401)
|
||
|
|
|
||
|
|
token = auth_header.split(' ')[1]
|
||
|
|
tk = Token.objects.filter(token=token).first()
|
||
|
|
if not tk:
|
||
|
|
return JsonResponse({'error': 'Invalid or missing token'}, status=401)
|
||
|
|
|
||
|
|
try:
|
||
|
|
data = json.loads(request.body)
|
||
|
|
except json.JSONDecodeError:
|
||
|
|
return JsonResponse({'error': 'Invalid JSON'}, status=400)
|
||
|
|
|
||
|
|
typ = data.get("type")
|
||
|
|
vc = data.get("data")
|
||
|
|
try:
|
||
|
|
vc = json.dumps(vc)
|
||
|
|
except Exception:
|
||
|
|
return JsonResponse({'error': 'Invalid JSON'}, status=400)
|
||
|
|
|
||
|
|
func = verify_vp
|
||
|
|
if typ == "credential":
|
||
|
|
func = verify_vc
|
||
|
|
|
||
|
|
if func(vc):
|
||
|
|
return JsonResponse({'status': 'success'}, status=200)
|
||
|
|
|
||
|
|
return JsonResponse({'status': 'fail'}, status=200)
|
||
|
|
|
||
|
|
return JsonResponse({'error': 'Invalid request method'}, status=400)
|
||
|
|
|
||
|
|
|
||
|
|
class WebHookTokenView(AdminView, SingleTableView):
|
||
|
|
template_name = "token.html"
|
||
|
|
title = _('Token')
|
||
|
|
section = ""
|
||
|
|
subtitle = _('Managament Tokens')
|
||
|
|
icon = 'bi bi-key'
|
||
|
|
model = Token
|
||
|
|
table_class = TokensTable
|
||
|
|
|
||
|
|
def get_queryset(self):
|
||
|
|
"""
|
||
|
|
Override the get_queryset method to filter events based on the user type.
|
||
|
|
"""
|
||
|
|
return Token.objects.filter().order_by("-id")
|
||
|
|
|
||
|
|
def get_context_data(self, **kwargs):
|
||
|
|
context = super().get_context_data(**kwargs)
|
||
|
|
context.update({
|
||
|
|
'tokens': Token.objects,
|
||
|
|
})
|
||
|
|
return context
|
||
|
|
|
||
|
|
|
||
|
|
class TokenDeleteView(AdminView, DeleteView):
|
||
|
|
model = Token
|
||
|
|
|
||
|
|
def get(self, request, *args, **kwargs):
|
||
|
|
self.check_valid_user()
|
||
|
|
self.pk = kwargs['pk']
|
||
|
|
self.object = get_object_or_404(self.model, pk=self.pk)
|
||
|
|
self.object.delete()
|
||
|
|
|
||
|
|
return redirect('webhook:tokens')
|
||
|
|
|
||
|
|
|
||
|
|
class TokenNewView(AdminView, View):
|
||
|
|
|
||
|
|
def get(self, request, *args, **kwargs):
|
||
|
|
self.check_valid_user()
|
||
|
|
Token.objects.create(token=uuid4())
|
||
|
|
|
||
|
|
return redirect('webhook:tokens')
|
||
|
|
|