aboutsummaryrefslogtreecommitdiff
path: root/src/background/completion/impl/HistoryRepositoryImpl.ts
blob: 3bf064e10be7aeff230439ce190d8e0b30f6bf08 (plain) (blame)
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
import * as filters from "./filters";
import HistoryRepository, { HistoryItem } from "../HistoryRepository";
import PrefetchAndCache from "./PrefetchAndCache";

const COMPLETION_ITEM_LIMIT = 10;

export default class CachedHistoryRepository implements HistoryRepository {
  private historyCache: PrefetchAndCache<browser.history.HistoryItem>;

  constructor() {
    this.historyCache = new PrefetchAndCache(this.getter, this.filter, 10);
  }

  async queryHistories(keywords: string): Promise<HistoryItem[]> {
    const items = await this.historyCache.get(keywords);

    const filterOrKeep = <T>(
      source: T[],
      filter: (items: T[]) => T[],
      min: number
    ): T[] => {
      const filtered = filter(source);
      if (filtered.length < min) {
        return source;
      }
      return filtered;
    };

    return [items]
      .map((items) =>
        filterOrKeep(items, filters.filterByPathname, COMPLETION_ITEM_LIMIT)
      )
      .map((items) =>
        filterOrKeep(items, filters.filterByOrigin, COMPLETION_ITEM_LIMIT)
      )[0]
      .sort((x, y) => Number(y.visitCount) - Number(x.visitCount))
      .slice(0, COMPLETION_ITEM_LIMIT)
      .map((item) => ({
        title: item.title!,
        url: item.url!,
      }));
  }

  private async getter(
    keywords: string
  ): Promise<browser.history.HistoryItem[]> {
    const items = await browser.history.search({
      text: keywords,
      startTime: 0,
    });

    return [items]
      .map(filters.filterBlankTitle)
      .map(filters.filterHttp)
      .map(filters.filterByTailingSlash)[0];
  }

  private filter(items: browser.history.HistoryItem[], query: string) {
    return items.filter((item) => {
      return query.split(" ").every((keyword) => {
        return (
          item.title!.toLowerCase().includes(keyword.toLowerCase()) ||
          item.url!.includes(keyword)
        );
      });
    });
  }
}