faucet/main.py

147 lines
4.0 KiB
Python
Raw Normal View History

2023-11-07 23:04:54 +11:00
from flask import Flask, make_response, redirect, request, jsonify, render_template, send_from_directory
import os
import dotenv
import requests
import gift
2023-11-08 12:48:17 +11:00
import json
import schedule
import time
2023-11-07 23:04:54 +11:00
app = Flask(__name__)
dotenv.load_dotenv()
2023-11-07 23:35:06 +11:00
address = 'hs1qr7d0xqsyatls47jf28gvm97twe8k606gspfpsz'
2023-11-07 23:04:54 +11:00
2023-11-07 23:35:06 +11:00
if os.getenv('address') != None:
address = os.getenv('address')
2023-11-07 23:04:54 +11:00
#Assets routes
@app.route('/assets/<path:path>')
def send_report(path):
return send_from_directory('templates/assets', path)
@app.route('/')
def index():
params = request.args
2023-11-08 13:33:17 +11:00
if 'r' in request.cookies:
print("Referer: " + request.cookies['r'])
return render_template('index.html', hidden=request.cookies['r'],address=address)
2023-11-07 23:04:54 +11:00
if 'r' in params:
print("Referer: " + params['r'])
2023-11-08 13:33:17 +11:00
# Set cookie
resp = make_response(render_template('index.html', hidden=params['r'],address=address))
resp.set_cookie('r', params['r'], max_age=60*60*24)
return resp
2023-11-07 23:04:54 +11:00
2023-11-07 23:35:06 +11:00
return render_template('index.html',address=address)
2023-11-07 23:04:54 +11:00
@app.route('/', methods=['POST'])
def submit():
name = request.form['name']
email = request.form['email']
hidden = request.form['hi']
ip = request.remote_addr
2023-11-07 23:14:36 +11:00
if 'X-REAL-IP' in request.headers:
ip = request.headers['X-REAL-IP']
if 'X-Real-Ip' in request.headers:
ip = request.headers['X-Real-Ip']
2023-11-07 23:04:54 +11:00
if hidden == '':
hidden = 'None'
2023-11-08 13:33:17 +11:00
if 'r' in request.cookies:
hidden = request.cookies['r']
2023-11-07 23:04:54 +11:00
status = gift.gift(name, email, hidden, ip)
2023-11-07 23:11:08 +11:00
print(status,flush=True)
2023-11-07 23:04:54 +11:00
if status == True:
2023-11-07 23:35:06 +11:00
return render_template('success.html',address=address)
2023-11-07 23:04:54 +11:00
else:
2023-11-07 23:35:06 +11:00
return render_template('error.html',error=status,address=address)
2023-11-07 23:04:54 +11:00
# Special routes
@app.route('/.well-known/wallets/<token>')
def send_wallet(token):
address = requests.get('https://nathan.woodburn.au/.well-known/wallets/'+token).text
return make_response(address, 200, {'Content-Type': 'text/plain'})
2023-11-08 12:48:17 +11:00
@app.route('/stats')
def stats():
# Read the file
path = '/data/gifts.json'
if os.getenv('local') == 'true':
path = './gifts.json'
# Load file
gifts = []
if os.path.isfile(path):
with open(path, 'r') as f:
gifts = json.load(f)
# Loop through gifts
referals = {}
2023-11-08 12:55:10 +11:00
for gift_item in gifts:
if gift_item['referer'] not in referals:
referals[gift_item['referer']] = 1
2023-11-08 12:48:17 +11:00
else:
2023-11-08 12:55:10 +11:00
referals[gift_item['referer']] += 1
2023-11-08 12:48:17 +11:00
statsHTML = 'Total gifts: ' + str(len(gifts)) + '<br><br>'
statsHTML += 'Referals:<br>'
for referal in referals:
statsHTML += referal + ': ' + str(referals[referal]) + '<br>'
2023-11-08 13:35:21 +11:00
statsHTML += '<br>Remaining balance: ' + str(gift.balance()) + ' HNS<br>'
2023-11-08 12:48:17 +11:00
return render_template('stats.html',address=address,stats=statsHTML)
2023-11-07 23:04:54 +11:00
@app.route('/<path:path>')
def catch_all(path):
2023-11-08 12:48:17 +11:00
# # If file exists, load it
# if os.path.isfile('templates/' + path):
# return render_template(path)
2023-11-07 23:04:54 +11:00
2023-11-08 12:48:17 +11:00
# # Try with .html
# if os.path.isfile('templates/' + path + '.html'):
# return render_template(path + '.html')
2023-11-08 13:30:13 +11:00
return redirect('/?r='+path)
2023-11-07 23:04:54 +11:00
# 404 catch all
@app.errorhandler(404)
def not_found(e):
return redirect('/')
def update_address():
global address
payload = {
"asset": "HNS",
# "timestamp": 1699411892673,
"timestamp": int(round(time.time() * 1000)),
"receiveWindow": 10000
}
nbcookie = os.getenv('cookie')
cookies = {"namebase-main": nbcookie}
headers = {"Accept": "application/json", "Content-Type": "application/json"}
r = requests.post('https://www.namebase.io/api/v0/deposit/address', data=json.dumps(payload), headers=headers, cookies=cookies)
address = r.json()['address']
print("Address updated: " + address,flush=True)
update_address()
# Schedule address update every hour
schedule.every(1).hour.do(update_address)
2023-11-07 23:04:54 +11:00
if __name__ == '__main__':
app.run(debug=False, port=5000, host='0.0.0.0')