overleaf/services/contacts/prepopulate.coffee

50 lines
No EOL
1.9 KiB
CoffeeScript

# Usage: coffee preprocess.coffee projects.json done.csv
# where projects.json is the output of
# mongoexport <CREDENTIALS> --db sharelatex-staging --collection projects --type=json --fields owner_ref,collaberator_refs,readOnly_refs --query '{ $or: [{collaberator_refs: { $not : {$size: 0} }}, {readOnly_refs: { $not: {$size: 0}}}]}'
fs = require "fs"
projects = fs.readFileSync(process.argv[2]).toString()
projects = projects.split("\n").filter((p) -> p!="").map (p) -> JSON.parse(p)
contact_pairs = []
for project in projects
project_id = project._id.$oid
owner_id = project.owner_ref.$oid
contact_ids = project.collaberator_refs.concat(project.readOnly_refs).map (r) -> r.$oid
for contact_id in contact_ids
contact_pairs.push [project_id, owner_id, contact_id]
# Done list is a list of pairs owner_id:contact_id
DONE_FILE = process.argv[3]
done_list = fs.readFileSync(DONE_FILE).toString()
done_contacts = {}
for done_pair in done_list.split("\n")
done_contacts[done_pair] = true
workers = []
for contact_pair in contact_pairs
do (contact_pair) ->
workers.push (cb) ->
if done_contacts[contact_pair.join(":")]
console.log "ALREADY DONE", contact_pair.join(":"), "SKIPPING"
cb()
else
[project_id, owner_id, contact_id] = contact_pair
console.log "PINGING CONTACT API (OWNER: #{owner_id}, CONTACT: #{contact_id})..."
require("request").post {
url: "http://localhost:3036/user/#{owner_id}/contacts"
json: { contact_id }
}, (error, response, body) ->
return cb(error) if error?
if response.statusCode != 204
return cb(new Error("bad status code: #{response.statusCode}"))
console.log "DONE, WRITING TO DONE FILE..."
fs.appendFile DONE_FILE, contact_pair.join(":") + "\n", (error) ->
return cb(error) if error?
console.log "WRITTEN"
cb()
require("async").series workers, (error) ->
console.error error if error?
console.log "DONE"