require "sqlite3"
require "active_record"

module RspecProfiling
  module Collectors
    class SQL
      def self.install
        new.install
      end

      def self.uninstall
        new.uninstall
      end

      def self.reset
        new.results.destroy_all
      end

      def initialize
        RspecProfiling.config.db_path ||= 'tmp/rspec_profiling'
        establish_connection
      end

      def install
        return if prepared?

        connection.create_table(table) do |t|
          t.string    :branch, index: true
          t.string    :commit_hash, index: true
          t.datetime  :date, index: true
          t.text      :file, index: true
          t.integer   :line_number, index: true
          t.text      :description
          t.decimal   :time, index: true
          t.string    :status, index: true
          t.text      :exception
          t.integer   :query_count, index: true
          t.decimal   :query_time, index:true
          t.integer   :request_count, index: true
          t.decimal   :request_time, index: true
          t.timestamps null: true
        end
      end

      def uninstall
        connection.drop_table(table)
      end

      def insert(attributes)
        results.create!(attributes.except(:created_at, :events, :event_counts, :event_times, :event_events))
      end

      def results
        @results ||= begin
          establish_connection

          Result.table_name = table
          Result.attr_protected if Result.respond_to?(:attr_protected)
          Result
        end
      end

      private

      def prepared?
        connection.table_exists?(table)
      end

      def connection
        @connection ||= results.connection
      end

      def establish_connection
        Result.establish_connection(
          :adapter  => 'sqlite3',
          :database => database
        )
      end

      def table
        RspecProfiling.config.table_name
      end

      def database
        RspecProfiling.config.db_path
      end

      class Result < ActiveRecord::Base
        def to_s
          [description, location].join(" - ")
        end

        private

        def location
          [file, line_number].join(":")
        end
      end
    end
  end
end