app.js 5.83 KB

Raw Blame History Permalink

var createError = require('http-errors');
var express = require('express');
var formidable = require('formidable');
var path = require('path');
var cookieParser = require('cookie-parser');
var logger = require('morgan');
var fs = require('fs')
var AWS = require('aws-sdk');


// 이미지를 저장할 버킷 이름
var bucketName = "kindofyourdogimage";
// s3 버킷의 엔드 포인트
var bucketRegion = 'ap-northeast-2';
// access key
var accessId= 'your_accessId';
// access secret key
var secretKey = 'your_secretKey';
// AWS Cognito 인증
//var identityPoolId = "ap-northeast-2:7cba9a17-588b-40d6-8c70-eb8ba4d573be";


// AWS의 config 정보를 Update해준다 -> 이용자가 S3 버킷에 접근 가능하다
/*
AWS.config.update({
  region:bucketRegion,
  credentials:new AWS.CognitoIdentityCredentials({
    IdentityPoolId:identityPoolId,
  })
})
*/
AWS.config.update({
  region:bucketRegion,
  accessKeyId:accessId,
  secretAccessKey:secretKey
});

// 내가 사용할 S3
var s3 = new AWS.S3({
  apiVersion: "2006-03-01",
  params: {Bucket: bucketName}
});

const fileName = 'puppy.jpg';

// 파일 업로드
const uploadFile = (fileName) => {
  const fileContent = fs.readFileSync(fileName);
  const params = {
    Bucket: bucketName,
    Key: fileName, // File name you want to save as in S3
    Body: fileContent }
;
  s3.upload(params, function(err, data) {
    if (err) { throw err; }
    console.log(`File uploaded successfully. ${data.Location}`);
  });
};
uploadFile(fileName);

/*
// rekognition 객체
const client = new AWS.Rekognition();

const params = {
  "Image": {
      "S3Object": {
          "Bucket": bucketName,
          "Name": fileName
      }
  },
  "MaxLabels": 10,
  "MinConfidence": 75
}


// 이미지 분석하기
client.detectLabels(params, function(err, response) {
  if (err) {
    console.log(err, err.stack); // an error occurred
  } else {
    console.log(`Detected labels for: ${photo}`)
    response.Labels.forEach(label => {
      console.log(`Label:      ${label.Name}`)
      console.log(`Confidence: ${label.Confidence}`)
      console.log("Instances:")
      label.Instances.forEach(instance => {
        let box = instance.BoundingBox
        console.log("  Bounding box:")
        console.log(`    Top:        ${box.Top}`)
        console.log(`    Left:       ${box.Left}`)
        console.log(`    Width:      ${box.Width}`)
        console.log(`    Height:     ${box.Height}`)
        console.log(`  Confidence: ${instance.Confidence}`)
      })
      console.log("Parents:")
      label.Parents.forEach(parent => {
        console.log(`  ${parent.Name}`)
      })
      console.log("------------")
      console.log("")
    }) // for response.labels
  } // if
});
*/

var indexRouter = require('./routes/index');
var usersRouter = require('./routes/users');
// image업로드 시 해당 route기능 사용

var app = express();

// view engine setup
app.set('views', path.join(__dirname, 'views'));
app.set('view engine', 'ejs');

app.use(logger('dev'));
app.use(express.json());
app.use(express.urlencoded({ extended: false }));
app.use(cookieParser());
app.use(express.static(path.join(__dirname, 'public')));

app.use('/', indexRouter);
app.use('/users', usersRouter);

// image 올리는 화면
app.get('/form', function(req, res){
  var output = `
  <html>
  <body>
    <form enctype="multipart/form-data" method="post" action="upload_receiver">
      <input type="file" name="userfile">
      <input type="submit">
    </form>
  </body>
  </html>
  `;
  res.send(output);
  console.log(output);
});

// image를 받았을 때
app.post('/upload_receiver', function(req,res){
  var form = new formidable.IncomingForm();
  console.log(form);
  const client = new AWS.Rekognition();

  const params = {
    "Image": {
        "S3Object": {
            "Bucket": bucketName,
            "Name": fileName
        }
    },
    "MaxLabels": 10,
    "MinConfidence": 75
  }

  // 이미지 분석하기
  client.detectLabels(params, function(err, response) {
    if (err) {
      console.log(err, err.stack); // an error occurred
    } else {
      console.log(`Detected labels for: ${form.name}`)
      response.Labels.forEach(label => {
        console.log(`Label:      ${label.Name}`)
        console.log(`Confidence: ${label.Confidence}`)
        console.log("Instances:")
        label.Instances.forEach(instance => {
          let box = instance.BoundingBox
          console.log("  Bounding box:")
          console.log(`    Top:        ${box.Top}`)
          console.log(`    Left:       ${box.Left}`)
          console.log(`    Width:      ${box.Width}`)
          console.log(`    Height:     ${box.Height}`)
          console.log(`  Confidence: ${instance.Confidence}`)
        })
        console.log("Parents:")
        label.Parents.forEach(parent => {
          console.log(`  ${parent.Name}`)
        })
        console.log("------------")
        console.log("")
      }) // for response.labels
    } // if
  });

  // S3에 upload해주기
  form.parse(req, function(err, fields, files){
    var s3 = new AWS.S3();
    var params = {
      Bucket:bucketName,
      ACL:'public-read',
      Key:files.userfile.name,
      Body: require('fs').createReadStream(files.userfile.path)
    }
    s3.upload(params, function(err, data){
      var result='';
      if(err)
        result = 'Fail';
      else
        result = `<img src="${data.Location}>`;
      res.send(`<html><body>${result}</body></html`)
    });
    console.log(err, fields,files);
  });
});

// catch 404 and forward to error handler
app.use(function(req, res, next) {
  next(createError(404));
});

// error handler
app.use(function(err, req, res, next) {
  // set locals, only providing error in development
  res.locals.message = err.message;
  res.locals.error = req.app.get('env') === 'development' ? err : {};

  // render the error page
  res.status(err.status || 500);
  res.render('error');
});

module.exports = app;