wp_wrapper 0.0.2

data/lib/wp_wrapper/modules/plugins.rb

@@ -0,0 +1,57 @@
+module WpWrapper
+  module Modules
+    module Plugins
+      include ::WpWrapper::Modules::Plugins::Akismet
+      include ::WpWrapper::Modules::Plugins::Gocodes
+      include ::WpWrapper::Modules::Plugins::W3TotalCache
+      include ::WpWrapper::Modules::Plugins::WordpressSeo
+      include ::WpWrapper::Modules::Plugins::TrackingCode
+      
+      def manage_plugins(plugin_identifiers, action = :activate)
+        plugin_identifiers        =   (plugin_identifiers.is_a?(Array)) ? plugin_identifiers : [plugin_identifiers.to_s]
+        
+        plugin_identifiers.each do |plugin_identifier|
+          manage_plugin(plugin_identifier, action)
+        end
+      end
+      
+      def manage_plugin(plugin_identifier, action = :activate)
+        success                   =   false
+      
+        if (login)
+          activation_link         =   nil
+          plugins_page            =   self.mechanize_client.open_url(get_url(:plugins))
+
+          if (plugins_page)
+            plugin_links          =   plugins_page.parser.css("table.plugins tbody tr td span.#{action} a")
+            regex                 =   Regexp.new("plugin=#{plugin_identifier}", Regexp::IGNORECASE)
+        
+            plugin_links.each do |link|
+              href                =   link["href"]
+          
+              if (regex.match(href))
+                activation_link   =   href
+                break
+              end
+            end if (plugin_links && plugin_links.any?)
+            
+            if (activation_link && activation_link.present?)
+              url                 =   "#{get_url(:admin)}/#{activation_link}"
+              self.mechanize_client.open_url(url)
+              puts "#{Time.now}: Url: #{self.url}. Plugin '#{plugin_identifier}' has been #{action}d!"
+              success             =   true
+            else
+              puts "#{Time.now}: Url: #{self.url}. Couldn't find the plugin #{plugin_identifier}'s #{action}-link."
+            end
+            
+          end
+        else
+          puts "#{Time.now}: Failed to login for url #{self.url}, will not proceed to #{action} plugins"
+        end
+
+        return success
+      end
+
+    end
+  end
+end

data/lib/wp_wrapper/modules/profiles.rb

@@ -0,0 +1,66 @@
+module WpWrapper
+  module Modules
+    module Profiles
+    
+      def add_user(login, email, password, role = :editor, send_password = false)
+        options   =   {
+          :user_login     =>  {:value   => login,         :type => :input}, 
+          :email          =>  {:value   => email,         :type => :input},
+          :pass1          =>  {:value   => password,      :type => :input},
+          :pass2          =>  {:value   => password,      :type => :input},
+          :role           =>  {:value   => role,          :type => :select},
+          :send_password  =>  {:checked => send_password, :type => :checkbox},
+        }
+      
+        url       =   get_url(:new_user, absolute: false, admin_prefix: '')
+      
+        return set_options_and_submit(url, {name: "createuser"}, options)
+      end
+    
+      def change_display_name(display_name)
+        options   =   {
+          :nickname         =>    {:value => display_name, :type => :input}
+        }
+        
+        update_profile(options)
+      
+        options   =   {
+          :display_name     =>    {:value => display_name, :type => :select}
+        }
+        
+        update_profile(options)
+      end
+        
+      def change_profile_password
+        options   =   {
+          :pass1    =>  {:value => new_password, :type => :input}, 
+          :pass2    =>  {:value => new_password, :type => :input}
+        }
+      
+        return update_profile(options)
+      end
+    
+      def update_profile(options = {})
+        url       =   get_url(:profile, absolute: false, admin_prefix: '')
+      
+        return set_options_and_submit(url, {action: /profile\.php/i}, options)
+      end
+    
+      # Deprec?
+      def change_profile_password(new_password)
+        if (login)
+          profile_page    =   self.mechanize_client.open_url(get_url(:profile))
+
+          if (profile_page)
+            profile_form  =   profile_page.form_with(:action => "#{self.url}/wp-admin/profile.php")
+            profile_form.field_with(:name => 'pass1').value   =   new_password
+            profile_form.field_with(:name => 'pass2').value   =   new_password
+          
+            updated_page  =   profile_form.submit(profile_form.buttons.first)
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/wp_wrapper/modules/setup.rb

