#    Copyright 2020 Couchbase, Inc.
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.

require "couchbase"

include Couchbase # rubocop:disable Style/MixinUsage for brevity

options = Cluster::ClusterOptions.new
options.authenticate("Administrator", "password")
cluster = Cluster.connect("couchbase://localhost", options)

bucket_name = "default"
# create index definition, if it does not exist already
search_index_name = "knob_search"
begin
  cluster.search_indexes.get_index(search_index_name)
rescue Error::IndexNotFound
  index = Management::SearchIndex.new
  index.type = "fulltext-index"
  index.name = search_index_name
  index.source_type = "couchbase"
  index.source_name = bucket_name
  index.params = {
    mapping: {
      types: {
        "knob" => {
          properties: {
            "name" => {
              fields: [
                {
                  name: "name",
                  type: "text",
                  include_in_all: true,
                  include_term_vectors: true,
                  index: true,
                  store: true,
                  docvalues: true,
                },
              ],
            },
          },
        },
      },
    },
  }

  cluster.search_indexes.upsert_index(index)

  num_indexed = 0
  loop do
    sleep(1)
    num = cluster.search_indexes.get_indexed_documents_count(search_index_name)
    break if num_indexed == num

    num_indexed = num
    puts "indexing #{search_index_name.inspect}: #{num_indexed} documents"
  end
end

collection = cluster.bucket(bucket_name).default_collection

# The application need to to know exactly which mutation affect the result,
# and supply mutation tokens from those operations.
random_string = ("a".."z").to_a.sample(10).join
res = collection.upsert("user:#{random_string}", {
  "name" => "Brass Doorknob",
  "email" => "brass.doorknob@example.com",
  "data" => random_string,
})

state = MutationState.new(res.mutation_token)
# state.add(*tokens) could be used to add more tokens

query = Cluster::SearchQuery.term("doorknob")
options = Cluster::SearchOptions.new
options.timeout = 10_000
options.consistent_with(state)
res = cluster.search_query(search_index_name, query, options)

res.rows.each do |row|
  puts "--- Found our newly created document!" if row.id == "user:#{random_string}"
  if ENV["REMOVE_DOOR_KNOBS"]
    puts "Removing #{row.id} (requested via env)"
    collection.remove(row.id)
  end
end