pinkeen · March 19, 2025 18:55
diff --git a/edge-router.md b/edge-router.md
diff --git a/route-aws-region-to-interface.sh b/route-aws-region-to-interface.sh
 #!/bin/bash

 ##################################################################
 #        *** VyOS / EdgeRouter Configuration Script ***          #
 #                                                                #
 # Licensed under MIT (c) 2020 Filip Sobalski <[email protected]> #
 ##################################################################

 set -e

 if (( $# != 2 )); then

 cat<<EOF

 *** VyOS / EdgeRouter Configuration Script ***"

 # Licensed under MIT (c) 2020 Filip Sobalski <[email protected]>

 Job: Route traffic for AWS region subnets to an interface
 Usage: $0 <aws-region> <target-interface> [mode]
 Args:
  aws-region           name of target AWS region
  target-interface     name of interface to route traffic to
  mode (default: set)  set or delete the routes

 EOF

  exit 1
 fi

 MODE="${3:-set}"
 TARGET_IFACE="$2"
 AWS_REGION="$1"

 aws_region_subnets() {
    local AWS_REGION="$1"
    curl -sL https://ip-ranges.amazonaws.com/ip-ranges.json | python -c 'import sys, json; region = sys.argv[1]; sys.stdout.write(str.join("\n", sorted([prefix["ip_prefix"] for prefix in (json.loads(sys.stdin.read()))["prefixes"] if "ip_prefix" in prefix and prefix["region"] == region])) + "\n");' "$AWS_REGION"
 }

 vrun() { echo "[CFG CMD] $@"; /opt/vyatta/bin/vyatta-op-cmd-wrapper "$@"; }
 vcfg() { echo "[OP CMD] $@"; /opt/vyatta/sbin/vyatta-cfg-cmd-wrapper "$@"; }

 # Ask for confirmation when interactive shell
 if [[ ! "$-" != *i* ]] || [[ -z "$PS1" ]] ; then
    read -p "Will route traffic for AWS subnets in region $AWS_REGION to interface $TARGET_IFACE, ar you sure? [Y\N] " CONTINUE

    if [[ ! $CONTINUE =~ ^[Yy] ]] ; then
        echo "Aborted" >&2
        exit 9
    fi
 fi

 AWS_SUBNETS="$(aws_region_subnets "$AWS_REGION")"

 vcfg begin

 # Disable error exit as the python script returns duplicates and some commands will fai
 # there's no `unique` command in bash and I'm too lazy to fix the script 😅
 set +e 
 for AWS_SUBNET in $AWS_SUBNETS ; do
    vcfg $MODE protocols static interface-route $AWS_SUBNET next-hop-interface $TARGET_IFACE
 done
 set -e

 echo -e "\n---\n"

 vcfg commit
 vcfg end
diff --git a/vhk.sh b/vhk.sh
 ##################################################################
 #       *** VyOS / EdgeRouter Interactive Shell Setup ***        #
 #                                                                #
 # Licensed under MIT (c) 2020 Filip Sobalski <[email protected]> #
 ##################################################################


 # Make triple-sure we're in interactive shell not to break anything
 [[ "$-" != *i* ]] && return
 [[ ! -z "$PS1" ]] && return

 export HISTFILE="$HOME/.bash_history"
 export HIST_STAMPS="yyyy-mm-dd"
 export HISTSIZE=200
 export HISTFILESIZE=10000
 export HISTCONTROL="erasedups:ignoreboth"
 export HISTTIMEFORMAT='%F %T '

 # Enable history niceties completion search
 shopt -s cmdhist
 shopt -s lithist
 shopt -s histappend
 shopt -s histreedit
 shopt -s histverify

 # History search with arrow up/down
 bind '"\e[A": history-search-backward'
 bind '"\e[B": history-search-forward'

 # Jump to next/prev words with option(alt)+left/right
 bind '"\e[C": forward-char'
 bind '"\e[D": backward-char'
 bind '"\e[1;5C": forward-word'
 bind '"\e[1;5D": backward-word'

 if which ccze >/dev/null 2>&1 ; then
  ccze() { command ccze -A -o nolookups ; }

  function ll() {
    ls -alsh "$@" | ccze
  }

  function psa() {
    ps auxf "$@" | ccze
  }

  function h() {
    history "$@" | ccze
  }

 else
  alias ll='ls -alsh'
  alias psa='ps auxf'
 fi

 alias dmesg='dmesg -t'
 alias l='ll'

 # Bash function overloading code via https://stackoverflow.com/a/18839557
 vhk_func_cpy() {
  test -n "$(declare -f "$1")" || return
  eval "${_/$1/$2}"
 }

 vhk_func_ren() {
  copy_function "$@" || return
  unset -f "$1"
 }

 vhk_cfg_prompt() {
    # Vyatta configuration edit mode labels
    # See: https://wiki.vyos.net/wiki/Cli-shell-api
    cli-shell-api inSession || return

    echo -n "\[\e[0;36m\][edit]\[\e[0m\] "

    cli-shell-api sessionChanged >/dev/null 2>&1 && echo -n "\[\e[0;33m\]#changed\[\e[0m\] "
    cli-shell-api sessionUnsaved >/dev/null 2>&1 && echo -n "\[\e[0;91m\]!saved\[\e[0m\] "

    echo -n "\n"
 }

 vhk_prompt() {
    history -a; PS1="$(vhk_cfg_prompt)${VHK_PS1}"
 }

 if [ "`id -u`" -eq 0 ]; then
  export VHK_PS1="\[\e[1;31m\]\u\[\e[0;32m\]@\h\[\e[0m\] \[\e[0;94m\]\w\[\e[0m\] \[\e[1;32m\]->\[\e[0m\] "
 else
  export VHK_PS1="\[\e[0;33m\]\u\[\e[0;32m\]@\h\[\e[0m\] \[\e[0;94m\]\w\[\e[0m\] \[\e[1;32m\]->\[\e[0m\] "
 fi

 unset PS1

 export VHK_PROMPT_COMMAND_ORIG="$PROMPT_COMMAND"; export PROMPT_COMMAND="vhk_prompt"
	#!/bin/bash

	##################################################################
	# * VyOS / EdgeRouter Configuration Script * #
	# #
	# Licensed under MIT (c) 2020 Filip Sobalski <[email protected]> #
	##################################################################

	set -e

	if (( $# != 2 )); then

	cat<<EOF

	* VyOS / EdgeRouter Configuration Script *"

	# Licensed under MIT (c) 2020 Filip Sobalski <[email protected]>

	Job: Route traffic for AWS region subnets to an interface
	Usage: $0 <aws-region> <target-interface> [mode]
	Args:
	aws-region name of target AWS region
	target-interface name of interface to route traffic to
	mode (default: set) set or delete the routes

	EOF

	exit 1
	fi

	MODE="${3:-set}"
	TARGET_IFACE="$2"
	AWS_REGION="$1"

	aws_region_subnets() {
	local AWS_REGION="$1"
	curl -sL https://ip-ranges.amazonaws.com/ip-ranges.json \| python -c 'import sys, json; region = sys.argv[1]; sys.stdout.write(str.join("\n", sorted([prefix["ip_prefix"] for prefix in (json.loads(sys.stdin.read()))["prefixes"] if "ip_prefix" in prefix and prefix["region"] == region])) + "\n");' "$AWS_REGION"
	}

	vrun() { echo "[CFG CMD] $@"; /opt/vyatta/bin/vyatta-op-cmd-wrapper "$@"; }
	vcfg() { echo "[OP CMD] $@"; /opt/vyatta/sbin/vyatta-cfg-cmd-wrapper "$@"; }

	# Ask for confirmation when interactive shell
	if [[ ! "$-" != i ]] \|\| [[ -z "$PS1" ]] ; then
	read -p "Will route traffic for AWS subnets in region $AWS_REGION to interface $TARGET_IFACE, ar you sure? [Y\N] " CONTINUE

	if [[ ! $CONTINUE =~ ^[Yy] ]] ; then
	echo "Aborted" >&2
	exit 9
	fi
	fi

	AWS_SUBNETS="$(aws_region_subnets "$AWS_REGION")"

	vcfg begin

	# Disable error exit as the python script returns duplicates and some commands will fai
	# there's no `unique` command in bash and I'm too lazy to fix the script 😅
	set +e
	for AWS_SUBNET in $AWS_SUBNETS ; do
	vcfg $MODE protocols static interface-route $AWS_SUBNET next-hop-interface $TARGET_IFACE
	done
	set -e

	echo -e "\n---\n"

	vcfg commit
	vcfg end
	##################################################################
	# * VyOS / EdgeRouter Interactive Shell Setup * #
	# #
	# Licensed under MIT (c) 2020 Filip Sobalski <[email protected]> #
	##################################################################


	# Make triple-sure we're in interactive shell not to break anything
	[[ "$-" != i ]] && return
	[[ ! -z "$PS1" ]] && return

	export HISTFILE="$HOME/.bash_history"
	export HIST_STAMPS="yyyy-mm-dd"
	export HISTSIZE=200
	export HISTFILESIZE=10000
	export HISTCONTROL="erasedups:ignoreboth"
	export HISTTIMEFORMAT='%F %T '

	# Enable history niceties completion search
	shopt -s cmdhist
	shopt -s lithist
	shopt -s histappend
	shopt -s histreedit
	shopt -s histverify

	# History search with arrow up/down
	bind '"\e[A": history-search-backward'
	bind '"\e[B": history-search-forward'

	# Jump to next/prev words with option(alt)+left/right
	bind '"\e[C": forward-char'
	bind '"\e[D": backward-char'
	bind '"\e[1;5C": forward-word'
	bind '"\e[1;5D": backward-word'

	if which ccze >/dev/null 2>&1 ; then
	ccze() { command ccze -A -o nolookups ; }

	function ll() {
	ls -alsh "$@" \| ccze
	}

	function psa() {
	ps auxf "$@" \| ccze
	}

	function h() {
	history "$@" \| ccze
	}

	else
	alias ll='ls -alsh'
	alias psa='ps auxf'
	fi

	alias dmesg='dmesg -t'
	alias l='ll'

	# Bash function overloading code via https://stackoverflow.com/a/18839557
	vhk_func_cpy() {
	test -n "$(declare -f "$1")" \|\| return
	eval "${_/$1/$2}"
	}

	vhk_func_ren() {
	copy_function "$@" \|\| return
	unset -f "$1"
	}

	vhk_cfg_prompt() {
	# Vyatta configuration edit mode labels
	# See: https://wiki.vyos.net/wiki/Cli-shell-api
	cli-shell-api inSession \|\| return

	echo -n "\[\e[0;36m\][edit]\[\e[0m\] "

	cli-shell-api sessionChanged >/dev/null 2>&1 && echo -n "\[\e[0;33m\]#changed\[\e[0m\] "
	cli-shell-api sessionUnsaved >/dev/null 2>&1 && echo -n "\[\e[0;91m\]!saved\[\e[0m\] "

	echo -n "\n"
	}

	vhk_prompt() {
	history -a; PS1="$(vhk_cfg_prompt)${VHK_PS1}"
	}

	if [ "`id -u`" -eq 0 ]; then
	export VHK_PS1="\[\e[1;31m\]\u\[\e[0;32m\]@\h\[\e[0m\] \[\e[0;94m\]\w\[\e[0m\] \[\e[1;32m\]->\[\e[0m\] "
	else
	export VHK_PS1="\[\e[0;33m\]\u\[\e[0;32m\]@\h\[\e[0m\] \[\e[0;94m\]\w\[\e[0m\] \[\e[1;32m\]->\[\e[0m\] "
	fi

	unset PS1

	export VHK_PROMPT_COMMAND_ORIG="$PROMPT_COMMAND"; export PROMPT_COMMAND="vhk_prompt"