flame/utils/importBookmark.js

const parse = require('node-bookmarks-parser');
var sqlite3 = require('sqlite3');
const File = require('./File');
const databaseFilePath = 'data/db.sqlite';

let bookmarks = [];

function saveBookmarks() {
  const db = new sqlite3.Database(databaseFilePath);
  const importDate = new Date().toISOString();

  db.serialize(() => {
    db.all(`SELECT id, name FROM categories`, (err, data) => {
      if (err) {
      }

      for (const bookmark of bookmarks) {
        // Found in db.
        let stmt = db.prepare(
          'INSERT INTO bookmarks (name, url, categoryId, icon, createdAt, updatedAt) VALUES(?,?,?,?,?,?)'
        );
        stmt.run(
          bookmark.title,
          bookmark.url,
          data.find(
            (r) => r.name.toLowerCase() === bookmark.category.toLowerCase()
          )?.id,
          bookmark.icon,
          importDate,
          importDate,
          (err) => {
            if (err) {
              console.error(err);
            }
          }
        );
        stmt.finalize();
      }
    });
  });
}

function saveCategories() {
  return new Promise((resolve, reject) => {
    let db = new sqlite3.Database(databaseFilePath);
    let uniqueCats = [...new Set(bookmarks.map((r) => r.category))];
    const importDate = new Date().toString();

    let tasks = {};

    db.serialize(() => {
      db.all(`SELECT * FROM categories`, (err, data) => {
        if (err) {
          console.error(err);
          return;
        }

        for (const newCaterory of uniqueCats) {
          for (const category of data) {
            if (category.name.toLowerCase() === newCaterory.toLowerCase()) {
              continue;
            }
          }

          let stmt = db.prepare(
            'INSERT INTO categories (name, createdAt, updatedAt) VALUES(?,?,?)'
          );
          tasks[newCaterory] = false;
          stmt.run(newCaterory, importDate, importDate, (err) => {
            tasks[newCaterory] = true;

            if (err) {
              console.error(err);
              reject(err);
            }

            if(Object.keys(tasks).every(function(k){ return tasks[k] })){
              resolve();
            }
          });
          stmt.finalize();
        }
      });
    });
  });
}

function crawlBookmarks(bookmark, category) {
  if (bookmark.type === 'bookmark') {
    bookmarks.push({
      title: bookmark.title,
      url: bookmark.url,
      icon: bookmark.icon,
      category: category,
    });
  }

  if (bookmark.type === 'folder') {
    for (const child of bookmark.children) {
      crawlBookmarks(child, bookmark.title);
    }
  }
}

module.exports = function importBookmark(path) {
  const fileContent = new File(path).read();
  const bookmarkData = parse(fileContent);

  for (const bookmark of bookmarkData) {
    crawlBookmarks(bookmark);
  }

  saveCategories()
    .then((r) => {
      saveBookmarks();
    })
};