require 'bundler/inline'

gemfile do
  source 'https://rubygems.org'
  gem 'mongoid', '9.0.8'
end

Mongoid.configure do |config|
  config.clients.default = {
    hosts: ['localhost:27017'],
    database: 'mongoid_test',
  }
end

class Post
  include Mongoid::Document

  has_many :comments, inverse_of: :post
  embeds_many :embedded_comments, inverse_of: :post

  accepts_nested_attributes_for :comments, allow_destroy: true
  accepts_nested_attributes_for :embedded_comments, allow_destroy: true

  field :title, type: String
end

class Comment
  include Mongoid::Document

  belongs_to :post, inverse_of: :comments

  field :title, type: String
end

class EmbeddedComment
  include Mongoid::Document

  embedded_in :post, inverse_of: :embedded_comments

  field :title, type: String
end

post = Post.create(title: 'Post')
comment = post.comments.create(title: 'Comment')
embedded_comment = post.embedded_comments.create(title: 'Embedded Comment')

puts 'Before:'
puts comment.reload.title
puts embedded_comment.reload.title
puts ''

post.assign_attributes(
  comments_attributes: [
    {
      _id: comment.id,
      title: 'Edited Comment',
    },
  ],
  embedded_comments_attributes: [
    {
      _id: embedded_comment.id,
      title: 'Edited Embedded Comment',
    },
  ],
)

puts 'After:'
puts comment.reload.title
puts embedded_comment.reload.title
puts ''