summaryrefslogtreecommitdiffstats
path: root/chrome/browser/browsing_data/autofill_counter.h
blob: 5072dd92f32dc816c4b7b6d31d7633ef6b1a5950 (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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
// Copyright 2015 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.

#ifndef CHROME_BROWSER_BROWSING_DATA_AUTOFILL_COUNTER_H_
#define CHROME_BROWSER_BROWSING_DATA_AUTOFILL_COUNTER_H_

#include "base/macros.h"
#include "base/threading/thread_checker.h"
#include "base/time/time.h"
#include "chrome/browser/browsing_data/browsing_data_counter.h"
#include "components/webdata/common/web_data_service_consumer.h"

namespace autofill {
class AutofillWebDataService;
}

class AutofillCounter: public BrowsingDataCounter,
                       public WebDataServiceConsumer {
 public:
  class AutofillResult : public FinishedResult {
   public:
    AutofillResult(const AutofillCounter* source,
                   ResultInt num_suggestions,
                   ResultInt num_credit_cards,
                   ResultInt num_addresses);
    ~AutofillResult() override;

    ResultInt num_credit_cards() const { return num_credit_cards_; }
    ResultInt num_addresses() const { return num_addresses_; }

   private:
    ResultInt num_credit_cards_;
    ResultInt num_addresses_;

    DISALLOW_COPY_AND_ASSIGN(AutofillResult);
  };

  AutofillCounter();
  ~AutofillCounter() override;

  // BrowsingDataCounter implementation.
  void OnInitialized() override;
  const std::string& GetPrefName() const override;

  // Whether the counting is in progress.
  bool HasPendingQuery() {
    return suggestions_query_ || credit_cards_query_ || addresses_query_;
  }

  // Set the beginning of the time period for testing. AutofillTable does not
  // allow us to set time explicitly, and BrowsingDataCounter recognizes
  // only predefined time periods, out of which the lowest one is one hour.
  // Obviously, the test cannot run that long.
  // TODO(msramek): Consider changing BrowsingDataCounter to use arbitrary
  // time periods instead of BrowsingDataRemover::TimePeriod.
  void SetPeriodStartForTesting(const base::Time& period_start_for_testing);

 private:
  const std::string pref_name_;
  base::ThreadChecker thread_checker_;

  scoped_refptr<autofill::AutofillWebDataService> web_data_service_;

  WebDataServiceBase::Handle suggestions_query_;
  WebDataServiceBase::Handle credit_cards_query_;
  WebDataServiceBase::Handle addresses_query_;

  ResultInt num_suggestions_;
  ResultInt num_credit_cards_;
  ResultInt num_addresses_;

  base::Time period_start_for_testing_;

  void Count() override;

  // WebDataServiceConsumer implementation.
  void OnWebDataServiceRequestDone(WebDataServiceBase::Handle handle,
                                   const WDTypedResult* result) override;

  // Cancel all pending requests to AutofillWebdataService.
  void CancelAllRequests();

  DISALLOW_COPY_AND_ASSIGN(AutofillCounter);
};

#endif  // CHROME_BROWSER_BROWSING_DATA_AUTOFILL_COUNTER_H_