root/branch/k0p/umitGUI/ProfileManager.py @ 894

#!/usr/bin/env python
# Copyright (C) 2005 Insecure.Com LLC.
#
# Author: Luis A. Bastiao Silva <luis.kop@gmail.com>
#
# 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 gtk

from higwidgets.higwindows import HIGWindow
from higwidgets.higboxes import HIGVBox, HIGHBox
from higwidgets.higscrollers import HIGScrolledWindow
from higwidgets.higbuttons import HIGButton
from higwidgets.higlabels import HIGSectionLabel, HIGEntryLabel, HIGDialogLabel
from higwidgets.higentries import HIGTextEntry
from higwidgets.higframe import HIGFrame
from higwidgets.higdialogs import HIGDialog

from umitGUI.Wizard import Wizard

# Testing at devel 
from os.path import split, join

from umitCore.Paths import Path
Path.set_umit_conf(join(split(__file__)[0], 'config', 'umit.conf'))
#END DEV TEST

from ProfileEditor import ProfileEditor
from umitCore.UmitConf import CommandProfile
from umitCore.I18N import _

import gobject

profile_editor = Path.profile_editor

class ProfileManager(HIGWindow):
    """
    Create a Profile Manager 
    """
    def __init__(self):
        HIGWindow.__init__(self, type=gtk.WINDOW_TOPLEVEL)
        self.set_title('Profile Manager')
        self.set_position(gtk.WIN_POS_CENTER)
        self.__create_widgets()
        self.add(self.vbox_main)
        self.__fill_widgets()
        self.__pack_widgets()

    
    def __create_widgets(self):
        
        self.vbox_main = HIGVBox()
        
        self.main_frame = HIGFrame("Profiles")
        self.main_frame.set_shadow_type(gtk.SHADOW_ETCHED_OUT)
        
        
        self.vbox = HIGVBox()
        self.profiles_sw = HIGScrolledWindow()
        #TreeView
        self.hbox_aux_tv = HIGHBox()
        self.model = gtk.TreeStore(gobject.TYPE_STRING)
        self.profiles_tv = gtk.TreeView(self.model)
        renderer = gtk.CellRendererText()
        column = gtk.TreeViewColumn("Name", renderer, text=0)
        self.profiles_tv.append_column(column)
        

        #Info 
        self.hbox_info = HIGHBox()
        self.command_label = HIGEntryLabel('Command: ')
        self.command_entry = HIGTextEntry()
        self.command_entry.set_editable(False)
        
        #Buttons
        self.hbox_buttons = HIGHBox()
        self.edit_button = HIGButton(stock='gtk-edit')
        self.edit_button.connect("clicked", self.open_peditor)
        self.new_button = HIGButton(stock='gtk-new')
        self.new_button.connect("clicked", self.new_wiz)
        self.copy_button = HIGButton(stock='gtk-copy')
        self.copy_button.connect("clicked", self.copy_profiles)
        self.delete_button = HIGButton(stock=gtk.STOCK_DELETE)
        self.delete_button.connect('clicked', self.delete_profile)      
        
        #Apply Buttons
        self.hbox_apply = HIGHBox()
        self.cancel_button = HIGButton(stock='gtk-cancel')
        self.cancel_button.connect("clicked", self.quit)
        self.ok_button = HIGButton(stock='gtk-ok')

    def __fill_widgets(self):

                
        self.profiles = CommandProfile()
        profiles = self.profiles.sections()
        profiles.sort()
        
       
        for command in profiles:
            myiter = self.model.insert_after(None, None)
            self.model.set_value(myiter, 0, command)
            
        #selection = self.profiles_tv.get_selection()
        #selection.connect("changed", self.change_nmap_command)
        self.profiles_tv.connect("cursor-changed", self.change_nmap_command)

        

    def __pack_widgets(self):
        """
        Pack all widgets of windows 
        """
        self.vbox_main.pack_start(self.main_frame)
        self.main_frame.add(self.vbox)
        
        self.vbox.pack_start(self.profiles_sw, True, True, 0)

        self.hbox_info.pack_start(self.command_label, False,False,0)
        self.hbox_info.pack_start(self.command_entry, True, True, 0)
        self.vbox.pack_start(self.hbox_info, True, True,0)
        
        self.vbox.pack_start(self.hbox_buttons, False, True, 0)
        
        self.hbox_buttons.pack_end(self.copy_button, False, True)
        self.hbox_buttons.pack_end(self.edit_button, False, True)
        self.hbox_buttons.pack_end(self.delete_button, False, True)
        self.hbox_buttons.pack_end(self.new_button, False, True)

        
        #self.hbox_aux_tv .pack_start(self.profiles_tv,False, False)
        self.profiles_sw.set_size_request(400,170)
        self.profiles_sw.add_with_viewport(self.profiles_tv) 

        self.hbox_apply.pack_end(self.ok_button)
        self.hbox_apply.pack_end(self.cancel_button)
        self.vbox_main.pack_start(self.hbox_apply)
        
    def get_selected_profile(self):
        """
        Returns the string with name of selected profile 
        """
        treeselection = self.profiles_tv.get_selection()
        (model,iter) = treeselection.get_selected()
        return model.get_value(iter,0)
        
        
    def change_nmap_command(self,widget_tv):
        """
        Change a nmap command at command entry
        """
        assert widget_tv != None
        
        self.command_entry.set_text(self.profiles.get_command(self.get_selected_profile()))     
    
    def new_wiz(self,widget):
        w = Wizard()
        w.set_notebook(None)
        
        w.show_all()

    def open_peditor(self, widget):
        """
        Open Profile Editor with a Selected or Non-Selected(New) Item
        """
        assert widget != None
        
        #widget = HIGButton()
        
        if widget.get_label() == "gtk-edit":
            # Edit profile selected    
            
            pe = ProfileEditor(self.get_selected_profile())
            pe.show_all()
        else:
            # New Profile
            pe = ProfileEditor()
            pe.show_all()
    
    def copy_profiles(self, widget):
        """
        Copy selected Profile
        """
        d = ProfileName(_("Insert a profile name"))

    def delete_profile(self, widget=None):
        """
        delete profile only at treeview 
        """
        #self.profiles.remove_profile()
        #Update treeview
        treeselection = self.profiles_tv.get_selection()
        (model,iter) = treeselection.get_selected()
        model.remove(iter)      
        
        
    
    def quit(self, widget):
        self.destroy()
        
