#!/bin/env python

import subprocess
import os
import re

STAGING_LOCAL="/tmp/blogstaging/"
ASSETS_LOCAL="/home/gabu/Documents/Obsidian_Vaults/commonplace/z_Assets/"
STATIC_REMOTE="/var/www/gabusite/static/"
CONTENT_REMOTE="/var/www/gabusite/content/"
GH_REPO_URL="https://github.com/bonkeryonker/blog_vault.git"

'''
Run shell commands to display outputs via notify-send
Args:
    msg (string) - The message to output
    title (string) - The title of the message
'''
def notify_send(msg, title="Sync Blog"):
    subprocess.run(["notify-send", title, msg])

'''
Clean up staging directory via shell commands
'''
def cleanStaging():
    subprocess.run(["rm", "-rf", STAGING_LOCAL])

'''
Run shell commands to clone github repo
Returns:
    int - Return code of `git clone` command
'''
def clone_from_github():
    # Create staging directory if it doesn't already exist
    subprocess.run(["mkdir", "-p", STAGING_LOCAL])

    # Clone remote repo to staging directory
    result = subprocess.run(["git", "clone", GH_REPO_URL, STAGING_LOCAL])
    
    return result

'''
Walk through the cloned repo and return a list containing the paths
of all .md files
Returns:
    list - list of filepaths to all .md files in STAGING_LOCAL
'''
def collect_md_filepaths():
    mdfiles = []
    for root, dirs, files in os.walk(STAGING_LOCAL):
        for file in files:
            if file.endswith(".md"):
                mdfiles.append(os.path.join(root, file))
    return mdfiles

'''
Scan the content of the passed file for Obsidian format image embeds, and convert them to the correct format.
(Eg. turn "![Image Description](/img/myImage.jpg)" into "![Image Description](/img/myImage.jpg)")
Args:
    file - path to the markdown file to patch
Returns:
    int - The amount of patchable strings in the file
'''
def patch_img_tag(file):
    content = []
    with open(file, 'r') as infile:
        content = infile.read()

    # Create a list containing all Obsidian image embed links
    images = re.findall(r'\!\[\[([^]]*\.(png|jpg|gif))\]\]', content) #Returns a list of tuples of format: ('image_name.jpg', 'jpg')
    
    # If there are no image embeds, return early
    if len(images) == 0:
        return 0

    # Modify image embed links to Hugo format
    for image in images:
        modified_img_string = f"![Image Description](/img/{image[0].replace(' ', '%20')})"
        content = content.replace(f"![[{image[0]}]]", modified_img_string)

    # Write out the modified content to the .md files
    with open(file, 'w') as outfile:
        outfile.write(content)

    return len(images)

def sync_local2remote(localRootDir, remoteRootDir):
    result = subprocess.run(["rsync", "-avzP", "-e", "ssh", 
                                 "--exclude", ".git/",
                                 "--exclude", ".gitignore",
                                 "--exclude", ".obsidian/",
                                 localRootDir, f"gabu@blogbox:{remoteRootDir}"
                                 ])
    return result

if __name__ == "__main__":
    if clone_from_github().returncode != 0:
        print("An error occurred cloning from github.")
        notify_send("An error occurred cloning from github.")
        cleanStaging()
        exit()

    # Variables to hold the total patch counts for notifications
    tags_patched = 0
    files_patched = 0

    mdfiles = collect_md_filepaths()
    for file in mdfiles:
        patchcount = patch_img_tag(file)
        print(f"{patchcount} image tags patched in {file}")
        if patchcount > 0:
            files_patched += 1
            tags_patched += patchcount

    returncode = sync_local2remote(STAGING_LOCAL, CONTENT_REMOTE).returncode
    if returncode in (0, 23, 24):
        print(f"Content sync completed successfully. (Code: {returncode})")
    else:
        print(f"Failed syncing content to remote. (Code: {returncode})")
        notify_send(f"Failed syncing content to remote. (Code: {returncode})")
        exit()
    
    returncode = sync_local2remote(ASSETS_LOCAL, STATIC_REMOTE).returncode
    if returncode in (0, 23, 24):
        print(f"Assets sync completed successfully. (Code: {returncode})")
    else:
        print(f"Failed syncing assets to remote. (Code: {returncode})")
        notify_send(f"Failed syncing assets to remote. (Code: {returncode})")
        exit()

    notify_send(f"Sync successful.\n(Patched {tags_patched} image tags in {files_patched} files)")
    cleanStaging()