MyGuide/db-cli.js at main · Miamiohlibs/MyGuide · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
/* Command-line interface for favorites */
/*
Usage:
  node db-cli.js add <userId> <favType> <favId>
  node db-cli.js remove <userId> <favType> <favId>
  node db-cli.js delete <userId>
  node db-cli.js empty

  favType: g, db, s
*/

const mongoose = require('mongoose');
const db = require('./helpers/database');
require('./models/userFavorites/UserFavorites');
const Crud = mongoose.model('userFavorites');
// const Crud = require('./models/userFavorites/crudUserFavorites');
const cmd = process.argv.slice(2)[0];
const arg = process.argv.slice(2)[1];
const arg2 = process.argv.slice(2)[2];
const arg3 = process.argv.slice(2)[3];

(async () => {
  await db.connect();
  await DbCmd(cmd);
  console.log(await Crud.find());
  //   await DbCmd(cmd);
  //   const allFavorites = await Favorite.find();
  //   console.log(allFavorites);
  await db.disconnect();
})();

async function DbCmd(cmd) {
  switch (cmd) {
    case 'add':
      return await UpdateUserAdd(arg, arg2, arg3);
      break;
    case 'remove':
      return await UpdateUserRemove(arg, arg2, arg3);
      break;
    case 'delete':
      return await DeleteUser(arg);
      break;
    case 'empty':
      return await Crud.deleteMany({});
      break;
  }
  return;
}

function BuildUpdateAttr(favType, favId) {
  let attr = {};
  switch (favType) {
    case 'g':
      attr['favoriteGuides'] = favId;
      break;
    case 'db':
      attr['favoriteDatabases'] = favId;
      break;
    case 's':
      attr['favoriteSubjects'] = favId;
      break;
  }
  return attr;
}

async function DeleteUser(userId) {
  return await Crud.deleteOne({ userId: userId });
}
async function UpdateUserAdd(userId, favType, favId) {
  let attr = BuildUpdateAttr(favType, favId);
  return await Crud.findOneAndUpdate(
    { userId: userId },
    { $addToSet: attr, updated: Date.now() },
    { upsert: true }
  );
}
async function UpdateUserRemove(userId, favType, favId) {
  let attr = BuildUpdateAttr(favType, favId);
  return await Crud.findOneAndUpdate(
    { userId: userId },
    { $pull: attr, updated: Date.now() },
    { upsert: true }
  );
}