X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=monitor-server.init;h=30cd6e2714c45978f75d37ba5a41e831ec7f3126;hb=da62714c855a89f1b50d21844fa8f1e709850ebb;hp=df007518799adbb8c7e4d199ff8d53c7f1270179;hpb=170be1fae6fd5956641668443245c50fbf7fd566;p=monitor.git

diff --git a/monitor-server.init b/monitor-server.init
index df00751..30cd6e2 100644
--- a/monitor-server.init
+++ b/monitor-server.init
@@ -16,7 +16,7 @@
 . /etc/planetlab/plc_config
 local_config=/etc/planetlab/configs/site.xml
 
-MONITORPATH=/usr/share/monitor-server
+MONITORPATH=/usr/share/monitor
 
 # Be verbose
 set -x
@@ -33,34 +33,46 @@ export PGPORT=$PLC_DB_PORT
 ZABBIX_DB_USER="zabbixuser"
 ZABBIX_DB_NAME="zabbix"
 
+MONITOR_DB_USER="monitoruser"
+MONITOR_DB_NAME="monitor"
+
+WROTE_PG_CONFIG=
+
+if [ -z "$PLC_MONITOR_IP" ] ; then
+	PLC_MONITOR_IP=$( gethostbyname $PLC_MONITOR_HOST )
+fi
+
 function check_pg_hba ()
 {
+	NAME=$1
+	USER=$2
 	#### SETUP ACCESS to this user and database
 	mkdir -p $PGDATA/pg_hba.conf.d
-	ZABCONF=$PGDATA/pg_hba.conf.d/zabbix.conf
-	if [ ! -f $ZABCONF ] ; then
-		echo "host $ZABBIX_DB_NAME $ZABBIX_DB_USER 127.0.0.1/32 password"   > $ZABCONF
-		echo "host $ZABBIX_DB_NAME $ZABBIX_DB_USER $PLC_MONITOR_IP/32 password" >> $ZABCONF
+	CONF=$PGDATA/pg_hba.conf.d/${NAME}.conf
+	if [ ! -f $CONF ] ; then
+		echo "host $NAME $USER 127.0.0.1/32 password"   > $CONF
+		echo "host $NAME $USER $PLC_MONITOR_IP/32 password" >> $CONF
 
-		# NOTE: restart db to enable access by users granted above.
-		service plc restart posgresql
+		WROTE_PG_CONFIG="true"
 	fi
 }
 
 function check_user_and_db()
 {
     CREATED=
+	NAME=$1
+	USER=$2
     # confirm user is present or create it
-    user_present=$( psql -U postgres -c "select * from pg_user;" -d template1 | grep $ZABBIX_DB_USER )
+    user_present=$( psql -U postgres -c "select * from pg_user;" -d template1 | grep $USER )
     if [ -z $user_present ] ; then 
-        createuser --no-superuser --no-createdb --no-createrole --login --unencrypted --echo $ZABBIX_DB_USER -U postgres
+        createuser --no-superuser --no-createdb --no-createrole --login --unencrypted --echo $USER -U postgres
 		CREATED="true"
     fi
     
     # confirm database is present or create it
-    db_present=$( psql -U postgres -c "select * from pg_database;" -d template1 | grep $ZABBIX_DB_NAME )
+    db_present=$( psql -U postgres -c "select * from pg_database;" -d template1 | grep $NAME )
     if [ -z $db_present ] ; then
-        createdb --owner=$ZABBIX_DB_USER $ZABBIX_DB_NAME -U postgres
+        createdb --owner=$USER $NAME -U postgres
 		CREATED="true"
     fi
 
@@ -73,7 +85,7 @@ function check_user_and_db()
 		CREATED="true"
     fi
     if [ -n "$CREATED" ] ; then
-        psql -d template1 -U postgres -c "ALTER USER $ZABBIX_DB_USER WITH PASSWORD '$PLC_MONITOR_DBPASSWORD';"
+        psql -d template1 -U postgres -c "ALTER USER $USER WITH PASSWORD '$PLC_MONITOR_DBPASSWORD';"
     fi
 }
 
@@ -84,8 +96,13 @@ function if_present_load ()
         psql -d $ZABBIX_DB_NAME -U $ZABBIX_DB_USER < $file
     fi
 }
+function check_monitor_schema_and_data() 
+{
+	# NOTE: call create_all() to setup the database from the infovacuum model.
+	python -c "from monitor.database.infovacuum import *; from elixir import create_all; create_all()"
+}
 
-function check_schema_and_data() 
+function check_zabbix_schema_and_data() 
 {
     schema_present=$( psql -U $ZABBIX_DB_USER $ZABBIX_DB_NAME -c "\d;" < /dev/null | grep hosts )
     if [ -z $schema_present ] ; then
@@ -97,13 +114,50 @@ function check_schema_and_data()
     fi
 }
 
