common libs go into the lib/common directory

lib/common/updater/git_updater.rb

@@ -0,0 +1,54 @@
+# encoding: UTF-8
+#--
+# WPScan - WordPress Security Scanner
+# Copyright (C) 2012-2013
+#
+# 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 3 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, see <http://www.gnu.org/licenses/>.
+#++
+
+require File.expand_path(File.dirname(__FILE__) + '/updater')
+
+class GitUpdater < Updater
+
+  def is_installed?
+    %x[git #{repo_directory_arguments()} status 2>&1] =~ /On branch/ ? true : false
+  end
+
+  # Git has not a revsion number like SVN,
+  # so we will take the 7 first chars of the last commit hash
+  def local_revision_number
+    git_log = %x[git #{repo_directory_arguments()} log -1 2>&1]
+    git_log[/commit ([0-9a-z]{7})/i, 1].to_s
+  end
+
+  def update
+    %x[git #{repo_directory_arguments()} pull]
+  end
+
+  def has_local_changes?
+    %x[git #{repo_directory_arguments()} diff --exit-code 2>&1] =~ /diff/ ? true : false
+  end
+
+  def reset_head
+    %x[git #{repo_directory_arguments()} reset --hard HEAD]
+  end
+
+  protected
+  def repo_directory_arguments
+    if @repo_directory
+      return "--git-dir=\"#{@repo_directory}/.git\" --work-tree=\"#{@repo_directory}\""
+    end
+  end
+
+end

lib/common/updater/svn_updater.rb

@@ -0,0 +1,40 @@
+# encoding: UTF-8
+#--
+# WPScan - WordPress Security Scanner
+# Copyright (C) 2012-2013
+#
+# 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 3 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, see <http://www.gnu.org/licenses/>.
+#++
+
+require File.expand_path(File.dirname(__FILE__) + '/updater')
+
+class SvnUpdater < Updater
+
+  REVISION_PATTERN = /revision="(\d+)"/i
+  TRUNK_URL        = 'https://github.com/wpscanteam/wpscan'
+
+  def is_installed?
+    %x[svn info "#@repo_directory" --xml 2>&1] =~ /revision=/ ? true : false
+  end
+
+  def local_revision_number
+    local_revision = %x[svn info "#@repo_directory" --xml 2>&1]
+    local_revision[REVISION_PATTERN, 1].to_s
+  end
+
+  def update
+    %x[svn up "#@repo_directory"]
+  end
+
+end

lib/common/updater/updater.rb

@@ -0,0 +1,42 @@
+# encoding: UTF-8
+#--
+# WPScan - WordPress Security Scanner
+# Copyright (C) 2012-2013
+#
+# 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 3 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, see <http://www.gnu.org/licenses/>.
+#++
+
+# This class act as an absract one
+class Updater
+
+  attr_reader :repo_directory
+
+  # TODO : add a last '/ to repo_directory if it's not present
+  def initialize(repo_directory = nil)
+    @repo_directory = repo_directory
+  end
+
+  def is_installed?
+    raise NotImplementedError
+  end
+
+  def local_revision_number
+    raise NotImplementedError
+  end
+
+  def update
+    raise NotImplementedError
+  end
+
+end

lib/common/updater/updater_factory.rb

@@ -0,0 +1,40 @@
+# encoding: UTF-8
+#--
+# WPScan - WordPress Security Scanner
+# Copyright (C) 2012-2013
+#
+# 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 3 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, see <http://www.gnu.org/licenses/>.
+#++
+
+class UpdaterFactory
+
+  def self.get_updater(repo_directory)
+    self.available_updaters_classes().each do |updater_symbol|
+      updater = Object.const_get(updater_symbol).new(repo_directory)
+
+      if updater.is_installed?
+        return updater
+      end
+    end
+    nil
+  end
+
+  protected
+
+  # return array of class symbols
+  def self.available_updaters_classes
+    Object.constants.grep(/^.+Updater$/)
+  end
+
+end