chore(style): add shebang, set bash strict mode, create function
[webi-installers/.git] / vps-addswap / install.sh
index f1f559f859ba2e40b96c4062fb363406f62256ff..64c849635ac2775a8ce358e1fbaf4cc77424c0bd 100644 (file)
@@ -1,12 +1,22 @@
 #!/bin/bash
+set -e
+set -u
 
-{
-    set -e
-    set -u
+function __init_vps_addswap() {
 
     default_size=2G
     my_size=${1:-$default_size}
 
+    if [ "0" != "$EUID" ] && [ "0" != "$UID" ]; then
+        echo Usage:
+        echo '        sudo env PATH="$PATH" vps-addswap' $my_size
+        exit 1
+    fi
+
+    if [ -f "/var/swapfile" ]; then
+        swapoff /var/swapfile
+    fi
+
     # Allocate a swapfile
     fallocate -l "$my_size" /var/swapfile
 
@@ -20,5 +30,6 @@
 
     # Cause swap to be activated on boot
     echo '/var/swapfile none swap sw 0 0' | tee -a /etc/fstab
-
 }
+
+__init_vps_addswap "$@"