require 'active_support/inflector'
module ConceptQL
  module Behaviors
    module Dottable
      @@counter = 0
      TYPE_COLORS = {
        person: 'blue',
        visit_occurrence: 'orange',
        condition_occurrence: 'red',
        procedure_occurrence: 'green3',
        procedure_cost: 'gold',
        death: 'brown',
        payer_plan_period: 'blue',
        drug_exposure: 'purple',
        observation: 'magenta',
        misc: 'black'
      }
      def node_name
        @__node_name ||= self.class.name.split('::').last.underscore.gsub(/\W/, '_').downcase + '_' + (@@counter += 1).to_s
      end

      def display_name
        @__display_name ||= begin
          output = self.class.name.split('::').last.titleize
          output += ": #{arguments.join(', ')}" unless arguments.empty?
          if output.length > 100
            parts = output.split
            output = parts.each_slice(output.length / parts.count).map do |subparts|
              subparts.join(' ')
            end.join ('\n')
          end
          output += "\n#{options.map{|k,v| "#{k}: #{v}"}.join("\n")}" unless options.nil? || options.empty?
          output
        end
      end

      def type_color(*types)
        types.flatten!
        types.length == 1 ? TYPE_COLORS[types.first] || 'black' : 'black'
      end

      def graph_node(g)
        @__graph_node ||= begin
          me = g.add_nodes(node_name)
          me[:label] = display_name
          me[:color] = type_color(types)
          me[:shape] = shape if respond_to?(:shape)
          me
        end
      end

      def link_to(g, dest_node, db = nil)
        edge_options = {}

        types.each do |type|
          if db
            my_n = my_n(db, type)
            label = [' rows=' + my_count(db, type).to_s + ' ']
            label << ' n=' + my_n.to_s + ' '
            edge_options[:label] = label.join("\n")
            edge_options[:style] = 'dashed' if my_n.zero?
          end
          e = g.add_edges(graph_node(g), dest_node, edge_options)
          e[:color] = type_color(type)
        end
      end

      def graph_it(g, db)
        graph_prep(db) if respond_to?(:graph_prep)
        children.each do |child|
          child.graph_it(g, db)
        end
        graph_node(g)
        children.each do |child|
          child.link_to(g, graph_node(g), db)
        end
      end

      def my_count(db, type)
        evaluate(db).from_self.where(criterion_type: type.to_s).count
      end

      def my_n(db, type)
        evaluate(db).from_self.where(criterion_type: type.to_s).select_group(:person_id).count
      end
    end
  end
end