ruby-wordpress 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 38f5112cc2886be6398ae4573f806025b0b8b6c9
-  data.tar.gz: 3e4dbf0b9de205b21e7f33407d01b62985e02675
+  metadata.gz: 70e9d73a18077c70dd4974d042ecc80a068a97d4
+  data.tar.gz: fc64bc3a3c5e0d77a861c5c0180a4c76f58e973a
 SHA512:
-  metadata.gz: 9f379fd8208d157b17811e2192a9101ebd54584177ddbbb517e8a4b8aaf4d88be76e2611ae95c11468997accaab6ed0b04f95dd779148a5d5ffd78e4d7b44ca3
-  data.tar.gz: 8d095d6e1475059603430adebc374e8162946a85c0e835960a90c426d1ddbc6576c7aab288383c0b21d13ee5ee08e5348e018613b3eaf4404d787803aeccc62e
+  metadata.gz: fbeae0d88a1d4701ba0dcce78111342f674a673f4c14f40437e25b78fe3c38f80005465f83badb7721637cefc11cba0956cbc78b3d9405b9e69b14b7f80ce693
+  data.tar.gz: e54c1d9e11a1e3f421412a743d685084f48f602e0c9ecbaf5b4ff3195b950e67b62009ca98bf368ed10906a9ca47e0b6ffcdd5ae4c1f88f68b6d9f71a8135105

data/lib/wordpress.rb

@@ -1,5 +1,11 @@
 require "wordpress/version"
 
+require "wordpress/error"
+require "wordpress/base"
+require "wordpress/options"
+require "wordpress/post"
+require "wordpress/post/meta"
+
 require 'mysql2'
 require 'php_serialize'
 require 'cgi'
@@ -23,9 +29,124 @@ class WordPress
     @conn = Mysql2::Client.new options
     @conn.query_options.merge!(symbolize_keys: true)
 
+    @configuration = options
+
     # The WordPress options table
-    @options = WP::Options.new @conn, @tbl
+    @options = WordPress::Options.new self
   end
 
   attr_reader :options
