From a553740db55b30e1a7cc13f8fed224ee9e582e32 Mon Sep 17 00:00:00 2001
From: AJ ONeal <aj@therootcompany.com>
Date: Mon, 6 Jul 2020 04:15:25 +0000
Subject: [PATCH] minor fixes

---
 vps-addswap/install.sh | 11 ++++++++++-
 vps-utils/install.sh   | 12 ++++++++----
 2 files changed, 18 insertions(+), 5 deletions(-)

diff --git a/vps-addswap/install.sh b/vps-addswap/install.sh
index f1f559f..e3e3702 100644
--- a/vps-addswap/install.sh
+++ b/vps-addswap/install.sh
@@ -7,6 +7,16 @@
     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,4 @@
 
     # Cause swap to be activated on boot
     echo '/var/swapfile none swap sw 0 0' | tee -a /etc/fstab
-
 }
diff --git a/vps-utils/install.sh b/vps-utils/install.sh
index 029807b..1f664e9 100644
--- a/vps-utils/install.sh
+++ b/vps-utils/install.sh
@@ -1,10 +1,14 @@
+#!/bin/bash
+
 {
-    rm -f "$HOME/.local/bin/vps-myip" "$HOME/.local/bin/vps-addswap" "$HOME/.local/bin/cap-net-bind"
-    #webi_download "$WEBI_HOST/packages/vps-utils/" "$HOME/.local/bin/"
+    set -e
+    set -u
+
+    rm -f "$HOME/.local/bin/myip" "$HOME/.local/bin/vps-myip" "$HOME/.local/bin/vps-addswap" "$HOME/.local/bin/cap-net-bind"
     webi_download "$WEBI_HOST/packages/vps-utils/cap-net-bind.sh" "$HOME/.local/bin/cap-net-bind"
-    webi_download "$WEBI_HOST/packages/vps-utils/vps-myip.sh" "$HOME/.local/bin/vps-myip"
-    webi_download "$WEBI_HOST/packages/vps-utils/myip.sh" "$HOME/.local/bin/myip"
     webi_download "$WEBI_HOST/packages/vps-utils/vps-addswap.sh" "$HOME/.local/bin/vps-addswap"
+    webi_download "$WEBI_HOST/packages/myip/myip.sh" "$HOME/.local/bin/myip"
+    ln -s "$HOME/.local/bin/myip" "$HOME/.local/bin/vps-myip"
     chmod a+x "$HOME/.local/bin/cap-net-bind"
     chmod a+x "$HOME/.local/bin/myip"
     chmod a+x "$HOME/.local/bin/vps-"*
-- 
2.25.1