overleaf/migrations/2_doc_lines_delete_from_project.coffee

184 lines
5.7 KiB
CoffeeScript

Settings = require "settings-sharelatex"
fs = require("fs")
mongojs = require("mongojs")
ObjectId = mongojs.ObjectId
db = mongojs(Settings.mongo.url, ['projects', 'docs'])
_ = require("lodash")
async = require("async")
exec = require("child_process").exec
finished_projects_path = "/tmp/finished-projects-2"
all_projects_path = "/tmp/all-projects-2"
unmigrated_docs_path = "/tmp/unmigrated-2"
printProgress = ->
exec "wc #{finished_projects_path}", (error, results) ->
setTimeout printProgress, 1000 * 30
checkIfFileHasBeenProccessed = (project_id, callback)->
exec "grep #{project_id} #{finished_projects_path}", (error, results) ->
hasBeenProcessed = _.include(results, project_id)
callback(error, hasBeenProcessed)
loadProjectIds = (callback)->
console.log "loading project ids from #{all_projects_path}"
fs.readFile all_projects_path, "utf-8", (err, data)->
ids = data.split("\n")
console.log "loaded #{ids.length} project ids from #{all_projects_path}"
callback err, ids
getAndWriteProjectids = (callback)->
console.log "finding all project id's - #{new Date().toString()}"
db.projects.find {}, {_id:1}, (err, ids)->
console.log "total found projects in mongo #{ids.length} - #{new Date().toString()}"
ids = _.pluck ids, '_id'
ids = _.filter ids, (id)-> id?
fileData = ids.join("\n")
fs.writeFile all_projects_path, fileData, ->
callback(err, ids)
markDocAsUnmigrated = (project_id, doc_id, callback)->
console.log "#{project_id} #{doc_id} unmigrated"
markProjectAsProcessed project_id, (err)->
fs.appendFile unmigrated_docs_path, "#{project_id} #{doc_id}\n", callback
markUnmigratedDocs = (project_id, docs, callback)->
console.log docs.length, project_id, "unmigrated"
jobs = _.map docs, (doc)->
(cb)->
markDocAsUnmigrated project_id, doc._id, cb
async.series jobs, callback
getProjectIds = (callback)->
exists = fs.existsSync all_projects_path
if exists
loadProjectIds callback
else
getAndWriteProjectids callback
markProjectAsProcessed = (project_id, callback)->
fs.appendFile finished_projects_path, "#{project_id}\n", callback
getAllDocs = (project_id, callback = (error, docs) ->) ->
excludes = {}
for i in [0..12]
excludes["rootFolder#{Array(i).join(".folders")}.docs.lines"] = 0
db.projects.findOne _id: ObjectId(project_id.toString()), excludes, (error, project) ->
return callback(error) if error?
if !project?
console.log "no such project #{project_id}"
return callback()
findAllDocsInProject project, (error, docs) ->
return callback(error) if error?
return callback null, docs, project
findAllDocsInProject = (project, callback = (error, docs) ->) ->
callback null, _findAllDocsInFolder project.rootFolder[0]
findDocInProject = (project, doc_id, callback = (error, doc, mongoPath) ->) ->
result = _findDocInFolder project.rootFolder[0], doc_id, "rootFolder.0"
if result?
callback null, result.doc, result.mongoPath
else
callback null, null, null
_findDocInFolder = (folder = {}, doc_id, currentPath) ->
for doc, i in folder.docs or []
if doc?._id? and doc._id.toString() == doc_id.toString()
return {
doc: doc
mongoPath: "#{currentPath}.docs.#{i}"
}
for childFolder, i in folder.folders or []
result = _findDocInFolder childFolder, doc_id, "#{currentPath}.folders.#{i}"
return result if result?
return null
_findAllDocsInFolder = (folder = {}) ->
docs = folder.docs or []
for childFolder in folder.folders or []
docs = docs.concat _findAllDocsInFolder childFolder
return docs
isDocInDocCollection = (doc, callback)->
if !doc?._id? or doc._id.length == 0
return callback(null, true)
db.docs.find({_id: ObjectId(doc._id+"")}, {_id: 1}).limit 1, (err, foundDocs)->
exists = foundDocs.length > 0
callback err, exists
getWhichDocsCanBeDeleted = (docs, callback = (err, docsToBeDeleted, unmigratedDocs)->)->
docsToBeDeleted = []
unmigratedDocs = []
jobs = _.map docs, (doc)->
return (cb)->
isDocInDocCollection doc, (err, exists)->
if exists
docsToBeDeleted.push doc
else
unmigratedDocs.push doc
cb(err)
async.series jobs, (err)->
callback err, docsToBeDeleted, unmigratedDocs
whipeDocLines = (project_id, mongoPath, callback)->
update =
$unset: {}
update.$unset["#{mongoPath}.lines"] = ""
update.$unset["#{mongoPath}.rev"] = ""
db.projects.update _id: ObjectId(project_id+''), update, callback
removeDocLinesFromProject = (docs, project, callback)->
jobs = _.map docs, (doc)->
(cb)->
findDocInProject project, doc._id, (err, doc, mongoPath)->
whipeDocLines project._id, mongoPath, cb
async.parallelLimit jobs, 5, callback
processNext = (project_id, callback)->
if !project_id? or project_id.length == 0
return callback()
checkIfFileHasBeenProccessed project_id, (err, hasBeenProcessed)->
if hasBeenProcessed
console.log "#{project_id} already procssed, skipping"
return callback()
console.log "#{project_id} processing"
getAllDocs project_id, (err, docs, project)->
if err?
console.error err, project_id, "could not get all docs"
return callback(err)
else
getWhichDocsCanBeDeleted docs, (err, docsToBeDeleted, unmigratedDocs)->
if err?
console.error err, project_id, "could not save docs into mongo"
return callback(err)
markUnmigratedDocs project_id, unmigratedDocs, (err)->
removeDocLinesFromProject docsToBeDeleted, project, (err)->
if err?
return callback(err)
markProjectAsProcessed project_id, (err)->
setTimeout(
-> callback(err)
,0)
exports.migrate = (client, done = ->)->
getProjectIds (err, ids)->
printProgress()
jobs = _.map ids, (id)->
return (cb)->
processNext(id, cb)
async.series jobs, (err)->
if err?
console.error err, "at end of jobs"
else
console.log "finished"
done(err)
exports.rollback = (next)->
next()