-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathindex.js
99 lines (82 loc) · 2.56 KB
/
index.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
const mysql = require('mysql2');
const express = require('express');
const cors = require('cors');
const app = express();
app.use(cors());
const connection = mysql.createConnection({
host : process.env.DATABASE_HOST,
port : process.env.DATABASE_PORT,
user : process.env.DATABASE_USERNAME,
password : process.env.DATABASE_PASSWORD,
database : process.env.DATABASE_DATABASE
});
const requiredPrefixes = process.env.URL_PREFIXES.split(';');
const installerUrl = process.env.INSTALLER_URL;
connection.connect(function (err) {
if (err) {
console.error('error connecting: ' + err.stack);
return;
}
console.log('connected as id ' + connection.threadId);
connection.execute(`CREATE TABLE IF NOT EXISTS downloads
(
id int auto_increment,
timestamp datetime(6) default NULL null,
url varchar(1000) default NULL null,
CONSTRAINT downloads_pk
PRIMARY KEY (id)
)`, [], function (err, result) {
if (err) {
console.error(err);
process.exit(1);
}
app.listen(3000, function () {
console.log('Web server listening on port 3000');
});
});
});
app.get('/api/v1/download/_count', function (req, res) {
connection.execute(
'SELECT COUNT(*) AS cnt FROM downloads',
[],
function (err, result) {
if (err) {
console.error(err);
res.json({}).status(500);
// Force restart
throw err;
}
console.log("Total downloads", result[0].cnt);
res.json(result[0].cnt);
}
);
});
app.get('/api/v1/download', function (req, res) {
const url = req.query.url;
if (!requiredPrefixes.some(x => url.startsWith(x))) {
return res.status(400).json({error: "Invalid URL"});
}
connection.execute(
'INSERT INTO downloads (timestamp, url) VALUES (CURRENT_TIMESTAMP, ?)',
[url],
function (err) {
if (err) {
console.error(err);
}
}
);
res.redirect(url);
});
app.get('/installer', function (req, res) {
res.redirect(installerUrl);
});
app.get('/health', function (req, res) {
connection.ping(function (err) {
if (err) {
console.error(err);
res.status(500).json({});
return;
}
res.json();
})
})