4 files changed, 235 insertions, 0 deletions
diff --git a/src/background/shared/completions/bookmarks.js b/src/background/shared/completions/bookmarks.js
new file mode 100644
index 0000000..1adb350
--- /dev/null
+++ b/src/background/shared/completions/bookmarks.js
@@ -0,0 +1,15 @@
+const getCompletions = (keywords) => {
+  return browser.bookmarks.search({ query: keywords }).then((items) => {
+    return items.filter((item) => {
+      let url = undefined;
+      try {
+        url = new URL(item.url);
+      } catch (e) {
+        return false;
+      }
+      return item.type === 'bookmark' && url.protocol !== 'place:';
+    }).slice(0, 10);
+  });
+};
+
+export { getCompletions };
diff --git a/src/background/shared/completions/histories.js b/src/background/shared/completions/histories.js
new file mode 100644
index 0000000..a7d3d47
--- /dev/null
+++ b/src/background/shared/completions/histories.js
@@ -0,0 +1,83 @@
+const filterHttp = (items) => {
+  const httpsHosts = items
+    .filter(item => item[1].protocol === 'https:')
+    .map(item => item[1].host);
+  const httpsHostSet = new Set(httpsHosts);
+  return items.filter(
+    item => !(item[1].protocol === 'http:' && httpsHostSet.has(item[1].host))
+  );
+};
+
+const filterEmptyTitle = (items) => {
+  return items.filter(item => item[0].title && item[0].title !== '');
+};
+
+const filterClosedPath = (items) => {
+  const allSimplePaths = items
+    .filter(item => item[1].hash === '' && item[1].search === '')
+    .map(item => item[1].origin + item[1].pathname);
+  const allSimplePathSet = new Set(allSimplePaths);
+  return items.filter(
+    item => !(item[1].hash === '' && item[1].search === '' &&
+      (/\/$/).test(item[1].pathname) &&
+      allSimplePathSet.has(
+        (item[1].origin + item[1].pathname).replace(/\/$/, '')
+      )
+    )
+  );
+};
+
+const reduceByPathname = (items, min) => {
+  let hash = {};
+  for (let item of items) {
+    let pathname = item[1].origin + item[1].pathname;
+    if (!hash[pathname]) {
+      hash[pathname] = item;
+    } else if (hash[pathname][1].href.length > item[1].href.length) {
+      hash[pathname] = item;
+    }
+  }
+  let filtered = Object.values(hash);
+  if (filtered.length < min) {
+    return items;
+  }
+  return filtered;
+};
+
+const reduceByOrigin = (items, min) => {
+  let hash = {};
+  for (let item of items) {
+    let origin = item[1].origin;
+    if (!hash[origin]) {
+      hash[origin] = item;
+    } else if (hash[origin][1].href.length > item[1].href.length) {
+      hash[origin] = item;
+    }
+  }
+  let filtered = Object.values(hash);
+  if (filtered.length < min) {
+    return items;
+  }
+  return filtered;
+};
+
+const getCompletions = (keyword) => {
+  return browser.history.search({
+    text: keyword,
+    startTime: 0,
+  }).then((historyItems) => {
+    return [historyItems.map(item => [item, new URL(item.url)])]
+      .map(filterEmptyTitle)
+      .map(filterHttp)
+      .map(filterClosedPath)
+      .map(items => reduceByPathname(items, 10))
+      .map(items => reduceByOrigin(items, 10))
+      .map(items => items
+        .sort((x, y) => x[0].visitCount < y[0].visitCount)
+        .slice(0, 10)
+        .map(item => item[0])
+      )[0];
+  });
+};
+
+export { getCompletions };
diff --git a/src/background/shared/completions/index.js b/src/background/shared/completions/index.js
new file mode 100644
index 0000000..728cee7
--- /dev/null
+++ b/src/background/shared/completions/index.js
@@ -0,0 +1,129 @@
+import * as tabs from './tabs';
+import * as histories from './histories';
+import * as bookmarks from './bookmarks';
+
+const getSearchCompletions = (command, keywords, searchConfig) => {
+  let engineNames = Object.keys(searchConfig.engines);
+  let engineItems = engineNames.filter(name => name.startsWith(keywords))
+    .map(name => ({
+      caption: name,
+      content: command + ' ' + name
+    }));
+  return Promise.resolve(engineItems);
+};
+
+const getHistoryCompletions = (command, keywords) => {
+  return histories.getCompletions(keywords).then((pages) => {
+    return pages.map((page) => {
+      return {
+        caption: page.title,
+        content: command + ' ' + page.url,
+        url: page.url
+      };
+    });
+  });
+};
+
+const getBookmarksCompletions = (command, keywords) => {
+  return bookmarks.getCompletions(keywords).then((items) => {
+    return items.map((item) => {
+      return {
+        caption: item.title,
+        content: command + ' ' + item.url,
+        url: item.url,
+      };
+    });
+  });
+};
+
+const getOpenCompletions = (command, keywords, searchConfig) => {
+  return Promise.all([
+    getSearchCompletions(command, keywords, searchConfig),
+    getHistoryCompletions(command, keywords),
+    getBookmarksCompletions(command, keywords),
+  ]).then(([engineItems, historyItems, bookmarkItems]) => {
+    let completions = [];
+    if (engineItems.length > 0) {
+      completions.push({
+        name: 'Search Engines',
+        items: engineItems
+      });
+    }
+    if (historyItems.length > 0) {
+      completions.push({
+        name: 'History',
+        items: historyItems
+      });
+    }
+    if (bookmarkItems.length > 0) {
+      completions.push({
+        name: 'Bookmarks',
+        items: bookmarkItems
+      });
+    }
+    return completions;
+  });
+};
+
+const getBufferCompletions = (command, keywords, excludePinned) => {
+  return tabs.getCompletions(keywords, excludePinned).then((got) => {
+    let items = got.map((tab) => {
+      return {
+        caption: tab.title,
+        content: command + ' ' + tab.title,
+        url: tab.url,
+        icon: tab.favIconUrl
+      };
+    });
+    return [
+      {
+        name: 'Buffers',
+        items: items
+      }
+    ];
+  });
+};
+
+const getCompletions = (line, settings) => {
+  let typedWords = line.trim().split(/ +/);
+  let typing = '';
+  if (!line.endsWith(' ')) {
+    typing = typedWords.pop();
+  }
+
+  if (typedWords.length === 0) {
+    return Promise.resolve([]);
+  }
+  let name = typedWords.shift();
+  let keywords = typedWords.concat(typing).join(' ');
+
+  switch (name) {
+  case 'o':
+  case 'open':
+  case 't':
+  case 'tabopen':
+  case 'w':
+  case 'winopen':
+    return getOpenCompletions(name, keywords, settings.search);
+  case 'b':
+  case 'buffer':
+    return getBufferCompletions(name, keywords, false);
+  case 'bd!':
+  case 'bdel!':
+  case 'bdelete!':
+  case 'bdeletes!':
+    return getBufferCompletions(name, keywords, false);
+  case 'bd':
+  case 'bdel':
+  case 'bdelete':
+  case 'bdeletes':
+    return getBufferCompletions(name, keywords, true);
+  }
+  return Promise.resolve([]);
+};
+
+const complete = (line, settings) => {
+  return getCompletions(line, settings);
+};
+
+export { complete };
diff --git a/src/background/shared/completions/tabs.js b/src/background/shared/completions/tabs.js
new file mode 100644
index 0000000..bdb2741
--- /dev/null
+++ b/src/background/shared/completions/tabs.js
@@ -0,0 +1,8 @@
+import * as tabs from '../tabs';
+
+const getCompletions = (keyword, excludePinned) => {
+  return tabs.queryByKeyword(keyword, excludePinned);
+};
+
+
+export { getCompletions };