@@ -0,0 +1,65 @@
+module WpWrapper
+  module Modules
+    module Setup
+    
+      def setup(title: nil, email: nil)
+        success           =   false
+        step_1_page       =   self.mechanize_client.open_url(get_url(:home))
+      
+        if step_1_page
+          step_1_form     =   step_1_page.form_with(action: '?step=1')
+
+          if step_1_form
+            step_2_page   =   step_1_form.submit
+            success       =   set_setup_form_options(step_2_page, title: title, email: email)
+          else
+            success       =   set_setup_form_options(step_1_page, title: title, email: email)
+          end
+        end
+      
+        return success
+      end
+      
+      def set_setup_form_options(setup_page, title: nil, email: nil)
+        success       =   false
+        setup_form    =   setup_page.form_with(action: 'install.php?step=2')
+
+        if (setup_form && title.present? && self.username.present? && self.password.present? && email.present?)
+          puts "#{Time.now}: Url: #{self.url}. Setting up site..."
+
+          setup_form.field_with(name: 'weblog_title').value      =   title
+          setup_form.field_with(name: 'user_name').value         =   self.username
+          setup_form.field_with(name: 'admin_password').value    =   self.password
+          setup_form.field_with(name: 'admin_password2').value   =   self.password
+          setup_form.field_with(name: 'admin_email').value       =   email
+
+          confirmation_page                                      =   setup_form.submit
+
+          puts "#{Time.now}: Url: #{self.url}. The WordPress-blog has now been installed!"
+          success                                                =   true
+        else
+          puts "#{Time.now}: Url: #{self.url}. The blog has already been setup or the registration form couldn't be found or some data is missing."
+          puts "#{Time.now}: Url: #{self.url}. Information supplied:\nTitle: #{title}.\nUsername: #{self.username}.\nPassword: #{self.password}.\nEmail: #{email}."
+        end
+        
+        return success
+      end
+    
+      def set_permalinks_options(options = {})
+        permalink_structure       =     options.fetch(:permalink_structure, '/%postname%/')
+        category_base             =     options.fetch(:category_base,       'kategori')
+        tag_base                  =     options.fetch(:tag_base,            'etikett')
+      
+        opts                      =     {
+          :custom_selection       =>  {:identifier => :id,              :checked => true,         :type => :radiobutton},
+          :permalink_structure    =>  {:value => permalink_structure,   :type => :input},
+          :category_base          =>  {:value => category_base,         :type => :input},
+          :tag_base               =>  {:value => tag_base,              :type => :input},
+        }
+      
+        return set_options_and_submit("options-permalink.php", {:action => 'options-permalink.php'}, opts, :first, {:should_reset_radio_buttons => true})
+      end
+
+    end
+  end
+end

data/lib/wp_wrapper/modules/themes.rb

@@ -0,0 +1,71 @@
+module WpWrapper
+  module Modules
+    module Themes
+      
+      def activate_random_theme
+        theme_links               =   get_theme_links
+        random_link               =   (theme_links && theme_links.any?) ? theme_links.to_a.sample : nil
+
+        perform_activation(random_link["href"]) if (random_link)
+      end
+    
+      def activate_theme(theme_identifier = 'twentytwelve')
+        success                   =   false
+        theme_links               =   get_theme_links
+        
+        if (theme_links && theme_links.any?)
+          activation_link         =   nil
+          regex                   =   Regexp.new("stylesheet=#{theme_identifier}", Regexp::IGNORECASE)
+      
+          theme_links.each do |link|
+            href                  =   link["href"]
+        
+            if (regex.match(href))
+              activation_link     =   href
+              break
+            end
+          end
+      
+          success                 =   perform_activation(activation_link)
+          
+          if success
+            puts "[WpWrapper::Modules::Themes] - #{Time.now}: Url: #{self.url}. Theme '#{theme_identifier}' has been activated!"
+          else
+            puts "[WpWrapper::Modules::Themes] - #{Time.now}: Url: #{self.url}. Couldn't find the theme #{theme_identifier}'s activation-link."
+          end
+        end
+      
+        return success
+      end
+      
+      def perform_activation(url)
+        success                 =   false
+        
+        if (url && url.present?)
+          puts "[WpWrapper::Modules::Themes] - #{Time.now}: Will activate theme with url #{url}."
+          self.mechanize_client.open_url(url)
+          success               =   true
+        end
+        
+        return success
+      end
+      
+      def get_theme_links
+        theme_links           =   []
+        
+        if (login)
+          themes_page         =   self.mechanize_client.open_url(get_url(:themes))
+          
+          if (themes_page)
+            theme_links       =   themes_page.parser.css("div.themes div.theme div.theme-actions a.activate")
+          end
+        end
+        
+        puts "[WpWrapper::Modules::Themes] - #{Time.now}: Found a total of #{theme_links.try(:size)} installed themes."
+        
+        return theme_links
+      end
+
+    end
+  end
+end