+
+  attr_reader :tbl
+  attr_reader :conn
+
+  attr_accessor :configuration
+
+  def query args
+    # Main query function.
+    # The goal is to support as much as the original WP_Query API as possible.
+    raise ArgumentError.new('Hash arguments are supported.') unless args.kind_of?(Hash)
+
+    # Defaults
+    args = {
+      post_type: 'post',
+      post_status: 'publish'
+    }.merge(args)
+
+    wheres_and = []
+    inner_joins = []
+
+    # Page finders
+
+    if args[:page_id]
+      args[:p] = args[:page_id]
+      args[:post_type] = 'page'
+    end
+
+    if args[:pagename]
+      args[:name] = args[:pagename]
+      args[:post_type] = 'page'
+    end
+
+    # Post finders
+
+    if args[:post_type]
+      wheres_and << "`#{@tbl[:posts]}`.`post_type`='#{ @conn.escape args[:post_type] }'"
+    end
+
+    if args[:p]
+      wheres_and << "`#{@tbl[:posts]}`.`ID`='#{ args[:p].to_i }'"
+    end
+
+    if args[:post_parent]
+      wheres_and << "`#{@tbl[:posts]}`.`post_parent`='#{ args[:post_parent].to_i }'"
+    end
+
+    if args[:name]
+      wheres_and << "`#{@tbl[:posts]}`.`post_name`='#{ @conn.escape args[:name] }'"
+    end
+
+    if args[:post_status]
+      wheres_and << "`#{@tbl[:posts]}`.`post_status`='#{ @conn.escape args[:post_status] }'"
+    end
+
+    if args[:post__in]
+      raise ArgumentError.new(':post__in should be an array.') unless args[:post__in].kind_of?(Array)
+      wheres_and << "`#{@tbl[:posts]}`.`ID` IN (#{ args[:post__in].map { |e| "'#{ e.to_i }'" }.join ', ' })"
+    end
+
+    if args[:post__not_in]
+      raise ArgumentError.new(':post__not_in should be an array.') unless args[:post__not_in].kind_of?(Array)
+      wheres_and << "`#{@tbl[:posts]}`.`ID` NOT IN (#{ args[:post__not_in].map { |e| "'#{ e.to_i }'" }.join ', ' })"
+    end
+
+    # Meta finders
+
+    if (mqs = args[:meta_query]) and mqs.kind_of?(Array)
+      inner_joins << "`#{@tbl[:postmeta]}` ON `#{@tbl[:posts]}`.`ID`=`#{@tbl[:postmeta]}`.`post_id`"
+      mqs.each do |mq|
+        mq_params = {
+          compare: '=',
+          type: 'CHAR' # Ignored for now
+        }.merge(mq)
+
+        # Allowed compares
+        mq_params[:compare] = '=' unless ['=', '!=', '>', '>=', '<', '<=', 'LIKE', 'NOT LIKE', 'IN', 'NOT IN', 'BETWEEN', 'NOT BETWEEN'].include?(mq_params[:compare])
+
+        wheres_and << "(`#{@tbl[:postmeta]}`.`meta_key`='#{@conn.escape mq_params[:key].to_s}' AND `#{@tbl[:postmeta]}`.`meta_value`='#{@conn.escape mq_params[:value].to_s}')"
+      end
+    end
+
+    query = "SELECT `#{@tbl[:posts]}`.* FROM `#{@tbl[:posts]}` "
+    if inner_joins.length > 0
+      query += "INNER JOIN #{inner_joins.join ', '} "
+    end
+
+    query += "WHERE #{ wheres_and.join ' AND ' }"
+
+    @conn.query(query).map do |row|
+      WordPress::Post.build self, row
+    end
+  end
+
+  def new_post args
+    # 'args' is a hash of attributes that WordPress::Post responds to
+    # See wordpress/post.rb
+    WordPress::Post.build self, args
+  end
+
+  def update_taxonomy_counts *taxes
+    taxes.each do |taxonomy|
+      @conn.query("SELECT `term_taxonomy_id`, `count` FROM `#{@tbl[:termtax]}` WHERE `taxonomy`='#{@conn.escape taxonomy.to_s}'").each do |term_tax|
+        termtax_id = term_tax[:term_taxonomy_id]
+        count = 0
+        @conn.query("SELECT COUNT(*) as `c` FROM `#{@tbl[:posts]}`, `#{@tbl[:termrel]}` WHERE `#{@tbl[:posts]}`.`ID`=`#{@tbl[:termrel]}`.`object_id` AND `#{@tbl[:termrel]}`.`term_taxonomy_id`='#{ termtax_id.to_i }'").each do |x|
+          count = x[:c]
+        end
+        if count != term_tax[:count]
+          @conn.query("UPDATE `#{@tbl[:termtax]}` SET `count`='#{count.to_i}' WHERE `term_taxonomy_id`='#{ termtax_id.to_i }'")
+        end
+      end
+    end
+  end
 end

data/lib/wordpress/base.rb

@@ -3,9 +3,15 @@
 require 'unicode_utils/nfc'
 
 class WordPress::Base
-  def initialize(connection, wp_tables)
-    @tbl = wp_tables
-    @conn = connection
+  def initialize root
+    @wp = root
+
+    @conn = @wp.conn
+    @tbl = @wp.tbl
+  end
+
+  def inspect
+    nil
   end
 
   def insert(table, content)
@@ -67,7 +73,6 @@ class WordPress::Base
   end
 
   def set_post_terms(post_id, terms, taxonomy, append=false)
-    terms.map! { |e| UnicodeUtils.nfc(e) }
     terms_esc = terms.map { |e| "'#{@conn.escape e.to_s}'" }
     terms_slugs = terms.map { |e| "'#{@conn.escape(CGI::escape e.to_s)}'"}
     raise ArgumentError, 'Terms must be an array with more than zero elements' unless terms_esc.count > 0

data/lib/wordpress/error.rb

@@ -0,0 +1,4 @@
+# Encoding: UTF-8
+
+class WordPress::Error < RuntimeError
+end

data/lib/wordpress/post.rb

