app.js

var fs = require('fs')
var express = require('express');
var cookieParser = require('cookie-parser');
var bodyParser = require('body-parser');
var util = require('util');
var libUser = require('./lib/User');
var libProject = require('./lib/project');

var dump = function(i){
  console.log(util.inspect(i));
};

var app = express();
var config = require('./config');
console.log(config.dbURI)
require('camo').connect(config.dbURI)

//配置路由，静态路径，错误处理
app.use('/statics',express.static(__dirname+'/statics'));
app.use('/favicon.ico',express.static(__dirname+'/favicon.ico'));

//设置
app.set('views', __dirname+'/views');
app.set('view engine', 'jade');
app.use(cookieParser());
// parse application/x-www-form-urlencoded
app.use(bodyParser.urlencoded({ extended: false,limit: '5mb' }));
// parse application/json
app.use(bodyParser.json({limit: '5mb'}));

//用户信息
app.use(function(req, res, next){
  if(req.cookies.user){
    req.user = req.cookies.user+'';
  }
  next();
});

function checkLoginGoUserPage(req,res,next){
  if(req.user){
    res.redirect('/user/' + req.user);
  }else{
    next()
  }
}

//登陆
app.get('/login',checkLoginGoUserPage,function(req, res){
    res.render('login');
});
app.post('/login',function(req, res){
    libUser.checkUserPassword(req.body.email,req.body.password)
    .then(function(){
      libUser.setLoinCookie(res,req.body.email);
      res.redirect('/user/' + req.body.email);
    },function(err){
      res.render('login',{errormessage:err+', please try again'});
    });
});

app.get('/logout',function(req, res){
  libUser.logOut(res)
  res.render('index');
});


//注册
app.get('/regist',checkLoginGoUserPage,function(req, res){
  res.render('regist');
});
app.post('/regist',function(req, res){
  libUser.createUser(req.body).then(function(user){
    console.log('not rejected!');
    libUser.setLoinCookie(res,req.body.email);
    res.redirect('/user/' + req.body.email);
  },function(err){
    console.log('error?');
    res.render('regist',{error:err+', please retry'});
  });
});

//首页
app.get('/',checkLoginGoUserPage,function(req, res, next){  
  
    res.render('index');
});

//以下页面需要登录，用户，编辑
app.use(function(req, res, next){
  console.log(req.url);
  if(/$\/(user|edit)\//.test(req.url) && !req.user){
    res.render(
      'error',
      {
        errormessage:'need login first',
        linkTitle:'go to login page',
        linkUri:'/login'
      }
    );  
  }else{
    next();  
  }
});

app.param(function(name, fn){
  if (fn instanceof RegExp) {
    return function(req, res, next, val){
      var captures;
      if (captures = fn.exec(String(val))) {
        req.params[name] = captures;
        next();
      } else {
        next('route');
      }
    };
  }
});

app.param('user', /^[0-9a-zA-Z\._@-]+$/);
app.param('project', /^[0-9a-zA-Z]+$/);

//用户页
app.get('/user/:user',function(req, res, next){

  if(req.user === req.params.user+''){
    libProject.getUserProjects(req.user)
    .then(function(resultprojects){
      res.render('user',{
          user:req.params.user,
          projects: resultprojects
        }
      );
    },function(err){
      res.render('user',{
        user:req.params.user,
        errormessage: 'failed to load projects'+err
      });
    });
  }else{
    res.render('error',{
        errormessage:'this is '+req.params.user+'\'s page',
        linkTitle:'go to my page',
        linkUri:'/user/'+req.user
      }
    );
  }
  
});

//项目页
app.get('/project/:project',function(req, res){
  libProject.loadProject(req.params.project[0])
  .then(function(result){
    
    result.pageCount = result.pageCount||1;
    res.render('project',{project:result});
  },function(err){
    res.render('error',{
      errormessage:'error:'+err
    });
  });
});

//编辑页面
app.get('/edit',function(req, res){

  var ready = function(template){
    //存入一个
    libProject.newUserProject(req.user,template)
    .then(function(result){
      res.redirect('/edit/' + result._id);
    },function(err){
      console.log(err)
      res.render('error',{
            errormessage:'db error, please try again',
            linkTitle:'try again!',
            linkUri:'/edit'
          }
        );
    });
  };

  //如果有模板
  var templid = req.params.template || req.query['template'];
  if(templid){
    libProject.loadProject(templid)
    .then(function(result){
      if(!result){
        res.render('error',{
            errormessage:'template project not found',
            linkTitle:'create a empty one',
            linkUri:'/edit'
          }
        );
      }else{
        ready(result);
      }  
    },function(err){
        res.render('error',{
            errormessage: err+', please try again',
            linkTitle:'try again!',
            linkUri:req.url
          }
        );
    });
  }else{
    ready();
  }
  
});
/**
 * 编辑页面
 */
app.get('/edit/:project',function(req, res){
  libProject.loadProject(req.params.project[0])
  .then(function(result){
    if(!result){
      res.render('error',{
          errormessage:'project not found',
          linkTitle:'show my projects',
          linkUri:'/user/'+req.user
        }
      );
    }else{
      res.render('edit',{
          project:result
        });
    }
  },function(err){
    res.render('error',{
          errormessage: err + ', please try again',
          linkTitle:'try again!',
          linkUri:req.url
        }
      );
  });
});

/**
 * 删除逻辑
 */
app.get('/delete/:project',function(req, res){
  libProject.removeProject(req.params.project[0])
  .then(function(){
    res.redirect('/user/'+req.user);
  },function(err){
    res.render('error',{
          errormessage: err + ', please try again',
          linkTitle:'try again!',
          linkUri:req.url
        }
      );
  });
});


/**
 * 获取项目数据
 */
app.get('/json/:project',function(req, res){
  res.setHeader('Content-Type', 'application/json');
  libProject.loadProject(req.params.project[0])
  .then(function(result){
    res.end(JSON.stringify({error:0,data:result}, null, 2));    
  },function(err){
    res.end(JSON.stringify({error:err}, null, 2));    
  });
});

/**
 * 项目编辑页面保存
 */
app.post('/json/:project',function(req, res){
  res.setHeader('Content-Type', 'application/json');
  libProject.updateProject(req.body)
  .then(function(result){
    res.end(JSON.stringify({error:0,data:result}, null, 2));
  },function(err){
    res.end(JSON.stringify({error:err}, null, 2));
  });
});

//所有项目
app.get('/view',function(req, res){
  var page = parseInt(req.query.page)|| 0
  var limit = parseInt(req.query.pageSize)||3
  var skip = page*limit
  var baseUrl =  req.protocol + '://' + req.get('host') 
  var QRCode = require('qrcode')

  libProject.getProjects(limit,skip)
  .then((arr)=>{
    var projects = arr[0]
    var total = arr[1]
    
    var pageCount = parseInt(total/limit)
    total%limit && pageCount++

    Promise.all(projects.map((x)=>getQrImg(x._id)))
    .then((qrcodes)=>{
      res.render('view',{
        user:req.params.user,
        projects,
        qrcodes,
        page,
        pageCount
      })
    })    
  })

  function getQrImg(projectId){
    var url = baseUrl+'/project/'+projectId
    return new Promise((resolve,reject)=>{
      QRCode.toDataURL(url, function (err, datUri) {
        if(err){
          reject(err)
        }else{
          resolve(datUri)
        }
      })
    })
  }
});

//下载页
app.get('/download/:project',function(req, res){
  libProject.loadProject(req.params.project[0])
    .then(function(result){
      return new Promise((resolve,reject)=>{
        result.pageCount = result.pageCount||1;
        res.render('project',{project:result},function(err,viewHtml){
          if(err){
            reject(err)
            return
          }
          if(!viewHtml){
            reject('render fail')
            return
          }
          var archiver = require('archiver');
          var archive = archiver.create('zip', {});
          res.attachment('source.zip');
          var fileDic = {}
          archive.pipe(res)
          //js\css
          viewHtml = viewHtml.replace(/(href|src)="(.*?)"/g,function(match){
            var linkContent = match.substring(match.indexOf('"')+1,match.lastIndexOf('"'))
            if(linkContent.startsWith('http')||linkContent.startsWith('//')){
              return match
            }
            try{
              var filePath = linkContent.substr(1)
              if(!fileDic[filePath] && filePath){
                archive.file(filePath,{name:filePath})
                fileDic[filePath] = true
              }
              return match.replace(linkContent,filePath)
            }catch(e){
              return match
            }
          })
          //pictures
          viewHtml = viewHtml.replace(/url\((.*?)\)/g,function(match){
            var filePath = match.substr(5,match.length-6)            
            if(!fileDic[filePath] && filePath && fs.existsSync(filePath)){
                archive.file(filePath,{name:filePath})
                fileDic[filePath] = true
              }
            return match.replace('/'+filePath,filePath)
          })

          archive.append(string2stream(viewHtml), { name: 'index.html' });
          archive.finalize();
          resolve(true)
        });
      })
      
      function string2stream(str){
        var stream = new require('stream').Readable()
        stream._read = function(){}
        stream.push(str)
        stream.push(null)
        return stream
      }
    })
    .then(()=>{},(err)=>{
      res.render('error',{
        errormessage:'error:'+err
      });
    });
});

/**
 * 上传文件
 */
var fileUpload = require('express-fileupload');
app.use(fileUpload());
app.post('/upload', function(req, res) {  
  res.setHeader('Content-Type', 'application/json');

  if (!req.files) {
    res.end(JSON.stringify({error:'No files were uploaded.'}));
    return;
  }

  var fx = require('mkdir-recursive'),
  path = require('path'),
  d = new Date(),
  targetPath = path.join('statics','uploaded',''+d.getFullYear(),''+(d.getMonth()+1),''+d.getDate());
  
  fx.mkdir(path.join(__dirname,targetPath), function(err) {
    if(err){
      res.end(JSON.stringify({error:'can not save file.'}));
    }

    var imgFile = req.files.img,
    crypto = require('crypto'),
    fileMd5 = crypto.createHash('md5').update(imgFile.data).digest("hex");
    fileExt = path.extname(imgFile.name).toLowerCase(),
    allowExts = ['.jpg','.png','.gif','.webp','.bmp','.svg'];

    if(!(allowExts.indexOf(fileExt)>=0)){
      res.end(JSON.stringify({error:'bad format, plase use:'+allowExts.join(',')}));
    }

    var fullPath = path.join(targetPath,fileMd5+fileExt);
    imgFile.mv(path.join(__dirname,fullPath), function(err) {
      if (err) {
        res.end(JSON.stringify({error:err}));
      } else {
        res.end(JSON.stringify({url:'/'+fullPath.replace(/\\/g,'/')}));
      }
    });
  });

  
});

//启动服务
app.listen(config.http.port,(err)=>{
  if(err){
    console.log(err)
    throw err
  }else{
    console.log('service started at port:'+config.http.port);
  }
});