data/lib/wp_wrapper/modules/upgrade.rb

@@ -0,0 +1,137 @@
+module WpWrapper
+  module Modules
+    
+    module Upgrade
+    
+      def upgrade(type = :core)
+        success              =   self.send("upgrade_#{type}")
+
+        puts "#{Time.now}: Will upgrade #{type} for site #{self.url}."
+
+        if (success)
+          puts "#{Time.now}: Successfully upgraded #{type} for site #{self.url}."
+        end
+      
+        return success
+      end
+    
+      def upgrade_core(retries: 3)
+        success               =   false
+      
+        if (login)
+          update_page         =   self.mechanize_client.open_url(get_url(:upgrade))
+        
+          if (update_page)
+            response_header   =   update_page.parser.at_css('h3.response')
+            upgrade_form      =   update_page.forms_with(:name => 'upgrade').try(:first)
+
+            begin
+              if (response_header && upgrade_form)
+                puts "#{Time.now}: Url: #{self.url}. Upgrading WordPress..."
+                upgraded_page =   upgrade_form.submit(upgrade_form.button_with(:name => 'upgrade'))
+              
+                puts "#{Time.now}: Url: #{self.url}. WordPress was upgraded!"
+                success       =   true
+              else
+                puts "#{Time.now}: Url: #{self.url}. Will not upgrade WordPress, already at latest version."
+                success       =   true
+              end
+            rescue Exception => e
+              puts "#{Time.now}: Url: #{self.url}. An error occurred while trying to upgrade WordPress. Exception: #{e.class.name}. Message: #{e.message}. Stacktrace: #{e.backtrace.join("\n")}"
+              retries        -=   1
+              retry if (retries > 0)
+            end
+          end
+        end
+
+        return success
+      end
+    
+      def upgrade_plugins
+        return upgrade_themes_or_plugins(:plugins)
+      end
+    
+      def upgrade_themes
+        return upgrade_themes_or_plugins(:themes)
+      end
+    
+      def upgrade_themes_or_plugins(type = :plugins, retries: 3)
+        success                 =   false
+        form_identifier         =   get_upgrade_form_identifier(type)
+      
+        if (login)
+          update_page           =   self.mechanize_client.open_url(get_url(:upgrade))
+        
+          if (update_page)
+            upgrade_form        =   update_page.form_with(:name => form_identifier)
+
+            if (upgrade_form)
+              puts "#{Time.now}: Url: #{self.url}. Upgrading #{type}..."
+
+              upgrade_form.checkboxes.each do |checkbox|
+                checkbox.checked = true
+              end
+
+              begin
+                upgraded_page   =   upgrade_form.submit(upgrade_form.buttons.first)
+                upgrade_url     =   upgraded_page.iframes.first.src
+              
+                puts "Upgrade #{type} url: #{upgrade_url.inspect}"
+
+                if (upgrade_url)
+                  self.mechanize_client.agent.get("#{get_url(:admin)}/#{upgrade_url}")
+                  puts "#{Time.now}: Url: #{self.url}. #{type.to_s.capitalize} were upgraded!"
+                  success       =   true
+                end
+              rescue Exception => e
+                retries        -=   1
+                retry if (retries > 0)
+              end
+
+            else
+              puts "#{Time.now}: Url: #{self.url}. Will not upgrade any #{type}, they are all already at the latest version."
+              success           =   true
+            end
+          end
+        end
+      
+        return success
+      end
+    
+      def upgrade_database
+        success                 =   false
+        response                =   self.http_client.retrieve_parsed_html(get_url(:admin)).parsed_body
+      
+        puts "#{Time.now}: Url: #{self.url}. Will check if database upgrades needs to be performed..."
+      
+        if (response)
+          should_upgrade        =   !(response.at_css('body').content =~ /Database Update Required/i).nil?
+        
+          if (should_upgrade)
+            puts "#{Time.now}: Url: #{self.url}. Should Upgrade WordPress database"
+            upgrade_url         =   response.at_css("p.step a.button")['href']
+
+            if (upgrade_url.present?)
+              upgrade_url       =   "#{self.url}wp-admin/#{upgrade_url}" unless upgrade_url =~ /^http/i
+              upgrade_response  =   self.http_client.retrieve_parsed_html(upgrade_url).parsed_body
+              success           =   !(upgrade_response.at_css('body').content =~ /Update Complete/i).nil?
+            
+              puts "#{Time.now}: Url: #{self.url}. Successfully updated WordPress database." if success
+            end
+          end
+        end
+      
+        return success
+      end
+    
+      def get_upgrade_form_identifier(type = :plugins)
+        return case type.to_sym
+          when :plugins then 'upgrade-plugins'
+          when :themes  then 'upgrade-themes' 
+        end
+      end
+
+    end
+    
+  end
+end

