summaryrefslogtreecommitdiff
path: root/chromium/components/ntp_snippets/category.cc
blob: 1ef1bed938549c06555240582e287ce8e47406e9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/ntp_snippets/category.h"

#include "base/logging.h"

namespace ntp_snippets {

// static
Category Category::FromKnownCategory(KnownCategories known_category) {
  return FromIDValue(static_cast<int>(known_category));
}

// static
Category Category::FromRemoteCategory(int remote_category) {
  DCHECK_GT(remote_category, 0);
  return Category(static_cast<int>(KnownCategories::REMOTE_CATEGORIES_OFFSET) +
                  remote_category);
}

// static
Category Category::FromIDValue(int id) {
  DCHECK(IsValidIDValue(id)) << id
                             << " is not a valid category ID. This may have "
                                "been caused by removal of a local "
                                "KnownCategory.";
  return Category(id);
}

// static
bool Category::IsValidIDValue(int id) {
  return (id >= 0) &&
         ((id < static_cast<int>(KnownCategories::LOCAL_CATEGORIES_COUNT) ||
           id > static_cast<int>(KnownCategories::REMOTE_CATEGORIES_OFFSET)));
}

Category::Category(int id) : id_(id) {}

bool Category::IsKnownCategory(KnownCategories known_category) const {
  DCHECK_NE(known_category, KnownCategories::LOCAL_CATEGORIES_COUNT);
  DCHECK_NE(known_category, KnownCategories::REMOTE_CATEGORIES_OFFSET);
  return id_ == static_cast<int>(known_category);
}

bool operator==(const Category& left, const Category& right) {
  return left.id() == right.id();
}

bool operator!=(const Category& left, const Category& right) {
  return !(left == right);
}

bool Category::CompareByID::operator()(const Category& left,
                                       const Category& right) const {
  return left.id() < right.id();
}

std::ostream& operator<<(std::ostream& os, const Category& obj) {
  os << obj.id();
  return os;
}

}  // namespace ntp_snippets