Initial commit

app/models/carto/analysis_node.rb

@@ -0,0 +1,100 @@
+require_dependency 'carto/query_rewriter'
+
+class Carto::AnalysisNode
+  include Carto::QueryRewriter
+
+  def initialize(definition)
+    @definition = definition
+  end
+
+  attr_reader :definition
+
+  def self.find_by_natural_id(visualization_id, natural_id)
+    analyses = Carto::Analysis.where(visualization_id: visualization_id).all
+    analyses.lazy.map { |analysis| analysis.analysis_node.find_by_id(natural_id) }.find(&:present?)
+  end
+
+  def id
+    definition[:id]
+  end
+
+  def type
+    definition[:type]
+  end
+
+  def params
+    definition[:params]
+  end
+
+  def non_child_params
+    param_locations = children_and_location.keys.select { |cal| cal.first == :params }.map(&:second)
+    params && params.reject { |key| param_locations.include?(key) }
+  end
+
+  def options
+    definition[:options] ||= Hash.new
+  end
+
+  def children
+    children_and_location.values
+  end
+
+  def children_and_location
+    @children ||= get_children(@definition)
+  end
+
+  def find_by_id(node_id)
+    return self if node_id == id
+    children.lazy.map { |child| child.find_by_id(node_id) }.find { |child| child }
+  end
+
+  def source?
+    type == 'source'
+  end
+
+  def table_source?(table_name)
+    # Maybe check params[:query]
+    source? && options && options[:table_name] == table_name
+  end
+
+  def source_descendants
+    return [self] if source?
+    children.map(&:source_descendants).flatten
+  end
+
+  def descendants
+    [self] + children.map(&:descendants).flatten
+  end
+
+  def fix_analysis_node_queries(old_username, new_user, renamed_tables)
+    if options && options.key?(:table_name)
+      old_table_name = options[:table_name]
+      old_username, old_table_name = old_table_name.split('.') if old_table_name.include?('.')
+      options[:table_name] = renamed_tables.fetch(old_table_name, old_table_name)
+    end
+
+    if params && old_username
+      query = params[:query]
+      params[:query] = rewrite_query(query, old_username, new_user, renamed_tables) if query.present?
+    end
+
+    children.each { |child| child.fix_analysis_node_queries(old_username, new_user, renamed_tables) }
+  end
+
+  private
+
+  MANDATORY_KEYS_FOR_ANALYSIS_NODE = [:id, :type, :params].freeze
+  def get_children(definition, path = [])
+    children = definition.map do |k, v|
+      if v.is_a?(Hash)
+        this_path = path + [k]
+        if (MANDATORY_KEYS_FOR_ANALYSIS_NODE - v.keys).empty?
+          { this_path => Carto::AnalysisNode.new(v) }
+        else
+          get_children(v, this_path)
+        end
+      end
+    end
+    children.flatten.compact.reduce({}, :merge)
+  end
+end