data/lib/wp_wrapper/railtie.rb

@@ -0,0 +1,12 @@
+require 'wp_wrapper'
+require 'rails'
+
+module WpWrapper
+  class Railtie < Rails::Railtie
+    
+    #rake_tasks do
+    #  Dir[File.join(File.dirname(__FILE__), '../tasks/*.rake')].each { |ext| load ext }
+    #end
+    
+  end
+end

data/lib/wp_wrapper.rb

@@ -0,0 +1,24 @@
+module WpWrapper
+  VERSION = "0.0.2"
+
+  require File.join(File.dirname(__FILE__), 'wp_wrapper/railtie') if defined?(Rails)
+
+  require File.join(File.dirname(__FILE__), 'wp_wrapper/modules/authorization')
+  require File.join(File.dirname(__FILE__), 'wp_wrapper/modules/setup')
+  require File.join(File.dirname(__FILE__), 'wp_wrapper/modules/upgrade')
+  require File.join(File.dirname(__FILE__), 'wp_wrapper/modules/themes')
+  
+  require File.join(File.dirname(__FILE__), 'wp_wrapper/modules/plugins/akismet')
+  require File.join(File.dirname(__FILE__), 'wp_wrapper/modules/plugins/gocodes')
+  require File.join(File.dirname(__FILE__), 'wp_wrapper/modules/plugins/w3_total_cache')
+  require File.join(File.dirname(__FILE__), 'wp_wrapper/modules/plugins/wordpress_seo')
+  require File.join(File.dirname(__FILE__), 'wp_wrapper/modules/plugins/tracking_code')
+  
+  require File.join(File.dirname(__FILE__), 'wp_wrapper/modules/plugins')
+  
+  require File.join(File.dirname(__FILE__), 'wp_wrapper/modules/options')
+  require File.join(File.dirname(__FILE__), 'wp_wrapper/modules/profiles')
+  require File.join(File.dirname(__FILE__), 'wp_wrapper/modules/api')
+  
+  require File.join(File.dirname(__FILE__), 'wp_wrapper/client')
+end

data/spec/spec_helper.rb

