1 files changed, 22 insertions, 6 deletions
diff --git a/mullvad-vpn-beta.install b/mullvad-vpn-beta.install
index a60c2504f804..49a8146b3ad2 100644
--- a/mullvad-vpn-beta.install
+++ b/mullvad-vpn-beta.install
@@ -1,13 +1,29 @@
-post_upgrade() {
+post_install() {
+    systemctl enable --now mullvad-daemon
+
     echo -------------------------------------------------------------
-    echo 'You need to restart the mullvad daemon'
-    echo '# systemctl restart mullvad-daemon'
+    echo 'mullvad-daemon has been enabled & started.'
     echo -------------------------------------------------------------
 }
 
-post_install() {
+post_upgrade() {
+    systemctl daemon-reload
+    systemctl restart mullvad-daemon
+
     echo -------------------------------------------------------------
-    echo 'Make sure to enable and run the mullvad daemon'
-    echo '# systemctl enable --now mullvad-daemon'
+    echo 'mullvad-daemon has been restarted.'
     echo -------------------------------------------------------------
 }
+
+pre_remove() {
+    systemctl disable --now mullvad-daemon
+}
+
+post_remove() {
+    # Remove logs & cache
+    rm -rf /var/log/mullvad-vpn/
+    rm -rf /var/cache/mullvad-vpn/
+
+    # Remove config. See backup()
+    #rm -rf /etc/mullvad-vpn
+}
+\ No newline at end of file