From b31fb5823fc33cd935656b434b32b327bf9b5e63 Mon Sep 17 00:00:00 2001
From: Daniil Baturin <daniil@baturin.org>
Date: Sat, 16 Nov 2013 14:56:14 +0100
Subject: Bug #25: Add template for running conf and op mode commands from
shell scripts.
---
Makefile.am | 3 +
functions/wrapper/script-template | 166 ++++++++++++++++++++++++++++++++++++++
2 files changed, 169 insertions(+)
create mode 100644 functions/wrapper/script-template
diff --git a/Makefile.am b/Makefile.am
index 8f84603..c02dc43 100644
--- a/Makefile.am
+++ b/Makefile.am
@@ -23,6 +23,9 @@ logrotate_DATA = etc/logrotate.d/vyatta-config-logs
interpdir = $(datadir)/vyatta-cfg/functions/interpreter/
interp_DATA = functions/interpreter/vyatta-cfg-run
+wrapperdir = /opt/vyatta/etc/functions/
+wrapper_DATA = functions/wrapper/script-template
+
src/cparse/cparse.cpp: src/cparse/cparse.ypp src/cparse/cparse_def.h
bison -p cparse_ --defines=src/cparse/cparse.h -o $@ $<
diff --git a/functions/wrapper/script-template b/functions/wrapper/script-template
new file mode 100644
index 0000000..51cb2c0
--- /dev/null
+++ b/functions/wrapper/script-template
@@ -0,0 +1,166 @@
+#!/bin/vbash
+#
+# Set up aliases and functions for running Vyatta commands from scripts
+#
+# Copyright (C) 2012 Vyatta, Inc.
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License version 2 as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+#
+# Authors: John Southworth, Daniil Baturin
+#
+# Usage: source /opt/vyatta/etc/functions/script-template
+
+export _OFR_CONFIGURE='ok'
+source /etc/bash_completion.d/vyatta-cfg
+unset _OFR_CONFIGURE
+
+shopt -s expand_aliases
+
+SBIN_PATH=/opt/vyatta/sbin
+BIN_PATH=/opt/vyatta/bin
+
+API=/bin/cli-shell-api
+
+# "pipe" functions
+count ()
+{
+ wc -l
+}
+
+match ()
+{
+ grep -E -e "$1"
+}
+
+no-match ()
+{
+ grep -E -v -e "$1"
+}
+
+no-more ()
+{
+ cat
+}
+
+function vyatta_configure()
+{
+ session_env=$($API getSessionEnv $PPID)
+ eval $session_env
+ $API setupSession
+ echo $session_env
+}
+
+function list()
+{
+ local -a expanded_api_args
+ local -a args=( "$@" )
+ local path='/opt/vyatta/share/vyatta-cfg/templates'
+ vyatta_config_expand_compwords show "${args[@]}"
+ for elem in "${expanded_api_args[@]:1}"; do
+ path+="/$elem"
+ done
+ if [[ ! -f $path/node.def ]]; then
+ return 1
+ fi
+ if grep -q -e 'tag:' "$path/node.def"; then
+ echo $($API listEffectiveNodes "${expanded_api_args[@]:1}")
+ elif grep -q -e 'multi:' "$path/node.def"; then
+ echo $($API returnEffectiveValues "${expanded_api_args[@]:1}")
+ else
+ echo $($API returnEffectiveValue "${expanded_api_args[@]:1}")
+ fi
+}
+
+function vyatta_exit_configure ()
+{
+ $API teardownSession
+ echo -n 'export -n VYATTA_CONFIG_TMP; '
+ echo -n 'export -n VYATTA_CHANGES_ONLY_DIR; '
+ echo -n 'export -n VYATTA_ACTIVE_CONFIGURATION_DIR; '
+ echo -n 'export -n VYATTA_TEMPLATE_LEVEL; '
+ echo -n 'export -n VYATTA_CONFIG_TEMPLATE; '
+ echo -n 'export -n VYATTA_TEMP_CONFIG_DIR; '
+ echo -n 'export -n VYATTA_EDIT_LEVEL; '
+}
+
+function load ()
+{
+ if [[ $# -eq 0 ]]; then
+ echo -e "You must provide a file name to load the config from"
+ return 1
+ fi
+
+ # don't load if there are uncommitted changes.
+ if $API sessionChanged; then
+ echo "Cannot load: configuration modified."
+ echo "Commit or discard the changes before loading a config file."
+ return 1
+ fi
+ # return to top level.
+ reset_edit_level
+ ${vyatta_sbindir}/vyatta-load-config.pl "$@"
+}
+
+function save ()
+{
+ if [[ $# -eq 0 ]]; then
+ echo -e "You must provide a file name to save the config to"
+ return 1
+ fi
+ if $API sessionChanged; then
+ echo -e "Warning: you have uncommitted changes that will not be saved.\n"
+ fi
+ # return to top level.
+ reset_edit_level
+ # transform individual args into quoted strings
+ local arg=''
+ local save_cmd="${vyatta_sbindir}/vyatta-save-config.pl"
+ for arg in "$@"; do
+ save_cmd+=" '$arg'"
+ done
+ eval "sudo sg vyattacfg \"umask 0002 ; $save_cmd\""
+ $API unmarkSessionUnsaved
+}
+
+function reset_edit_level ()
+{
+ edit_env=$($API getEditResetEnv)
+ eval $edit_env
+ return $?
+}
+
+function edit ()
+{
+ edit_env=$($API getEditEnv "$@")
+ eval $edit_env
+}
+
+function top ()
+{
+ if $API editLevelAtRoot; then
+ echo "Already at the top level"
+ return 0
+ fi
+
+ # go to the top level.
+ reset_edit_level
+}
+
+function up ()
+{
+ edit_env=$($API getEditUpEnv "$@")
+ eval $edit_env
+}
+
+alias configure='eval $(vyatta_configure)'
+alias exit='eval $(vyatta_exit_configure)'
--
cgit v1.2.3