gnms 2.1.0.rc1

data/lib/gui/syslog_ref_window.rb

@@ -0,0 +1,288 @@
+class SyslogRefWindow
+
+	def clear_ls()
+		@syslog_ref_ls.clear()
+	end
+	#populate listview with saved values
+	def populate_ls()
+   	$syslog_refs.each_key do |name|
+		  iter = @syslog_ref_ls.append
+		  iter[0] = name
+		  iter[1] = $syslog_refs[name].description
+		end
+	end
+
+	def initialize()
+		#save syslogref hash
+		@syslogrefs_backup=$syslog_refs.dup
+
+		@window=Gtk::Window.new
+
+   	@window.set_title("Syslog Ref configuration")
+   	@window.set_size_request(300, 200)
+
+      @window.signal_connect("key_press_event") {|w,e|
+        if e.keyval == Gdk::Keyval::GDK_Escape
+			  @window.destroy
+        end
+   	}
+
+		frame = Gtk::Frame::new
+		frame.border_width=5
+		frame.show
+		vbox_root=Gtk::VBox::new
+		vbox_root.show
+		@window.add vbox_root
+		vbox_root.add frame
+
+		vbox = Gtk::VBox.new(false,0)
+		vbox.show
+		frame.add vbox
+
+	   @syslog_ref_ls = Gtk::ListStore.new(String, String)
+  	   @syslog_ref_view = Gtk::TreeView.new(@syslog_ref_ls)
+		@syslog_ref_view.show
+ 		column1 = Gtk::TreeViewColumn.new("Name",
+					Gtk::CellRendererText.new, {:text => 0})
+  		column1.set_sort_column_id(0)
+  		column2 = Gtk::TreeViewColumn.new("Description",
+					Gtk::CellRendererText.new, {:text => 1})
+  		column2.set_sort_column_id(1)
+
+  		@syslog_ref_view.append_column(column1)
+  		@syslog_ref_view.append_column(column2)
+  		
+		scroll = Gtk::ScrolledWindow.new(nil, nil)
+		scroll.border_width=2
+		scroll.set_policy(Gtk::POLICY_NEVER, Gtk::POLICY_AUTOMATIC)  
+		scroll.add @syslog_ref_view
+		scroll.show
+
+		# Create a menu
+		menu = Gtk::Menu.new
+
+		item_add = Gtk::ImageMenuItem.new("Add Syslog Ref")
+ 		image_item_add = Gtk::Image.new() 
+ 		image_item_add.set_stock(Gtk::Stock::ADD) 
+		item_add.set_image(image_item_add)
+		item_add.signal_connect("activate") { add_gui_syslog_ref() }
+		menu.append(item_add)
+		item_del = Gtk::ImageMenuItem.new("Delete selected Syslog Ref")
+ 		image_item_del = Gtk::Image.new() 
+ 		image_item_del.set_stock(Gtk::Stock::REMOVE) 
+		item_del.set_image(image_item_del)
+		item_del.signal_connect("activate") {
+			if @syslog_ref_view.selection.selected!=nil
+				if SyslogRef::del_syslog_ref(@syslog_ref_view.selection.selected[0])
+					@syslog_ref_view.model.remove(@syslog_ref_view.selection.selected)
+				else
+   				dialog = Gtk::MessageDialog.new(@window, Gtk::Dialog::DESTROY_WITH_PARENT,
+                             Gtk::MessageDialog::ERROR,
+                             Gtk::MessageDialog::BUTTONS_OK,
+                             "You can't delete! Syslog Ref in use")
+					dialog.run
+					dialog.destroy
+				end
+			end
+		}
+		menu.append(item_del)
+		item_edit = Gtk::ImageMenuItem.new("Edit Syslog Ref")
+ 		image_item_edit = Gtk::Image.new() 
+ 		image_item_edit.set_stock(Gtk::Stock::EDIT) 
+		item_edit.set_image(image_item_edit)
+		item_edit.signal_connect("activate") { 
+			if @syslog_ref_view.selection.selected != nil
+				edit_syslog_ref(@syslog_ref_view.selection.selected[0])
+			end
+		}
+		menu.append(item_edit)
+
+		menu.show_all
+
+		@syslog_ref_view.signal_connect("button_press_event") do |widget, event|
+		  if event.kind_of? Gdk::EventButton and event.button == 3
+			 if @syslog_ref_view.selection.selected == nil
+				 item_del.set_sensitive(false)
+				 item_edit.set_sensitive(false)
+			 else
+				 item_del.set_sensitive(true)
+				 item_edit.set_sensitive(true)
+			 end
+   		 menu.popup(nil, nil, event.button, event.time)
+		  elsif (@syslog_ref_view.selection.selected != nil) && (event.event_type == Gdk::Event::BUTTON2_PRESS)
+				edit_syslog_ref(@syslog_ref_view.selection.selected[0])	
+		  end
+		end
+
+		populate_ls()
+		vbox.add(scroll)
+
+		bbox = Gtk::HBox::new(FALSE, 10)
+		bbox.border_width=10
+		bbox.show
+
+		ok_button = Gtk::Button.new(Gtk::Stock::OK)
+		ok_button.show
+		ok_button.signal_connect("clicked") {
+			SyslogRef::write_db_syslogref()
+			@window.destroy
+		}
+
+		cancel_button = Gtk::Button.new(Gtk::Stock::CANCEL)
+		cancel_button.show
+		cancel_button.signal_connect("clicked") {
+		  $syslog_refs=@syslogrefs_backup
+   	  @window.destroy
+		}
+		ok_button.set_flags(Gtk::Widget::CAN_DEFAULT)
+		bbox.pack_start(ok_button, TRUE, TRUE, 0)
+		cancel_button.set_flags(Gtk::Widget::CAN_DEFAULT)
+		bbox.pack_start(cancel_button, TRUE, TRUE, 0)
+
+		separator = Gtk::HSeparator::new()
+		separator.show
+		vbox_root.pack_start(separator, FALSE, TRUE, 0)
+		vbox_root.pack_start(bbox, FALSE, TRUE, 0)
+
+		@window.show
+   end
+
+	def add_gui_syslog_ref()
+		add_modify_syslog_ref(true)
+	end
+
+	def edit_syslog_ref(syslog_ref_name)
+		add_modify_syslog_ref(false, syslog_ref_name)
+	end
+
+	def add_modify_syslog_ref(add_boolean, syslog_ref_name=nil)
+		window=Gtk::Window.new
+		window.set_modal(true)
+		lbl="Edit Syslog Ref"
+		if add_boolean
+			lbl="Add Syslog Ref"
+		end
+		
+   	window.set_title(lbl)
+   	window.set_size_request(460, 160)
+
+      window.signal_connect("key_press_event") {|w,e|
+        if e.keyval == Gdk::Keyval::GDK_Escape
+			  window.destroy
+        end
+   	}
+
+		frame = Gtk::Frame::new
+		frame.border_width=10
+		frame.show
+		vbox_root=Gtk::VBox::new
+		vbox_root.show
+		window.add vbox_root
+		vbox_root.add frame
+
+		vbox = Gtk::VBox.new
+		vbox.show
+		frame.add vbox
+
+		table = Gtk::Table.new(3,2,false)
+		table.show
+
+		label_name = Gtk::Label.new "Name"
+		label_name.show
+		if add_boolean
+			entry_name = Gtk::Entry.new			
+		else
+			entry_name = Gtk::Entry.new	
+			entry_name.set_editable false
+			entry_name.set_text(syslog_ref_name)
+		end
+		entry_name.show
+
+		table.attach(label_name,0,1,0,1)
+		table.attach(entry_name,1,2,0,1)
+
+		entry_description = Gtk::Entry.new
+		if !add_boolean
+			entry_description.set_text($syslog_refs[entry_name.text()].description) unless entry_name.text() == ""
+		end
+		entry_description.show
+		label_description = Gtk::Label.new "Description"
+		label_description.show
+
+		table.attach(label_description,0,1,1,2)
+		table.attach(entry_description,1,2,1,2)
+
+		entry_value = Gtk::Entry.new
+		if !add_boolean
+			entry_value.set_text($syslog_refs[entry_name.text()].value.source) unless entry_name.text() == ""
+		end
+		entry_value.show
+		label_value = Gtk::Label.new "Value"
+		label_value.show
+
+		table.attach(label_value,0,1,2,3)
+		table.attach(entry_value,1,2,2,3)
+		if !add_boolean
+    	  entry_name.signal_connect("changed") {|e|
+      	  if e.text() != ""
+				  entry_description.set_text($syslog_refs[e.text()].description)
+				  entry_value.set_text($syslog_refs[e.text()].value)
+			  end
+    	  }
+		end
+		bbox =Gtk::HBox::new(FALSE, 10)
+		bbox.border_width=10
+		bbox.show
+
+		ok_button = Gtk::Button.new(Gtk::Stock::OK)
+		ok_button.show
+		ok_button.signal_connect("clicked") {
+			if add_boolean
+				#new entry!
+				if $syslog_refs[entry_name.text()] == nil
+					SyslogRef::add_syslog_ref(entry_name.text(),entry_description.text(),entry_value.text())
+				   iter = @syslog_ref_ls.append
+				   iter[0] = entry_name.text()
+				   iter[1] = entry_description.text()
+					window.destroy 
+				else
+   				dialog = Gtk::MessageDialog.new(window, Gtk::Dialog::DESTROY_WITH_PARENT,
+                             Gtk::MessageDialog::ERROR,
+                             Gtk::MessageDialog::BUTTONS_OK,
+                             "Syslog Ref '#{entry_name.text()}' already exists!")
+					dialog.run
+					dialog.destroy
+				end
+			else
+				#modify existing entry
+				$syslog_refs[entry_name.text()].description=entry_description.text()
+				$syslog_refs[entry_name.text()].set_value(entry_value.text())
+			   @syslog_ref_ls.each do |model,path,iter|
+					if iter[0] == entry_name.text()
+						iter[1]=entry_description.text()
+						break
+					end
+				end
+				window.destroy 
+			end
+		}
+
+		cancel_button = Gtk::Button.new(Gtk::Stock::CANCEL)
+		cancel_button.show
+		cancel_button.signal_connect("clicked") {
+   	  window.destroy
+		}
+
+		ok_button.set_flags(Gtk::Widget::CAN_DEFAULT)
+		bbox.pack_start(ok_button, TRUE, TRUE, 0)
+		cancel_button.set_flags(Gtk::Widget::CAN_DEFAULT)
+		bbox.pack_start(cancel_button, TRUE, TRUE, 0)
+
+		vbox.add(table)
+		separator = Gtk::HSeparator::new()
+		separator.show
+		vbox_root.pack_start(separator, FALSE, TRUE, 0)
+		vbox_root.pack_start(bbox, FALSE, TRUE, 0)
+		window.show
+	end
+end

