class RailsAiAdmin::SchemaSuggestions

# File lib/rails_ai_admin/schema_suggestions.rb, line 30
def self.array?(column)
  column.respond_to?(:array) && column.array
end

# File lib/rails_ai_admin/schema_suggestions.rb, line 34
def initialize(model)
  @model = model
end

# File lib/rails_ai_admin/schema_suggestions.rb, line 25
def self.sensitive?(model, name)
  encrypted = model.respond_to?(:encrypted_attributes) ? model.encrypted_attributes : []
  name.to_s.match?(SENSITIVE) || Array(encrypted&.to_a).map(&:to_s).include?(name.to_s)
end

# File lib/rails_ai_admin/schema_suggestions.rb, line 38
def call
  return unavailable unless @model.is_a?(Class) && @model < ActiveRecord::Base
  columns = @model.columns
  foreign_keys = @model.reflect_on_all_associations(:belongs_to).flat_map { |association| Array(association.foreign_key) }.map(&:to_s)
  foreign_keys.concat(@model.reflect_on_all_associations(:belongs_to).filter_map { |association| association.foreign_type if association.polymorphic? })
  foreign_keys.concat(@model.connection.foreign_keys(@model.table_name).flat_map { |foreign_key| Array(foreign_key.column) }.map(&:to_s))
  # A declared non-polymorphic belongs_to may become a commented reference.
  references = @model.reflect_on_all_associations(:belongs_to).reject(&:polymorphic?).to_h { |association| [ association.foreign_key.to_s, association ] }
  omitted = 0
  suggestions = columns.filter_map do |column|
    name = column.name
    next if name == @model.primary_key || name == "id"
    if (reference = suggest_reference(references[name], column))
      next reference
    end
    if !name.match?(IDENTIFIER) || [ "lock_version", @model.inheritance_column ].include?(name) || name.end_with?("_id") || foreign_keys.include?(name) || self.class.sensitive?(@model, name) || self.class.array?(column)
      omitted += 1
      next
    end
    type = TYPES[column.type]
    unless type
      omitted += 1
      next
    end
    options = {}
    enum = @model.defined_enums[name]
    if enum && (!enum.keys.all? { |value| value.length <= 100 } || !enum.size.between?(1, 100))
      omitted += 1
      next
    end
    type = :email if type == :text && name.match?(EMAIL_NAME)
    type = :url if type == :text && name.match?(URL_NAME)
    if enum
      type = :enum
      options[:values] = enum.keys
    end
    options[:nullable] = false unless column.null
    options[:max_length] = column.limit if %i[text textarea email url].include?(type) && column.limit.is_a?(Integer) && column.limit.positive?
    # Explicit limits keep the declaration reviewable; coercion would
    # otherwise read the same values from the column on first use.
    if type == :decimal && column.precision.is_a?(Integer) && column.precision.between?(1, 1000)
      options[:precision] = column.precision
      options[:scale] = column.scale if column.scale.is_a?(Integer) && column.scale.between?(0, column.precision)
    end
    options[:readonly] = true if %w[created_at updated_at].include?(name)
    Suggestion.new(name, type, options.freeze)
  end
  Report.new(suggestions.freeze, omitted, true)
rescue ActiveRecord::ActiveRecordError
  unavailable
end