class ProfileName:
    def __init__(self, text):
        d = HIGDialog(title=_('Profile\'s Name'),
                      buttons=(gtk.STOCK_OK, gtk.RESPONSE_ACCEPT))
        dialog_label = HIGDialogLabel(text)
        dialog_label.show()
        d.vbox.pack_start(dialog_label)

        entry_text = HIGTextEntry()
        entry_text.show()
        d.vbox.pack_start(entry_text)
        
        d.run()
        
        d.destroy()
        
class ProfileChosse(HIGHBox):
    def __init__(self):
        HIGHBox.__init__(self)        
        self._create_profile()
        
    #def _create_profile(self):
    def __init__(self):
        HIGHBox.__init__(self)

        self._create_profile()
        
        self.change_button = gtk.Button(_("Change"))
        
        self._pack_noexpand_nofill(self.profile_label)
        self._pack_expand_fill(self.profile_entry)
        
        self._pack_noexpand_nofill(self.change_button)

        self.profile_entry.set_focus_child(self.profile_entry.child)

        # Events
        self.profile_entry.child.connect('activate',
                        lambda x: self.change_button.clicked())

    def _create_profile(self):
        self.profile_label = HIGEntryLabel(_('Profile:'))
        self.profile_entry = ProfileCombo()
        self.update()
    def update(self):
        self.update_profile()
    def update_profile(self):
        self.profile_entry.update()
        
        
        
        
if __name__=="__main__":
    pm = ProfileManager()
    pm.show_all()
    gtk.main()