Parse partials path in jbuilder module

Author: gazay

lib/gon/global.rb

@@ -14,6 +14,24 @@ def inspect
         'Gon::Global'
       end
 
+      def rabl(*args)
+        unless Gon.constants.include?(:Rabl)
+          raise "Possible wrong require order problem - try to add `gem 'rabl'` before `gem 'gon'` in your Gemfile"
+        end
+        data, options = Gon::Rabl.handler(args, true)
+
+        store_builder_data 'rabl', data, options
+      end
+
+      def jbuilder(*args)
+        unless Gon.constants.include?(:Jbuilder)
+          raise "Possible wrong require order problem - try to add `gem 'jbuilder'` before `gem 'gon'` in your Gemfile"
+        end
+        data, options = Gon::Jbuilder.handler(args, true)
+
+        store_builder_data 'jbuilder', data, options
+      end
+
       private
 
       def get_variable(name)

lib/gon/jbuilder.rb

@@ -7,12 +7,14 @@ def handler(args, global = false)
         if global && !options[:template]
           raise 'You should provide :template when use rabl with global variables'
         end
+        controller = Gon::Base.get_controller(options)
+        @_controller_name = global ? '' : controller.controller_path
 
         include_helpers
 
         data = parse_jbuilder \
           Gon::Base.get_template_path(options,'jbuilder'),
-          Gon::Base.get_controller(options)
+          controller
 
         [data, options]
       end
@@ -69,6 +71,7 @@ def parse_source(source, controller)
 
       def parse_partial(partial_line)
         path = partial_line.match(/['"]([^'"]*)['"]/)[1]
+        path = parse_path path
         options_hash = partial_line.match(/,(.*)/)[1]
         if options_hash.present?
           options = eval '{' + options_hash + '}'
@@ -80,10 +83,40 @@ def parse_partial(partial_line)
         find_partials File.readlines(path)
       end
 
+      def parse_path(path)
+        return path if File.exists?(path)
+
+        splitted = path.split('/')
+        if splitted.size == 2
+          tmp_path = construct_path(splitted[0], splitted[1])
+          return tmp_path if tmp_path
+        elsif splitted.size == 1
+          tmp_path = construct_path @_controller_name, splitted[0]
+          return tmp_path if tmp_path
+        end
+
+        raise 'Something wrong with partial path in your jbuilder templates'
+      end
+
+      def construct_path(part1, part2)
+        tmp_path = "app/views/#{part1}/_#{part2}"
+        tmp_path = path_with_ext(tmp_path)
+        return tmp_path if tmp_path
+        tmp_path = "app/views/#{part1}/#{part2}"
+        tmp_path = path_with_ext(tmp_path)
+        return tmp_path if tmp_path
+      end
+
+      def path_with_ext(path)
+        return path if File.exists?(path)
+        return "#{path}.jbuilder" if File.exists?("#{path}.jbuilder")
+        return "#{path}.json.jbuilder" if File.exists?("#{path}.json.jbuilder")
+      end
+
       def find_partials(lines = [])
         lines.map do |line|
           if line =~ /partial!/
-            parse_partial(line)
+            parse_partial line
           else
             line
           end