/*
 * Copyright 2016 Google Inc. All Rights Reserved.
 *
 * 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.
 */

#include "cobalt/browser/storage_upgrade_handler.h"

#include <vector>

#include "base/logging.h"
#include "base/memory/ref_counted.h"
#include "cobalt/dom/local_storage_database.h"
#include "cobalt/dom/storage_area.h"
#include "cobalt/network/persistent_cookie_store.h"
#include "cobalt/storage/storage_manager.h"
#include "cobalt/storage/upgrade/upgrade_reader.h"
#include "net/cookies/canonical_cookie.h"

namespace cobalt {
namespace browser {
namespace {
void OnCookiesLoaded(const std::vector<net::CanonicalCookie*>& cookies) {
  // We don't care about the local copies of the cookies returned to us.
  for (size_t i = 0; i < cookies.size(); i++) {
    delete cookies[i];
  }
}
}  // namespace

StorageUpgradeHandler::StorageUpgradeHandler(const GURL& url)
    : default_local_storage_id_(
          dom::StorageArea::GetLocalStorageIdForUrl(url)) {}

void StorageUpgradeHandler::OnUpgrade(storage::StorageManager* storage,
                                      const char* data, int size) {
  storage::upgrade::UpgradeReader upgrade_reader(data, size);
  int num_cookies = upgrade_reader.GetNumCookies();
  int num_local_storage_entries = upgrade_reader.GetNumLocalStorageEntries();
  DLOG(INFO) << "Upgrading legacy save data: " << num_cookies << " cookies, "
             << num_local_storage_entries << " local storage entries.";

  if (num_cookies > 0) {
    scoped_refptr<network::PersistentCookieStore> cookie_store(
        new network::PersistentCookieStore(storage));
    // Load the current cookies to ensure the database table is initialized.
    cookie_store->Load(base::Bind(OnCookiesLoaded));
    for (int i = 0; i < num_cookies; i++) {
      const net::CanonicalCookie* cookie = upgrade_reader.GetCookie(i);
      DCHECK(cookie);
      cookie_store->AddCookie(*cookie);
    }
  }

  if (num_local_storage_entries > 0) {
    dom::LocalStorageDatabase local_storage_database(storage);
    for (int i = 0; i < num_local_storage_entries; i++) {
      const storage::upgrade::UpgradeReader::LocalStorageEntry*
          local_storage_entry = upgrade_reader.GetLocalStorageEntry(i);
      DCHECK(local_storage_entry);
      local_storage_database.Write(default_local_storage_id_,
                                   local_storage_entry->key,
                                   local_storage_entry->value);
    }
  }
}

}  // namespace browser
}  // namespace cobalt