cnpmjs.org/controllers/registry/package/download_total.js

Private npm registry and web for Enterprise, base on MySQL and Simple Store Service

/*!
 * cnpmjs.org - controllers/registry/package/download_total.js
 *
 * Copyright(c) cnpmjs.org and other contributors.
 * MIT Licensed
 *
 * Authors:
 *   dead_horse <dead_horse@qq.com> (https://github.com/dead-horse)
 */

'use strict';

/**
 * Module dependencies.
 */

var debug = require('debug')('cnpmjs.org:controllers:registry:package:download_total');
var DownloadTotal = require('../../../services/download_total');
var moment = require('moment');

var DATE_REG = /^\d{4}-\d{2}-\d{2}$/;

module.exports = function* downloadTotal () {
  var range = this.params.range;
  var name = this.params.name;

  range = range.split(':');
  if (range.length !== 2
      || !range[0].match(DATE_REG)
      || !range[1].match(DATE_REG)) {
    this.status = 400;
    this.body = {
      error: 'range_error',
      reason: 'range must be YYYY-MM-DD:YYYY-MM-DD style'
    };
    return;
  }

  this.body = name
    ? yield getPackageTotal(name, range[0], range[1])
    : yield getTotal(range[0], range[1]);
}

function* getPackageTotal(name, start, end) {
  var res = yield DownloadTotal.getModuleTotal(name, start, end);
  var downloads = res.map(function (row) {
    return {
      day: row.date,
      downloads: row.count
    };
  });

  downloads.sort(function (a, b) {
    return a.day > b.day ? 1 : -1;
  });

  return {
    downloads: downloads,
    package: name,
    start: start,
    end: end
  };
}

function* getTotal(start, end) {
  var res = yield DownloadTotal.getTotal(start, end);
  var downloads = res.map(function (row) {
    return {
      day: row.date,
      downloads: row.count
    };
  });

  downloads.sort(function (a, b) {
    return a.day > b.day ? 1 : -1;
  });

  return {
    downloads: downloads,
    start: start,
    end: end
  };
}