module Msf
  module Ui
    module Gtk2

      ##
      # This class describe all search stuff into the module treeview
      ##
      class ModuleSearch
        include Msf::Ui::Gtk2::MyControls

        RUNNING, CLEAR = *(0..2).to_a

        @@state = nil

        #
        # Initialize all stuff to perform a search
        #
        def initialize(search_entry, search_button, search_cancel_button)
          @search_entry = search_entry
          @search_button = search_button
          @cancel_button = search_cancel_button

          # Active completion
          completion()

          # Init state
          @@state = CLEAR

          # Signals
          @search_entry.signal_connect('activate') do
            @search_button.activate
          end

          @search_button.signal_connect('clicked') do
            if @search_entry.text.match(/[a-zA-Z0-9\(\)]/)
              search(@search_entry.text)
            end
          end

          @cancel_button.signal_connect('clicked') do
            cancel()
          end
        end


        #
        # Perform a search throught the module treeview,
        # and return the array result to MyModuleTree::remove
        #
        def search(text)
          # If current state set to RUUNING, call cancel method
          cancel if @@state == RUNNING

          # Set current state to RUNNING
          @@state = RUNNING

          # Perform the search
          found = []
          filter = Regexp.new(text, Regexp::IGNORECASE)
          $gtk2driver.module_model.each do |model, path, iter|
            if (not iter[1][filter])
              found.push(iter)
            end
          end

          # Colorize the Gtk::Entry
          state(RUNNING)

          # pass the found array to the MyModuleTree and remove all not matched iter
          # and finish by expanding the treeview
          $gtk2driver.module_tree.remove(found)
          $gtk2driver.module_tree.expand
        end


        #
        # Clean the Gtk::Entry and refresh the modules treeview
        #
        def cancel
          # clear the Gtk::Entry
          @search_entry.set_text("")

          # Colorize the Gtk::Entry
          state(CLEAR)

          # Refresh the modules treeview
          $gtk2driver.module_tree.refresh

          # Register the current state
          @@state = CLEAR
        end


        #
        # Colorize the Gtk::Entry by state parameter
        #
        def state(state)
          if (state == RUNNING)
            @search_entry.modify_base(Gtk::STATE_NORMAL, Gdk::Color.parse('gray'))
          elsif (state == CLEAR)
            @search_entry.modify_base(Gtk::STATE_NORMAL, Gdk::Color.parse('white'))
          end
        end

        #
        # Display completion for @search_entry
        #
        def completion
          # set completion
          completion = Gtk::EntryCompletion.new
          @search_entry.completion = completion

          # Describe the model completion
          model = Gtk::ListStore.new(String)
          $gtk2driver.module_completion.each do |v|
            iter = model.append
            iter[0] = v
          end

          # Attach the model completion to the completion object
          completion.model = model
          completion.text_column = 0
        end
      end

    end
  end
end