upload.js

//引入express模块
var express = require('express');

//引入multer模块
var multer = require ('multer');
var path = require('path');

var fs = require('fs');
//引入child_process
var cp=require('child_process');

var geoip = require('geoip-country');

var ip = "118.195.210.89";
var geo = geoip.lookup(ip);

var upload = multer({ dest:  path.join(__dirname,'upload-single')});
var app = express();
var mysql      = require('mysql');
var connection = mysql.createConnection({
    host     : 'localhost',
    user     : 'root',
    password : 'Gongclub123',
    database : 'HCP_CC'
});

connection.connect();
app.use(express.static(path.join(__dirname, './')));
app.use(express.json());
app.use(express.text());
app.use(express.urlencoded({ extended: true }));

function getClientIp(req) {
    return req.headers['x-forwarded-for'] ||
        req.connection.remoteAddress ||
        req.socket.remoteAddress ||
        req.connection.socket.remoteAddress;
};

app.get('/download/*', (req, res) => {
    console.log('download test');
    console.log(getClientIp(req));
    console.log(req.url)
    var url=req.url;
    var arr=url.split('/');
    var b = arr.splice(2);
    var file=b.join('/');
    if (typeof(file) !== undefined){
        console.log(file);
        var path = __dirname  + '/'+file;
        console.log(path);
        var size = fs.statSync(path).size;
        var f = fs.createReadStream(path);
        res.writeHead(200,{
            'Content-Type':'application/force-download',
            'Content-Disposition':'attachment;filename='+file,
            'Content-Length':size
        })
        f.pipe(res);
    }

})
app.post('/multiSurface',function(req,res){
    let parcellation=req.body.parcellation;
    let parcelID=req.body.parcelID;
    console.log(parcellation);
    console.log(parcelID);
    var r = Math.random().toString(36).substr(2, 3) + "-" + Math.random().toString(36).substr(2, 3) + "-" + Math.random().toString(36).substr(2, 4);
    console.log('python ' + path.join(__dirname, 'py/multiSurfResult.py') + ' ' + parcellation+' '+parcelID+' '+r.toUpperCase()+' '+__dirname)
    cp.exec('python ' + path.join(__dirname, 'py/multiSurfResult.py') + ' ' + parcellation+' '+parcelID+' '+r.toUpperCase()+' '+__dirname, (err, stdout, stderr) => {
        if (err) console.log('stderr', err);
        if (stdout) {
            console.log('stdout', stdout);
            if (stderr) console.log('stderr', stderr);
            res.json({
                path: r.toUpperCase(),
                std: stdout,
                err: err,
                stderr: stderr
            })
        }
        res.end();

    });

})
app.get('/threshold',function(req,res)     {
    console.log(req.query.type);
    console.log(req.query.name);
    connection.query("SELECT value from HCP_CC."+req.query.type+" where name='"+req.query.name+"';",function (error,results){
        if (error) throw error;
        if (results){
            let dataString = JSON.stringify(results);
            let data = JSON.parse(dataString);
            res.json({
                value:data[0].value,
            });
        }
        res.end();
    })
})
app.get('/parcelmeta',function(req,res){
    console.log(req.query.template);
    //console.log("SELECT zscore,behavior,pname from HCP_CC.meta_cognition where template='"+req.query.template+"' and pname in ("+req.query.roi1+","+req.query.roi2+") order by pname,zscore desc;")
    connection.query("SELECT zscore,behavior,pname from HCP_CC.meta_cognition where template='"+req.query.template+"' and pname in ("+req.query.roi1+","+req.query.roi2+") order by pname,zscore;", function (error, results, fields) {
        if (error) throw error;
        if (results){
            let dataString = JSON.stringify(results);
            let data = JSON.parse(dataString);
            //console.log('The solution is: ', data);
            let zscore=[],behavior=[];
            for(i=0;i<data.length;i++){
                zscore.push(data[i].zscore);
                behavior.push(data[i].behavior);
            }
            res.json({
                zscore:zscore,
                behavior:behavior,
            });
        }
        res.end();
    });
});

