pkg/generate_homebrew_formula.py: Calculate SHA256 digest of archive

1 files changed, 11 insertions, 1 deletions

diff --git a/pkg/generate_homebrew_formula.py b/pkg/generate_homebrew_formula.py
index 122c56e..35162be 100755
--- a/pkg/generate_homebrew_formula.py
+++ b/pkg/generate_homebrew_formula.py
@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 
+from hashlib import sha256
 from string import Template
 import os
 import re
@@ -12,6 +13,15 @@ def get_version():
                 version = re.search('"([\d.]+)"', line)[1]
                 return version
 
+def archive_sha():
+    with open(
+        os.path.join(script_dir, '../dome-key_{}.tar.bz2'.format(get_version())),
+        'rb'
+    ) as f:
+        m = sha256()
+        m.update(f.read())
+        return m.hexdigest()
+
 
 script_dir = os.path.dirname(__file__)
 
@@ -28,7 +38,7 @@ with open(os.path.join(script_dir, 'com.teddywing.dome-key.plist'), 'r') as f:
 template = Template(homebrew_template)
 formula = template.substitute(
     VERSION=get_version(),
-    SHA256='unknown',
+    SHA256=archive_sha(),
     PLIST=plist.rstrip(),
 )