Migrate from koa-route to koa-router
This commit is contained in:
parent
8474cca94c
commit
a4bf3dd51d
17 changed files with 206 additions and 291 deletions
|
@ -20,39 +20,39 @@ const hotlinkCheck = function(file, userAgent, referrer) {
|
|||
return !userAgentCheck(userAgent) && !file.width && (!referrer || !(referrer.match(/^https:\/\/hostr.co/) || referrer.match(/^http:\/\/localhost:4040/)))
|
||||
};
|
||||
|
||||
export function* get(id, name, size) {
|
||||
const file = yield this.db.Files.findOne({_id: id, 'file_name': name, 'status': 'active'});
|
||||
export function* get() {
|
||||
const file = yield this.db.Files.findOne({_id: this.params.id, 'file_name': this.params.name, 'status': 'active'});
|
||||
this.assert(file, 404);
|
||||
|
||||
if (hotlinkCheck(file, this.headers['user-agent'], this.headers['referer'])) {
|
||||
return this.redirect('/' + id);
|
||||
return this.redirect('/' + file._id);
|
||||
}
|
||||
|
||||
if (!file.width && this.request.query.warning != 'on') {
|
||||
return this.redirect('/' + id);
|
||||
return this.redirect('/' + file._id);
|
||||
}
|
||||
|
||||
if (file.malware) {
|
||||
let alert = this.request.query.alert;
|
||||
if (!alert || !alert.match(/i want to download malware/i)) {
|
||||
return this.redirect('/' + id);
|
||||
return this.redirect('/' + file._id);
|
||||
}
|
||||
}
|
||||
|
||||
let localPath = path.join(storePath, file._id[0], file._id + '_' + file.file_name);
|
||||
let remotePath = path.join(file._id[0], file._id + '_' + file.file_name);
|
||||
if (size > 0) {
|
||||
localPath = path.join(storePath, file._id[0], size, file._id + '_' + file.file_name);
|
||||
remotePath = path.join(size, file._id + '_' + file.file_name);
|
||||
if (this.params.size > 0) {
|
||||
localPath = path.join(storePath, file._id[0], this.params.size, file._id + '_' + file.file_name);
|
||||
remotePath = path.join(this.params.size, file._id + '_' + file.file_name);
|
||||
}
|
||||
|
||||
console.log(localPath);
|
||||
if (file.malware) {
|
||||
this.statsd.incr('file.malware.download', 1);
|
||||
}
|
||||
|
||||
let type = 'application/octet-stream';
|
||||
if (file.width > 0) {
|
||||
if (size) {
|
||||
if (this.params.size) {
|
||||
this.statsd.incr('file.view', 1);
|
||||
}
|
||||
type = mime.lookup(file.file_name);
|
||||
|
@ -71,19 +71,18 @@ export function* get(id, name, size) {
|
|||
this.body = yield hostrFileStream(localPath, remotePath);
|
||||
}
|
||||
|
||||
export function* resized(size, id, name) {
|
||||
yield get.call(this, id, name, size);
|
||||
export function* resized() {
|
||||
yield get.call(this);
|
||||
}
|
||||
|
||||
export function* landing(id, next) {
|
||||
if (id === 'config.js') {
|
||||
return yield next;
|
||||
}
|
||||
const file = yield this.db.Files.findOne({_id: id});
|
||||
export function* landing() {
|
||||
const file = yield this.db.Files.findOne({_id: this.params.id, status: 'active'});
|
||||
this.assert(file, 404);
|
||||
if(userAgentCheck(this.headers['user-agent'])) {
|
||||
return yield get.call(this, file._id, file.file_name);
|
||||
this.params.name = file.file_name;
|
||||
return yield get.call(this);
|
||||
}
|
||||
console.log('incr')
|
||||
this.statsd.incr('file.landing', 1);
|
||||
const formattedFile = formatFile(file);
|
||||
yield this.render('file', {file: formattedFile});
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue