Boostnote/browser/main/lib/dataApi/migrateFromV5Storage.js

const _ = require('lodash')
const keygen = require('browser/lib/keygen')
const resolveStorageData = require('./resolveStorageData')
const consts = require('browser/lib/consts')
const CSON = require('@rokt33r/season')
const path = require('path')
const sander = require('sander')

function migrateFromV5Storage (storageKey, data) {
  let targetStorage
  try {
    const cachedStorageList = JSON.parse(localStorage.getItem('storages'))
    if (!_.isArray(cachedStorageList)) throw new Error('Target storage doesn\'t exist.')

    targetStorage = _.find(cachedStorageList, {key: storageKey})
    if (targetStorage == null) throw new Error('Target storage doesn\'t exist.')
  } catch (e) {
    return Promise.reject(e)
  }
  return resolveStorageData(targetStorage)
    .then(function (storage) {
      return importAll(storage, data)
    })
}

function importAll (storage, data) {
  const oldArticles = data.articles
  const notes = []
  data.folders
    .forEach(function (oldFolder) {
      let folderKey = keygen()
      while (storage.folders.some((folder) => folder.key === folderKey)) {
        folderKey = keygen()
      }
      const newFolder = {
        key: folderKey,
        name: oldFolder.name,
        color: consts.FOLDER_COLORS[Math.floor(Math.random() * 7) % 7]
      }

      storage.folders.push(newFolder)

      const articles = oldArticles.filter((article) => article.FolderKey === oldFolder.key)
      articles.forEach((article) => {
        let noteKey = keygen()
        let isUnique = false
        while (!isUnique) {
          try {
            sander.statSync(path.join(storage.path, 'notes', noteKey + '.cson'))
            noteKey = keygen()
          } catch (err) {
            if (err.code === 'ENOENT') {
              isUnique = true
            } else {
              console.error('Failed to read `notes` directory.')
              throw err
            }
          }
        }

        if (article.mode === 'markdown') {
          const newNote = {
            tags: article.tags,
            createdAt: article.createdAt,
            updatedAt: article.updatedAt,
            folder: folderKey,
            storage: storage.key,
            type: 'MARKDOWN_NOTE',
            isStarred: false,
            title: article.title,
            content: '# ' + article.title + '\n\n' + article.content,
            key: noteKey,
            linesHighlighted: article.linesHighlighted
          }
          notes.push(newNote)
        } else {
          const newNote = {
            tags: article.tags,
            createdAt: article.createdAt,
            updatedAt: article.updatedAt,
            folder: folderKey,
            storage: storage.key,
            type: 'SNIPPET_NOTE',
            isStarred: false,
            title: article.title,
            description: article.title,
            key: noteKey,
            snippets: [{
              name: article.mode,
              mode: article.mode,
              content: article.content,
              linesHighlighted: article.linesHighlighted
            }]
          }
          notes.push(newNote)
        }
      })
    })

  notes.forEach(function (note) {
    CSON.writeFileSync(path.join(storage.path, 'notes', note.key + '.cson'), _.omit(note, ['storage', 'key']))
  })

  CSON.writeFileSync(path.join(storage.path, 'boostnote.json'), _.pick(storage, ['version', 'folders']))

  return {
    storage,
    notes
  }
}

module.exports = migrateFromV5Storage