ccollect/tools/ccollect_add_source.sh

#!/bin/sh
# 
# 2007-2008 Nico Schottelius (nico-ccollect at schottelius.org)
# 
# This file is part of ccollect.
#
# ccollect is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
# 
# ccollect 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 ccollect. If not, see <http://www.gnu.org/licenses/>.
#
# 2007-08-16: Written for Netstream (www.netstream.ch)
#
# Creates a source from standard values specified in
# /etc/ccollect/defaults/sources
#

# standard values
CCOLLECT_CONF="${CCOLLECT_CONF:-/etc/ccollect}"
CSOURCES="${CCOLLECT_CONF}/sources"
CDEFAULTS="${CCOLLECT_CONF}/defaults"
SCONFIG="${CDEFAULTS}/sources"

# standard options: variable2filename
exclude="exclude"
summary="summary"
intervals="intervals"
pre_exec="pre_exec"
post_exec="post_exec"
rsync_options="rsync_options"
verbose="verbose"
very_verbose="very_verbose"

# options that we simply copy over
standard_opts="exclude summary intervals pre_exec post_exec rsync_options verbose very_verbose"

# options not in standard ccollect, used only for source generation
src_prefix="${SCONFIG}/source_prefix"
src_postfix="${SCONFIG}/source_postfix"
destination_base="${SCONFIG}/destination_base"

self="$(basename $0)"

# functions first
_echo()
{
   echo "${self}> $@"
}

_exit_err()
{
   _echo "$@"
   rm -f "$TMP"
   exit 1
}

# argv
if [ $# -lt 1 ]; then
   _exit_err "<hostnames to create sources for>"
fi

_echo "Reading defaults from ${SCONFIG} ..."

while [ $# -gt 0 ]; do
   source="$1"; shift

   # Create
   _echo "Creating ${source} ..."

   fullname="${CSOURCES}/${source}"

   # create source
   if [ -e "${fullname}" ]; then
      _echo "${fullname} already exists. Skipping."
      continue
   fi
   mkdir -p "${fullname}" || _exit_err Cannot create \"${fullname}\".

   # copy standard files
   for file in $standard_opts; do
      eval rfile=\"\$$file\"
      eval filename="${SCONFIG}/${rfile}"
      if [ -e "${filename}" ]; then
         _echo Copying \"$rfile\" to ${fullname} ...
         cp -r "${filename}" "${fullname}/${rfile}"
      fi
   done

   # create source entry
   if [ -f "${src_prefix}" ]; then
      source_source="$(cat "${src_prefix}")" || _exit_err "${src_prefix}: Reading failed."
   fi
   source_source="${source_source}${source}"
   if [ -f "${src_postfix}" ]; then
      source_source="${source_source}$(cat "${src_postfix}")" || _exit_err "${src_postfix}: Reading failed."
   fi
   _echo "Adding \"${source_source}\" as source for ${source}"
   echo "${source_source}" > "${fullname}/source" || _exit_err "Creating ${fullname}/source: failed."

   # create destination directory
   absbase=$(cd "${destination_base}" 2>/dev/null && pwd -P) || \
      _exit_err "${destination_base} must exist before creating sources."

   dest="${absbase}/${source}"
   _echo "Creating ${dest} ..."
   mkdir -p "${dest}" || _exit_err "${dest}: Cannot create."

   # link destination directory
   dest_abs=$(cd "${dest}" && pwd -P) || _exit_err "${dest}: Changing to newly created directory failed."
   echo "${dest_abs}" > "${fullname}/destination" || \
      _exit_err "${fullname}/destination: Failed to create."

done

exit 0