arvis-chrome-workflow/src/fetchBookmark.js

Chrome kernel based broswer workflow for Arvis

const arvish = require('arvish');
const psl = require('psl');
const _ = require('lodash');
require('./init.js');
const { parseArgv } = require('./argHandler');
const {
  extractHostname,
  existsAsync,
  getHistoryDB,
  bookmarkDFS,
  getExecPath,
  getChromeBookmark
} = require('./utils');
const path = require('path');
const sep = path.sep;

const conf = arvish.config.get('setting');

(async function () {
  const bookmarkRoot = await getChromeBookmark();
  const { options, input } = parseArgv(process.argv);

  let bookmarks;

  if (options['folderId']) {
    const folders = bookmarkDFS(bookmarkRoot, { targets: ['folder'] });
    const targetFolder = _.filter(
      folders,
      (item) => item.id == options['folderId']
    )[0];
    bookmarks = bookmarkDFS(targetFolder.children, {
      targets: ['url'],
      depth: 1,
    });
  } else {
    bookmarks = bookmarkDFS(bookmarkRoot);
  }

  if (input !== '') {
    bookmarks = bookmarks.filter((item) => {
      const name = item.name.toLowerCase();
      const url = item.url.toLowerCase();
      const loweredInput = input.normalize().toLowerCase();

      if (name.includes(loweredInput) || url.includes(loweredInput)) {
        return true;
      }
      return false;
    });
  }

  const result = await Promise.all(
    _.map(bookmarks, async (item) => {
      const hostname = psl.get(extractHostname(item.url));
      const ret = {
        title: item.name,
        subtitle: item.url,
        arg: item.url,
        quicklookurl: item.url
      };

      (await existsAsync(`cache${sep}${hostname}.png`)) &&
        (ret.icon = {
          path: `cache${sep}${hostname}.png`,
        });

      return ret;
    })
  );

  if (conf.chb.sort === 'VISIT_FREQ') {
    const historyDB = getHistoryDB();
    let visitHistorys = historyDB.prepare('SELECT url FROM urls').all();

    const freqs = {};
    for (const history of visitHistorys) {
      const key = history.url;
      if (freqs[key]) {
        ++freqs[key];
      } else {
        freqs[key] = 1;
      }
    }

    result.sort((a, b) => {
      const key1 = a.subtitle;
      const key2 = b.subtitle;

      if (freqs[key1] && freqs[key2]) return freqs[key2] - freqs[key1];
      else if (freqs[key1]) return -1;
      else if (freqs[key2]) return 1;
      else a.title > b.title ? 1 : -1;
    });
  } else {
    result.sort((a, b) => (a.title > b.title ? 1 : -1));
  }

  if (result.length === 0) {
    result.push({
      valid: true,
      title: 'No bookmarks were found.',
      autocomplete: 'No bookmarks were found.',
      subtitle: '',
      text: {
        copy: 'No bookmarks were found.',
        largetype: 'No bookmarks were found.',
      },
    });
  } else {
    result.splice(0, 0, {
      valid: true,
      arg: 'chrome://bookmarks/',
      title: `${result.length} bookmarks were found.`,
    });
  }

  if (options['folderId']) {
    result.splice(0, 0, {
      title: 'Back',
      arg: 'back',
      icon: {
        path: `${getExecPath()}${sep}assets${sep}back-button.png`
      }
    });
  }

  arvish.output(result);
})();