@@ -0,0 +1,43 @@
+$LOAD_PATH << "." unless $LOAD_PATH.include?(".")
+
+begin
+  require "rubygems"
+  require "bundler"
+
+  if Gem::Version.new(Bundler::VERSION) <= Gem::Version.new("0.9.5")
+    raise RuntimeError, "Your bundler version is too old." +
+     "Run `gem install bundler` to upgrade."
+  end
+
+  # Set up load paths for all bundled gems
+  Bundler.setup
+rescue Bundler::GemNotFound
+  raise RuntimeError, "Bundler couldn't find some gems." +
+    "Did you run \`bundlee install\`?"
+end
+
+Bundler.require
+
+require File.expand_path('../../lib/wp_wrapper', __FILE__)
+
+RSpec.configure do |config|
+  config.mock_with :mocha
+end
+
+def load_config
+  config_path   =   File.exists?(File.expand_path('../wordpress.yml', __FILE__)) ? File.expand_path('../wordpress.yml', __FILE__) : File.expand_path('../wordpress.yml.example', __FILE__)
+  config        =   YAML.load_file(config_path)
+  config.symbolize_keys! if config.respond_to?(:symbolize_keys!)
+  
+  return config
+end
+
+def init_admin_connection
+  config = load_config
+  WpWrapper::Client.new(config[:admin])
+end
+
+def init_invalid_connection
+  config = load_config
+  WpWrapper::Client.new(config[:invalid])
+end

data/spec/wordpress.yml.example

@@ -0,0 +1,8 @@
+:admin: 
+  :url: http://domain.com
+  :username: UsernameToLoginWith
+  :password: PasswordToLoginWith
+:invalid: 
+  :url: http://domain.com
+  :username: foofoofoo
+  :password: foobar

data/spec/wp_wrapper/authentication_spec.rb

@@ -0,0 +1,18 @@
+require File.expand_path('../../spec_helper', __FILE__)
+
+describe "Authentication using WpWrapper" do
+  
+  before(:each) do
+    @client       =   init_admin_connection
+  end
+  
+  it 'can successfully login given proper credentials' do
+    @client.login.should == true
+  end
+  
+  it 'can\'t login with invalid credentials' do
+    invalid_client = init_invalid_connection
+    invalid_client.login.should == false
+  end
+
+end

data/wp_wrapper.gemspec

@@ -0,0 +1,51 @@
+Gem::Specification.new do |s|
+  s.specification_version     = 2 if s.respond_to? :specification_version=
+  s.required_rubygems_version = Gem::Requirement.new(">= 1.3.5") if s.respond_to? :required_rubygems_version=
+
+  s.name = 'wp_wrapper'
+  s.version = '0.0.2'
+
+  s.homepage      =   "http://github.com/Agiley/wp_wrapper"
+  s.email         =   "sebastian@agiley.se"
+  s.authors       =   ["Sebastian Johnsson"]
+  s.description   =   "Wrapper to interact with WordPress using Mechanize"
+  s.summary       =   "Wrapper to interact with WordPress using Mechanize"
+
+  s.add_dependency "nokogiri", ">= 1.5.9"
+  s.add_dependency "http_utilities", ">= 1.0.1"
+
+  s.add_development_dependency 'rake'
+  s.add_development_dependency 'rspec'
+  s.add_development_dependency 'mocha'
+  
+  # = MANIFEST =
+ s.files = %w[
+ Gemfile
+ LICENSE.txt
+ README.markdown
+ Rakefile
+ lib/wp_wrapper.rb
+ lib/wp_wrapper/client.rb
+ lib/wp_wrapper/modules/api.rb
+ lib/wp_wrapper/modules/authorization.rb
+ lib/wp_wrapper/modules/options.rb
+ lib/wp_wrapper/modules/plugins.rb
+ lib/wp_wrapper/modules/plugins/akismet.rb
+ lib/wp_wrapper/modules/plugins/gocodes.rb
+ lib/wp_wrapper/modules/plugins/tracking_code.rb
+ lib/wp_wrapper/modules/plugins/w3_total_cache.rb
+ lib/wp_wrapper/modules/plugins/wordpress_seo.rb
+ lib/wp_wrapper/modules/profiles.rb
+ lib/wp_wrapper/modules/setup.rb
+ lib/wp_wrapper/modules/themes.rb
+ lib/wp_wrapper/modules/upgrade.rb
+ lib/wp_wrapper/railtie.rb
+ spec/spec_helper.rb
+ spec/wordpress.yml.example
+ spec/wp_wrapper/authentication_spec.rb
+ wp_wrapper.gemspec
+ ]
+ # = MANIFEST =
+
+  s.test_files = s.files.select { |path| path =~ %r{^spec/*/.+\.rb} }
+end