Pi66

merodi
A markdown-based static site generator.
git clone https://git.pi66.xyz/merodi

src/hash.py 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
import hashlib
import json
from os import path

from src.fileops import read_file, write_file
from src.hooks import hook_call
from src.log import warn

HASH_FILE_CONTENT: dict | None = None


def read_hash_file(config):
    global HASH_FILE_CONTENT
    if HASH_FILE_CONTENT is None:
        try:
            with open(config.cache.hash, "r") as f:
                HASH_FILE_CONTENT = json.load(f)
        except (FileNotFoundError, json.JSONDecodeError):
            HASH_FILE_CONTENT = {}


def write_hash_file(config):
    global HASH_FILE_CONTENT
    write_file(config.cache.hash, json.dumps(HASH_FILE_CONTENT, indent=2))

def clear_all_hashes(config):
    global HASH_FILE_CONTENT
    HASH_FILE_CONTENT = {}
    write_hash_file(config)

def clear_file_hash(md_path):
    global HASH_FILE_CONTENT
    if HASH_FILE_CONTENT is not None and md_path in HASH_FILE_CONTENT:
        del HASH_FILE_CONTENT[md_path]

def md5_handler(md_path):
    md_content = read_file(md_path)
    return hashlib.md5(md_content.encode()).hexdigest()


def hash_is_modified(md_path, new_hash):
    global HASH_FILE_CONTENT
    old_hash = HASH_FILE_CONTENT.get(md_path)
    if old_hash == new_hash:
        return None
    return new_hash


def append_md5(key, value):
    global HASH_FILE_CONTENT
    HASH_FILE_CONTENT[key] = value


def handle_hash_sync(config, md_path):
    global HASH_FILE_CONTENT
    try:
        hash_dir = config.cache.hash
        if config is None:
            return
        if not (path.exists(hash_dir) and path.isfile(hash_dir)):
            write_file(hash_dir, "{}")

        read_hash_file(config)

        if HASH_FILE_CONTENT is None:
            HASH_FILE_CONTENT = {}

        current_hash = md5_handler(md_path)
        md_rel_path = path.relpath(md_path, config.tree.markdown)
        hook_call("on_hash_check", md_rel_path, current_hash)
        new_hash = hash_is_modified(md_rel_path, current_hash)

        if new_hash is None:
            return None

        append_md5(md_rel_path, new_hash)
        write_hash_file(config)
        hook_call("on_hash_written", md_rel_path, new_hash)
        return True

    except Exception as e:
        warn(f"Hashing failed: {e}")