source: community/mandrou/script_autoadd_screens.rb @ 226

#!/usr/bin/ruby

# Generate all screen for all host of zabbix DB ( with all graph owned by the hosts )
# Written by Py MIMAULT <iMandrou@gmail.com>
# personal site : http://mandrou.fr
# company site oooo OVEA : www.oveaconnect.me

require 'rexml/document'
require "mysql"

include REXML


#Catch cli argument
$mysql_host = ARGV[0]
$mysql_user = ARGV[1]
$mysql_passwd = ARGV[2]

#Global variables
$doc
$screenitems
$screen
$screens
$nbr
$nbr_rows

#Function
def debut_xml()
        $doc = Document.new
        $screens = Element.new("screens")
        $doc.elements << $screens
end 
  
def boucle_xml_screen(hn) 
        $screen = Element.new("screen")    
        name = Element.new("name")
                name.text = "#{hn}"
        hsize = Element.new("hsize")
                hsize.text = "1"
        vsize = Element.new("vsize")
                vsize.text = $nbr_rows
  
        $screen.elements << name
        $screen.elements << hsize
        $screen.elements << vsize
        
        $screens.elements << $screen    
        
        $screenitems = Element.new("screenitems")
        $screen.elements << $screenitems
end 
    
def boucle_xml_item(hn,y,gr_name)
        screenitem = Element.new("screenitem")
        screenitem.add_element("resourcetype")
                screenitem.elements["resourcetype"].text = "0"
        screenitem.add_element("width")
                screenitem.elements["width"].text = "500"
        screenitem.add_element("height")
                screenitem.elements["height"].text = "100"
        screenitem.add_element("x")
                screenitem.elements["x"].text = "0"
        screenitem.add_element("y")
                screenitem.elements["y"].text= "#{y}"
        screenitem.add_element("colspan")
                screenitem.elements["colspan"].text = "0"
        screenitem.add_element("rowspan")
                screenitem.elements["rowspan"].text = "0"
        screenitem.add_element("elements")
                screenitem.elements["elements"].text = "0"
        screenitem.add_element("valign")
                screenitem.elements["valign"].text = "0"
        screenitem.add_element("halign")
                screenitem.elements["halign"].text = "0"
        screenitem.add_element("style")
                screenitem.elements["style"].text = "0"
        screenitem.add_element("dynamic")
                screenitem.elements["dynamic"].text = "0"
        
        resourceid = Element.new("resourceid")
        resourceid.add_element("host")
                resourceid.elements["host"].text ="#{hn}"
        resourceid.add_element("name")
                resourceid.elements["name"].text = "#{gr_name}"
        
        screenitem.elements << resourceid
        $screenitems.elements << screenitem
end



def get_graph_for_host(hostid)
        graphs = $my.query("SELECT DISTINCT g.name FROM hosts h, graphs g, graphs_items gi, items i WHERE h.hostid = i.hostid AND gi.graphid = g.graphid AND gi.itemid = i.itemid AND h.status = 0 AND h.hostid = #{hostid} order by h.host ASC;")
        
        $graphs_row = Array.new
        while row = graphs.fetch_row do
                $graphs_row << row[0]
        end
        $nbr_rows = graphs.num_rows
        return $graphs_row
end

#-- MAIN --
if ARGV.length == 3 # if all three arguments are here.
        $my = Mysql::new($mysql_host, $mysql_user,$mysql_passwd, "zabbix")

        hosts = $my.query("SELECT DISTINCT h.hostid,h.host FROM hosts h, graphs g, graphs_items gi, items i WHERE h.hostid = i.hostid AND gi.graphid = g.graphid AND gi.itemid = i.itemid AND h.status = 0 AND h.hostid = i.hostid order by h.host ASC;")

        

#XML Generation
        debut_xml()
        while row = hosts.fetch_row do
                hnid = row[0]
                hn = row[1]
                boucle_xml_screen(hn)
                y=0
                get_graph_for_host(hnid) 
                for gr in $graphs_row
                        boucle_xml_item(hn,y,gr)
                        y=y+1   
                end
        end 

#XML File creation
        File::new("screens_to_import.xml", "w+")
        xml_file = File.open("screens_to_import.xml", "w")
        $doc.write(xml_file,-1)
else
        #error message
        puts "not enougth argument : \n host - user - passwd"
end