# vim: set fileencoding=utf-8 : # Copyright (C) 2008 Insecure.Com LLC. # # Author: João Paulo de Souza Medeiros # # This program 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 2 of the License, or # (at your option) any later version. # # 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, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA import os import gtk import gobject from radialnet.bestwidgets.buttons import * from radialnet.gui.SaveDialog import SaveDialog from radialnet.gui.Dialogs import AboutDialog from radialnet.gui.HostsViewer import HostsViewer from zenmapGUI.higwidgets.higdialogs import HIGAlertDialog SHOW = True HIDE = False REFRESH_RATE = 500 class ToolsMenu(gtk.Menu): """ """ def __init__(self, radialnet): """ """ gtk.Menu.__init__(self) self.radialnet = radialnet self.__create_items() def __create_items(self): """ """ self.__hosts = gtk.ImageMenuItem(_('Hosts viewer')) self.__hosts.connect("activate", self.__hosts_viewer_callback) self.__hosts_image = gtk.Image() self.__hosts_image.set_from_stock(gtk.STOCK_INDEX, gtk.ICON_SIZE_MENU) self.__hosts.set_image(self.__hosts_image) self.append(self.__hosts) self.__hosts.show_all() def __hosts_viewer_callback(self, widget): """ """ window = HostsViewer(self.radialnet.get_scanned_nodes()) window.show_all() window.set_keep_above(True) def enable_dependents(self): """ """ self.__hosts.set_sensitive(True) def disable_dependents(self): """ """ self.__hosts.set_sensitive(False) class Toolbar(gtk.HBox): """ """ def __init__(self, radialnet, window, control, fisheye): """ """ gtk.HBox.__init__(self) #self.set_style(gtk.TOOLBAR_BOTH_HORIZ) #self.set_tooltips(True) self.radialnet = radialnet self.__window = window self.__control_widget = control self.__fisheye_widget = fisheye self.__control_widget.show_all() self.__control_widget.set_no_show_all(True) self.__control_widget.hide() self.__fisheye_widget.show_all() self.__fisheye_widget.set_no_show_all(True) self.__fisheye_widget.hide() self.__save_chooser = None self.__create_widgets() def __create_widgets(self): """ """ self.__tooltips = gtk.Tooltips() #self.__tools_menu = ToolsMenu(self.radialnet) #self.__tools_button = gtk.MenuToolButton(gtk.STOCK_PREFERENCES) #self.__tools_button.set_label(_('Tools')) #self.__tools_button.set_is_important(True) #self.__tools_button.set_menu(self.__tools_menu) #self.__tools_button.connect('clicked', self.__tools_callback) self.__save_button = BWStockButton(gtk.STOCK_SAVE, _("Save Graphic")) self.__save_button.connect("clicked", self.__save_image_callback) self.__hosts_button = BWStockButton(gtk.STOCK_INDEX, _("Hosts Viewer")) self.__hosts_button.connect("clicked", self.__hosts_viewer_callback) self.__control = BWToggleStockButton(gtk.STOCK_PROPERTIES, _("Controls")) self.__control.connect('clicked', self.__control_callback) self.__control.set_active(False) self.__fisheye = BWToggleStockButton(gtk.STOCK_ZOOM_FIT, _("Fisheye")) self.__fisheye.connect('clicked', self.__fisheye_callback) self.__fisheye.set_active(False) #self.__fullscreen = gtk.ToggleToolButton(gtk.STOCK_FULLSCREEN) #self.__fullscreen.set_label(_('Fullscreen')) #self.__fullscreen.set_is_important(True) #self.__fullscreen.connect('clicked', self.__fullscreen_callback) #self.__fullscreen.set_tooltip(self.__tooltips, _('Toggle fullscreen')) #self.__about = gtk.ToolButton(gtk.STOCK_ABOUT) #self.__about.set_label(_('About')) #self.__about.set_is_important(True) #self.__about.connect('clicked', self.__about_callback) #self.__about.set_tooltip(self.__tooltips, _('About RadialNet')) self.__separator = gtk.SeparatorToolItem() self.__expander = gtk.SeparatorToolItem() self.__expander.set_expand(True) self.__expander.set_draw(False) #self.insert(self.__open, 0) #self.insert(self.__separator, 1) #self.insert(self.__tools_button, 2) #self.insert(self.__expander, 3) #self.insert(self.__control, 4) #self.insert(self.__fisheye, 5) #self.insert(self.__fullscreen, 6) #self.insert(self.__about, 7) #self.pack_start(self.__tools_button, False) self.pack_start(self.__hosts_button, False) self.pack_start(self.__fisheye, False) self.pack_start(self.__control, False) self.pack_end(self.__save_button, False) def disable_controls(self): """ """ self.__control.set_sensitive(False) self.__fisheye.set_sensitive(False) self.__hosts_button.set_sensitive(False) #self.__tools_menu.disable_dependents() def enable_controls(self): """ """ self.__control.set_sensitive(True) self.__fisheye.set_sensitive(True) self.__hosts_button.set_sensitive(True) #self.__tools_menu.enable_dependents() def __tools_callback(self, widget): """ """ self.__tools_menu.popup(None, None, None, 1, 0) def __hosts_viewer_callback(self, widget): """ """ window = HostsViewer(self.radialnet.get_scanned_nodes()) window.show_all() window.set_keep_above(True) def __save_image_callback(self, widget): """ """ if self.__save_chooser is None: self.__save_chooser = SaveDialog() response = self.__save_chooser.run() if response == gtk.RESPONSE_OK: filename = self.__save_chooser.get_filename() filetype = self.__save_chooser.get_filetype() try: self.radialnet.save_drawing_to_file(filename, filetype) except Exception, e: alert = HIGAlertDialog(parent=self.__save_chooser, type=gtk.MESSAGE_ERROR, message_format=_("Error saving snapshot"), secondary_text=unicode(e)) alert.run() alert.destroy() self.__save_chooser.hide() def __control_callback(self, widget=None): """ """ if self.__control.get_active(): self.__control_widget.show() else: self.__control_widget.hide() def __fisheye_callback(self, widget=None): """ """ if not self.radialnet.is_in_animation(): if self.__fisheye.get_active(): self.__fisheye_widget.active_fisheye() self.__fisheye_widget.show() else: self.__fisheye_widget.deactive_fisheye() self.__fisheye_widget.hide() def __about_callback(self, widget): """ """ self.__about_dialog = AboutDialog() self.__about_dialog.show_all() def __fullscreen_callback(self, widget=None): """ """ if self.__fullscreen.get_active(): self.__window.fullscreen() else: self.__window.unfullscreen()