url50r/index.js

http://50r.cn 短网址 node.js SDK

/**
 * Created by wdj on 16/8/11.
 * index
 */
var http = require('http');
var https = require('https');
var url = require('url');
var Promise = require('promise');
var crypto = require('crypto');

var __ = {};
var _ = {
    getCache: function (key) {
        return __[key];
    },
    
    setCache: function (key, value) {
        __[key] = value;
    },
    
    md5: function (str) {
        var md5 = crypto.createHash('md5');
        md5.update(str);
        return md5.digest('hex');
    }
};

function URL50R(opts) {
    this.ak = '';
    this.https = false;
    for(var k in opts){
        if(opts.hasOwnProperty(k)){
            if('ak' === k){
                this.setAppKey(opts[k]);
            }else{
                this[k] = opts[k];
            }
        }
    }
}

URL50R.prototype = {
    
    setAppKey: function (ak) {
        this.ak = ak;
        return this;
    },

    convert: function (url) {
        var self = this;
        var reqUrl = (self.https ? 'https' : 'http') + '://50r.cn/urls/add.json?' + ( this.ak ? 'ak=' + this.ak + '&' : '') + 'url=' + encodeURIComponent(url);
        
        return new Promise(function (resolve, reject) {

            var k = _.md5(reqUrl);
            var result = _.getCache(k);
            if(result){
                resolve(result);
            }else{
                self.httpRequest(reqUrl).then(function (data) {
                    if(typeof data === 'string' && data.charAt(0) === '{'){
                        data = JSON.parse(data);
                        if(data.error){
                            reject(Error(data.error));
                        }else{
                            _.setCache(k, data.url);
                            resolve(data.url);
                        }
                    }else{
                        reject(Error(data));
                    }
                }).catch(reject);
            }
        })
    },

    httpRequest: function (reqUrl) {
        var uri = url.parse(reqUrl);

        var httpOpt = {
            host: uri.host,
            port: uri.port,
            path: uri.path
        };
        var isHttps = this.https;
        return new Promise(function (resolve, reject) {
            var req = (isHttps ? https : http).request(httpOpt, function (res) {
                if(typeof res.setEncoding === 'function'){
                    res.setEncoding('utf8');
                }
                var chunks = [];
                res.on('data', function(chunk){
                    chunks.push(chunk);
                });
                res.on('end', function(){
                    resolve(chunks.join(''));
                });
                res.on('error', function(err){
                    reject(err);
                });
            });

            req.on('error', function (err) {
                reject(err);
            });

            req.end();
        })
    }
};

module.exports = function (opts) {
    return new URL50R(opts);
};