Delete removes all data now

This commit is contained in:
Clayton Wilson 2019-11-01 00:26:01 -04:00
parent 56323801aa
commit e8e69cec31

View File

@ -70,12 +70,14 @@ exports.signup = (req, res) => {
}) })
.then((idToken) => { .then((idToken) => {
token = idToken; token = idToken;
const defaultImageUrl = `https://firebasestorage.googleapis.com/v0/b/${config.storageBucket}/o/no-img.png?alt=media`;
const userCred = { const userCred = {
email: newUser.email, email: newUser.email,
handle: newUser.handle, handle: newUser.handle,
createdAt: newUser.createdAt, createdAt: newUser.createdAt,
userId, userId,
followedTopics: [] followedTopics: [],
imageUrl: defaultImageUrl
}; };
return db.doc(`/users/${newUser.handle}`).set(userCred); return db.doc(`/users/${newUser.handle}`).set(userCred);
}) })
@ -179,57 +181,76 @@ exports.login = (req, res) => {
} }
}; };
//Deletes user account //Deletes user account and all associated data
exports.deleteUser = (req, res) => { exports.deleteUser = (req, res) => {
var currentUser; // Get the profile image filename
firebase.auth().onAuthStateChanged(function(user) { // `https://firebasestorage.googleapis.com/v0/b/${config.storageBucket}/o/${imageFileName}?alt=media`
currentUser = user; const imageFileName = req.userData.imageUrl.split('/o/')[1].split('?alt=')[0];
if (currentUser) { const userId = req.userData.userId;
var post_query = db.collection("posts").where("userHandle", "==", req.user.handle); let errors = {};
post_query.get()
.then(function(myPosts) { function thenFunction(data) {
myPosts.forEach(function(doc) { console.log(`${data} data for ${req.userData.handle} has been deleted.`);
doc.ref.delete(); }
});
return; function catchFunction(data, err) {
console.error(err);
error[data] = err;
}
// Deletes user from authentication
let auth = admin.auth().deleteUser(userId);
// Deletes database data
let data = db.collection("users").doc(`${req.user.handle}`).delete();
// Deletes any custom profile image
let image;
if (imageFileName !== 'no-img.png') {
image = admin.storage().bucket().file(imageFileName).delete()
} else {
image = Promise.resolve();
}
// Deletes all users posts
let posts = db.collection("posts")
.where("userHandle", "==", req.user.handle)
.get()
.then((query) => {
query.forEach((snap) => {
snap.ref.delete();
}) })
.then(function() { })
res.status(200).send("Successfully removed all user's posts from database.");
return; let promises = [
}) auth
.catch(function(err) { .then(thenFunction('auth'))
res.status(500).send("Failed to remove all user's posts from database.", err); .catch((err) => catchFunction('auth', err)),
}); data
.then(thenFunction('data'))
.catch((err) => catchFunction('data', err)),
image
.then(thenFunction('image'))
.catch((err) => catchFunction('image', err)),
posts
.then(thenFunction('posts'))
.catch((err) => catchFunction('image', err))
];
// Wait for all promises to resolve
let waitPromise = Promise.all(promises);
db.collection("users").doc(`${req.user.handle}`).delete() waitPromise.then(() => {
.then(function() { if (Object.keys(errors) > 0) {
res.status(200).send("Sucessfully removed user from database."); return res.status(500).json(errors);
return; } else {
}) return res.status(200).json({message: `All data for ${req.userData.handle} has been deleted.`});
.catch(function(err) {
res.status(500).send("Failed to remove user from database.", err);
});
currentUser.delete()
.then(function() {
console.log("Successfully deleted user.");
res.status(200).send("Sucessfully deleted user.");
return;
})
.catch(function(err) {
console.log("Failed to delete user.", err);
res.status(500).send("Failed to delete user.");
});
} }
else { })
console.log("Failed to deleter user or cannot get user."); .catch((err) => {
res.status(500).send("Failed to deleter user or cannot get user."); return res.status(500).json({error: err});
} })
});
}; };
// Returns all data in the database for the user who is currently signed in // Returns all data in the database for the user who is currently signed in