@@ -0,0 +1,266 @@
+# Encoding: UTF-8
+
+require 'cgi'
+require 'mime/types'
+require 'RMagick'
+
+class WordPress::Post < WordPress::Base
+  # Left: DB, right: our symbol
+  DB_MAP = {
+    :ID => :post_id
+  }
+
+  READONLY_ATTRIBUTES = [
+    :post_id
+  ]
+
+  WORDPRESS_ATTRIBUTES = {
+    :post_id => -1,
+    :post_author => 1,
+    :post_date => Time.new,
+    :post_date_gmt => Time.new.utc,
+    :post_content => '',
+    :post_title => '',
+    :post_excerpt => '',
+    :post_status => 'draft',
+    :comment_status => 'open',
+    :ping_status => 'open',
+    :post_password => '',
+    :post_name => '',
+    :to_ping => '',
+    :pinged => '',
+    :post_modified => Time.new,
+    :post_modified_gmt => Time.new.utc,
+    :post_content_filtered => '',
+    :post_parent => 0,
+    :guid => '0',
+    :menu_order => 0,
+    :post_type => 'post',
+    :post_mime_type => '',
+    :comment_count => 0
+  }
+
+  def to_h
+    Hash[ WORDPRESS_ATTRIBUTES.keys.map { |e| [e, instance_variable_get(:"@#{e}")] }]
+  end
+
+  def inspect
+    to_h.to_s
+  end
+
+  def persisted?
+    @post_id != -1 and !unsaved_changes?
+  end
+
+  def unsaved_changes
+    WORDPRESS_ATTRIBUTES.keys.select do |k|
+      false
+      true if instance_variable_get(:"@#{k}") != @in_database[k]
+    end
+  end
+
+  def unsaved_changes?
+    # Not an empty array of changes means there are unsaved changes.
+    unsaved_changes != []
+  end
+
+  WORDPRESS_ATTRIBUTES.keys.each do |att|
+    if READONLY_ATTRIBUTES.include? att
+      attr_reader att
+    else
+      attr_accessor att
+    end
+  end
+
+  def save
+    # We don't need to save because nothing has changed
+    return self if persisted?
+
+    new_id = update_or_insert @tbl[:posts], "`#{@tbl[:posts]}`.`ID`='#{ post_id.to_i }'", Hash[ WORDPRESS_ATTRIBUTES.keys.map { |e| [e, instance_variable_get(:"@#{e}")] }].reject { |k, v| READONLY_ATTRIBUTES.include? k }
+
+    # We'll assume everything went OK since no errors were thrown.
+    @in_database = Hash[ WORDPRESS_ATTRIBUTES.keys.map { |e| [e, instance_variable_get(:"@#{e}")] }]
+    if new_id and new_id > 0
+      @in_database[:post_id] = @post_id = new_id
+    end
+
+    self
+  end
+
+  def save!
+    save || raise(WordPress::Error.new('Save failed.'))
+  end
+
+  # Post Meta
+
+  def post_meta
+    raise 'Post must be saved before manipulating metadata' if @post_id == -1
+    @post_meta ||= WordPress::Post::Meta.new @wp, self
+  end
+
+  # Taxonomies
+
+  def get_the_terms taxonomy
+    raise 'Post must be saved before manipulating taxonomies' if @post_id == -1
+    super @post_id, taxonomy
+  end
+
+  def set_post_terms terms, taxonomy, append=false
+    raise 'Post must be saved before manipulating taxonomies' if @post_id == -1
+    terms = [terms] unless terms.kind_of?(Array)
+    current_terms = get_the_terms(taxonomy)
+    return current_terms if current_terms.sort == terms.sort && append == false
+    super @post_id, terms, taxonomy, append=false
+  end
+
+  # Attachments
+
+  def attach_featured_image image
+    img_id = attach_image(image).post_id
+    post_meta['_thumbnail_id'] = img_id
+  end
+
+  def attach_image image
+    raise 'Post must be saved before manipulating attached images' if @post_id == -1
+
+    # Make a new post with the "attachment" format
+    if image.respond_to? :open
+      handle = image.open
+    else
+      handle = image
+    end
+
+    title = (0...10).map{(65+rand(26)).chr}.join
+    if image.respond_to? :path
+      path = Pathname.new image.path
+      title = path.each_filename.to_a[-1]
+    end
+
+    mimetype = nil
+    ext = ''
+    if image.respond_to? :meta
+      mimetype = (image.meta['content-type'] || '').split(';')[0]
+      type = MIME::Types[mimetype].first
+      if type
+        ext = '.' + type.extensions.first
+      end
+    else
+      if type = MIME::Types.type_for(title).first
+        mimetype = types.content_type
+        # ext = types.extensions.first
+      end
+    end
+
+    # Build the pathname where this will go.
+    file_basename = File.join(@wp.configuration[:wordpress_wp_content], '/uploads')
+    uri_basename = @wp.configuration[:wordpress_wp_content_url] || (@wp.options['siteurl'] + '/wp-content/uploads')
+
+    today = Date.today
+    relative_filepath = "#{'%02d' % today.year}/#{'%02d' % today.month}/#{title}"
+
+    # Copy the file
+    local_filepath = Pathname.new(File.join(file_basename, relative_filepath + ext))
+    FileUtils.mkdir_p local_filepath.dirname.to_s
+
+    buffer = handle.read.force_encoding('BINARY')
+    out = File.open(local_filepath.to_s, 'wb')
+    out.write buffer
+    out.close
+
+    attachment = self.class.build @wp, {
+      post_title: title,
+      post_name: CGI::escape(title.downcase),
+      post_status: 'inherit',
+      post_parent: @post_id,
+      post_type: 'attachment',
+      post_mime_type: mimetype,
+      guid: uri_basename + '/' + relative_filepath + ext
+    }
+    attachment.save
+
+    attachment.post_meta['_wp_attached_file'] = relative_filepath + ext
+
+    # Get image metadata
+
+    begin
+      img = Magick::Image::read(local_filepath.to_s).first
+      size_hash = {}
+
+      thumbnail_filename = title + '-150x150' + ext
+      thumb_img = img.resize_to_fill(150, 150)
+      thumb_img.write File.join(file_basename, "#{'%02d' % today.year}/#{'%02d' % today.month}/#{thumbnail_filename}")
+
+      size_hash[:thumbnail] = {
+        file: thumbnail_filename,
+        width: 150,
+        height: 150
+      }
+
+      size_hash[:medium] = {
+        file: title + ext,
+        height: img.rows,
+        width: img.columns
+      }
+
+      size_hash[:large] = {
+        file: title + ext,
+        height: img.rows,
+        width: img.columns
+      }
+
+      attachment.post_meta['_wp_attachment_metadata'] = {
+        file: title + ext,
+        height: img.rows,
+        width: img.columns,
+        sizes: size_hash
+      }
+    rescue Exception => e
+      # raise e
+      puts "Warn: Ignoring exception #{e.to_s}"
+    end
+
+    attachment
+  end
+
+  def featured_image
+    thumb_id = post_meta['_thumbnail_id']
+    @wp.query(post_type: 'attachment', post_parent: @post_id, post_status: 'inherit', p: thumb_id).first if thumb_id
+  end
+
+  def attached_files *args
+    attach_args = {
+      post_type: 'attachment', post_parent: @post_id, post_status: 'inherit'
+      }.merge(args[0] || {})
+    @wp.query attach_args
+  end
+
+  # Initializators
+
+  def initialize root
+    super
+
+    WORDPRESS_ATTRIBUTES.each do |att, default|
+      instance_variable_set :"@#{att}", default
+    end
+
+    @in_database = {}
+  end
+
+  def self.build root, values
+    post = new root
+    # Use the map
+    values = Hash[values.map { |k, v| [DB_MAP[k] || k, v] }]
+
+
+    from_db = false
+    # Because the post ID is greater than zero, let's assume that this is a persisted record.
+    from_db = true if values[:post_id] and values[:post_id] > 0
+
+    values.select { |key, value| WORDPRESS_ATTRIBUTES.keys.include? key }.each do |key, value|
+      post.instance_variable_set(:"@#{key}", value)
+      post.instance_variable_get(:@in_database)[key] = value if from_db
+    end
+    post
+  end
+
+end

