254 lines
8.1 KiB
Python
254 lines
8.1 KiB
Python
|
import mysql.connector
|
||
|
|
from mysql.connector import Error
|
||
|
|
import dotenv
|
||
|
|
import os
|
||
|
|
import json
|
||
|
|
import time
|
||
|
|
import smtplib
|
||
|
|
from email.mime.text import MIMEText
|
||
|
|
from email.mime.multipart import MIMEMultipart
|
||
|
|
import requests
|
||
|
|
import re
|
||
|
|
|
||
|
|
dotenv.load_dotenv()
|
||
|
|
|
||
|
|
db_config = {
|
||
|
|
'host': os.getenv('DB_HOST'),
|
||
|
|
'database': os.getenv('DB_NAME'),
|
||
|
|
'user': os.getenv('DB_USER'),
|
||
|
|
'password': os.getenv('DB_PASSWORD')
|
||
|
|
}
|
||
|
|
|
||
|
|
def addDomain(userID:int, domain:str):
|
||
|
|
# Get user domains
|
||
|
|
conn = mysql.connector.connect(**db_config)
|
||
|
|
cursor = conn.cursor()
|
||
|
|
cursor.execute("SELECT domains FROM users WHERE id = %s", (userID,))
|
||
|
|
user = cursor.fetchall()
|
||
|
|
cursor.close()
|
||
|
|
conn.close()
|
||
|
|
|
||
|
|
# Read json domains from first user
|
||
|
|
user = user[0][0]
|
||
|
|
userDomains = json.loads(user)
|
||
|
|
|
||
|
|
# Check if domain is already in use
|
||
|
|
for userDomain in userDomains:
|
||
|
|
if (userDomain['name'] == domain):
|
||
|
|
return False
|
||
|
|
|
||
|
|
# Add domain to user
|
||
|
|
userDomains.append({
|
||
|
|
"name": domain,
|
||
|
|
"status": "pending"
|
||
|
|
})
|
||
|
|
|
||
|
|
# Update user domains
|
||
|
|
conn = mysql.connector.connect(**db_config)
|
||
|
|
cursor = conn.cursor()
|
||
|
|
cursor.execute("UPDATE users SET domains = %s WHERE id = %s", (json.dumps(userDomains), userID))
|
||
|
|
conn.commit()
|
||
|
|
cursor.close()
|
||
|
|
conn.close()
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
def verifyDomain(domain:str):
|
||
|
|
# Remove any trailing slashes
|
||
|
|
domain = domain.rstrip('/')
|
||
|
|
# Remove any protocol
|
||
|
|
domain = domain.replace('https://', '')
|
||
|
|
domain = domain.replace('http://', '')
|
||
|
|
domain = domain.lower()
|
||
|
|
# Verify domain contains only valid characters
|
||
|
|
if (re.match("^[a-z0-9.-]*$", domain) == None):
|
||
|
|
return False
|
||
|
|
return domain
|
||
|
|
|
||
|
|
def deleteDomain(userID:int, domain:str):
|
||
|
|
# Get user domains
|
||
|
|
conn = mysql.connector.connect(**db_config)
|
||
|
|
cursor = conn.cursor()
|
||
|
|
cursor.execute("SELECT domains FROM users WHERE id = %s", (userID,))
|
||
|
|
user = cursor.fetchall()
|
||
|
|
cursor.close()
|
||
|
|
conn.close()
|
||
|
|
|
||
|
|
# Read json domains from first user
|
||
|
|
user = user[0][0]
|
||
|
|
userDomains = json.loads(user)
|
||
|
|
|
||
|
|
# Check if domain is already in use
|
||
|
|
for userDomain in userDomains:
|
||
|
|
if (userDomain['name'] == domain):
|
||
|
|
userDomains.remove(userDomain)
|
||
|
|
break
|
||
|
|
|
||
|
|
# Update user domains
|
||
|
|
conn = mysql.connector.connect(**db_config)
|
||
|
|
cursor = conn.cursor()
|
||
|
|
cursor.execute("UPDATE users SET domains = %s WHERE id = %s", (json.dumps(userDomains), userID))
|
||
|
|
conn.commit()
|
||
|
|
cursor.close()
|
||
|
|
conn.close()
|
||
|
|
return True
|
||
|
|
|
||
|
|
def syncDomains():
|
||
|
|
# Verify connection to HSD node
|
||
|
|
try:
|
||
|
|
r = requests.get('http://x:' + os.getenv('HSD_API_KEY') + '@' + os.getenv('HSD_IP')
|
||
|
|
+ ':' + os.getenv('HSD_PORT'))
|
||
|
|
if (r.status_code != 200):
|
||
|
|
return "HSD node is not responding"
|
||
|
|
# Check to make sure the node is synced
|
||
|
|
data = r.json()
|
||
|
|
if (data['chain']['progress'] < 0.999):
|
||
|
|
return "HSD node is not synced"
|
||
|
|
|
||
|
|
except:
|
||
|
|
return False
|
||
|
|
|
||
|
|
# Get all users
|
||
|
|
conn = mysql.connector.connect(**db_config)
|
||
|
|
cursor = conn.cursor()
|
||
|
|
cursor.execute("SELECT id, domains FROM users")
|
||
|
|
users = cursor.fetchall()
|
||
|
|
cursor.close()
|
||
|
|
conn.close()
|
||
|
|
|
||
|
|
# Loop through users
|
||
|
|
for user in users:
|
||
|
|
userID = user[0]
|
||
|
|
userDomains = json.loads(user[1])
|
||
|
|
# Loop through user domains
|
||
|
|
for userDomain in userDomains:
|
||
|
|
# Check if domain is pending
|
||
|
|
if (userDomain['status'] == 'pending'):
|
||
|
|
try:
|
||
|
|
print('Importing domain: ' + userDomain['name'])
|
||
|
|
r = requests.post('http://x:' + os.getenv('HSD_API_KEY') + '@' + os.getenv('HSD_IP')
|
||
|
|
+ ':' + os.getenv('HSD_WALLET_PORT'), json={'method': 'importname', 'params': [userDomain['name']]})
|
||
|
|
|
||
|
|
if (r.status_code == 200):
|
||
|
|
# Update domain status
|
||
|
|
userDomain['status'] = 'added'
|
||
|
|
|
||
|
|
# Set some defaults
|
||
|
|
userDomain['transfering'] = 0
|
||
|
|
userDomain['next'] = 'none'
|
||
|
|
userDomain['when'] = 0
|
||
|
|
|
||
|
|
# Update user domains
|
||
|
|
conn = mysql.connector.connect(**db_config)
|
||
|
|
cursor = conn.cursor()
|
||
|
|
cursor.execute("UPDATE users SET domains = %s WHERE id = %s", (json.dumps(userDomains), userID))
|
||
|
|
conn.commit()
|
||
|
|
cursor.close()
|
||
|
|
conn.close()
|
||
|
|
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
|
||
|
|
# Get domain info
|
||
|
|
r = requests.post('http://x:' + os.getenv('HSD_API_KEY') + '@' + os.getenv('HSD_IP')
|
||
|
|
+ ':' + os.getenv('HSD_PORT'), json={'method': 'getnameinfo', 'params': [userDomain['name']]})
|
||
|
|
print(json.dumps(r.json(), indent=4))
|
||
|
|
if (r.status_code == 200):
|
||
|
|
data = r.json()
|
||
|
|
# Check if domain is registered
|
||
|
|
info = data['result']['info']
|
||
|
|
if (userDomain['transfering'] != info['transfer']):
|
||
|
|
# Update domain status
|
||
|
|
alert('transfer', userDomain['name'], userID)
|
||
|
|
userDomain['transfering'] = info['transfer']
|
||
|
|
|
||
|
|
if 'stats' in info:
|
||
|
|
if 'blocksUntilExpire' in info['stats']:
|
||
|
|
# Update domain status
|
||
|
|
userDomain['next'] = 'expires'
|
||
|
|
|
||
|
|
previous = userDomain['when']
|
||
|
|
userDomain['when'] = info['stats']['blocksUntilExpire']
|
||
|
|
|
||
|
|
if (crossTimeAlert(userDomain['next'], previous, userDomain['when'])):
|
||
|
|
alert('expire', userDomain['name'], userID)
|
||
|
|
|
||
|
|
elif 'blocksUntilBidding' in info['stats']:
|
||
|
|
# Update domain status
|
||
|
|
userDomain['next'] = 'opens for bidding'
|
||
|
|
userDomain['when'] = info['stats']['blocksUntilBidding']
|
||
|
|
|
||
|
|
# Update user domains
|
||
|
|
conn = mysql.connector.connect(**db_config)
|
||
|
|
cursor = conn.cursor()
|
||
|
|
cursor.execute("UPDATE users SET domains = %s WHERE id = %s", (json.dumps(userDomains), userID))
|
||
|
|
conn.commit()
|
||
|
|
cursor.close()
|
||
|
|
conn.close()
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
else:
|
||
|
|
return "Failed to get info about domain: " + userDomain['name'] + "<br>" + str(r.text)
|
||
|
|
return "Finished syncing domains"
|
||
|
|
|
||
|
|
def alert(event,domain,user):
|
||
|
|
# TODO this later
|
||
|
|
pass
|
||
|
|
|
||
|
|
def crossTimeAlert(event,was, now):
|
||
|
|
# Check for each of these times to see if we should alert
|
||
|
|
month = 4320
|
||
|
|
week = 1008
|
||
|
|
|
||
|
|
# If the time crossed the month mark
|
||
|
|
if (was > month and now <= month):
|
||
|
|
return True
|
||
|
|
# If the time crossed the week mark
|
||
|
|
if (was > week and now <= week):
|
||
|
|
return True
|
||
|
|
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def getCachedDomainInfo(domain):
|
||
|
|
# Get domain info from user domains
|
||
|
|
conn = mysql.connector.connect(**db_config)
|
||
|
|
cursor = conn.cursor()
|
||
|
|
cursor.execute("SELECT domains FROM users")
|
||
|
|
users = cursor.fetchall()
|
||
|
|
cursor.close()
|
||
|
|
conn.close()
|
||
|
|
|
||
|
|
# Loop through users
|
||
|
|
for user in users:
|
||
|
|
userDomains = json.loads(user[0])
|
||
|
|
# Loop through user domains
|
||
|
|
for userDomain in userDomains:
|
||
|
|
if (userDomain['name'] == domain):
|
||
|
|
return userDomain
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def blocksToTime(blocks):
|
||
|
|
# Get minutes
|
||
|
|
minutes = blocks * 10
|
||
|
|
years, minutes = divmod(minutes, 525600)
|
||
|
|
days, minutes = divmod(minutes, 1440)
|
||
|
|
hours, minutes = divmod(minutes, 60)
|
||
|
|
|
||
|
|
# Build the string
|
||
|
|
time_string = ""
|
||
|
|
if years:
|
||
|
|
time_string += f"{years} {'year' if years == 1 else 'years'}, "
|
||
|
|
if days:
|
||
|
|
time_string += f"{days} {'day' if days == 1 else 'days'}, "
|
||
|
|
if hours and not years:
|
||
|
|
time_string += f"{hours} {'hour' if hours == 1 else 'hours'}, "
|
||
|
|
if minutes and not years and not days:
|
||
|
|
time_string += f"{minutes} {'min' if minutes == 1 else 'mins'}"
|
||
|
|
|
||
|
|
return time_string.rstrip(', ')
|