forked from ForeseTech/OAuthKeeper2.0
-
Notifications
You must be signed in to change notification settings - Fork 0
/
seeder.js
52 lines (44 loc) · 1.17 KB
/
seeder.js
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
const fs = require('fs');
const mongoose = require('mongoose');
const colors = require('colors');
const dotenv = require('dotenv').config();
// Load models
const Contact = require('./models/Contact');
const User = require('./models/User');
// Connect to DB
mongoose.connect(process.env.MONGO_URI, {
useNewUrlParser: true,
useCreateIndex: true,
useUnifiedTopology: true,
useFindAndModify: false,
});
// Read JSON files
const users = JSON.parse(fs.readFileSync(`${__dirname}/resources/_data/users.json`, 'utf-8'));
const contacts = JSON.parse(fs.readFileSync(`${__dirname}/resources/_data/contacts.json`, 'utf-8'));
// Import into DB
const importData = async () => {
try {
await User.create(users);
await Contact.create(contacts);
console.log('Data Imported...'.green);
process.exit();
} catch (err) {
console.log(err);
}
};
// Delete Data
const deleteData = async () => {
try {
await User.deleteMany();
await Contact.deleteMany();
console.log('Data Destroyed...'.red);
process.exit();
} catch (err) {
console.log(err);
}
};
if (process.argv[2] === '-i') {
importData();
} else if (process.argv[2] === '-d') {
deleteData();
}