require 'set'
require 'active_support/concern'

module Protobuf
  module ActiveRecord
    module Columns
      extend ::ActiveSupport::Concern

      included do
        include ::Heredity

        inheritable_attributes :_protobuf_columns,
                               :_protobuf_column_types,
                               :_protobuf_mapped_columns
      end

      module ClassMethods
        def _protobuf_columns
          _protobuf_map_columns unless _protobuf_mapped_columns?

          @_protobuf_columns
        end

        def _protobuf_column_types
          _protobuf_map_columns unless _protobuf_mapped_columns?

          @_protobuf_column_types
        end

        # :nodoc:
        def _protobuf_date_column?(key)
          _protobuf_column_types.fetch(:date, false) && _protobuf_column_types[:date].include?(key)
        end

        # :nodoc:
        def _protobuf_datetime_column?(key)
          _protobuf_column_types.fetch(:datetime, false) && _protobuf_column_types[:datetime].include?(key)
        end

        # Map out the columns for future reference on type conversion
        # :nodoc:
        def _protobuf_map_columns(force = false)
          ::Thread.exclusive do
            @_protobuf_mapped_columns = false if force

            return unless table_exists?
            return if _protobuf_mapped_columns?

            @_protobuf_columns = {}
            @_protobuf_column_types = ::Hash.new { |h,k| h[k] = ::Set.new }

            columns.map do |column|
              @_protobuf_columns[column.name.to_sym] = column
              @_protobuf_column_types[column.type.to_sym] << column.name.to_sym
            end

            @_protobuf_mapped_columns = true
          end
        end

        def _protobuf_mapped_columns?
          @_protobuf_mapped_columns
        end

        # :nodoc:
        def _protobuf_time_column?(key)
          _protobuf_column_types.fetch(:time, false) && _protobuf_column_types[:time].include?(key)
        end

        # :nodoc:
        def _protobuf_timestamp_column?(key)
          _protobuf_column_types.fetch(:timestamp, false) && _protobuf_column_types[:timestamp].include?(key)
        end
      end
    end
  end
end