build.py

import hashlib
import yaml


# Read YAML file
with open("data.yaml", 'r') as stream:
    data = yaml.safe_load(stream)


################################ index.html ################################

with open("index_top.html", 'r') as f:
    top = f.read()
with open("index_bottom.html", 'r') as f:
    bottom = f.read()

middle = ""
for i, enigma in enumerate(data['enigmas']):
    middle += f"""
        <div class="enigma" id="enigma{i}">
            <div class="title">
                <h4>Enigma {i+1}:</h4>
                <p>{enigma['description']}</p>
            </div>
            <div class="input">
                <label for="password{i}">Answer:</label>
                <input type="password{i}" id="password{i}" name="password{i}" />
                <button type="button" onclick="checkPassword{i}()">Submit</button>
            </div>
            <div class="status">
                <img src="checked.svg" class="finished" id="enigma{i}-check">
                <img src="not_checked.svg" class="not-finished" id="enigma{i}-no-check">
            </div>
        </div>

"""

full = top + middle + bottom
with open("index.html", 'w') as f:
    f.write(full)


################################ script.js ################################

script = ""
for i, enigma in enumerate(data['enigmas']):
    m = hashlib.sha256()
    clean_answer = str(enigma['answer']).replace(' ', '')
    m.update(clean_answer.encode('utf8'))
    m.digest()
    print(i, enigma['answer'], m.hexdigest())
    script += f"""
function checkPassword{i}() {{
  var password = document.getElementById("password{i}").value;
  var sha_password = SHA256(password.replace(/ /g, ''));
  if (
    sha_password ==
    "{m.hexdigest()}"
  ) {{
    document.getElementById("enigma{i}-check").style = "display: block";
    document.getElementById("enigma{i}-no-check").style = "display: none";
  }} else {{
    document.getElementById("enigma{i}-no-check").style = "display: block";
    document.getElementById("enigma{i}-check").style = "display: none";
    document.getElementById("password{i}").value = "";
  }}
  final_code();
}}

checkPassword{i}();

"""

with open("script.js", 'w') as f:
    f.write(script)


################################ finish_script.js ################################

sum_answers = ""
for i, enigma in enumerate(data['enigmas']):
    clean_answer = str(enigma['answer']).replace(' ', '')
    sum_answers += clean_answer
sha_password  = hashlib.sha256(sum_answers.encode('utf8')).hexdigest()[:8].upper()
print("Final:", sum_answers, sha_password)
script = """function final_code() {
  var password =
"""
for i in range(len(data['enigmas'])-1):
    script += f"""    document.getElementById("password{i}").value + \n"""
script += f"""    document.getElementById("password{len(data['enigmas'])-1}").value;\n"""
script += f"""
  var sha_password = SHA256(
    password.replace(/ /g, "")
  );
  var user_code = sha_password.substring(0, 8).toUpperCase();
  if (
    SHA256(user_code) ==
    "{hashlib.sha256(sha_password.encode('utf8')).hexdigest()}"
  ) {{
    document.getElementById("finish-code").innerHTML =
      'Finish code: <span id="code">' + user_code + "</span>";
  }} else {{
    document.getElementById("finish-code").innerHTML = "";
  }}
}}
"""


with open("finish_script.js", 'w') as f:
    f.write(script)