#!/usr/bin/env python3
"""Prepara a pagina (HTML unico de export do Bitsy) de um jogo: aplica as mudancas nossas e poe uma CSP.
uso: make.py <slug> <entrada.html> <saida.html> [<entrada.bitsydata> <saida.bitsydata>]"""
import re, sys

slug, src, out = sys.argv[1:4]
page = open(src, encoding='utf-8').read().replace('\r\n', '\n')

def cut(pattern, flags=re.S, count=1):
    global page
    page, n = re.subn(pattern, '', page, flags=flags)
    assert n == count, (pattern, n)

if slug == 'left-undetected':
    # sem os links do rodape (repositorio, itch, Twitter e site do autor)
    cut(r'<div id=\'links\'>.*?</div>\n</div>\n')
elif slug == 'a-cozy-winterwalk':
    # sem a trilha de fundo, que o jogo tocava de um video do YouTube (script externo, musica de terceiros)
    cut(r'    <!--LOAD YouTube "iframe_api" ASYNCHRONOUSLY.-->\n    <script async src="https://www.youtube.com/iframe_api"></script>\n\n')
    cut(r'    <div style="display:none;" id="div_YouTube"></div>\n')
    cut(r'<script>\n    function onYouTubeIframeAPIReady\(\) \{.*?\n    \}\n</script>\n')
    assert 'youtube' not in page.lower()

csp = ("default-src 'none'; script-src 'unsafe-inline' 'unsafe-eval'; style-src 'unsafe-inline'; img-src data: blob:; "
       "media-src data: blob:; font-src data:; connect-src 'self'")
assert page.count('<head>') == 1
page = page.replace('<head>', '<head>\n<meta http-equiv="Content-Security-Policy" content="%s">' % csp, 1)
open(out, 'w', encoding='utf-8', newline='\n').write(page)

if len(sys.argv) > 4:
    data = open(sys.argv[4], encoding='utf-8').read().replace('\r\n', '\n')
    if slug == 'left-undetected':
        old = 'LEFT UNDETECTED{br}by MILDMOJO{br}Twitter: @mildmojo{br}Web: mildmojo.com'
        assert data.count(old) == 1
        data = data.replace(old, 'LEFT UNDETECTED{br}by MILDMOJO')
    open(sys.argv[5], 'w', encoding='utf-8', newline='\n').write(data)