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

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

function migrateFromV5Storage (storagePath) {
  var boostnoteJSONPath = path.join(storagePath, 'boostnote.json')
  return Promise.resolve()
    .then(function readBoostnoteJSON () {
      return sander.readFile(boostnoteJSONPath, {
        encoding: 'utf-8'
      })
    })
    .then(function verifyVersion (rawData) {
      var boostnoteJSONData = JSON.parse(rawData)
      if (boostnoteJSONData.version === '1.0') throw new Error('Target storage seems to be transformed already.')
      if (!_.isArray(boostnoteJSONData.folders)) throw new Error('the value of folders is not an array.')

      return boostnoteJSONData
    })
    .then(function setVersion (boostnoteJSONData) {
      boostnoteJSONData.version = '1.0'
      return sander.writeFile(boostnoteJSONPath, JSON.stringify(boostnoteJSONData))
        .then(() => boostnoteJSONData)
    })
    .then(function fetchNotes (boostnoteJSONData) {
      var fetchNotesFromEachFolder = boostnoteJSONData.folders
        .map(function (folder) {
          const folderDataJSONPath = path.join(storagePath, folder.key, 'data.json')
          return sander
            .readFile(folderDataJSONPath, {
              encoding: 'utf-8'
            })
            .then(function (rawData) {
              var data = JSON.parse(rawData)
              if (!_.isArray(data.notes)) throw new Error('value of notes is not an array.')
              return data.notes
                .map(function setFolderToNote (note) {
                  note.folder = folder.key
                  return note
                })
            })
            .catch(function failedToReadDataJSON (err) {
              console.warn('Failed to fetch notes from ', folderDataJSONPath, err)
              return []
            })
        })

      return Promise.all(fetchNotesFromEachFolder)
        .then(function flatten (folderNotes) {
          return folderNotes
            .reduce(function concatNotes (sum, notes) {
              return sum.concat(notes)
            }, [])
        })
        .then(function saveNotes (notes) {
          notes.forEach(function renewKey (note) {
            var newKey = keygen()
            while (notes.some((_note) => _note.key === newKey)) {
              newKey = keygen()
            }
            note.key = newKey
          })

          const noteDirPath = path.join(storagePath, 'notes')
          notes
            .map(function saveNote (note) {
              CSON.writeFileSync(path.join(noteDirPath, note.key) + '.cson', note)
            })
          return true
        })
        .then(function deleteFolderDir (check) {
          if (check) {
            boostnoteJSONData.folders.forEach((folder) => {
              sander.rimrafSync(path.join(storagePath, folder.key))
            })
          }
          return check
        })
    })
    .catch(function handleError (err) {
      console.warn(err)
      return false
    })
}

module.exports = migrateFromV5Storage