wherelizer 0.0.1

data/lib/wherelizer/sexp_extensions.rb

@@ -0,0 +1,62 @@
+module SexpExtensions
+  def is_type?(expected_type)
+    type = self[0]
+    expected_type = [expected_type] unless expected_type.respond_to?(:include?)
+    expected_type.include? type
+  end
+
+  def check_type(expected_type)
+    raise "Unexpected type: #{expected_type} in #{self}" unless is_type?(expected_type)
+  end
+
+  def extract_val
+    check_type [:lit, :str]
+    self[1]
+  end
+
+  def extract_const
+    check_type :const
+    self[1]
+  end
+
+  def extract_assignment_target
+    check_type [:lasgn, :iasgn]
+    self[1].to_s
+  end
+
+  ###
+  # Extract the receiver of a method call.
+  def extract_receiver
+    check_type :call
+    self[1]
+  end
+
+  ###
+  # Extract the method name of a method call.
+  def extract_method_name
+    check_type :call
+    self[2]
+  end
+
+  ###
+  # Turn a sexp "hash" node into an actual Ruby hash.
+  # If process_keys is true, then we assume the hash keys are strings or symbols,
+  #   and extract the actual string or symbol to be the key in the resulting hash.
+  # Otherwise, we make the key the whole sexp node that is the key.
+  def to_hash(process_keys=true)
+    check_type :hash
+    result = {}
+    self[1..-1].each_slice(2) do |key, val|
+      result[process_keys ? key.extract_val : key] = val
+    end
+    result
+  end
+
+  ###
+  # Turns a sexp "array" node into an actual Ruby array of just the elements
+  def to_array
+    check_type :array
+    self[1..-1]
+  end
+
+end

data/lib/wherelizer.rb

@@ -0,0 +1,85 @@
+require 'ruby_parser'
+require 'ruby2ruby'
+require 'wherelizer/sexp_extensions'
+
+class Sexp
+  include SexpExtensions
+end
+
+class Wherelizer
+  attr_accessor :orig, :parsed, :final, :ruby2ruby
+
+  def initialize(orig)
+    @orig = orig
+    @ruby2ruby = Ruby2Ruby.new
+  end
+
+  def convert
+    @parsed = RubyParser.new.parse(orig)
+    assignment = handle_assignment
+
+    receiver = ruby2ruby.process(parsed.extract_receiver)
+    method_name = handle_method_name
+    options_hash = parsed.to_hash
+
+    where = parse_conditions(options_hash.delete(:conditions)) if options_hash[:conditions]
+    others = options_hash.map{|key, value| parse_basic_param(key, value)}.join
+
+    "#{assignment}#{receiver}#{where}#{others}#{method_name}"
+  end
+
+  def handle_assignment
+    if parsed.is_type? [:lasgn, :iasgn]
+      assignment = final_assignment(parsed.extract_assignment_target)
+      @parsed = @parsed[2]
+    end
+    assignment
+  end
+
+  def handle_method_name
+    method_name = parsed.extract_method_name
+    if method_name == :find
+      first_param = parsed[3]
+      @parsed = @parsed[4]
+      final_method_name(first_param.extract_val)
+    else
+      @parsed = @parsed[3]
+      final_method_name(method_name)
+    end
+  end
+
+  def final_method_name(method_name)
+    ".#{method_name}" unless method_name == :all
+  end
+
+  def final_assignment(assignment_target)
+    "#{assignment_target} = "
+  end
+
+  def parse_conditions node
+    where = ''
+
+    if node.is_type? :hash
+      node.to_hash(false).each do |key, val|
+        if key.is_type? :lit
+          where += ".where(#{key.extract_val}: #{ruby2ruby.process(val)})"
+        else
+          where += ".where(#{ruby2ruby.process(key)} => #{ruby2ruby.process(val)})"
+        end
+      end
+    elsif node.is_type? :array
+      where = ".where(" + node.to_array.map{ |el| ruby2ruby.process(el) }.join(', ') + ")"
+    elsif node.is_type? :str
+      where = ".where(#{ruby2ruby.process(node)})"
+    else
+      raise "Unexpected type for conditions parameter. Expecting hash, array, or string."
+    end
+
+    where
+  end
+
+  def parse_basic_param name, value
+    method_name = name.to_s == 'include' ? 'includes' : name.to_s
+    ".#{method_name}(#{ruby2ruby.process(value)})"
+  end
+end

metadata

@@ -0,0 +1,80 @@
+--- !ruby/object:Gem::Specification
+name: wherelizer
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Melinda Weathers
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-05-22 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: ruby_parser
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: ruby2ruby
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A gem for converting pre-arel ActiveRecord queries to their non-deprecated
+  equivalents
+email: melinda@agileleague.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/wherelizer.rb
+- lib/wherelizer/sexp_extensions.rb
+homepage: http://github.com/agileleague/wherelizer.git"
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.25
+signing_key: 
+specification_version: 3
+summary: ActiveRecord Query Updater
+test_files: []
+has_rdoc: