watobo 0.9.19 → 0.9.20

data/lib/watobo/gui/login_wizzard.rb

@@ -1,24 +1,12 @@
-# .
+#.
 # login_wizzard.rb
-# 
-# Copyright 2013 by siberas, http://www.siberas.de
-# 
-# This file is part of WATOBO (Web Application Tool Box)
-#        http://watobo.sourceforge.com
-# 
-# WATOBO 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 version 2 of the License.
-# 
-# WATOBO 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 WATOBO; if not, write to the Free Software
-# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
-# .
+#.
+# Copyright 2014 by siberas, http://www.siberas.de
+# This file is part of WATOBO (Web Application Tool Box) http://watobo.sourceforge.com
+# WATOBO 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 version 2 of the License.
+# WATOBO 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 WATOBO; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+
 require 'fox16'
 
 include Fox

data/lib/watobo/gui/main_window.rb

@@ -1,24 +1,12 @@
-# .
+#.
 # main_window.rb
-# 
-# Copyright 2013 by siberas, http://www.siberas.de
-# 
-# This file is part of WATOBO (Web Application Tool Box)
-#        http://watobo.sourceforge.com
-# 
-# WATOBO 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 version 2 of the License.
-# 
-# WATOBO 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 WATOBO; if not, write to the Free Software
-# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
-# .
+#.
+# Copyright 2014 by siberas, http://www.siberas.de
+# This file is part of WATOBO (Web Application Tool Box) http://watobo.sourceforge.com
+# WATOBO 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 version 2 of the License.
+# WATOBO 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 WATOBO; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+
 # @private 
 module Watobo#:nodoc: all
   module Gui
@@ -353,6 +341,7 @@ module Watobo#:nodoc: all
 
         if vuln.details[:proof_pattern] then
         pattern = vuln.details[:proof_pattern].strip
+        
         @response_viewer.highlight(pattern)
         end
         @switcher.current = 0
@@ -666,12 +655,13 @@ module Watobo#:nodoc: all
         @statusBar.portNumber = Watobo::Conf::Interceptor.port.to_s
         @statusBar.forwardingProxy = "-"
       #  puts Watobo::Conf::ForwardingProxy.default_proxy
-        unless Watobo::Conf::ForwardingProxy.default_proxy.empty?
-          default_proxy = Watobo::Conf::ForwardingProxy.default_proxy
-          ps = Watobo::Conf::ForwardingProxy.to_h
-          proxy = ps[default_proxy]
-          @statusBar.forwardingProxy = "#{proxy[:name]} (#{proxy[:host]}:#{proxy[:port]})"
-        end
+        
+        #unless Watobo::Conf::ForwardingProxy.default_proxy.empty?
+        #  default_proxy = Watobo::Conf::ForwardingProxy.default_proxy
+        #  ps = Watobo::Conf::ForwardingProxy.to_h
+        #  proxy = ps[default_proxy]
+        #  @statusBar.forwardingProxy = "#{proxy[:name]} (#{proxy[:host]}:#{proxy[:port]})"
+        #end
         
         @statusBar.update_proxy_mode
       end
@@ -1216,6 +1206,11 @@ module Watobo#:nodoc: all
         FXMenuTitle.new(menu_bar, "File" , :popupMenu => file_menu_pane)
         @file_new_menu = FXMenuCommand.new(file_menu_pane, "New/Open" )
         @file_new_menu.connect(SEL_COMMAND, method(:onNewProject))
+        
+         export_menu = FXMenuCommand.new(file_menu_pane, "Export" )
+        #FXMenuCommand.new(file_menu_pane, "Exit", nil, getApp(), FXApp::ID_QUIT)
+        export_menu.connect(SEL_COMMAND, method(:onExport))
+
 
         exit_menu = FXMenuCommand.new(file_menu_pane, "Exit" )
         #FXMenuCommand.new(file_menu_pane, "Exit", nil, getApp(), FXApp::ID_QUIT)
