130 lines
3.5 KiB
Python
130 lines
3.5 KiB
Python
import os
|
|
import time
|
|
import logging
|
|
import subprocess
|
|
|
|
from datetime import datetime
|
|
|
|
# Define file paths
|
|
GIT_REPO_PATH = "/app/site"
|
|
PUB_PATH = "/app/public"
|
|
FOOTER_FILE_PATH = "/app/site/themes/kitsune-theme/layouts/partials/footer.html"
|
|
|
|
|
|
def get_git_hash() -> str:
|
|
try:
|
|
result = subprocess.run(
|
|
["git", "rev-parse", "HEAD"],
|
|
cwd=GIT_REPO_PATH,
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
except FileNotFoundError:
|
|
logging.warning(
|
|
f"No site found at {GIT_REPO_PATH}, assuming first-start. Will clone."
|
|
)
|
|
return None
|
|
|
|
r_hash = result.stdout.strip()
|
|
return r_hash if len(r_hash) > 3 else None
|
|
|
|
|
|
def git_fetch() -> bool:
|
|
# Fetch the latest changes
|
|
subprocess.run(["git", "fetch"], cwd=GIT_REPO_PATH)
|
|
|
|
# Check if the local branch is behind the remote branch
|
|
result = subprocess.run(
|
|
["git", "status", "-uno"],
|
|
cwd=GIT_REPO_PATH,
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
|
|
# If output contains "Your branch is behind", there are updates
|
|
if "Your branch is behind" in result.stdout:
|
|
return True
|
|
return False
|
|
|
|
|
|
def update_repo():
|
|
# Make dir
|
|
logging.debug(f"Making dirs {GIT_REPO_PATH}")
|
|
if get_git_hash() is None and os.path.exists(GIT_REPO_PATH):
|
|
logging.warning("Clearing file path")
|
|
subprocess.run(["rm", "-rv", f"{GIT_REPO_PATH}/*"])
|
|
os.makedirs(GIT_REPO_PATH, exist_ok=True)
|
|
|
|
# Clone
|
|
url = os.environ.get("GIT_REPO_URL", "").strip('"')
|
|
if get_git_hash() is None:
|
|
logging.info(f"Cloning '{url}'.")
|
|
subprocess.run(["git", "clone", url, GIT_REPO_PATH, "--recurse-submodules"])
|
|
else:
|
|
# Check if fetch indicates a need for an update
|
|
if git_fetch():
|
|
logging.info(f"Fetch and pull '{url}'.")
|
|
subprocess.run(["git", "reset", "--hard"], cwd=GIT_REPO_PATH)
|
|
subprocess.run(["git", "pull"], cwd=GIT_REPO_PATH)
|
|
logging.info(f"Update submodules '{url}'.")
|
|
subprocess.run(
|
|
["git", "submodule", "update", "--init", "--recursive"],
|
|
cwd=GIT_REPO_PATH,
|
|
)
|
|
|
|
return True
|
|
return False
|
|
|
|
|
|
def build_pages():
|
|
logging.info("Cleaning files")
|
|
subprocess.run(["rm", "-rf", f"{PUB_PATH}/*"])
|
|
subprocess.run(["rm", "-rf", f"{GIT_REPO_PATH}/public/*"])
|
|
|
|
logging.info("Building pages")
|
|
subprocess.run(["hugo"], cwd=GIT_REPO_PATH)
|
|
|
|
logging.info("Copying files")
|
|
subprocess.run(["cp", "-r", f"{GIT_REPO_PATH}/public/.", PUB_PATH])
|
|
|
|
|
|
def update_footer(last_build_time, git_hash):
|
|
with open(FOOTER_FILE_PATH, "r") as file:
|
|
footer_content = file.read()
|
|
|
|
build_info = f"Last Build: {last_build_time}, Git Hash: {git_hash}"
|
|
new_footer_content = footer_content.replace("<!-- BUILD_METADATA -->", build_info)
|
|
|
|
with open(FOOTER_FILE_PATH, "w") as file:
|
|
file.write(new_footer_content)
|
|
|
|
|
|
def main():
|
|
# Time!
|
|
startime = time.time()
|
|
|
|
# Get the current Git hash
|
|
current_git_hash = get_git_hash()
|
|
|
|
# Update repo
|
|
logging.info("Updating repo")
|
|
if not update_repo():
|
|
logging.debug("Nothing to do")
|
|
return
|
|
|
|
# Update footer with build metadata
|
|
last_build_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
|
update_footer(last_build_time, current_git_hash)
|
|
|
|
# Rebuild site
|
|
logging.info("Building the site")
|
|
build_pages()
|
|
|
|
logging.info(f"Done in {time.time() - startime}!")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
main()
|