/*
  This is a JavaScript script for mongodb.
  It constructs a new version of the Artworks collection of the
  moma database from the one created by mongoimport and the
  Artworks.json file downloaded from the Museum of Modern Art:

  https://github.com/MuseumofModernArt/collection

  It makes the following changes:

  The auto-generated _id field is replaced by the ObjectID field

  The parallel arrays representing Artist, ConstituentID,
  etc. are replaced by an array of embedded documents
*/
conn = new Mongo("mongodb.cs.oberlin.edu")
db = conn.getDB("moma")
db1 = conn.getDB("moma1")
db1.Artworks.drop()
db1.createCollection("Artworks")
cursor = db.Artworks.find()
var n = 0
var docs = []
while(cursor.hasNext()){
    doc = cursor.next()
    doc._id = doc.ObjectID

    artistArray = []
    for(i=0; i<doc.Artist.length; i++) {
	artistArray.push({ "DisplayName" : doc.Artist[i] })
    }
    for(i=0; i<doc.ConstituentID.length; i++) {
	artistArray[i]["ConstituentID"] = doc.ConstituentID[i]
    }
    for(i=0; i<doc.ArtistBio.length; i++) {
	artistArray[i]["ArtistBio"] = doc.ArtistBio[i]
    }
    for(i=0; i<doc.Nationality.length; i++) {
	artistArray[i]["Nationality"] = doc.Nationality[i]
    }
    for(i=0; i<doc.BeginDate.length; i++) {
	artistArray[i]["BeginDate"] = doc.BeginDate[i]
    }
    for(i=0; i<doc.EndDate.length; i++) {
	artistArray[i]["EndDate"] = doc.EndDate[i]
    }
    for(i=0; i<doc.Gender.length; i++) {
	artistArray[i]["Gender"] = doc.Gender[i]
    }
    doc.Artist = artistArray
    delete doc.ConstituentID
    delete doc.ArtistBio
    delete doc.Nationality
    delete doc.BeginDate
    delete doc.EndDate
    delete doc.Gender
    docs.push(doc)
}
print("calling insertMany")
db1.Artworks.insertMany(docs)