gem 'amber_odm'

require 'amber_odm'

AmberODM.databases_settings[:dev] = {
  url: 'http://localhost:9200',
  log: true
}

require 'amber_odm'

# Define a model for the "users" index
class User < AmberODM::Document
  
  attr_accessor :name, :email, :posts, :keywords_count

  def self.index_name
    :users
  end
  
  def self.db_name
    :dev
  end
  
  def self.posts=(posts)
    @posts = posts.map { |post| Post.new(post)}
  end
  
  class Post < AmberODM::AttrInitializer
    attr_accessor :id, :title, :body, :created_at, :updated_at
    
    def created_at
      Time.parse(@created_at)
    end

    def updated_at
      Time.parse(@updated_at)
    end
    
    def keywords
      body.scan(/\w+/)
    end
  end
  
end

# Find users using a query
users = User.search(
  {
    bool: {
      must: [
        {match: {name: 'John Doe'}}
      ]
    }
  }, # filter query is required
  _source: %w[name email posts keywords_count], # optional, the default is to return all fields defined in the document
  size: 10, # optional, the default is to return all documents
)

# users is an array of User objects like Array<User>
bulk_users = []
users.each do |user|
  posts = user.posts
  all_user_keywords = posts.map(&:keywords).flatten.uniq
  user.keywords_count = all_user_keywords.count
  bulk_hash = user.get_bulk_update_hash(:keywords_count) # returns a hash that can be used to update the document field ':keywords_count' in bulk

  bulk_users << bulk_hash
end

# Update the documents in bulk
User.client.bulk(body: bulk_users) unless bulk_users.empty?

# 1. Get the first page of users
users = User.search(
  { bool: { must: [{match: {name: 'John Doe'}}] } },
  size: 10,
  sort: [{name: 'asc'}],
)

# 2. Get the last sort value
search_after = users.last&.sort

# 3. Loop through the pages
while users.any?
  users = User.search(
    { bool: { must: [{match: {name: 'John Doe'}}] } },
    size: 10,
    sort: [{name: 'asc'}],
    search_after: search_after
  )
  
  # 4. Do something with the users documents...
  puts users.map(&:name).join(', ')
  
  # 5. Update the last sort value
  search_after = users.last&.sort
end

User.client # returns the Elasticsearch client from 'elasticsearch-ruby' gem