require 'benchmark'
require 'neo4j/migrations/helpers/id_property'

module Neo4j
  class Migration
    def migrate
      fail 'not implemented'
    end

    def output(string = '')
      puts string unless !!ENV['MIGRATIONS_SILENCED']
    end

    def print_output(string)
      print string unless !!ENV['MIGRATIONS_SILENCED']
    end

    def default_path
      Rails.root if defined? Rails
    end

    def joined_path(path)
      File.join(path.to_s, 'db', 'neo4j-migrate')
    end

    def setup
      FileUtils.mkdir_p('db/neo4j-migrate')
    end

    def query(*args)
      Neo4j::ActiveBase.current_driver.query(*args)
    end

    class AddIdProperty < Neo4j::Migration
      include Neo4j::Migrations::Helpers::IdProperty

      attr_reader :models_filename

      def initialize(path = default_path)
        @models_filename = File.join(joined_path(path), 'add_id_property.yml')
      end

      def migrate
        ActiveSupport::Deprecation.warn '`AddIdProperty` task is deprecated and may be removed from future releases. '\
                                        'Create a new migration and use the `populate_id_property` helper.', caller
        models = ActiveSupport::HashWithIndifferentAccess.new(YAML.load_file(models_filename))[:models]
        output 'This task will add an ID Property every node in the given file.'
        output 'It may take a significant amount of time, please be patient.'
        models.each do |model|
          output
          output
          output "Adding IDs to #{model}"
          populate_id_property model
        end
      end

      def query(*args)
        ActiveBase.magic_query(*args)
      end

      def execute(*args)
        ActiveBase.query(*args)
      end

      def setup
        super
        return if File.file?(models_filename)

        File.open(models_filename, 'w') do |file|
          message = <<MESSAGE
# Provide models to which IDs should be added.
# # It will only modify nodes that do not have IDs. There is no danger of overwriting data.
# # models: [Student,Lesson,Teacher,Exam]\nmodels: []
MESSAGE
          file.write(message)
        end
      end

      # Might need some of this...
      # private

      # def add_ids_to(model)
      #  max_per_batch = (ENV['MAX_PER_BATCH'] || default_max_per_batch).to_i

      #  label = model.mapped_label_name
      #  last_time_taken = nil

      #  until (nodes_left = idless_count(label, model.primary_key)) == 0
      #    print_status(last_time_taken, max_per_batch, nodes_left)

      #    count = [nodes_left, max_per_batch].min
      #    last_time_taken = Benchmark.realtime do
      #      max_per_batch = id_batch_set(label, model.primary_key, Array.new(count) { new_id_for(model) }, count)
      #    end
      #  end
      # end

      # def idless_count(label, id_property)
      #  Neo4j::ActiveBase.new_query.match(n: label).where("NOT EXISTS(n.#{id_property})").pluck('COUNT(n) AS ids').first
      # end

      # def print_status(last_time_taken, max_per_batch, nodes_left)
      #  time_per_node = last_time_taken / max_per_batch if last_time_taken
      #  message = if time_per_node
      #              eta_seconds = (nodes_left * time_per_node).round
      #              "#{nodes_left} nodes left.  Last batch: #{(time_per_node * 1000.0).round(1)}ms / node (ETA: #{eta_seconds / 60} minutes)\r"
      #            else
      #              "Running first batch...\r"
      #            end

      #  print_output message
      # end


      # def id_batch_set(label, id_property, new_ids, count)
      #  tx = Neo4j::ActiveBase.new_transaction

      #  Neo4j::ActiveBase.current_driver.query("MATCH (n:`#{label}`) WHERE NOT EXISTS(n.#{id_property})
      #    with COLLECT(n) as nodes, #{new_ids} as ids
      #    FOREACH(i in range(0,#{count - 1})|
      #      FOREACH(node in [nodes[i]]|
      #        SET node.#{id_property} = ids[i]))
      #    RETURN distinct(true)
      #    LIMIT #{count}")

      #  count
      # rescue Neo4j::Server::CypherResponse::ResponseError, Faraday::TimeoutError
      #  new_max_per_batch = (max_per_batch * 0.8).round
      #  output "Error querying #{max_per_batch} nodes.  Trying #{new_max_per_batch}"
      #  new_max_per_batch
      # ensure
      #  tx.close
      # end

      # def default_max_per_batch
      #  900
      # end

      # def new_id_for(model)
      #  if model.id_property_info[:type][:auto]
      #    SecureRandom.uuid
      #  else
      #    model.new.send(model.id_property_info[:type][:on])
      #  end
      # end
    end
  end
end