class RubyConversations::MessagePrompt

def attributes
  base_attributes.merge(message_inputs_attributes_hash)
end

def attributes_for_api
  {
    id: id,
    prompt_version_id: prompt_version_id,
    name: name,
    role: role,
    content: content,
    metadata: metadata,
    draft: draft,
    ai_message_inputs_attributes: message_inputs.map(&:attributes_for_api)
  }.compact
end

def base_attributes
  identity_attributes.merge(content_attributes).merge(timestamp_attributes)
end

def content_attributes
  {
    'name' => name,
    'role' => role,
    'content' => content,
    'metadata' => metadata,
    'draft' => draft
  }
end

def extract_key_attributes(attributes, key)
  attributes.delete(key) || attributes.delete(key.to_s) || attributes.delete(key.to_sym) || []
end

def extract_nested_attributes!(attributes, key)
  nested = extract_key_attributes(attributes, key)
  nested_attributes = extract_key_attributes(attributes, "#{key}_attributes")
  nested.concat(nested_attributes)
end

def identity_attributes
  {
    'id' => id,
    'message_id' => message_id,
    'prompt_version_id' => prompt_version_id,
    'message' => message
  }
end

def initialize(attributes = {})
  @message_inputs = [] # Initialize inputs array
  # Extract nested inputs before super
  inputs_attributes = extract_nested_attributes!(attributes, :message_inputs)
  super
  initialize_message_inputs(inputs_attributes)
end

def initialize_message_inputs(attributes_array)
  (attributes_array || []).each do |attrs|
    next if attrs.blank?
    @message_inputs << RubyConversations::MessageInput.new(attrs)
  end
end

def message_inputs_attributes=(attributes)
  @message_inputs ||= []
  attributes.each do |attrs|
    @message_inputs << RubyConversations::MessageInput.new(attrs)
  end
end

def message_inputs_attributes_hash
  {
    'message_inputs' => message_inputs.map(&:attributes)
  }
end

def timestamp_attributes
  {
    'created_at' => created_at,
    'updated_at' => updated_at
  }
end