[pkg] add helpers to repo, generate hooks in snap

1 files changed, 16 insertions, 1 deletions

diff --git a/branding/templates/snap/generate.py b/branding/templates/snap/generate.py
index b5ca641..e62b220 100755
--- a/branding/templates/snap/generate.py
+++ b/branding/templates/snap/generate.py
@@ -16,8 +16,14 @@ SNAPCRAFT = 'snapcraft.yaml'
 here = os.path.split(os.path.realpath(__file__))[0]

 data = json.load(open(os.path.join(here, 'data.json')))

 

+binaryName = data['binaryName']

+

 DESKTOP_TEMPLATE = 'local/app.desktop'

-DESKTOP = 'local/{}.desktop'.format(data['binaryName'])

+DESKTOP = 'local/{}.desktop'.format(binaryName)

+

+POLKIT_TEMPLATE = 'local/pre/se.leap.bitmask.snap.policy'

+POLKIT_FILE = 'se.leap.bitmask.{}.policy'.format(binaryName)

+POLKIT = 'local/pre/' + POLKIT_FILE

 

 template = Template(open(TEMPLATE).read())

 with open(SNAPCRAFT, 'w') as output:

@@ -28,5 +34,14 @@ with open(DESKTOP, 'w') as output:
     output.write(template.safe_substitute(data))

 os.remove(DESKTOP_TEMPLATE)

 

+template = Template(open(POLKIT_TEMPLATE).read())

+with open(POLKIT, 'w') as output:

+    output.write(template.safe_substitute(data))

+os.remove(POLKIT_TEMPLATE)

+

+os.putenv('POLKIT_FILE', POLKIT_FILE)

+os.putenv('APP_NAME', binaryName)

+os.system('cd local/pre && ./pack_installers')

+

 print("[+] Snapcraft spec written to {path}".format(

     path=os.path.abspath(SNAPCRAFT)))