From ebaac290e76eabb55c39176a1a4af8eeec51a8a9 Mon Sep 17 00:00:00 2001
From: Eric Callahan <arksine.code@gmail.com>
Date: Fri, 18 Jun 2021 19:11:09 -0400
Subject: [PATCH] utils: add a method to calculate the hash of a directory

Signed-off-by:  Eric Callahan <arksine.code@gmail.com>
---
 moonraker/utils.py | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/moonraker/utils.py b/moonraker/utils.py
index 04c3639..cc1073d 100644
--- a/moonraker/utils.py
+++ b/moonraker/utils.py
@@ -11,10 +11,12 @@ import os
 import sys
 import subprocess
 import asyncio
+import hashlib
 from queue import SimpleQueue as Queue
 
 # Annotation imports
 from typing import (
+    List,
     Optional,
     ClassVar,
     Tuple,
@@ -122,3 +124,28 @@ def setup_logging(log_file: str,
             queue, stdout_hdlr)
     listener.start()
     return listener, file_hdlr
+
+def hash_directory(dir_path: str,
+                   ignore_exts: List[str],
+                   ignore_dirs: List[str]
+                   ) -> str:
+    checksum = hashlib.blake2s()
+    if not os.path.exists(dir_path):
+        return ""
+    for dpath, dnames, fnames in os.walk(dir_path):
+        valid_dirs: List[str] = []
+        for dname in sorted(dnames):
+            if dname[0] == '.' or dname in ignore_dirs:
+                continue
+            valid_dirs.append(dname)
+        dnames[:] = valid_dirs
+        for fname in sorted(fnames):
+            ext = os.path.splitext(fname)[-1].lower()
+            if fname[0] == '.' or ext in ignore_exts:
+                continue
+            fpath = pathlib.Path(os.path.join(dpath, fname))
+            try:
+                checksum.update(fpath.read_bytes())
+            except Exception:
+                pass
+    return checksum.hexdigest()