1 files changed, 3 insertions, 1 deletions
diff --git a/setup.sh b/setup.sh
index 3e39d40..7c7a9f4 100755
--- a/setup.sh
+++ b/setup.sh
@@ -64,6 +64,8 @@ configure_firewall() {
   ufw limit 22/tcp
 }
 
+enable_firewall() { ufw enable; }
+
 configure_user() {
   echo "== User =="
   mkdir -p /home/git
@@ -88,7 +90,7 @@ update() {
 
 cmd="$1"; shift;
 case "$cmd" in
-  update|start|startfg|stop|services|core|cache_clear) "$cmd" "$@" ;;
+  update|start|startfg|stop|services|core|cache_clear|enable_firewall) "$cmd" "$@" ;;
   configure|configure_firewall|configure_user|configure_fs|configure_packages|configure_git) "$cmd" "$@" ;;
   *) echo "Invalid command: $cmd"; exit 1 ;;
 esac