From 89414577c71b75440fff728edffeb5d9881056d3 Mon Sep 17 00:00:00 2001
From: Joey Hafner <joey@jafner.net>
Date: Wed, 9 Oct 2024 16:17:51 -0700
Subject: [PATCH] Variablize vyos target for SSH commands, add function to run
 arbitrary op-mode commands

---
 homelab/vyos/op.sh   |  7 +++++++
 homelab/vyos/vyos.sh | 34 +++++++++++++++++++++++++---------
 2 files changed, 32 insertions(+), 9 deletions(-)
 create mode 100644 homelab/vyos/op.sh

diff --git a/homelab/vyos/op.sh b/homelab/vyos/op.sh
new file mode 100644
index 00000000..a0e14f1b
--- /dev/null
+++ b/homelab/vyos/op.sh
@@ -0,0 +1,7 @@
+#!/bin/vbash
+
+command="$@"
+
+source /opt/vyatta/etc/functions/script-template
+
+run $command
\ No newline at end of file
diff --git a/homelab/vyos/vyos.sh b/homelab/vyos/vyos.sh
index 910ee2d3..7d34683e 100755
--- a/homelab/vyos/vyos.sh
+++ b/homelab/vyos/vyos.sh
@@ -2,30 +2,44 @@
 
 SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
 
+# Change this to the user, host, (and optionally port) of your VyOS target.
+VYOS_TARGET="vyos@192.168.1.1"
+
 # Returns saved config file
 function get_config_saved () {
-  ssh vyos@192.168.1.1 'cat /config/config.boot'
+  ssh $VYOS_TARGET 'cat /config/config.boot'
 }
 
 # Returns active config file
 function get_config_active () {
-  scp -q ./get_config.sh vyos@192.168.1.1:/home/vyos/get_config.sh
-  ssh vyos@192.168.1.1 'chmod +x /home/vyos/get_config.sh; /home/vyos/get_config.sh; rm /home/vyos/get_config.sh'
+  scp -q ./get_config.sh $VYOS_TARGET:/home/vyos/get_config.sh
+  ssh $VYOS_TARGET 'chmod +x /home/vyos/get_config.sh; /home/vyos/get_config.sh; rm /home/vyos/get_config.sh'
 }
 
 # Push local ./config.boot to remote /home/vyos/config.boot
 function post_config () {
-  scp -q ./config.boot vyos@192.168.1.1:/home/vyos/config.boot
+  scp -q ./config.boot    :/home/vyos/config.boot
 }
 
 function load_config () {
-  scp -q ./load_config.sh vyos@192.168.1.1:/home/vyos/load_config.sh
-  ssh vyos@192.168.1.1 'chmod +x /home/vyos/load_config.sh; /home/vyos/load_config.sh; rm /home/vyos/load_config.sh'
+  scp -q ./load_config.sh $VYOS_TARGET:/home/vyos/load_config.sh
+  ssh $VYOS_TARGET 'chmod +x /home/vyos/load_config.sh; /home/vyos/load_config.sh; rm /home/vyos/load_config.sh'
 }
 
 function save_config () {
-  scp -q ./save_config.sh vyos@192.168.1.1:/home/vyos/save_config.sh
-  ssh vyos@192.168.1.1 'chmod +x /home/vyos/save_config.sh; /home/vyos/save_config.sh; rm /home/vyos/save_config.sh'
+  scp -q ./save_config.sh $VYOS_TARGET:/home/vyos/save_config.sh
+  ssh $VYOS_TARGET 'chmod +x /home/vyos/save_config.sh; /home/vyos/save_config.sh; rm /home/vyos/save_config.sh'
+}
+
+function get_dhcp_leases () {
+  scp -q ./op.sh $VYOS_TARGET:/home/vyos/op.sh
+  ssh $VYOS_TARGET 'chmod +x /home/vyos/op.sh; /home/vyos/op.sh "show dhcp server leases"; rm /home/vyos/op.sh'
+}
+
+function op () {
+  command="$@"
+  scp -q ./op.sh $VYOS_TARGET:/home/vyos/op.sh
+  ssh $VYOS_TARGET "chmod +x /home/vyos/op.sh; /home/vyos/op.sh $command; rm /home/vyos/op.sh"
 }
 
 function pull () {
@@ -44,4 +58,6 @@ function edit () {
   push
 }
 
-"$1"
\ No newline at end of file
+"$@"
+
+# Fair warning, this script is trash.
\ No newline at end of file