import requests
import os
import json
from subprocess import getoutput

# Replaces nginx version with actual version, provide URL to site to find nginx version
url = 'https://askiiart.net'
with open('stock-404.html', 'r') as html_404:
    html_404 = html_404.read()
    html_404 = html_404.replace('nginx/0.0.0', requests.get(url).headers['Server'])
    
boykisser_url = '/boykisser.png'
gandalf_url = '/you.gif'

try:
    os.mkdir('pages')
except FileExistsError:
    pass

with open('status-codes.json', 'r') as codes_file:
    codes_dict = json.load(codes_file)

for code in codes_dict:
    new_html = html_404.replace('404 Not Found', f'{code} {codes_dict[code]}')
    if code == "404":
        new_html = new_html.replace("<h1>404 You Made A Typo In The URL, Didn't You?</h1>",
                                    f"<h1>404 You Made A Typo In The URL, Didn't You?</h1><br><img src={boykisser_url} alt=\"boykisser\" width=\"350\"")
    elif code == "300":
        # 57% Yes
        # - 43% No
        # - (300 Multiple Choices is the actual error code? Geddit?)
        new_html = new_html.replace(
            '<h1>300 Poll</h1>', '<h1>300 Poll</h1><br>57% Yes, 43% No<br>(300 Multiple Choices is the actual error code? Geddit?)')
    elif code == "403":
        new_html = new_html.replace('<h1>403 You Shall Not Pass</h1>',
                                    f'<h1>403 You Shall Not Pass</h1><br><img src={gandalf_url} alt=\"gandalf saying you looped infinitely\" width=\"350\"')
    with open(f'pages/{code}.html', 'w') as new_file:
        new_file.write(new_html)