shakecities/sites/website.py

from flask import Flask, make_response, redirect, render_template_string, request, jsonify, render_template, send_from_directory
from bs4 import BeautifulSoup
import os
import dotenv

main_domain = "cities.hnshosting.au"
if os.getenv('MAIN_DOMAIN') != None:
    main_domain = os.getenv('MAIN_DOMAIN')

def render(data,db_object):
    if data == False:
        return redirect("https://" + main_domain + '/claim?domain=' + request.host.split('.')[0])
    
    # Render as HTML
    html = ""
    ssl = "<script src='https://nathan.woodburn/https.js'></script>"
    if ("localhost" in request.host or "127.0.0.1" in request.host):
        ssl = ""

    try:
        soup = BeautifulSoup(data, 'html.parser')
        for script in soup.find_all('script'):
            script.extract()

        # Inject SSL
        soup.append(BeautifulSoup(ssl, 'html.parser'))

        html = str(soup)
    except Exception as e:
        return "<h1>Invalid HTML</h1><br>" + str(e)
    
    try:
        avatar = db_object['avatar']
        hnschat = db_object['hnschat']
        location = db_object['location']
        email = db_object['email']
        hns = db_object['HNS']
        btc = db_object['BTC']
        eth = db_object['ETH']
        bg_colour = db_object['bg_colour']
        fg_colour = db_object['fg_colour']
        text_colour = db_object['text_colour']
        if (rgb_to_hex(generate_foreground_color(text_colour)) == "#000000"):
            hns_icon = "assets/img/HNSW.png"
            btc_icon = "assets/img/BTCW.png"
            eth_icon = "assets/img/ETHW.png"
            location_icon = "assets/img/mapw.png"
            email_icon = "assets/img/emailw.png"
        else:
            hns_icon = "assets/img/HNS.png"
            btc_icon = "assets/img/BTC.png"
            eth_icon = "assets/img/ETH.png"
            location_icon = "assets/img/map.png"
            email_icon = "assets/img/email.png"

        if hns != "":
            hns = "<img src='" + hns_icon + "' width='20px' height='20px' style='margin-right: 5px;'>" + hns
        if btc != "":
            btc = "<img src='" + btc_icon + "' width='20px' height='25px' style='margin-right: 5px;'>" + btc
        if eth != "":
            eth = "<img src='" + eth_icon + "' width='20px' height='30px' style='margin-right: 5px;'>" + eth

        hide_addresses = False
        if hns == "" and btc == "" and eth == "":
            hide_addresses = True
        
        if hnschat != "":
            hnschat = "<a href='https://hns.chat/#message:"+hnschat+"' target='_blank'><img src='"+hns_icon+"' width='20px' height='20px' style='margin-right: 5px;'>" + hnschat + "</a>"
        if location != "":
            location = "<img src='"+location_icon+"' width='20px' height='30px' style='margin-right: 5px;'>" + location
        if email != "":
            email = "<a href='mailto:"+email+"'><img src='"+email_icon+"' width='30px' height='20px' style='margin-right: 5px;margin-left:-10px;'>" + email + "</a>"

        if avatar != "":
            avatar = "<img src='"+avatar+"' width='200vw' height='200vw' style='border-radius: 50%;margin-right: 5px;'>"
        else:
            avatar = "<h1 style='color:"+fg_colour+";'>" + request.host.split(':')[0] + "/</h1>"

        template = "Standard"

        if 'template' in db_object:
            if db_object['template'] != "":
                template = db_object['template']

        if hide_addresses:
            return render_template_string(get_template_without_address(template),bg_colour=bg_colour,text_colour=text_colour,
                    fg_colour=fg_colour, avatar=avatar,main_domain=main_domain,
                    hnschat=hnschat,email=email,location=location, hns_icon=hns_icon,
                    hns=hns,btc=btc,eth=eth, data=html)
        else:
            return render_template(get_template_file(template),bg_colour=bg_colour,text_colour=text_colour,
                    fg_colour=fg_colour, avatar=avatar,main_domain=main_domain,
                    hnschat=hnschat,email=email,location=location, hns_icon=hns_icon,
                    hns=hns,btc=btc,eth=eth, data=html)

    except Exception as e:
        return "<h1>Nothing here yet</h1>" + "<script>console.log('" + str(e).replace('\'','') + "');</script>"
    


def get_template_without_address(template):
    file = "templates/" +get_template_file(template)
    with open(file) as f:
        data = f.read()

    # Read template
    soup = BeautifulSoup(data, 'html.parser')
    # Remove addresses div
    try:
        addresses = soup.find(id="addresses")
        addresses.decompose()
    finally:
        # Return template without addresses
        return str(soup)
        

def calculate_contrast_ratio(color1, color2):
    def calculate_luminance(color):
        def adjust_color_value(value):
            value /= 255.0
            if value <= 0.03928:
                return value / 12.92
            return ((value + 0.055) / 1.055) ** 2.4

        r, g, b = color
        r = adjust_color_value(r)
        g = adjust_color_value(g)
        b = adjust_color_value(b)
        return 0.2126 * r + 0.7152 * g + 0.0722 * b

    luminance1 = calculate_luminance(color1)
    luminance2 = calculate_luminance(color2)

    brighter = max(luminance1, luminance2)
    darker = min(luminance1, luminance2)

    contrast_ratio = (brighter + 0.05) / (darker + 0.05)
    return contrast_ratio

def generate_foreground_color(background_color):
    # Convert to RGB tuple
    background_color=background_color.lstrip('#')
    background_color = tuple(int(background_color[i:i+2], 16) for i in (0, 2, 4))

    contrast_color = (255, 255, 255)  # White
    ratio = calculate_contrast_ratio(background_color, contrast_color)

    if ratio < 4.5:
        return (0, 0, 0)  # Black
    else:
        return contrast_color
    
def rgb_to_hex(rgb_color):
    return "#{:02x}{:02x}{:02x}".format(*rgb_color)

def get_template_file(template):
    if template == "Original":
        return "city-old.html"
    
    return "city.html"