+function check_zabbix_templates_and_import ()
+{
+	# LOG IN
+	COOKIE_FILE=/tmp/cookiejar.txt
+	rm -f ${COOKIE_FILE}
+	TEMPLATES_DIR=${MONITORPATH}/zabbix/templates
+	curl -s --cookie $COOKIE_FILE --cookie-jar $COOKIE_FILE \
+			--form "enter=Enter" \
+			--form "name=Admin" \
+			--form "password=zabbix" \
+			"http://${PLC_MONITOR_HOST}/zabbix/index.php?login=1"
+	
+	deleted=$( grep 'deleted' $COOKIE_FILE )
+	if [ -n "$deleted" ] ; then
+		echo "Login to the zabbix web server failed!!!"
+		return 1
+	fi
+
+	for file in ${TEMPLATES_DIR}/*.xml ; do 
+		# 0 - update , 1 - skip, 0 - add
+		echo "############### IMPORTING $file" >> /var/log/monitor.log
+		curl -s --cookie $COOKIE_FILE --cookie-jar $COOKIE_FILE \
+			--form "config=1" \
+			--form "import_file=@${file}" \
+			--form "rules[host][exist]=0" \
+			--form "rules[host][missed]=0" \
+			--form "rules[template][exist]=0" \
+			--form "rules[template][missed]=1" \
+			--form "rules[item][exist]=0" \
+			--form "rules[item][missed]=0" \
+			--form "rules[trigger][exist]=0" \
+			--form "rules[trigger][missed]=0" \
+			--form "rules[graph][exist]=0" \
+			--form "rules[graph][missed]=0" \
+			--form "import=Import" \
+			"http://${PLC_MONITOR_HOST}/zabbix/exp_imp.php" >> /var/log/monitor.log
+	done
+}
+
+
 function check_monitor_conf ()
 {
 	MONITOR_CONFIG=/etc/monitor.conf
 
-	if [ -z "$PLC_MONITOR_IP" ] ; then
-		PLC_MONITOR_IP=$( gethostbyname $PLC_MONITOR_HOST )
-	fi
 
 	# Using plcsh add default, monitor user
 	plcsh <<EOF &>/dev/null 
@@ -135,9 +189,9 @@ API_AUTH_USER=${PLC_MONITOR_EMAIL}
 API_AUTH_PASSWORD=${PLC_MONITOR_DBPASSWORD}
 
 # SERVER PATHS
-MONITOR_SCRIPT_ROOT=/usr/share/monitor-server
-MONITOR_DATA_ROOT=/var/lib/monitor-server
-MONITOR_ARCHIVE_ROOT=/var/lib/monitor-server/archive-pdb
+MONITOR_SCRIPT_ROOT=${MONITORPATH}
+MONITOR_DATA_ROOT=/var/lib/monitor
+MONITOR_ARCHIVE_ROOT=/var/lib/monitor/archive-pdb
 
 MONITOR_HOSTNAME=${PLC_MONITOR_HOST}
 MONITOR_IP=${PLC_MONITOR_IP}
@@ -155,12 +209,15 @@ support_email=${PLC_MAIL_SUPPORT_ADDRESS}
 cc_email=
 
 [monitordatabase]
-monitor_dburi=postgres://monitoruser:${PLC_MONITOR_DBPASSWORD}@localhost:5432/monitor
-zabbix_dburi=postgres://zabbixuser:${PLC_MONITOR_DBPASSWORD}@localhost:5432/zabbix
+monitor_dburi=postgres://${MONITOR_DB_NAME}:${PLC_MONITOR_DBPASSWORD}@localhost:5432/${MONITOR_DB_NAME}
+zabbix_dburi=postgres://${ZABBIX_DB_USER}:${PLC_MONITOR_DBPASSWORD}@localhost:5432/${ZABBIX_DB_NAME}
 
 cachetime=60
 
+# Evaluated as true or false
 [commandline]
+cachecalls=True
+
 echo=False
 debug=False
 mail=True
@@ -239,11 +296,21 @@ case "$1" in
 		MESSAGE=$"Bootstrap Monitoring"
 		dialog "$MESSAGE"
 
-
 		# DATABASE acces, creation, and data loading
-		check_pg_hba
-		check_user_and_db 
-		check_schema_and_data
+		check_pg_hba $MONITOR_DB_NAME $MONITOR_DB_USER
+		check_user_and_db $MONITOR_DB_NAME $MONITOR_DB_USER
+		check_monitor_schema_and_data
+
+		check_pg_hba $ZABBIX_DB_NAME $ZABBIX_DB_USER
+		check_user_and_db $ZABBIX_DB_NAME $ZABBIX_DB_USER
+
+		if [ -n "$WROTE_PG_CONFIG" ] ; then
+			# NOTE: restart db to enable access by users granted above.
+			service plc restart postgresql
+		fi
+
+		check_zabbix_schema_and_data
+		check_zabbix_templates_and_import
 
 		# WRITE default /etc/monitor.conf
 		check_monitor_conf
@@ -253,7 +320,6 @@ case "$1" in
 		check_zab_agentd
 		check_zab_webconfig
 
-
 		result "$MESSAGE"
 	;;
 
@@ -261,7 +327,14 @@ case "$1" in
 		MESSAGE=$"Syncing PLC db with Zabbix DB"
 		dialog "$MESSAGE"
 
+		# turn off zabbix server, etc. before writing to the db.
+		service plc stop monitor 
+
 		$MONITORPATH/zabbix/zabbixsync.py --setupglobal &> /var/log/monitor-server
+		# import any templates
+		check_zabbix_templates_and_import
+
+		service plc start monitor 
 		
 		result "$MESSAGE"
 	;;