data/lib/wordpress/post/meta.rb

@@ -0,0 +1,64 @@
+# Encoding: UTF-8
+
+require 'php_serialize'
+
+class WordPress::Post::Meta < WordPress::Base
+
+  def initialize root, post
+    super root
+    @post = post
+  end
+
+  def [](key)
+    v = nil
+    @conn.query("SELECT `meta_value` FROM `#{@tbl[:postmeta]}` WHERE `meta_key`='#{@conn.escape key}' AND `post_id`='#{ @post.post_id.to_i }' LIMIT 1").each do |row|
+      v = row[:meta_value]
+    end
+    # Apply out-filters
+    if v
+      if v[0] == 'a' and v[-1] == '}'
+        # PHP-serialized array
+        v = PHP.unserialize v
+      end
+    end
+    v
+  end
+
+  def []=(key, value)
+    old_value = self[key]
+    # Apply in-filters
+
+    if value.kind_of?(Hash) or value.kind_of?(Array)
+      value = PHP.serialize value
+    end
+
+    if !value.nil? and !old_value.nil? and value != old_value
+      # Update operation.
+      @conn.query("UPDATE `#{@tbl[:postmeta]}` SET `meta_value`='#{@conn.escape value.to_s}' WHERE `meta_key`='#{@conn.escape key}' AND `post_id`='#{ @post.post_id.to_i }'")
+    elsif value.nil? and !old_value.nil?
+      # New value nil, old value not. Delete operation.
+      @conn.query("DELETE FROM `#{@tbl[:postmeta]}` WHERE `meta_key`='#{@conn.escape key}' AND `post_id`='#{ @post.post_id.to_i }'")
+    elsif !value.nil? and old_value.nil?
+      # New value non-nil, old value nil. Insert operation.
+      @conn.query("INSERT INTO `#{@tbl[:postmeta]}` (`meta_key`, `meta_value`, `post_id`) VALUES ('#{@conn.escape key}', '#{@conn.escape value.to_s}', '#{ @post.post_id.to_i }')")
+    end
+    value
+  end
+
+  def keys
+    all = []
+    @conn.query("SELECT `meta_key` FROM `#{@tbl[:postmeta]}` WHERE `post_id`='#{ @post.post_id.to_i }'").each { |x| all << x }
+    all.collect { |x| x[:meta_key] }
+  end
+
+  def to_s
+    to_h.to_s
+  end
+
+  def to_h
+    Hash[keys.map { |e| [e, self[e]] }]
+  end
+  alias_method :to_hash, :to_h
+  alias_method :inspect, :to_h
+
+end