app.post('/upload', upload.single('nifti1-file'), function (req, res) {
    //TODO:need consider .nii.gz
    console.log(req.file);
    console.log(req.body);
    var file = req.file;
    if (!req.file) {
        res.json({
            path: '',
            std: '',
            err: 'ERR: Plz choose a file to upload!',
            stderr: 'ERR: Plz choose a file to upload!'
        })
        res.end();
    } else{
    res.set({
        'content-type': 'application/json; charset=utf-8'
    });
    console.log("Request handler 'upload' was called.");
    if (path.extname(file.originalname) === '.nii') {
        var r = Math.random().toString(36).substr(2, 3) + "-" + Math.random().toString(36).substr(2, 3) + "-" + Math.random().toString(36).substr(2, 4);
        var upload_file = path.join(__dirname, "/upload-single/" + r.toUpperCase() + path.extname(file.originalname));
        fs.rename(file.path, upload_file, function () {
            var arg1 = upload_file;
            var arg2 = path.join(__dirname, '/save-file/', r.toUpperCase());
            var arg3 = path.join(__dirname, '/models');
            console.log('python ' + path.join(__dirname, 'py/genResult.py') + ' ' + arg1 + ' ' + arg2 + ' ' + arg3);
            cp.exec('python ' + path.join(__dirname, 'py/genResult.py') + ' ' + arg1 + ' ' + arg2 + ' ' + arg3, (err, stdout, stderr) => {
                let std;
                if (err) console.log('stderr', err);
                if (stdout) {
                    console.log('stdout', stdout);
                    std = stdout;
                }
                if (stderr) console.log('stderr', stderr);
                res.json({name: file.originalname, path: r.toUpperCase(), std: std, err: err, stderr: stderr})
                res.end();

            });

        });
    } else {
        res.json({
            name: file.originalname,
            path: '',
            std: '',
            err: 'ERR: You should upload a nifti-file!',
            stderr: 'ERR: You should upload a nifti-file!'
        })
        res.end();
    }
}
});
app.post('/uploadSurface', upload.single('gifti1-file'), function (req, res) {
    //TODO:need consider .nii.gz
    console.log('upload');
    console.log(req.file);
    var file = req.file;
    var index = req.body.opt;
    if (!req.file) {
        res.json({
            path: '',
            std: '',
            err: 'ERR: Plz choose a file to upload!',
            stderr: 'ERR: Plz choose a file to upload!'
        })
        res.end();
    } else {
        res.set({
            'content-type': 'application/json; charset=utf-8'
        });
        console.log("Request handler 'upload' was called.");
        if (path.extname(file.originalname) === '.gii') {
            var r = Math.random().toString(36).substr(2, 3) + "-" + Math.random().toString(36).substr(2, 3) + "-" + Math.random().toString(36).substr(2, 4);
            var upload_file = path.join(__dirname, "/upload-single/" + r.toUpperCase() + '.tex.gii');
            fs.rename(file.path, upload_file, function () {
                var arg1 = upload_file;
                var arg2 = index;
                console.log('python ' + path.join(__dirname, 'py/genSurfResult.py') + ' ' + arg1 + ' ' + arg2+ ' '+__dirname)
                cp.exec('python ' + path.join(__dirname, 'py/genSurfResult.py') + ' ' + arg1 + ' ' + arg2 + ' '+__dirname, (err, stdout, stderr) => {
                    let std;
                    console.log('err'+err)
                    console.log('stdout'+stdout)
                    console.log('stderr'+stderr)
                    if (stdout) {
                    cp.exec('ls ' + path.join(__dirname, 'save-file/' + r.toUpperCase() + '/json/*.nii'), (err0, stdout0, stderr0) => {
                        console.log(err0)
                        if (err0 || stderr0) {
                            res.json({
                                name: file.originalname,
                                path: '',
                                flag:[],
                                std: '',
                                err: 'ERR: There\'s something wrong with your mask! ',
                                stderr: 'ERR: There\'s something wrong with your mask! Nothing generated! '
                            })
                        }
                        else if (stdout0) {
                            console.log(stdout0.split("\n"));
                            std=stdout0.split("\n");
                            var resultno=stdout0.split("\n").length;
                            var result_arr=[];
                            for (i = 0; i < resultno; i++){
                                let fname = std[i].split("/").pop().split(".");
                                result_arr.push(fname[0]+'.'+fname[1]+'.'+fname[2])
                            }
                            console.log(stdout0)
                            if (stdout0.length === 0) {
                                res.json({
                                    name: file.originalname,
                                    path: '',
                                    flag:[],
                                    std: '',
                                    err: 'ERR: There\'s something wrong with your mask! ',
                                    stderr: 'ERR: There\'s something wrong with your mask! '
                                });
                                res.end();
                            } else if (stdout0.length > 0) {
                                console.log('stderr', stderr);
                                res.json({
                                    name: file.originalname,
                                    path: r.toUpperCase(),
                                    flag: result_arr,
                                    std: std,
                                    err: err,
                                    stderr: stderr
                                })
                                res.end();
                            }

                        }
                    })
                }
                });

            });
        } else {

            res.end();
        }
    }
});
//app.listen(5700)
app.listen(8080,'0.0.0.0',() => {
    // client.country('142.1.1.1').then(response => {
    //     console.log(response);
    //     console.log(response.country.isoCode); // 'CA'
    // });
    console.log(`Example app listening on port 8080`)
});
app.get('/', function(req,res){


    console.log('test_get');
    console.log(geo.country);
    console.log(getClientIp(req));
    if (getClientIp(req)=='CN'){
        console.log('yes');

        res.redirect('http://ccmapping.cn/');
    }else{
        res.redirect('http://ccmapping.org/');
    }
    //res.sendFile(path.join(__dirname,'./CCmapping_index.html'));
});
// app.get('/', function(req, res){
//     res.set('Content-Type', 'text/plain');
//     res.send('hello world');
// });