data/lib/gui/treenode_window.rb

@@ -0,0 +1,320 @@
+require "#{GNMSLIB}"+'/gui/color_macro'
+
+class TreeNodeWindow
+
+	#
+	#	check if the tree is already visible
+	#
+	def visible?()
+		return @treenode.visible?()
+	end
+
+	#
+	#	show the tree
+	#
+	def show()
+		if !$treenode_win.visible?
+  			 reload()
+		end
+		@treenode.show()
+	end
+
+	#
+	#	hide the tree
+	#
+	def hide()
+		@treenode.hide()
+	end
+
+	#
+	#	clear the tree
+	#
+	def clear()
+		@tree_store.clear()
+	end
+
+	#
+	#	del a row in the tree when a node is removed
+	#
+	def del_row(node)
+		iter=search_node(node)
+		if iter != nil	
+			@tree_store.remove(@tree_store.get_iter(iter))
+			
+			#reduce count value
+			if node
+				#search for parent iter
+			
+				endloop=false
+				begin
+					if node.map == ROOTMAPADDR
+						iter=0
+						endloop=true
+					else
+						iter=search_node($network[node.map])
+					end
+					#adjust network map nb hosts
+					str_tmp=@tree_store.get_value(@tree_store.get_iter("#{iter}"),1)
+					if str_tmp
+						str_tmp.scan(@re) do |network_ip, nb_host|
+							nb_host=1 unless nb_host.to_i() > 0
+							@tree_store.set_value(@tree_store.get_iter("#{iter}"),1,"#{network_ip} (#{nb_host.to_i()-1})")
+							node=$network[network_ip]
+						end
+					end
+				end while !endloop
+			end
+		end
+	end
+
+
+	#
+	#	add a row in the tree when a new node is found
+	#
+	def add_row(node)
+		if node
+			#test if node is already present
+			iter=search_node(node)
+			if !iter
+				#find where to insert it
+				if node.map == ROOTMAPADDR
+					iter_map=0
+					endloop=true
+				else
+					iter_map=search_node($network[node.map])
+				end
+
+				root_iter = @tree_store.append(@tree_store.get_iter("#{iter_map}"))
+  	   			hexa_rgb=$status_color[$status_value.index(node.status)]
+				root_iter[0] = hexa_to_rgb_color(hexa_rgb)
+				if node.class == Host
+					root_iter[1] = node.ip
+				else
+					root_iter[1] = "#{node.ip} (0)"
+				end
+				if node && node.class == Host
+					#search for network map
+
+					endloop=false
+					begin
+						if node.map == ROOTMAPADDR
+							iter_map=0
+							endloop=true
+						else
+							iter_map=search_node($network[node.map])
+						end
+
+						if iter_map != nil
+							#adjust network map nb hosts
+							str_tmp=@tree_store.get_value(@tree_store.get_iter("#{iter_map}"),1)
+							if str_tmp
+								str_tmp.scan(@re) do |network_ip, nb_host|
+									@tree_store.set_value(@tree_store.get_iter("#{iter_map}"),1,"#{network_ip} (#{nb_host.to_i()+1})")
+									node=$network[network_ip]
+								end
+							end
+						end
+					end while !endloop
+				end
+			end
+		end
+	end
+
+	#
+	#	return an iter if node is found
+	#
+	def search_node(node)
+		if node
+			search_sub_node(node, 0)
+		end
+
+	end
+	
+	#
+	#	recursive search function to find sub node
+	#
+	def search_sub_node(node, iter)
+		iter_val=iter		
+		iter_pos=0
+		while @tree_store.get_iter("#{iter_val}:#{iter_pos}") != nil do
+			value=@tree_store.get_value(@tree_store.get_iter("#{iter_val}:#{iter_pos}"),1)
+			#print "#{value}\n"
+			if value and value.index(node.ip) == 0
+				return 	"#{iter_val}:#{iter_pos}"		
+			end
+			#check if it's a network
+			if value and value.match(/\s\(\d+/)
+				result=search_sub_node(node, "#{iter_val}:#{iter_pos}")
+				if result
+					return result
+				end
+			end
+			iter_pos+=1
+		end
+		return nil
+	end
+
+	#
+	# update the row severity corresponding to the node
+	#
+	def update_row(node)
+		iter=search_node(node)
+		if iter != nil
+			@tree_store.set_value(@tree_store.get_iter("#{iter}"), 0, get_color_from_status($status[$status_value.index(node.status)]))
+		end
+	end
+
+	#
+	#	recursive reload the whole tree
+	#
+	def reload_sub(tree_root,network_node)
+	   if network_node
+		  root_iter = @tree_store.append(tree_root)
+		  hexa_rgb=$status_color[$status_value.index(network_node.status)]
+		  root_iter[0] = hexa_to_rgb_color(hexa_rgb)
+
+  		  cpt=0
+		  
+		  network_node.get_node().sort{|ip1,ip2|
+			ip1_tmp=fill_ip_str(ip1[0])
+			ip2_tmp=fill_ip_str(ip2[0])
+			ip1_tmp = "0.0.0.0" if ip1_tmp.nil? or ip1_tmp.empty?
+			ip2_tmp = "0.0.0.0" if ip2_tmp.nil? or ip2_tmp.empty?
+			ip1_tmp<=>ip2_tmp
+		  }.each {|ipt, node|
+			 if node.class != Network
+				 cpt+=1
+				 child_iter = @tree_store.append(root_iter)
+  	   			 hexa_rgb=$status_color[$status_value.index(node.status)]
+				 child_iter[0] = hexa_to_rgb_color(hexa_rgb)
+				 child_iter[1] = node.ip
+			 else
+				#network is also considered as a node
+				cpt+=1+reload_sub(root_iter, node)
+			 end
+ 	    		 root_iter[1] = "#{network_node.ip} (#{cpt})"
+		  }
+		  #special case when there is no node in network
+		  if cpt == 0
+	    		root_iter[1] = "#{network_node.ip} (#{cpt})"
+		  end
+		  return cpt
+          end
+	  return 0
+	      
+	end
+
+	#
+	#	reload the whole tree, used when no updates was done for a long time aka tree was hidden
+	#
+	def reload()
+	  clear()
+	  node=$network[ROOTMAPADDR]
+	  reload_sub(nil,node)
+	end
+
+	#
+	#	main
+	#
+	def initialize()
+
+	  @re=Regexp.new('^([\d.]+) \((\d+)\)$')
+	  @column_title_expand="Expand all networks"
+	  @column_title_collapse="Collapse all networks"
+	  @node_selected=nil
+
+	  @treenode = Gtk::Window.new
+	  @treenode.set_size_request(200,200)
+	  @treenode.set_title "Tree node"
+	  @treenode.set_icon("#{PIXMAP_PATH}/logo_icon.png")
+	  @tree_store=Gtk::TreeStore.new(Gdk::Color,String)
+
+	  vbox=Gtk::VBox.new
+	  vbox.show
+	  @treenode.add vbox
+
+	  scrolled_window = Gtk::ScrolledWindow.new(nil, nil)
+	  scrolled_window.border_width=2
+	  scrolled_window.set_policy(Gtk::POLICY_NEVER, Gtk::POLICY_AUTOMATIC)  
+
+	  vbox.pack_start(scrolled_window, true, true, 0)
+	  scrolled_window.show
+
+	  tv = Gtk::TreeView.new(@tree_store)
+	  renderer = Gtk::CellRendererText.new
+	  column = Gtk::TreeViewColumn.new(@column_title_expand, renderer, {
+                                 		 :background_gdk => 0,
+                                 		 :text => 1
+                                 	  })
+		tv.set_headers_clickable(true)
+		tv.set_headers_visible(true)
+	   column.set_clickable(true)
+		column.set_alignment(0.5)
+
+#collapse/expand button click events
+	   column.signal_connect("clicked") {
+			if column.title == @column_title_expand
+				column.title=@column_title_collapse
+				tv.expand_all
+			else
+				column.title=@column_title_expand
+				tv.collapse_all
+			end			
+		}
+	  tv.append_column(column)
+	  tv.show
+	  scrolled_window.add(tv)
+
+#select a node when a row is selected
+	tv.signal_connect("row-activated") {|trv,path|
+	#1 is the second column
+	value=trv.model.get_value(trv.model.get_iter(path),1)
+	if value and value.match(/\s\(\d+/)
+		#if it's a network
+		network_value=value.gsub(/\s\(.+$/,'')
+		if $network[network_value]
+			set_map(network_value)
+		end
+	else
+		if $host[value]
+			#change the map if it's needed
+			set_map($host[value].map)	
+			if @node_selected != nil
+				@node_selected.node_view.unselect()
+			end
+			@node_selected=$host[value]
+			$host[value].node_view.select()
+		end
+	end
+	}
+
+#delete node when delete key is pressed
+	tv.signal_connect("key_press_event") {|trv,e|
+		if e.keyval==Gdk::Keyval::GDK_Delete
+			path,focus_col=trv.cursor()
+			iter=trv.model.get_iter(path)
+
+			value=trv.model.get_value(iter,1)
+			#check if value of the cell contains a () so it's a network
+			if value and !value.match(/\s\(\d+/)
+				verify_delnodewin($host[value])
+			else
+                                if value
+				   network_val=value.gsub(/\s\(.+$/,'')
+				   if (get_map() != network_val) && !$network[network_val].contain_network(get_map())
+					   verify_delnodewin($network[network_val])
+				   end
+				end
+			end
+		end
+	}
+	  @treenode.signal_connect("delete_event") do
+			@treenode.hide
+	  end
+	  @treenode.signal_connect("key_press_event") {|w,e|
+		if e.keyval==Gdk::Keyval::GDK_Escape
+      	@treenode.hide()
+   	end
+	  }
+	end
+end