/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

const TEST_URI = "http://mozilla.com/";
const TEST_SUBDOMAIN_URI = "http://foobar.mozilla.com/";

async function checkEmptyHistory() {
  let db = await PlacesUtils.promiseDBConnection();
  let rows = await db.executeCached("SELECT count(*) FROM moz_historyvisits");
  return !rows[0].getResultByIndex(0);
}

add_task(async function test_addPage() {
  await PlacesTestUtils.addVisits(TEST_URI);
  Assert.ok(!(await checkEmptyHistory()), "History has entries");
});

add_task(async function test_removePage() {
  await PlacesUtils.history.remove(TEST_URI);
  Assert.ok(await checkEmptyHistory(), "History is empty");
});

add_task(async function test_removePages() {
  let pages = [];
  for (let i = 0; i < 8; i++) {
    pages.push(TEST_URI + i);
  }

  await PlacesTestUtils.addVisits(pages.map(uri => ({ uri })));
  // Bookmarked item should not be removed from moz_places.
  const ANNO_INDEX = 1;
  const ANNO_NAME = "testAnno";
  const ANNO_VALUE = "foo";
  const BOOKMARK_INDEX = 2;
  await PlacesUtils.history.update({
    url: pages[ANNO_INDEX],
    annotations: new Map([[ANNO_NAME, ANNO_VALUE]]),
  });
  await PlacesUtils.bookmarks.insert({
    parentGuid: PlacesUtils.bookmarks.unfiledGuid,
    url: pages[BOOKMARK_INDEX],
    title: "test bookmark",
  });
  await PlacesUtils.history.update({
    url: pages[BOOKMARK_INDEX],
    annotations: new Map([[ANNO_NAME, ANNO_VALUE]]),
  });

  await PlacesUtils.history.remove(pages);
  Assert.ok(await checkEmptyHistory(), "History is empty");

  // Check that the bookmark and its annotation still exist.
  let folder = await PlacesUtils.getFolderContents(
    PlacesUtils.bookmarks.unfiledGuid
  );
  Assert.equal(folder.root.childCount, 1);
  let pageInfo = await PlacesUtils.history.fetch(pages[BOOKMARK_INDEX], {
    includeAnnotations: true,
  });
  Assert.equal(pageInfo.annotations.get(ANNO_NAME), ANNO_VALUE);

  // Check the annotation on the non-bookmarked page does not exist anymore.
  await assertNoOrphanPageAnnotations();

  // Cleanup.
  await PlacesUtils.bookmarks.eraseEverything();
  await PlacesUtils.history.clear();
});

add_task(async function test_removePagesByTimeframe() {
  let visits = [];
  let startDate = (Date.now() - 10000) * 1000;
  for (let i = 0; i < 10; i++) {
    visits.push({
      uri: TEST_URI + i,
      visitDate: startDate + i * 1000,
    });
  }

  await PlacesTestUtils.addVisits(visits);

  // Delete all pages except the first and the last.
  await PlacesUtils.history.removeByFilter({
    beginDate: PlacesUtils.toDate(startDate + 1000),
    endDate: PlacesUtils.toDate(startDate + 8000),
  });

  // Check that we have removed the correct pages.
  for (let i = 0; i < 10; i++) {
    Assert.equal((await page_in_database(TEST_URI + i)) == 0, i > 0 && i < 9);
  }

  // Clear remaining items and check that all pages have been removed.
  await PlacesUtils.history.removeByFilter({
    beginDate: PlacesUtils.toDate(startDate),
    endDate: PlacesUtils.toDate(startDate + 9000),
  });
  Assert.ok(await checkEmptyHistory(), "History is empty");
});

add_task(async function test_removePagesFromHost() {
  await PlacesTestUtils.addVisits(TEST_URI);
  await PlacesUtils.history.removeByFilter({ host: ".mozilla.com" });
  Assert.ok(await checkEmptyHistory(), "History is empty");
});

add_task(async function test_removePagesFromHost_keepSubdomains() {
  await PlacesTestUtils.addVisits([
    { uri: TEST_URI },
    { uri: TEST_SUBDOMAIN_URI },
  ]);
  await PlacesUtils.history.removeByFilter({ host: "mozilla.com" });
  Assert.ok(!(await checkEmptyHistory()), "History has entries");
});

add_task(async function test_history_clear() {
  await PlacesUtils.history.clear();
  Assert.ok(await checkEmptyHistory(), "History is empty");
});
