Авторизация в Django при помощи Metamask
- воскресенье, 28 августа 2022 г. в 00:40:44
Сегодня мы напишем легкий сниппет для авторизации пользователей на сайте при помощи кошелька Metamask. Замечу, что данное решение максимально изолировано от фреймворка. Вы сможете легко адаптировать его не только к Django, но и к Flask, Sanic, Starlette, Aiohttp и т.п.
Сниппет будет состоять из двух частей - django template
и django view
.
Начнем с django template
.
Допустим у нас есть шаблон для авторизации templates/login.html
Создадим кнопку авторизации
при нажатии на которую будет вызван Metamask.
<button type="button" onclick="web3Login()">
Log in with Metamask
</button>
Создадим обслуживающий скрипт.
<script src="https://cdn.ethers.io/lib/ethers-5.2.umd.min.js" type="application/javascript"></script>
<script>
function web3Login() {
/* Подписываем сообщение */
try {
window.ethereum.enable().then(function () {
provider = new ethers.providers.Web3Provider(window.ethereum);
provider.getNetwork().then(function (result) {
if (result['chainId'] != 1) {
alert('Switch to Mainnet!');
} else {
provider.listAccounts().then(function (result) {
accountAddress = result[0];
signer = provider.getSigner();
signer.signMessage("Sign to auth {{ csrf_token }}").then((signature) => {web3LoginBackend(accountAddress, signature)});
})
}
})
})
} catch {
alert('Please install MetaMask for your browser.')
}
}
function web3LoginBackend(accountAddress, signature) {
/* Отправляем сообщение на функцию авторизации/регистрации */
var form = document.createElement('form');
form.action = '{% url "main:auth_web3" %}'; // TODO замените на свой адрес
form.method = 'POST';
var input = document.createElement('input');
input.type = 'hidden';
input.name = 'csrfmiddlewaretoken';
input.value = '{{ csrf_token }}';
form.appendChild(input);
var input = document.createElement('input');
input.type = 'hidden';
input.name = 'accountAddress';
input.value = accountAddress;
form.appendChild(input);
var input = document.createElement('input');
input.type = 'hidden';
input.name = 'signature';
input.value = signature;
form.appendChild(input);
document.body.appendChild(form);
form.submit();
}
</script>
Данный скрипт состоит из двух функций web3Login
и web3LoginBackend
.
При нажатии на кнопку авторизации
вызывается функция web3Login
, которая активирует окно «Запрос подписи» в Metamask. Подписывать мы будем сообщение вида Sign to auth {{ csrf_token }}.
После нажатия на кнопку «Подписать» в окне Metamask вызывается функция web3LoginBackend
, которая отправит данные (csrf token, адрес кошелька и подпись сообщения) в django view
.
Перейдем к django view
.
Установим необходимые зависимости.
pip install web3==5.30.0
pip install eth-account==0.5.9
pip install names==0.3.0
pip install shortuuid==1.0.9
Создадимdjango view
.
import datetime
import secrets
import names
import string
import shortuuid
from django.contrib import messages
from django.contrib.auth import authenticate
from django.contrib.auth import login
from django.contrib.auth.models import User
from django.http import HttpResponse
from django.http import HttpResponseRedirect
from django.shortcuts import render
from django.shortcuts import reverse
from eth_account.messages import defunct_hash_message
from web3.auto import w3
from user_profile.models import UserProfile # TODO измените на свою модель
def auth_web3(request):
public_address = request.POST["accountAddress"]
signature = request.POST["signature"]
csrf_token = request.POST["csrfmiddlewaretoken"]
original_message = f"Sign to auth {csrf_token}"
message_hash = defunct_hash_message(text=original_message)
signer = w3.eth.account.recoverHash(message_hash, signature=signature)
short_uuid = shortuuid.uuid()
if signer == public_address:
user_profile = UserProfile.objects.filter(eth_account_address=public_address).first()
if user_profile:
try:
user = user_profile.user
except:
messages.add_message(request, messages.WARNING, _("Профайл не найден"))
return HttpResponseRedirect(
reverse(
"main:user_login", # TODO измените на свой адрес
)
)
user.backend = "django.contrib.auth.backends.ModelBackend"
login(request, user)
return HttpResponseRedirect(reverse("main:dashboard")) # TODO измените на свой адрес
else:
alphabet = string.ascii_letters + string.digits
password = "".join(secrets.choice(alphabet) for i in range(20))
first_name = names.get_first_name()
last_name = names.get_last_name()
email = f"{short_uuid}@{HOST}"
user = User.objects.create_user(email=email, username=short_uuid, first_name=first_name, last_name=last_name, password=password)
user_profile = UserProfile()
user_profile.user = user
user_profile.eth_account_address = public_address
user_profile.save()
user.backend = "django.contrib.auth.backends.ModelBackend"
login(request, user)
messages.success(request, _("Успешная регистрация."))
return HttpResponseRedirect(reverse("main:dashboard")) # TODO измените на свой адрес
else:
messages.add_message(request, messages.WARNING, _("Обновите страницу и попробуйте еще раз"))
return HttpResponseRedirect(
reverse(
"main:user_login", # TODO измените на свой адрес
)
)
На строке 26 мы дублируем сообщение с django template
original_message = f"Sign to auth {csrf_token}"
И восстанавливаем адрес кошелька при помощи подписи signature
полученной из django tempate
после отправки данных в django view
.
message_hash = defunct_hash_message(text=original_message)
signer = w3.eth.account.recoverHash(message_hash, signature=signature)
Если восстановленный адрес signer
равен адресу public_address
(адресу кошелька пользователя) отправленному из django template
, то авторизация/регистрация прошла успешно и все что нам остается это сделать немного "магии" для авторизации/регистрации.
P.S. Рабочий пример можно найти по адресу workhours.space. А еще это удобный и бесплатный трекер времени с удобным ботом в телеграм - пользуйтесь.