@@ -1413,10 +1408,9 @@ module Watobo#:nodoc: all
         # R E Q U E S T I N F O
         requestInfo = FXVerticalFrame.new(@switcher, :opts => LAYOUT_FILL_X|LAYOUT_FILL_X|LAYOUT_FILL_Y, :padding => 0)
         request_splitter = FXSplitter.new(requestInfo, :opts => LAYOUT_SIDE_TOP|SPLITTER_HORIZONTAL|LAYOUT_FILL_Y|LAYOUT_FILL_X|SPLITTER_TRACKING|SPLITTER_REVERSED)
-request_splitter.connect(SEL_COMMAND){
-  puts "Request Splitter Resized!"
-  
-}
+#request_splitter.connect(SEL_COMMAND){
+  #puts "Request Splitter Resized!"
+  #}
 
         # C H A T  T A B L E  C O N T R O L L E R
        # @conversation_table_ctrl = ConversationTableCtrl.new(request_splitter,  :opts => LAYOUT_FILL_X|LAYOUT_FILL_Y|FRAME_SUNKEN)
@@ -1845,6 +1839,8 @@ request_splitter.connect(SEL_COMMAND){
         @sites_tree.subscribe(:show_chat){ |chat|
           showChat(chat)
         }
+        
+         @sites_tree.subscribe(:vuln_click){ |v| showVulnerability(v) }
       end
 
       def subscribeFindingsTree()
@@ -1924,6 +1920,14 @@ request_splitter.connect(SEL_COMMAND){
         end
         return settings
       end
+      
+      def onExport(sender,sel, item)
+          ccdlg = Watobo::Gui::ExportDialog.new(self)
+          if ccdlg.execute != 0
+            
+          end
+      
+      end
 
       def onExit(sender, sel, item)
         response = FXMessageBox.question(self, MBOX_YES_NO, "Finished?", "Are you sure?")
@@ -1969,9 +1973,6 @@ request_splitter.connect(SEL_COMMAND){
           if proxy_dialog.execute != 0 then
           proxy_prefs = proxy_dialog.getProxyPrefs
           Watobo::Conf::ForwardingProxy.set proxy_prefs
-          #@project.setProxyOptions(YAML.load(YAML.dump(proxy_prefs)))
-          puts "* current proxy:\n> #{@project.getCurrentProxy()}" if $DEBUG
-          
         # Watobo::Gui.save_settings()
          #Watobo::Conf::ForwardingProxy.save
          

data/lib/watobo/gui/manual_request_editor.rb

@@ -1,24 +1,12 @@
-# .
+#.
 # manual_request_editor.rb
-# 
-# Copyright 2013 by siberas, http://www.siberas.de
-# 
-# This file is part of WATOBO (Web Application Tool Box)
-#        http://watobo.sourceforge.com
-# 
-# WATOBO 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 version 2 of the License.
-# 
-# WATOBO 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 WATOBO; if not, write to the Free Software
-# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
-# .
+#.
+# Copyright 2014 by siberas, http://www.siberas.de
+# This file is part of WATOBO (Web Application Tool Box) http://watobo.sourceforge.com
+# WATOBO 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 version 2 of the License.
+# WATOBO 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 WATOBO; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+
 # @private 
 module Watobo#:nodoc: all
    module Gui      
@@ -60,10 +48,11 @@ module Watobo#:nodoc: all
            request = Watobo::Request.new(new_request)
            begin
               test_req, test_resp = self.doRequest(request, prefs)
-              rq = Watobo::Request.new test_req
-              rs = Watobo::Response.new test_resp
-              rs.unchunk
-              return rq, rs
+              #rq = Watobo::Request.new test_req
+             # rs = Watobo::Response.new test_resp
+              #rs.unchunk
+              #rs.unzip
+              return test_req, test_resp
            rescue => bang
               puts bang
               puts bang.backtrace if $DEBUG
@@ -211,10 +200,10 @@ module Watobo#:nodoc: all
                if quick_scan_options[:update_csrf_tokens] == true
                  unless csrf_requests.empty?
                   csrf_requests = Watobo::OTTCache.requests(req)
-                 else
-                  puts "* No CSRF requests defined for request:"
-                  puts request
-                  puts "---"
+                # else
+                #  puts "* No CSRF requests defined for request:"
+                #  puts req
+                #  puts "---"
                  end 
                end
 
@@ -421,13 +410,13 @@ module Watobo#:nodoc: all
                @csrf_settings_btn = FXButton.new(csrf_frame, "Settings", nil, nil, 0, FRAME_RAISED|FRAME_THICK)
                @csrf_settings_btn.connect(SEL_COMMAND, method(:openCSRFTokenDialog))
 
-               @updateCSRF.connect(SEL_COMMAND) do |sender, sel, item|
-                  if @updateCSRF.checked? then
-                     @csrf_settings_btn.enable
-                  else
-                     @csrf_settings_btn.disable
-                  end
-               end
+               #@updateCSRF.connect(SEL_COMMAND) do |sender, sel, item|
+               #   if @updateCSRF.checked? then
+               #      @csrf_settings_btn.enable
+               #   else
+               #      @csrf_settings_btn.disable
+               #   end
+               #end
 
                ##################################################
 
@@ -591,6 +580,7 @@ module Watobo#:nodoc: all
                 logger("send request")
                 last_request, last_response = @request_sender.sendRequest(nr, p )
                 logger("got answer")
+                
                 @chat_queue.push [ last_request, last_response ]
               rescue => bang
                 puts bang

data/lib/watobo/gui/master_pw_dialog.rb

@@ -1,24 +1,12 @@
-# .
+#.
 # master_pw_dialog.rb
-# 
-# Copyright 2013 by siberas, http://www.siberas.de
-# 
-# This file is part of WATOBO (Web Application Tool Box)
-#        http://watobo.sourceforge.com
-# 
-# WATOBO 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 version 2 of the License.
-# 
-# WATOBO 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 WATOBO; if not, write to the Free Software
-# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
-# .
+#.
+# Copyright 2014 by siberas, http://www.siberas.de
+# This file is part of WATOBO (Web Application Tool Box) http://watobo.sourceforge.com
+# WATOBO 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 version 2 of the License.
+# WATOBO 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 WATOBO; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+
 # @private 
 module Watobo#:nodoc: all
   module Gui

data/lib/watobo/gui/mixins/gui_settings.rb

@@ -1,52 +1,40 @@
-# .
+#.
 # gui_settings.rb
-# 
-# Copyright 2013 by siberas, http://www.siberas.de
-# 
-# This file is part of WATOBO (Web Application Tool Box)
-#        http://watobo.sourceforge.com
-# 
-# WATOBO 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 version 2 of the License.
-# 
-# WATOBO 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 WATOBO; if not, write to the Free Software
-# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
-# .
-# @private 
-module Watobo#:nodoc: all
-  module Gui
-    module Settings
-       def self.save_gui_settings(settings)
-        wd = Watobo.working_directory
-
-        dir_name = Watobo::Utils.snakecase self.class.to_s.gsub(/.*::/,'')
-        path = File.join(wd, "conf", "gui")
-        Dir.mkdir path unless File.exist? path
-        conf_dir = File.join(path, dir_name)
-        Dir.mkdir conf_dir unless File.exist? conf_dir
-        file = File.join(conf_dir, dir_name + "_settings.yml")
-        
-        Watobo::Utils.save_settings(file, config)
-      end
+#.
+# Copyright 2014 by siberas, http://www.siberas.de
+# This file is part of WATOBO (Web Application Tool Box) http://watobo.sourceforge.com
+# WATOBO 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 version 2 of the License.
+# WATOBO 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 WATOBO; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 
-      def load_gui_settings()
-        wd = Watobo.working_directory
-        dir_name = Watobo::Utils.snakecase self.class.to_s.gsub(/.*::/,'')
-        path = File.join(wd, "conf", "gui")
-        Dir.mkdir path unless File.exist? path
-        conf_dir = File.join(path, dir_name)
-        Dir.mkdir conf_dir unless File.exist? conf_dir
-        file = File.join(conf_dir, dir_name + "_settings.yml")
-        config = Watobo::Utils.load_settings(file)
-        config
-      end
-    end
-  end
+# @private 
+module Watobo#:nodoc: all
+  module Gui
+    module Settings
+       def self.save_gui_settings(settings)
+        wd = Watobo.working_directory
+
+        dir_name = Watobo::Utils.snakecase self.class.to_s.gsub(/.*::/,'')
+        path = File.join(wd, "conf", "gui")
+        Dir.mkdir path unless File.exist? path
+        conf_dir = File.join(path, dir_name)
+        Dir.mkdir conf_dir unless File.exist? conf_dir
+        file = File.join(conf_dir, dir_name + "_settings.yml")
+        
+        Watobo::Utils.save_settings(file, config)
+      end
+
+      def load_gui_settings()
+        wd = Watobo.working_directory
+        dir_name = Watobo::Utils.snakecase self.class.to_s.gsub(/.*::/,'')
+        path = File.join(wd, "conf", "gui")
+        Dir.mkdir path unless File.exist? path
+        conf_dir = File.join(path, dir_name)
+        Dir.mkdir conf_dir unless File.exist? conf_dir
+        file = File.join(conf_dir, dir_name + "_settings.yml")
+        config = Watobo::Utils.load_settings(file)
+        config
+      end
+    end
+  end
 end

data/lib/watobo/gui/page_tree.rb

@@ -1,239 +1,227 @@
-# .
+#.
 # page_tree.rb
-# 
-# Copyright 2013 by siberas, http://www.siberas.de
-# 
-# This file is part of WATOBO (Web Application Tool Box)
-#        http://watobo.sourceforge.com
-# 
-# WATOBO 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 version 2 of the License.
-# 
-# WATOBO 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 WATOBO; if not, write to the Free Software
-# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
-# .
- #require 'qcustomize.rb'
-
+#.
+# Copyright 2014 by siberas, http://www.siberas.de
+# This file is part of WATOBO (Web Application Tool Box) http://watobo.sourceforge.com
+# WATOBO 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 version 2 of the License.
+# WATOBO 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 WATOBO; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+
+ #require 'qcustomize.rb'
+
 # @private 
-module Watobo#:nodoc: all
-  module Gui
-    class PageTree < FXTreeList
-
-      attr :page
-
-      include Watobo::Constants
-      include Watobo::Gui::Icons
-      def subscribe(event, &callback)
-        (@event_dispatcher_listeners[event] ||= []) << callback
-      end
-
-      def reload()
-        self.clearItems
-
-      # @interface.updateRequestTable(@project)
-      end
-
-      def page=(p)
-        @page = p
-        refresh_tree
-      end
-
-      def refresh_tree()
-        self.clearItems
-        unless @page.nil?
-          if page.forms.length > 0
-          page.forms.each do |f|
-            add_form_item f
-          end
-          else
-            self.appendItem(nil, "Forms ()", nil, nil)
-          end
-        end
-      # @interface.updateRequestTable(@project)
-      end
-
-      def expandFullTree(item)
-        self.expandTree(item)
-        item.each do |c|
-          expandFullTree(c) if !self.itemLeaf?(c)
-        end
-      end
-
-      def collapseFullTree(item)
-        self.collapseTree(item)
-        item.each do |c|
-          collapseFullTree(c) if !self.itemLeaf?(c)
-        end
-      end
-
-      def hidden?(chat)
-
-        #TODO: Filter
-        false
-      end
-
-      def initialize(parent, page=nil, prefs={})
-        @page = page
-        @quick_filter = Hash.new
-        opts = LAYOUT_FILL_X|LAYOUT_FILL_Y|LAYOUT_TOP|TREELIST_SHOWS_LINES|TREELIST_SHOWS_BOXES|TREELIST_ROOT_BOXES|TREELIST_EXTENDEDSELECT
-        opts = prefs[:opts] if prefs.has_key? :opts
-        super(parent, :opts => opts)
-        @event_dispatcher_listeners = Hash.new
-        @filters = {
-          :element => []
-        }
-
-        forms = self.appendItem(nil, "load a new page first!", nil, nil)
-
-        self.connect(SEL_COMMAND) do |sender, sel, item|
-          
-          if item.data.is_a? Mechanize::Form
-               puts "Form Selected"
-               notify(:form_selected, item.data)
-          elsif item.data.is_a? Mechanize::Form::Field
-            puts "Field Selected"
-            notify(:form_selected, item.parent.data)
-            notify(:field_selected, item)
-          end
-
-        end
-
-        self.connect(SEL_RIGHTBUTTONRELEASE) do |sender, sel, event|
-          exclude_site = nil
-          unless event.moved?
-            FXMenuPane.new(self) do |menu_pane|
-
-            #
-              item = sender.getItemAt(event.win_x, event.win_y)
-
-              unless item.nil?
-
-                unless self.itemLeaf?(item)
-                  FXMenuSeparator.new(menu_pane)
-                  FXMenuCommand.new(menu_pane, "expand tree" ).connect(SEL_COMMAND) {
-                    expandFullTree(item)
-                  }
-
-                  FXMenuCommand.new(menu_pane, "collapse tree" ).connect(SEL_COMMAND) {
-                    self.collapseFullTree(item)
-                  }
-
-                end
-
-                data = self.getItemData(item)
-                
-                if item == "Forms" then
-                  puts "Right Click Forms"
-                  FXMenuSeparator.new(menu_pane)
-
-                  FXMenuCommand.new(menu_pane, "found Form Item" ).connect(SEL_COMMAND) {
-
-                    notify(:add_site_to_scope, item.to_s)
-                  }
-                else
-                  puts "Didn't click Forms"
-
-                end
-              # submenu = FXMenuPane.new(self) do |domain_menu|
-
-              #   @filtered_domains.each do |domain, filter|
-              #     hide_domain = FXMenuCommand.new(domain_menu, "#{domain}" )
-              #     hide_domain.connect(SEL_COMMAND) {
-              #       @interface.default_settings[:domain_filters].delete(filter)
-              #       @filtered_domains.clear
-              #       @interface.updateTreeLists
-              #     }
-              #   end
-              # end
-              # FXMenuCascade.new(menu_pane, "Unhide Domains", nil, submenu)
-
-              end
-              menu_pane.create
-              menu_pane.popup(nil, event.root_x, event.root_y)
-              app.runModalWhileShown(menu_pane)
-
-            end
-          end
-        end
-      end
-
-      private
-
-      def get_node_class(name)
-        return nil if name.nil?
-        return nil if name.empty?
-        puts "get_node_class #{name}"
-
-        ccname = Watobo::Utils.camelcase(name)
-        nc = nil
-        unless Watobo::Gui::PageTree.const_defined? ccname
-          puts "Creating new class #{ccname}"
-          Watobo::Gui::PageTree.class_eval("class #{ccname};end")
-          nc = Watobo::Gui::PageTree.const_get(ccname)
-        else
-          puts "Get existing class #{ccname}"
-          nc = Watobo::Gui::PageTree.const_get(ccname)
-        end
-        puts nc.name
-        nc
-      end
-
-      def add_form_item(form)
-
-        node_name = "Forms"
-        #nc = get_node_class node_name
-        # form_item = self.findItem(node_name, nil, SEARCH_FORWARD|SEARCH_IGNORECASE)
-        form_item = self.findItemByData(node_name.to_sym, nil, SEARCH_FORWARD)
-
-       
-        if form_item.nil?
-          # found new site
-          form_item = self.appendItem(nil, node_name, nil, nil)
-        #site = @findings_tree.moveItem(project.first,project,site)
-        self.setItemData(form_item, node_name.to_sym)
-        end
-
-        form_name_item = nil
-        name = "undefined"
-        unless form.name.nil?
-        name = form.name unless form.name.empty?
-        end
-        form_name_item = self.appendItem(form_item, name, nil, nil) 
-
-        unless form_name_item.nil?
-          self.setItemData(form_name_item, form)
-          form.fields.each do |ff|
-            form_field_item = self.appendItem(form_name_item, ff.name, nil, nil)
-            self.setItemData(form_field_item, ff)
-
-          end
-          form.buttons.each do |fb|
-            form_field_item = self.appendItem(form_name_item, fb.name, nil, nil)
-            self.setItemData(form_field_item, fb)
-          end
-        end
-
-        unless form_item.nil?
-          form_item.text = "#{node_name} (#{form_item.numChildren})"
-        end
-
-      end
-
-      def notify(event, *args)
-        if @event_dispatcher_listeners[event]
-          @event_dispatcher_listeners[event].each do |m|
-            m.call(*args) if m.respond_to? :call
-          end
-        end
-      end
-    end
-  # namespace end
-  end
-end
+module Watobo#:nodoc: all
+  module Gui
+    class PageTree < FXTreeList
+
+      attr :page
+
+      include Watobo::Constants
+      include Watobo::Gui::Icons
+      def subscribe(event, &callback)
+        (@event_dispatcher_listeners[event] ||= []) << callback
+      end
+
+      def reload()
+        self.clearItems
+
+      # @interface.updateRequestTable(@project)
+      end
+
+      def page=(p)
+        @page = p
+        refresh_tree
+      end
+
+      def refresh_tree()
+        self.clearItems
+        unless @page.nil?
+          if page.forms.length > 0
+          page.forms.each do |f|
+            add_form_item f
+          end
+          else
+            self.appendItem(nil, "Forms ()", nil, nil)
+          end
+        end
+      # @interface.updateRequestTable(@project)
+      end
+
+      def expandFullTree(item)
+        self.expandTree(item)
+        item.each do |c|
+          expandFullTree(c) if !self.itemLeaf?(c)
+        end
+      end
+
+      def collapseFullTree(item)
+        self.collapseTree(item)
+        item.each do |c|
+          collapseFullTree(c) if !self.itemLeaf?(c)
+        end
+      end
+
+      def hidden?(chat)
+
+        #TODO: Filter
+        false
+      end
+
+      def initialize(parent, page=nil, prefs={})
+        @page = page
+        @quick_filter = Hash.new
+        opts = LAYOUT_FILL_X|LAYOUT_FILL_Y|LAYOUT_TOP|TREELIST_SHOWS_LINES|TREELIST_SHOWS_BOXES|TREELIST_ROOT_BOXES|TREELIST_EXTENDEDSELECT
+        opts = prefs[:opts] if prefs.has_key? :opts
+        super(parent, :opts => opts)
+        @event_dispatcher_listeners = Hash.new
+        @filters = {
+          :element => []
+        }
+
+        forms = self.appendItem(nil, "load a new page first!", nil, nil)
+
+        self.connect(SEL_COMMAND) do |sender, sel, item|
+          
+          if item.data.is_a? Mechanize::Form
+               puts "Form Selected"
+               notify(:form_selected, item.data)
+          elsif item.data.is_a? Mechanize::Form::Field
+            puts "Field Selected"
+            notify(:form_selected, item.parent.data)
+            notify(:field_selected, item)
+          end
+
+        end
+
+        self.connect(SEL_RIGHTBUTTONRELEASE) do |sender, sel, event|
+          exclude_site = nil
+          unless event.moved?
+            FXMenuPane.new(self) do |menu_pane|
+
+            #
+              item = sender.getItemAt(event.win_x, event.win_y)
+
+              unless item.nil?
+
+                unless self.itemLeaf?(item)
+                  FXMenuSeparator.new(menu_pane)
+                  FXMenuCommand.new(menu_pane, "expand tree" ).connect(SEL_COMMAND) {
+                    expandFullTree(item)
+                  }
+
+                  FXMenuCommand.new(menu_pane, "collapse tree" ).connect(SEL_COMMAND) {
+                    self.collapseFullTree(item)
+                  }
+
+                end
+
+                data = self.getItemData(item)
+                
+                if item == "Forms" then
+                  puts "Right Click Forms"
+                  FXMenuSeparator.new(menu_pane)
+
+                  FXMenuCommand.new(menu_pane, "found Form Item" ).connect(SEL_COMMAND) {
+
+                    notify(:add_site_to_scope, item.to_s)
+                  }
+                else
+                  puts "Didn't click Forms"
+
+                end
+              # submenu = FXMenuPane.new(self) do |domain_menu|
+
+              #   @filtered_domains.each do |domain, filter|
+              #     hide_domain = FXMenuCommand.new(domain_menu, "#{domain}" )
+              #     hide_domain.connect(SEL_COMMAND) {
+              #       @interface.default_settings[:domain_filters].delete(filter)
+              #       @filtered_domains.clear
+              #       @interface.updateTreeLists
+              #     }
+              #   end
+              # end
+              # FXMenuCascade.new(menu_pane, "Unhide Domains", nil, submenu)
+
+              end
+              menu_pane.create
+              menu_pane.popup(nil, event.root_x, event.root_y)
+              app.runModalWhileShown(menu_pane)
+
+            end
+          end
+        end
+      end
+
+      private
+
+      def get_node_class(name)
+        return nil if name.nil?
+        return nil if name.empty?
+        puts "get_node_class #{name}"
+
+        ccname = Watobo::Utils.camelcase(name)
+        nc = nil
+        unless Watobo::Gui::PageTree.const_defined? ccname
+          puts "Creating new class #{ccname}"
+          Watobo::Gui::PageTree.class_eval("class #{ccname};end")
+          nc = Watobo::Gui::PageTree.const_get(ccname)
+        else
+          puts "Get existing class #{ccname}"
+          nc = Watobo::Gui::PageTree.const_get(ccname)
+        end
+        puts nc.name
+        nc
+      end
+
+      def add_form_item(form)
+
+        node_name = "Forms"
+        #nc = get_node_class node_name
+        # form_item = self.findItem(node_name, nil, SEARCH_FORWARD|SEARCH_IGNORECASE)
+        form_item = self.findItemByData(node_name.to_sym, nil, SEARCH_FORWARD)
+
+       
+        if form_item.nil?
+          # found new site
+          form_item = self.appendItem(nil, node_name, nil, nil)
+        #site = @findings_tree.moveItem(project.first,project,site)
+        self.setItemData(form_item, node_name.to_sym)
+        end
+
+        form_name_item = nil
+        name = "undefined"
+        unless form.name.nil?
+        name = form.name unless form.name.empty?
+        end
+        form_name_item = self.appendItem(form_item, name, nil, nil) 
+
+        unless form_name_item.nil?
+          self.setItemData(form_name_item, form)
+          form.fields.each do |ff|
+            form_field_item = self.appendItem(form_name_item, ff.name, nil, nil)
+            self.setItemData(form_field_item, ff)
+
+          end
+          form.buttons.each do |fb|
+            form_field_item = self.appendItem(form_name_item, fb.name, nil, nil)
+            self.setItemData(form_field_item, fb)
+          end
+        end
+
+        unless form_item.nil?
+          form_item.text = "#{node_name} (#{form_item.numChildren})"
+        end
+
+      end
+
+      def notify(event, *args)
+        if @event_dispatcher_listeners[event]
+          @event_dispatcher_listeners[event].each do |m|
+            m.call(*args) if m.respond_to? :call
+          end
+        end
+      end
+    end
+  # namespace end
+  end
+end