-
Notifications
You must be signed in to change notification settings - Fork 0
/
AccessToken.py
111 lines (101 loc) · 4.91 KB
/
AccessToken.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
import json
import requests
from eth_account.messages import encode_defunct
from web3 import Web3
def signRoninMessage(message, key, attempts2=0):
try:
mes = encode_defunct(text=message)
ronweb3 = Web3(Web3.HTTPProvider('https://api.roninchain.com/rpc', request_kwargs={"headers": {"content-type": "application/json", "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/95.0.4638.69 Safari/537.36"}}))
sig = ronweb3.eth.account.sign_message(mes, private_key=key)
signature = sig['signature'].hex()
temp = signature[-2:]
signature = signature[:-2]
if temp == "1c":
signature += "01"
elif temp == "1b":
signature += "00"
else:
print("something went wrong with signRoninMessage")
return signature
except Exception as e:
if attempts2 > 3:
print("Could not Sign Message. Are the servers having issues?")
print(e)
return None
else:
return signRoninMessage(message, key, attempts2 + 1)
def GenerateAccessToken(key, address, attempts=0):
def getRandomMessage(attempts2=0):
try:
url = "https://graphql-gateway.axieinfinity.com/graphql"
payload = "{\"query\":\"mutation CreateRandomMessage {\\r\\n createRandomMessage\\r\\n}\",\"variables\":{}}"
headers = {
'Content-Type': 'application/json'
}
response = requests.request("POST", url, headers=headers, data=payload)
json_data = json.loads(response.text)
return json_data['data']['createRandomMessage']
except Exception as e:
if attempts2 > 3:
print("Could not generate AccessToken Random Message. Are the servers having issues?")
print(e)
return None
else:
return getRandomMessage(attempts2 + 1)
def signRoninMessage(message, key, attempts2=0):
try:
mes = encode_defunct(text=message)
ronweb3 = Web3(Web3.HTTPProvider('https://api.roninchain.com/rpc', request_kwargs={"headers": {"content-type": "application/json", "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/95.0.4638.69 Safari/537.36"}}))
sig = ronweb3.eth.account.sign_message(mes, private_key=key)
signature = sig['signature'].hex()
temp = signature[-2:]
signature = signature[:-2]
if temp == "1c":
signature += "01"
elif temp == "1b":
signature += "00"
else:
print("something went wrong with signRoninMessage")
return signature
except Exception as e:
if attempts2 > 3:
print("Could not Sign Message. Are the servers having issues?")
print(e)
return None
else:
return signRoninMessage(message, key, attempts2 + 1)
def CreateAccessToken(message, signature, address, attempts2=0):
try:
url = "https://graphql-gateway.axieinfinity.com/graphql"
payload = "{\"query\":\"mutation CreateAccessTokenWithSignature($input: SignatureInput!)" \
"{\\r\\n createAccessTokenWithSignature(input: $input)" \
"{\\r\\n newAccount\\r\\n result\\r\\n accessToken\\r\\n __typename\\r\\n }" \
"\\r\\n}\\r\\n\",\"variables\":{\"input\":{\"mainnet\":\"ronin\",\"owner\":\"" + \
address + "\",\"message\":\"" + message + "\",\"signature\":\"" + signature + "\"}}}"
headers = {
'Content-Type': 'application/json',
'User-Agent': 'Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.0)'
}
response = requests.request("POST", url, headers=headers, data=payload)
json_data = json.loads(response.text)
return json_data['data']['createAccessTokenWithSignature']['accessToken']
except Exception as e:
if attempts2 > 3:
print("Could not Create Access Token. Are the servers having issues?")
print(e)
return None
else:
return CreateAccessToken(message, signature, address, attempts2 + 1)
try:
myResponse = getRandomMessage()
mySignature = signRoninMessage(myResponse, key)
token = CreateAccessToken(repr(myResponse).replace("\'", ""), mySignature, address)
return token
except Exception as e:
if attempts > 3:
# TODO add a website guide for common errors
print(e)
print("Unable To generate Access Token. This is gernerally an internet issue or a server issue.")
return None
else:
return GenerateAccessToken(key, address, attempts + 1)