-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathapp.js
117 lines (98 loc) · 2.71 KB
/
app.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
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
111
112
113
114
115
116
117
const express = require('express');
const exphbs = require('express-handlebars');
const hbs = require('hbs');
const cookieParser = require('cookie-parser');
const path = require('path');
const auth = require('./src/middleware/auth.middleware');
const { getQueries, getQuery } = require('./db/queries');
const fs = require('fs');
const { ExportToExcel } = require('./src/utils/exportToExcel');
require('dotenv').config();
const PORT = process.env.PORT || 5000;
const app = express();
app.set('view engine', 'hbs');
app.engine(
'hbs',
exphbs({
layoutsDir: __dirname + 'views',
extname: 'hbs',
defaultLayout: false
})
);
hbs.registerPartials(__dirname + '/views/partials');
app.use(express.urlencoded({ extended: true }));
app.use(express.json());
app.use(cookieParser());
app.use(express.static('public'));
app.use(express.static('outputs'));
app.use('/api/parser', require('./src/routes/parser.routes'));
app.use('/api/auth', require('./src/routes/auth.routes'));
app.get('/login', (req, res) => {
res.sendFile(path.join(__dirname, '/views/login.html'));
});
app.get('/register', (req, res) => {
if (process.env.REGISTRATION_ENABLED === 'true') {
res.sendFile(path.join(__dirname, '/views/register.html'));
} else {
res.status(403).json({ message: 'Registration disabled on this server' });
}
});
app.get('/dashboard', auth, async (req, res) => {
let queries = await getQueries();
res.render(path.join(__dirname, '/views/dashboard.hbs'), { queries });
});
app.get('/dashboard/details/:id', auth, async (req, res) => {
let data = await getQuery({ id: req.params.id });
let cl = data[0].CurrentLink || 0;
let lc = data[0].LinksCount || 0;
let progress = cl * (100 / lc);
let output = null;
try {
let outputExists = await new Promise((resolve, reject) => {
fs.access(
path.join(__dirname + '/outputs/' + data[0].Output),
fs.F_OK,
(err) => {
if (err) {
return reject(err);
}
resolve(true);
}
);
});
if (outputExists) {
let tmpOutput = await new Promise((resolve, reject) => {
fs.readFile(
path.join(__dirname + '/outputs/' + data[0].Output),
(err, data) => {
if (err) {
return reject(err);
}
resolve(data);
}
);
});
output = JSON.parse(tmpOutput);
}
} catch (err) {
console.log(err);
}
res.render(path.join(__dirname, '/views/details.hbs'), {
...data[0],
Progress: progress,
Output: output
});
});
app.get('/', auth, (req, res) => {
if (req.user) {
res.redirect(301, '/dashboard');
} else {
res.redirect(301, '/login');
}
});
app.get('*', (req, res) => {
res.render(path.join(__dirname, '/views/not_found.hbs'));
});
app.listen(PORT, () => {
console.log(`Server running on http://localhost:${PORT}`);
});