Nathanwoodburn.github.io/curl.py

from flask import render_template
from tools import error_response, getAddress, get_tools_data, getClientIP
import os
from functools import lru_cache
import requests
from blueprints.spotify import get_spotify_track


def clean_path(path:str):
    path = path.strip("/ ").lower()
    # Strip any .html extension
    if path.endswith(".html"):
        path = path[:-5]

    # If the path is empty, set it to "index"
    if path == "":
        path = "index"
    return path

@lru_cache(maxsize=1)
def get_header():
    with open("templates/header.ascii", "r") as f:
        return f.read()

@lru_cache(maxsize=1)
def get_current_project():
    git = requests.get(
        "https://git.woodburn.au/api/v1/users/nathanwoodburn/activities/feeds?only-performed-by=true&limit=1",
        headers={"Authorization": os.getenv("GIT_AUTH") if os.getenv("GIT_AUTH") else os.getenv("git_token")},
    )
    git = git.json()
    git = git[0]
    repo_name = git["repo"]["name"]
    repo_name = repo_name.lower()
    repo_description = git["repo"]["description"]    
    if not repo_description:
        return f"{repo_name}"
    return f"{repo_name} - {repo_description}"


@lru_cache(maxsize=1)
def get_projects():
    projectsreq = requests.get(
            "https://git.woodburn.au/api/v1/users/nathanwoodburn/repos"
        )

    projects = projectsreq.json()

    # Check for next page
    pageNum = 1
    while 'rel="next"' in projectsreq.headers["link"]:
        projectsreq = requests.get(
            "https://git.woodburn.au/api/v1/users/nathanwoodburn/repos?page="
            + str(pageNum)
        )
        projects += projectsreq.json()
        pageNum += 1

    # Sort by last updated
    projectsList = sorted(
        projects, key=lambda x: x["updated_at"], reverse=True)
    projects = ""
    projectNum = 0
    includedNames = []
    while len(includedNames) < 5 and projectNum < len(projectsList):
        # Avoid duplicates
        if projectsList[projectNum]["name"] in includedNames:
            projectNum += 1
            continue
        includedNames.append(projectsList[projectNum]["name"])
        project = projectsList[projectNum]
        projects += f"""{project['name']} - {project['description'] if project['description'] else 'No description'}
{project['html_url']}

"""
        projectNum += 1
        
    return projects

@lru_cache(maxsize=32)
def valid_curl_path(path: str) -> bool:
    """Check if the given path corresponds to a valid curl/ascii response."""
    path = clean_path(path)

    # Special cases
    special_paths = [
        "index",
        "projects",
        "donate",
        "donate/more",
        "tools"
    ]
    if path in special_paths:
        return True

    # Check for donate/<coin> pattern
    if path.startswith("donate/"):
        coin = path.split("/")[1]
        address = getAddress(coin)
        if address != "":
            return True

    # Check if .ascii template exists
    if os.path.exists(f"templates/{path}.ascii"):
        return True

    return False

def curl_response(request):
    # Check if <path>.ascii exists
    path = clean_path(request.path)
    
    # Handle special cases
    if path == "index":
        # Get current project
        return render_template("index.ascii",repo=get_current_project(), ip=getClientIP(request), spotify=get_spotify_track()), 200, {'Content-Type': 'text/plain; charset=utf-8'}
    if path == "projects":
        # Get projects
        return render_template("projects.ascii",header=get_header(),projects=get_projects()), 200, {'Content-Type': 'text/plain; charset=utf-8'}

    if path == "donate":
        # Get donation info
        return render_template("donate.ascii",header=get_header(),
                               HNS=getAddress("HNS"), BTC=getAddress("BTC"),
                               SOL=getAddress("SOL"), ETH=getAddress("ETH")
                               ), 200, {'Content-Type': 'text/plain; charset=utf-8'}
        
    if path == "donate/more":
        coinList = os.listdir(".well-known/wallets")
        coinList = [file for file in coinList if file[0] != "."]
        coinList.sort()
        return render_template("donate_more.ascii",header=get_header(),
                               coins=coinList
                               ), 200, {'Content-Type': 'text/plain; charset=utf-8'}        

    # For other donation pages, fall back to ascii if it exists
    if path.startswith("donate/"):
        coin = path.split("/")[1]
        address = getAddress(coin)
        if address != "":
            return render_template("donate_coin.ascii",header=get_header(),coin=coin.upper(),address=address), 200, {'Content-Type': 'text/plain; charset=utf-8'}
    
    if path == "tools":
        tools = get_tools_data()
        return render_template("tools.ascii",header=get_header(),tools=tools), 200, {'Content-Type': 'text/plain; charset=utf-8'}


    if os.path.exists(f"templates/{path}.ascii"):
        return render_template(f"{path}.ascii",header=get_header()), 200, {'Content-Type': 'text/plain; charset=utf-8'}
    
    return error_response(request)