1 files changed, 31 insertions, 0 deletions
diff --git a/machammer/hooks.py b/machammer/hooks.py
new file mode 100644
index 0000000..8826ad9
--- /dev/null
+++ b/machammer/hooks.py
@@ -0,0 +1,31 @@
+# -*- coding: utf-8 -*-
+
+import defaults
+
+PREF_DOMAIN = 'com.apple.loginwindow'
+
+
+def login(path=None):
+    """Set login hook to path, or disable login hook."""
+    if path is None:
+        return defaults.delete(PREF_DOMAIN, 'LoginHook')
+
+    return defaults.set(PREF_DOMAIN, 'LoginHook', path)
+
+
+def logout(path=None):
+    """Set logout hook to path, or disable logout hook."""
+    if path is None:
+        return defaults.delete(PREF_DOMAIN, 'LogoutHook')
+
+    return defaults.set(PREF_DOMAIN, 'LogoutHook', path)
+
+
+def reboot(path=None):
+    """Set reboot hook to path, or disable reboot hook."""
+    pass
+
+
+def shutdown(path=None):
+    """Set shutdown hook to path, or disable shutdown hook."""
+    pass