// Copyright 2024 The Pigweed Authors
//
// 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
//
//     https://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.

// DOCSTAG: [pw_allocator-examples-pmr]
#include <map>
#include <string_view>

#include "pw_allocator/allocator.h"
#include "pw_allocator/as_pmr_allocator.h"

namespace examples {

class LibraryIndex {
 public:
  using StringType = ::std::pmr::string;
  using MapType = ::std::pmr::multimap<StringType, StringType>;
  using Iterator = typename MapType::iterator;

  LibraryIndex(pw::allocator::Allocator& allocator)
      : allocator_(allocator), by_author_(allocator_) {}

  void AddBook(std::string_view title, std::string_view author) {
    by_author_.emplace(author, title);
  }

  std::pair<Iterator, Iterator> FindByAuthor(std::string_view author) {
    return by_author_.equal_range(StringType(author, allocator_));
  }

 private:
  pw::allocator::AsPmrAllocator allocator_;
  MapType by_author_;
};

}  // namespace examples
// DOCSTAG: [pw_allocator-examples-pmr]

// Unit tests.
#include "pw_allocator/testing.h"
#include "pw_unit_test/framework.h"

namespace {

using AllocatorForTest = ::pw::allocator::test::AllocatorForTest<4096>;

TEST(PmrExample, FindBookByAuthor) {
  AllocatorForTest allocator;
  examples::LibraryIndex books(allocator);

  // Books which sold over 60M copies.
  // inclusive-language: disable
  books.AddBook("A Tale of Two Cities", "Charles Dickens");
  books.AddBook("Le Petit Prince", "Antoine de Saint-Exupery");
  books.AddBook("O Alquimista", "Paulo Coelho");
  books.AddBook("Harry Potter and the Philosopher's Stone", "J. K. Rowling");
  books.AddBook("And Then There Were None", "Agatha Christie");
  books.AddBook("Dream of the Red Chamber", "Cao Xueqin");
  books.AddBook("The Hobbit", "J. R. R. Tolkien");
  books.AddBook("She: A History of Adventure", "H. Rider Haggard");
  books.AddBook("The Da Vinci Code", "Dan Brown");
  books.AddBook("Harry Potter and the Chamber of Secrets", "J. K. Rowling");
  books.AddBook("Harry Potter and the Prisoner of Azkaban", "J. K. Rowling");
  books.AddBook("Harry Potter and the Goblet of Fire", "J. K. Rowling");
  books.AddBook("Harry Potter and the Order of the Phoenix", "J. K. Rowling");
  books.AddBook("Harry Potter and the Half-Blood Prince", "J. K. Rowling");
  books.AddBook("Harry Potter and the Deathly Hallows", "J. K. Rowling");
  books.AddBook("The Catcher in the Rye", "J. D. Salinger");
  // inclusive-language: enable

  examples::LibraryIndex::Iterator begin, end;
  std::tie(begin, end) = books.FindByAuthor("J. K. Rowling");
  EXPECT_EQ(std::distance(begin, end), 7);
}

}  // namespace