summaryrefslogtreecommitdiffstats
path: root/chrome/browser/history/history_publisher_win.cc
blob: ef4fb7f7e3e30467cd19bd3c53a857d75f7aab1f (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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
// Copyright (c) 2008-2009 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 "chrome/browser/history/history_publisher.h"

#include <atlsafe.h>
#include <objbase.h>
#include <oleauto.h>
#include <wtypes.h>

#include "base/registry.h"
#include "base/scoped_comptr_win.h"

namespace history {

const wchar_t* const HistoryPublisher::kRegKeyRegisteredIndexersInfo =
    L"Software\\Google\\Google Chrome\\IndexerPlugins";

// static
double HistoryPublisher::TimeToUTCVariantTime(const base::Time& time) {
  double var_time = 0;
  if (!time.is_null()) {
    base::Time::Exploded exploded;
    time.UTCExplode(&exploded);

    // Create the system time struct representing our exploded time.
    SYSTEMTIME system_time;
    system_time.wYear = exploded.year;
    system_time.wMonth = exploded.month;
    system_time.wDayOfWeek = exploded.day_of_week;
    system_time.wDay = exploded.day_of_month;
    system_time.wHour = exploded.hour;
    system_time.wMinute = exploded.minute;
    system_time.wSecond = exploded.second;
    system_time.wMilliseconds = exploded.millisecond;
    SystemTimeToVariantTime(&system_time, &var_time);
  }

  return var_time;
}

HistoryPublisher::HistoryPublisher() {
  CoInitialize(NULL);
}

HistoryPublisher::~HistoryPublisher() {
  CoUninitialize();
}

bool HistoryPublisher::Init() {
  return ReadRegisteredIndexersFromRegistry();
}

bool HistoryPublisher::ReadRegisteredIndexersFromRegistry() {
  RegistryKeyIterator iter(HKEY_CURRENT_USER, kRegKeyRegisteredIndexersInfo);
  while (iter.Valid()) {
    // The subkey name is the GUID of the Indexer COM object which implements
    // the IChromeHistoryIndexer interface. We shall store that and use it to
    // send historical data to the indexer.
    CLSID clsid;
    CLSIDFromString(static_cast<LPOLESTR>(
        const_cast<TCHAR*>(iter.Name())), &clsid);
    ScopedComPtr<IChromeHistoryIndexer> indexer;
    HRESULT hr = CoCreateInstance(clsid, NULL, CLSCTX_INPROC,
                                  __uuidof(IChromeHistoryIndexer),
                                  reinterpret_cast<void**>(&indexer));
    if (SUCCEEDED(hr) && indexer != NULL)
      indexers_.push_back(indexer);
    ++iter;
  }
  return indexers_.size() > 0;
}

void HistoryPublisher::PublishDataToIndexers(const PageData& page_data)
    const {
  double var_time = TimeToUTCVariantTime(page_data.time);

  CComSafeArray<unsigned char> thumbnail_arr;
  if (page_data.thumbnail) {
    for(size_t i = 0; i < page_data.thumbnail->size(); ++i)
      thumbnail_arr.Add((*page_data.thumbnail)[i]);
  }

  // Send data to registered indexers.
  for(size_t i = 0; i < indexers_.size(); ++i) {
    indexers_[i]->SendPageData(
        CComVariant(var_time, VT_DATE),
        CComBSTR(page_data.url.spec().c_str()),
        CComBSTR(page_data.html),
        CComBSTR(page_data.title),
        CComBSTR(page_data.thumbnail_format),
        CComVariant(thumbnail_arr.m_psa));
  }
}

void HistoryPublisher::DeleteUserHistoryBetween(const base::Time& begin_time,
                                                const base::Time& end_time)
    const {
  double var_begin_time = TimeToUTCVariantTime(begin_time);
  double var_end_time = TimeToUTCVariantTime(end_time);
  for(size_t i = 0; i < indexers_.size(); ++i) {
    indexers_[i]->DeleteUserHistoryBetween(CComVariant(var_begin_time, VT_DATE),
                                           CComVariant(var_end_time, VT_DATE));
  }
}

}  // namespace history