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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
const mariadb = require("mariadb");
const bcrypt = require("bcrypt");
const nanoid = require("nanoid");
const fs = require("fs");
class DB {
constructor() {
this.pool = mariadb.createPool({
host: process.env.DBHost,
user: process.env.DBUser,
password: process.env.DBPassword,
database: process.env.DBName,
});
this.init();
}
connect() {
return this.pool.getConnection();
}
init() {
fs.readFile(__dirname + "/tables.sql", "utf8", async (err, data) => {
if (err) throw err;
const queries = data.split(";");
queries.pop();
for (const query of queries) await this.query(query);
console.log("Tables created!");
const res = await this.query("SELECT * FROM users");
if (res.length === 0) this.initValues();
});
}
initValues() {
fs.readFile(__dirname + "/names.csv", "utf8", async (err, data) => {
if (err) throw err;
await this.query("INSERT INTO types (name) VALUES ('teacher'), ('pupil')");
await this.query(
"INSERT INTO class (name) VALUES ('TGM13.1'), ('TGM13.2'), ('TGTM13.1'), ('TGI13.1'), ('TGI13.2')"
);
fs.readFile(__dirname + "/poll.txt", "utf8", (err, data) => {
if (err) throw err;
const questions = data.split("\n");
questions.forEach((question) => {
if (question.length > 0)
this.query("INSERT INTO ranking_questions (question, type_id) VALUE (?,?)", [question, 2]);
});
});
fs.readFile(__dirname + "/mottos.txt", "utf8", (err, data) => {
if (err) throw err;
const mottos = data.split("\n");
mottos.forEach(async (motto) => {
const [name, desc] = motto.split(" - ");
if (motto)
await this.query("INSERT INTO mottos (name, description) VALUES (?, ?)", [name, desc]);
});
});
const classes = data.split("--");
const userPasswords = {};
console.log("Generating users");
for (const [classIndex, clazz] of classes.entries()) {
const students = clazz.split("\n");
userPasswords[classIndex] = [];
// students.forEach(async (student) => {
for (const student of students) {
// console.log(".");
// Fix undefined
if (student && student.length > 3) {
const [_, surname, name] = student.split(",");
const names = name.split(" ");
const middlename = names.length > 1 && names[1] ? names.slice(1).join(" ") : null;
let username = surname.toLowerCase().slice(0, 6);
if (middlename) username += middlename[0].toLowerCase();
username += names[0].toLowerCase().slice(0, 2);
const pwd = nanoid.nanoid(8);
const password = await bcrypt.hash(pwd, 10);
userPasswords[classIndex].push({ username, pwd });
await this.query(
"INSERT INTO users (username, name, middlename, surname, password, class_id, type_id) VALUE (?,?,?,?,?,?,?)",
[username, names[0].replace("\r", ""), middlename, surname, password, classIndex + 1, 2]
);
}
}
}
fs.writeFile(__dirname + "/users.json", JSON.stringify(userPasswords), (err) => {
if (err) console.error(err);
});
console.log("Initialized users!");
});
}
async query(query, params) {
const conn = await this.connect();
try {
return await conn.query(query, params);
} catch (e) {
throw e;
} finally {
conn.release();
}
}
}
module.exports = new DB();
|