data/lib/wordpress/version.rb

@@ -1,5 +1,5 @@
 class WordPress
   module Version
-    VERSION = "0.0.1"
+    VERSION = "0.0.2"
   end
 end

data/ruby-wordpress.gemspec

@@ -21,6 +21,8 @@ Gem::Specification.new do |spec|
   spec.add_dependency "mysql2", "~> 0.3.11"
   spec.add_dependency "unicode_utils", "~> 1.4.0"
   spec.add_dependency "k-php-serialize", "~> 1.2.0"
+  spec.add_dependency "mime-types", "~> 1.23"
+  spec.add_dependency 'rmagick', '= 2.13.2'
 
   spec.add_development_dependency "bundler", "~> 1.3"
   spec.add_development_dependency "rake"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-wordpress
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Keitaroh Kobayashi
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-05-27 00:00:00.000000000 Z
+date: 2013-06-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mysql2
@@ -52,6 +52,34 @@ dependencies:
     - - ~>
       - !ruby/object:Gem::Version
         version: 1.2.0
+- !ruby/object:Gem::Dependency
+  name: mime-types
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.23'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.23'
+- !ruby/object:Gem::Dependency
+  name: rmagick
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 2.13.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 2.13.2
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -94,7 +122,10 @@ files:
 - Rakefile
 - lib/wordpress.rb
 - lib/wordpress/base.rb
+- lib/wordpress/error.rb
 - lib/wordpress/options.rb
+- lib/wordpress/post.rb
+- lib/wordpress/post/meta.rb
 - lib/wordpress/version.rb
 - ruby-wordpress.gemspec
 homepage: https://github.com/keichan34/ruby-wordpress