rgviz-rails 0.23 → 0.25

data/lib/rgviz_rails/view_helper.rb

@@ -0,0 +1,437 @@
+module Rgviz
+  module ViewHelper
+    def rgviz(options = {})
+      options = options.with_indifferent_access
+    
+      id = options[:id]
+      kind = options[:kind]
+      url = options[:url]
+      query = options[:query] || ''
+      events = options[:events] || {}
+      hidden = options[:hidden]
+      
+      html_prefix = (options[:html_prefix] || 'html').to_s
+      js_prefix = (options[:js_prefix] || 'js').to_s
+      param_prefix = (options[:param_prefix] || 'param').to_s
+      
+      html_prefix += '_'
+      js_prefix += '_'
+      param_prefix += '_'
+      
+      debug = options[:debug]
+      opts = options[:options] || {}
+      opts[:width] = 640 unless opts[:width]
+      opts[:height] = 480 unless opts[:height]
+      opts = opts.to_json
+      
+      raise "Must specify an :id" unless id
+      raise "Must specify a :kind" unless kind
+      raise "Must specify a :url" unless url
+      
+      url = url_for url
+      
+      # Parse the query
+      query = Parser.new(query).parse
+      
+      # And replace the html_ and javascript_ magic names
+      visitor = MagicNamesVisitor.new(html_prefix, js_prefix, param_prefix)
+      query.accept visitor
+      query_builder = visitor.query_builder
+      query_builder_var = visitor.query_builder_var
+      params = visitor.params
+      params = params.sort!.map{|i| "param_#{i}"}
+      
+      out = ''
+
+      # Output the google jsapi tag the first time
+      @first_time ||= 1
+      if @first_time == 1
+        out << "<script type=\"text/javascript\" src=\"http://www.google.com/jsapi\"></script>\n"
+      end
+      @first_time = 0
+      
+      # Now the real script
+      out << "<script type=\"text/javascript\">\n"
+      
+      # Load visualizations and the package, if not already loaded
+      pack = kind.downcase      
+      @packages ||= []
+      unless @packages.include?(pack)
+        out << "google.load(\"visualization\", \"1\", {'packages':['#{pack}']});\n"
+        @packages << pack
+      end
+      
+      callback = "draw_rgviz_#{id}"
+      
+      # Set the callback if the function doesn't have params and if the
+      # user didn't request to hide the visualization
+      if !hidden && params.empty?
+        out << "google.setOnLoadCallback(#{callback});\n"
+      end
+      
+      # Define the visualization var and data
+      out << "var rgviz_#{id} = null;\n"
+      out << "var rgviz_#{id}_data = null;\n"
+      
+      # And define the callback
+      out << "function #{callback}(#{params.join(', ')}) {\n"
+        out << "var query = new google.visualization.Query('#{url}');\n"
+        out << "#{query_builder}\n"
+        out << "alert(#{query_builder_var});\n" if debug
+        out << "query.setQuery(#{query_builder_var});\n"
+        out << "query.send(function(response) {\n"
+          out << "rgviz_#{id} = new google.visualization.#{kind}(document.getElementById('#{id}'));\n"
+          events.each do |name, handler|
+            out << "google.visualization.events.addListener(rgviz_#{id}, '#{name}', #{handler});\n"
+          end
+          out << "rgviz_#{id}_data = response.getDataTable();\n"
+          out << "rgviz_#{id}.draw(rgviz_#{id}_data, #{opts});\n"
+        out << "});\n"
+      out << "}\n"
+      
+      out << "</script>\n"
+      
+      # Write the div
+      out << "<div id=\"#{id}\"></div>\n"
+      
+      out
+    end
+    
+    module_function :rgviz
+  end
+  
+  class MagicNamesVisitor < Visitor
+    def initialize(html_prefix, js_prefix, param_prefix)
+      @html_prefix = html_prefix
+      @js_prefix = js_prefix
+      @param_prefix = param_prefix
+      @s = ''
+      @params = []
+    end
+    
+    def query_builder
+      @s.strip
+    end
+    
+    def query_builder_var
+      'q'
+    end
+    
+    def params
+      @params
+    end
+    
+    def visit_query(node)
+      @s << "var e = null;\n"
+      @s << "var q = '"
+      node.select.accept self if node.select
+      node.where.accept self if node.where
+      node.group_by.accept self if node.group_by
+      node.pivot.accept self if node.pivot
+      node.order_by.accept self if node.order_by
+      @s << "limit #{node.limit} " if node.limit
+      @s << "offset #{node.offset} " if node.offset
+      if node.labels  
+        @s << "label "
+        node.labels.each_with_index do |l, i|
+          @s << ', ' if i > 0 
+          l.accept self
+        end
+      end 
+      if node.formats
+        @s << "format "
+        node.formats.each_with_index do |f, i|
+          @s << ', ' if i > 0 
+          f.accept self
+        end
+      end 
+      if node.options
+        @s << "options "
+        @s << "no_values " if node.options.no_values
+        @s << "no_format " if node.options.no_format
+      end
+      @s << "';\n"
+      false
+    end
+    
+    def visit_select(node)
+      @s << "select ";
+      print_columns node
+      @s << " "
+      false
+    end
+    
+    def visit_where(node)
+      @s << "where "
+      node.expression.accept self
+      @s << " "
+      false
+    end
+    
+    def visit_group_by(node)
+      @s << "group by "
+      print_columns node
+      @s << " "
+      false
+    end
+    
+    def visit_pivot(node)
+      @s << "pivot "
+      print_columns node
+      @s << " "
+      false
+    end
+    
+    def visit_order_by(node)
+      @s << "order_by "
+      node.sorts.each_with_index do |s, i|
+        @s << ', ' if i > 0
+        s.column.accept self
+        @s << ' '
+        @s << s.order
+      end
+      @s << " "
+      false
+    end
+    
+    def visit_label(node)
+      node.column.accept self
+      @s << ' '
+      if node.label.include?("'")
+        val = node.label.gsub("'", "\\'")
+        @s << "\"#{val}\""
+      else
+        @s << "\\'#{node.label}\\'"
+      end
+      false
+    end
+    
+    def visit_format(node)
+      node.column.accept self
+      @s << ' '
+      if node.pattern.include?("'")
+        @s << "\"#{node.pattern}\""
+      else
+        @s << "'#{node.pattern}'"
+      end
+      false
+    end
+    
+    def visit_binary_expression(node)
+      if node.operator == BinaryExpression::Eq 
+        source = has_magic_name?(node.right)
+        if source
+          @s << "';\n"
+          case source[:source]
+          when :html
+            @s << "e = document.getElementById('#{source[:id]}');\n"
+            @s << "if (e.tagName.toLowerCase() == 'select' && e.multiple) {\n"
+              @s << "var s = [];\n" 
+              @s << "var o = e.options;\n"
+              @s << "for(var i = 0; i < o.length; i++) {\n"
+                @s << "if (o[i].selected)\n";
+                  @s << "s.push(o[i].value);\n" 
+              @s << "}\n"
+              append_selections node, source
+            @s << "} else {\n"
+              @s << "q += '"
+              node.left.accept self
+              @s << " #{node.operator} "
+              @s << "';\n"
+              append_before_source_type source[:type]
+              @s << "e.value"
+              append_after_source_type source[:type]
+            @s << "}\n";
+          when :js
+            @s << "var s = #{source[:id]}();\n"
+            @s << "if (typeof(s) != 'object') s = [s];\n"
+            append_selections node, source
+          when :param
+            @s << "var s = param_#{source[:id]};\n"
+            @s << "if (typeof(s) != 'object') s = [s];\n"
+            append_selections node, source
+            @params << source[:id].to_i unless @params.include?(source[:id])
+          end
+          @s << "q += '"
+          return false
+        end
+      end
+      node.left.accept self
+      @s << " #{node.operator} "
+      node.right.accept self
+      false
+    end
+    
+    def visit_unary_expression(node)
+      if node.operator == UnaryExpression::Not
+        @s << "not "
+        node.operand.accept self
+      else
+        node.operand.accept self
+        @s << " #{node.operator}"
+      end
+      false
+    end
+
+    def visit_id_column(node)
+      source = get_source node.name
+      case source[:source]
+      when nil
+        @s << "`#{node.name}`"
+      when :html
+        @s << "';\n"
+        append_before_source_type source[:type]
+        @s << "document.getElementById('#{source[:id]}').value"
+        append_after_source_type source[:type]
+        @s << "q += '"
+      when :js
+        @s << "';\n"
+        append_before_source_type source[:type]
+        @s << "#{source[:id]}()"
+        append_after_source_type source[:type]
+        @s << "q += '"
+      when :param
+        @s << "';\n"
+        append_before_source_type source[:type]
+        @s << "param_#{source[:id]}"
+        append_after_source_type source[:type]
+        @s << "q += '"
+        @params << source[:id].to_i unless @params.include?(source[:id])
+      end
+    end
+    
+    def visit_number_column(node)
+      @s << node.value.to_s
+    end
+    
+    def visit_string_column(node)
+      @s << node.value
+    end
+    
+    def visit_boolean_column(node)
+      @s << node.value.to_s
+    end
+    
+    def visit_date_column(node)
+      @s << node.to_s
+    end
+    
+    def visit_date_time_column(node)
+      @s << node.to_s
+    end
+    
+    def visit_time_of_day_column(node)
+      @s << node.to_s
+    end
+    
+    def visit_scalar_function_column(node)
+      case node.function
+      when ScalarFunctionColumn::Sum, ScalarFunctionColumn::Difference, 
+           ScalarFunctionColumn::Product, ScalarFunctionColumn::Quotient
+        node.arguments[0].accept node
+        @s << " #{node.function} "
+        node.arguments[1].accept node
+      else
+        @s << "#{node.function}("
+        node.arguments.each_with_index do |a, i|
+          @s << ', ' if i > 0
+          a.accept self
+        end
+        @s << ")"
+      end
+      false
+    end
+    
+    def visit_aggregate_column(node)
+      @s << "#{node.function}("
+      node.argument.accept self
+      @s << ")"
+      false
+    end
+    
+    def print_columns(node)
+      node.columns.each_with_index do |c, i|
+        @s << ', ' if i > 0
+        c.accept self
+      end
+    end
+    
+    def get_source(name)
+      if name.start_with?(@html_prefix)
+        get_source_type :html, name[@html_prefix.length .. -1]
+      elsif name.start_with?(@js_prefix)
+        get_source_type :js, name[@js_prefix.length .. -1]
+      elsif name.start_with?(@param_prefix)
+        get_source_type :param, name[@param_prefix.length .. -1]
+      else
+        {}
+      end
+    end
+    
+    def get_source_type(source, name)
+      if name.start_with?('number_')
+        {:source => source, :id => name[7 .. -1], :type => :number}
+      elsif name.start_with?('string_')
+        {:source => source, :id => name[7 .. -1], :type => :string}
+      elsif name.start_with?('date_')
+        {:source => source, :id => name[5 .. -1], :type => :date}
+      elsif name.start_with?('datetime_')
+        {:source => source, :id => name[9 .. -1], :type => :datetime}
+      elsif name.start_with?('timeofday_')
+        {:source => source, :id => name[10 .. -1], :type => :timeofday}
+      else
+        {:source => source, :id => name, :type => :string}
+      end
+    end
+    
+    def append_before_source_type(type)
+      case type
+      when :number
+        @s << "q += "
+      when :string
+        @s << "q += \"'\" +"
+      when :date
+        @s << "q += \"date '\" + "
+      when :datetime
+        @s << "q += \"datetime '\" + "
+      when :timeofday
+        @s << "q += \"timeofday '\" + "
+      end
+    end
+    
+    def append_after_source_type(type)
+      case type
+      when :number
+        return
+      else
+        @s << " + \"'\";\n"
+      end
+    end
+    
+    def append_selections(node, source)
+      @s << "if (s.length == 0) {\n"
+        @s << "q += '1 = 2';\n";
+      @s << "} else {\n"
+        @s << "q += '(';\n"
+        @s << "for(var i = 0; i < s.length; i++) {\n";
+          @s << "if (i > 0) q += ' or ';\n"
+          @s << "q += '"
+          node.left.accept self
+          @s << " #{node.operator} "
+          @s << "';\n"
+          append_before_source_type source[:type]
+          @s << "s[i]"
+          append_after_source_type source[:type]
+        @s << "}";
+        @s << "q += ')';\n"
+      @s << "}\n"
+    end
+    
+    def has_magic_name?(node)
+      return false unless node.kind_of?(IdColumn)
+      source = get_source node.name
+      return false unless source[:source]
+      return source
+    end
+  end
+end

metadata

@@ -1,12 +1,12 @@
 --- !ruby/object:Gem::Specification 
 name: rgviz-rails
 version: !ruby/object:Gem::Version 
-  hash: 37
+  hash: 57
   prerelease: false
   segments: 
   - 0
-  - 23
-  version: "0.23"
+  - 25
+  version: "0.25"
 platform: ruby
 authors: 
 - Ary Borenszweig
@@ -29,6 +29,7 @@ extra_rdoc_files:
 files: 
 - lib/rgviz_rails.rb
 - lib/rgviz_rails/executor.rb
+- lib/rgviz_rails/view_helper.rb
 - lib/rgviz_rails/adapters/mysql_adapter.rb
 - lib/rgviz_rails/adapters/postgresql_adapter.rb
 - lib/rgviz_rails/adapters/sqlite_adapter.rb