-
Notifications
You must be signed in to change notification settings - Fork 0
/
index.js
96 lines (82 loc) · 2.8 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
var express = require('express');
var app = express();
var npm = new require('npm-api')();
var _ = require('lodash');
var fs = require('fs');
var read = function(name) {return fs.readFileSync(name, { encoding: 'utf-8' })};
var help = read(__dirname + '/index.html');
var MongoClient = require('mongodb').MongoClient;
var mongoUrl = process.env.MONGODB_URI;
app.use(express.static('.'));
var customRouter = express.Router();
customRouter.get('*', function redirect(req, res, next) {
if (req.hostname === 'npm-api.herokuapp.com') {
console.log('redirect');
res.redirect(301, 'http://npmapi.invntrm.ru' + req.originalUrl);
res.end();
} else {
next();
}
});
app.use('*', customRouter)
app.get('/', function (req, res) {
res.send(help)
});
app.get('/p/:package', function (req, res) {
var package = req.params.package;
getPackageUrl(package)
.then(function(url) { res.end(url); })
.then(function() { writeLogDb(package); })
.catch(function(err) { res.status(500).end('error: package not found or server broken: ' + err); });
});
app.get('/go/:package', function (req, res) {
var package = req.params.package;
getPackageUrl(package)
.then(function(url) { res.redirect(url); })
.then(function() { writeLogDb(package); })
.catch(function(err) { res.status(500).end('error: package not found or server broken: ' + err); });
});
app.get('/logs', function(req, res) {
readLogsDb()
.then(function(docs) {res.send(docs)})
.catch(function(err) { res.status(500).end('error: server broken: ' + err); });
});
app.listen(process.env.PORT, function () {
console.log('Example app listening on port ' + process.env.PORT);
});
function getPackageUrl(name, cb) {
var repo = npm.repo(name);
return repo.package()
.then(function(pkg) {
var hostpath = pkg.repository.url
.replace(/^.+\/\//, '')
.replace(/^.+@/g, '')
.replace(/:/, '/')
.replace(/\.git$/, '');
return 'https://' + hostpath;
});
}
function writeLogDb(name) {
console.log('> log', name);
MongoClient.connect(mongoUrl, function(err, db) {
if (err) console.log('mongo err', err);
if (!db) return;
var collection = db.collection(process.env.MONGODB_COLLECTION);
collection.insertOne({"package": name}, function(err, result) {
db.close();
});
});
}
function readLogsDb() {
return new Promise(function(res, rej) {
MongoClient.connect(mongoUrl, function(err, db) {
if (err) rej(err);
if (!db) return;
var collection = db.collection(process.env.MONGODB_COLLECTION);
collection.find({}).toArray(function(err, docs) {
res(_(docs).map('package'));
db.close();
});
});
});
}