faucet/main.py

from flask import (
    Flask,
    make_response,
    redirect,
    request,
    jsonify,
    render_template,
    send_from_directory,
)
import os
import dotenv
import requests
import gift
import json
import schedule
import time
from email_validator import validate_email, EmailNotValidError


app = Flask(__name__)
dotenv.load_dotenv()
address = "hs1qr7d0xqsyatls47jf28gvm97twe8k606gspfpsz"

if os.getenv("address") != None:
    address = os.getenv("address")


# 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
    if "r" in request.cookies:
        print("Referer: " + request.cookies["r"])
        return render_template(
            "index.html", hidden=request.cookies["r"], address=address
        )

    if "r" in params:
        print("Referer: " + params["r"])
        # 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
    return render_template("index.html", address=address)


@app.route("/", methods=["POST"])
def submit():
    name = request.form["name"]
    email = request.form["email"]
    hidden = request.form["hi"]
    ip = request.remote_addr

    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"]

    if hidden == "":
        hidden = "None"

    if "r" in request.cookies:
        hidden = request.cookies["r"]

    # Validate email
    try:
        emailinfo = validate_email(email, check_deliverability=False)
        email = emailinfo.normalized
    except EmailNotValidError as e:
        return render_template(
            "error.html", error="Invalid email address", address=address
        )

    status = gift.gift(name, email, hidden, ip)
    print(status, flush=True)

    if status == True:
        return render_template("success.html", address=address)
    else:
        return render_template("error.html", error=status, address=address)


@app.route("/api", methods=["POST"])
def api():
    # Get from params
    params = request.args
    name = params["name"]
    email = params["email"]
    key = params["key"]

    if key != os.getenv("api_key"):
        return jsonify({"error": "Invalid API key", "success": False})

    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"]

    # Validate email
    try:
        emailinfo = validate_email(email, check_deliverability=False)
        email = emailinfo.normalized
    except EmailNotValidError as e:
        return jsonify({"error": "Invalid email address", "success": False})

    status = gift.gift(name, email, "api", ip, True)
    print(status, flush=True)

    if status == True:
        return jsonify({"success": True})
    else:
        return jsonify({"error": status, "success": False})


# 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"})


@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 = {}
    for gift_item in gifts:
        if gift_item["referer"] not in referals:
            referals[gift_item["referer"]] = 1
        else:
            referals[gift_item["referer"]] += 1

    statsHTML = "Total gifts: " + str(len(gifts)) + "<br><br>"
    statsHTML += "Referals:<br>"
    for referal in referals:
        statsHTML += referal + ": " + str(referals[referal]) + "<br>"

    statsHTML += "<br>Remaining balance: " + str(gift.balance()) + " HNS<br>"

    return render_template("stats.html", address=address, stats=statsHTML)


@app.route("/<path:path>")
def catch_all(path):
    return redirect("/?r=" + path)


# 404 catch all
@app.errorhandler(404)
def not_found(e):
    return redirect("/")


def update_address():
    global address
    payload = {
        "asset": "HNS",
        "timestamp": int(round(time.time() * 1000)),
        "receiveWindow": 20000,
    }
    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,
    )
    if "address" not in r.json():
        print("Error: " + r.text, flush=True)
        # Send alert via discord
        url = os.getenv("discord_webhook")
        if url == None:
            return "No webhook set"

        payload = {"content": "NB cookie has expired\n@everyone"}
        response = requests.post(
            url, data=json.dumps(payload), headers={"Content-Type": "application/json"}
        )

        # Check if the message was sent successfully
        if response.status_code == 204:
            print("Message sent successfully!")
        else:
            print(f"Failed to send message. Status code: {response.status_code}")
            print(response.text)
        return

    address = r.json()["address"]
    print("Address updated: " + address, flush=True)


update_address()


if __name__ == "__main__":
    app.run(debug=False, port=5000, host="0.0.0.0")