src/background/completion/impl/TabRepositoryImpl.ts


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

import TabRepository, { Tab } from "../TabRepository";

const COMPLETION_ITEM_LIMIT = 10;

export default class TabRepositoryImpl implements TabRepository {
  async queryTabs(query: string, excludePinned: boolean): Promise<Tab[]> {
    const tabs = await browser.tabs.query({ currentWindow: true });
    return tabs
      .filter((t) => {
        return (
          (t.url && t.url.toLowerCase().includes(query.toLowerCase())) ||
          (t.title && t.title.toLowerCase().includes(query.toLowerCase()))
        );
      })
      .filter((t) => {
        return !(excludePinned && t.pinned);
      })
      .filter((item) => item.id && item.title && item.url)
      .slice(0, COMPLETION_ITEM_LIMIT)
      .map(TabRepositoryImpl.toEntity);
  }

  async getAllTabs(excludePinned: boolean): Promise<Tab[]> {
    if (excludePinned) {
      return (
        await browser.tabs.query({ currentWindow: true, pinned: true })
      ).map(TabRepositoryImpl.toEntity);
    }
    return (await browser.tabs.query({ currentWindow: true })).map(
      TabRepositoryImpl.toEntity
    );
  }

  private static toEntity(tab: browser.tabs.Tab): Tab {
    return {
      id: tab.id!!,
      url: tab.url!!,
      active: tab.active,
      title: tab.title!!,
      faviconUrl: tab.favIconUrl,
      index: tab.index,
    };
  }
}