Temp

package.json

@@ -22,6 +22,7 @@
     "axios": "^1.4.0",
     "bcrypt": "^5.1.0",
     "bull": "^4.10.4",
+    "cheerio": "^1.0.0-rc.12",
     "cors": "^2.8.5",
     "dotenv": "^16.3.1",
     "express": "^4.18.2",
@@ -31,6 +32,7 @@
     "mongoose": "^7.3.0",
     "multer": "^1.4.5-lts.1",
     "multer-s3": "^3.0.1",
+    "nodemailer": "^6.9.3",
     "passport": "^0.6.0",
     "passport-local": "^1.0.0"
   }

src/main.js

@@ -18,7 +18,9 @@ const aiAPI = process.env.AI_API
 
 const allowedOrigins = [
   'http://localhost:3000',
-  'http://www.model-fit.kro.kr',
+  'http://localhost:3001',
+  process.env.WEB_API,
+  process.env.WEB_IP,
   aiAPI,
 ]
 
@@ -60,13 +62,29 @@ app.use(express.static(path.join(__dirname, '../style/build')))
 app.get('/', (req, res) => {
   res.sendFile(path.join(__dirname, '../style/build/index.html'))
 })
-app.get('*', (req, res) => {
-  res.sendFile(path.join(__dirname, '../style/build/index.html'))
-})
 
 const usersRouter = require('./routes/RegisterUser')
 const loginRouter = require('./routes/LoginUser')
 const logoutRouter = require('./routes/LogoutUser')
 app.use('/users', usersRouter)
 app.use('/users', loginRouter)
 app.use('/users', logoutRouter)
+
+const authRouter = require('./routes/MailAuthRouter')
+app.use('/auth', authRouter)
+
+// userInfo 조회
+const infoRouter = require('./routes/UserInfo')
+app.use('/userInfo', infoRouter)
+
+const fittingRouter = require('./routes/FittingImage')
+const clothUploadRouter = require('./routes/UploadClothImage')
+app.use('/api', fittingRouter)
+app.use('/api', clothUploadRouter)
+
+const clothRouter = require('./routes/Clothes')
+app.use('/cloth', clothRouter)
+
+app.get('*', (req, res) => {
+  res.sendFile(path.join(__dirname, '../style/build/index.html'))
+})

src/models/Closet.js

@@ -0,0 +1,11 @@
+const mongoose = require('mongoose')
+const { Schema } = mongoose
+
+const closetSchema = new Schema({
+  userId: { type: Schema.Types.ObjectId, ref: 'User', required: true },
+  clothesUrl: { type: String, required: true },
+  clothesImageLink: { type: String, required: true },
+  fittingImageLink: { type: String },
+})
+
+module.exports = mongoose.model('Closet', closetSchema)

src/models/SizeProfile.js

@@ -2,10 +2,10 @@ const mongoose = require('mongoose')
 const { Schema } = mongoose
 
 const sizeProfileSchema = new Schema({
-  userId: { type: Schema.Types.ObjectId, ref: 'User' },
-  length: { type: Number },
-  shoulderWidth: { type: Number },
-  chestWidth: { type: Number },
+  userId: { type: Schema.Types.ObjectId, ref: 'User', required: true },
+  length: { type: Number, required: true },
+  shoulderWidth: { type: Number, required: true },
+  chestWidth: { type: Number, required: true },
 })
 
 module.exports = mongoose.model('SizeProfile', sizeProfileSchema)

src/models/User.js

@@ -9,13 +9,14 @@ const userSchema = new Schema({
   gender: { type: String, required: true },
   height: { type: Number, required: true },
   weight: { type: Number, required: true },
-  file: { type: String },
+  file: { type: String, required: true },
   favoriteStyle: {
     style: { type: String, default: '' },
     color: { type: String, default: '' },
     fit: { type: String, default: '정핏' },
   },
   sizeProfile: { type: Schema.Types.ObjectId, ref: 'SizeProfile' },
+  clothes: [{ type: Schema.Types.ObjectId, ref: 'Clothes' }],
 })
 
 module.exports = mongoose.model('User', userSchema)

src/modules/AuthMail/AuthMail.js

@@ -0,0 +1,43 @@
+const nodemailer = require('nodemailer')
+const generateAuthCode = require('./generateAuthCode')
+
+/*
+  구글에서는 from 바꾸는 부분 동작하지 않음
+  참고 : https://nodemailer.com/usage/using-gmail/
+*/
+const noreply = '[email protected]'
+
+async function sendAuthMail(email) {
+  try {
+    // SMTP 전송을 위한 transporter 생성
+    const transporter = nodemailer.createTransport({
+      service: 'Gmail',
+      auth: {
+        user: process.env.AUTH_BOT_MAIL,
+        pass: process.env.AUTH_BOT_PASS,
+      },
+    })
+
+    // 이메일 옵션 설정
+    const code = generateAuthCode()
+    const mailOptions = {
+      from: `AuthBot<${noreply}>`,
+      to: email,
+      subject: 'The authorization code for model.fit.',
+      text: `code : ${code}`,
+    }
+
+    // 이메일 전송
+    const info = await transporter.sendMail(mailOptions)
+    console.log(`Email sent successfully : ${email}`)
+    return {
+      authCode: code,
+      result: { success: true, code: 'SEND_DONE', errno: 0 },
+    }
+  } catch (error) {
+    console.error(`Email sent failed : ${email}`)
+    return { success: false, code: error.code, errno: -1 }
+  }
+}
+
+module.exports = sendAuthMail

src/modules/AuthMail/generateAuthCode.js

@@ -0,0 +1,13 @@
+const crypto = require('crypto')
+
+// 6자리의 숫자로만 구성된 난수 생성
+function generateAuthCode() {
+  const byteLength = 4
+
+  const buffer = crypto.randomBytes(byteLength)
+  const randomCode = parseInt(buffer.toString('hex'), 16).toString().slice(0, 6)
+
+  return randomCode
+}
+
+module.exports = generateAuthCode

src/routes/Clothes.js

@@ -0,0 +1,67 @@
+const express = require('express')
+const router = express.Router()
+const Clothes = require('../models/Closet')
+const User = require('../models/User')
+const { getClothesImageUrls } = require('./crawling')
+const { deleteImageFromS3 } = require('./ImageUploader')
+
+router.get('/api/clothes', async (req, res) => {
+  const { userId } = req.query
+  try {
+    const user = await User.findById(userId)
+    if (!user) {
+      return res
+        .status(404)
+        .json({ success: false, message: 'The user was not found.' })
+    }
+    const clothes = await Clothes.find({ userId: userId })
+    res.json(
+      clothes.map((cloth) => ({
+        clothesImageLink: cloth.clothesImageLink,
+        _id: cloth._id,
+      }))
+    )
+  } catch (err) {
+    console.error(err.message)
+    res.status(500).send('Server Error')
+  }
+})
+
+// 입력받은 url에서 이미지 링크 검색 후 반환
+router.get('/api/clothesFromUrl', async (req, res) => {
+  const { url } = req.query
+
+  const result = await getClothesImageUrls(url)
+  if (result.success) {
+    console.log('success')
+    return res.status(200).json(result)
+  } else {
+    return res.status(500).json(result)
+  }
+})
+
+router.delete('/api/delete/:id', async (req, res) => {
+  const clothId = req.params.id
+  try {
+    const cloth = await Clothes.findById(clothId)
+    if (!cloth) {
+      return res.status(404).json({ error: 'No cloth found for given id' })
+    }
+    if (cloth.clothesImageLink && cloth.clothesImageLink.trim() !== '') {
+      console.log(cloth.clothesImageLink)
+      await deleteImageFromS3(cloth.clothesImageLink)
+    }
+
+    if (cloth.fittingImageLink && cloth.fittingImageLink.trim() !== '') {
+      console.log(cloth.fittingImageLink)
+      await deleteImageFromS3(cloth.fittingImageLink)
+    }
+
+    await Clothes.findByIdAndDelete(clothId)
+    res.status(200).json({ message: 'Cloth deleted successfully' })
+  } catch (error) {
+    res.status(400).json({ error: 'Error while deleting the cloth' })
+  }
+})
+
+module.exports = router

src/routes/FittingImage.js

@@ -0,0 +1,92 @@
+const express = require('express')
+const router = express.Router()
+const axios = require('axios')
+const https = require('https')
+const User = require('../models/User')
+const Clothes = require('../models/Closet')
+const fittingApiUrl = process.env.AI_FITTING_API_URL
+
+async function isImageValid(imageUrl) {
+  return new Promise((resolve, reject) => {
+    const options = {
+      method: 'HEAD',
+      url: imageUrl,
+    };
+
+    const req = https.request(imageUrl, options, (res) => {
+      if (res.statusCode === 200) {
+        resolve(true);
+      } else {
+        resolve(false);
+      }
+    });
+
+    req.on('error', (error) => {
+      console.error(`Error checking image URL: ${imageUrl}`, error);
+      resolve(false);
+    });
+
+    req.end();
+  });
+}
+router.post('/fitting', async (req, res) => {
+  try {
+    const { userId, clothID } = req.body
+
+    // 사용자 이미지 가져오기
+    const user = await User.findOne({ _id: userId })
+    if (!user) {
+      return res.status(404).json({ error: 'User not found' })
+    }
+    const userImageUrl = user.file
+    const cloth = await Clothes.findOne({ _id: clothID })
+    if (!cloth) {
+      return res.status(404).json({ error: 'Cloth not found' })
+    }
+    const clothImageUrl = cloth.clothesImageLink
+
+    // 이미 피팅이미지가 있으면
+    if (cloth && cloth.fittingImageLink) {
+      const isValid = await isImageValid(cloth.fittingImageLink)
+      if (isValid) {
+        return res
+          .status(200)
+          .json({ fittingImageLink: cloth.fittingImageLink })
+      } else {
+        console.warn(
+          'Stored fittingImageLink is not valid. Fetching new image.'
+        )
+      }
+    }
+    const response = await axios.post(fittingApiUrl, null, {
+      headers: {
+        'Content-Type': 'application/json',
+      },
+      params: {
+        ID: userId,
+        image_url: userImageUrl,
+        cloth_url: clothImageUrl,
+      },
+    })
+    if (response.data.success) {
+      console.log('Fitting request sent successfully.')
+
+      const imageUrl = response.data.file_name
+      await Clothes.updateOne(
+        { userId, clothesImageLink: clothImageUrl },
+        { $set: { fittingImageLink: imageUrl } }
+      )
+
+      res.status(200).json({ fittingImageLink: imageUrl })
+    } else {
+      console.log(error.response)
+      console.error('Error:', response.data.error)
+      res.status(500).json({ error: response.data.error })
+    }
+  } catch (error) {
+    console.error('Fitting Request Error:', error)
+    res.status(500).json({ error: 'Fitting Request Error' })
+  }
+})
+
+module.exports = router

src/routes/ImageUploader.js

@@ -2,8 +2,10 @@ const aws = require('@aws-sdk/client-s3')
 const multer = require('multer')
 const multers3 = require('multer-s3')
 const path = require('path')
+const { ObjectId } = require('mongodb')
+const url = require('url')
 
-const { S3Client, PutObjectCommand } = aws
+const { S3Client, PutObjectCommand, DeleteObjectCommand } = aws
 
 const s3 = new S3Client({
   region: 'us-east-2',
@@ -26,13 +28,32 @@ const ImageUploader = multer({
     s3: s3,
     bucket: 'bigprogect-bucket',
     key: (req, file, callback) => {
-      const uploadDirectory = req.query.directory ?? 'images'
+      /*
+      회원가입 시 신체 이미지 업로드 하면 아직 DB에 컬럼이 없음.
+        => userId 필드는 무조건 undefined가 됨.
+        => s3/undefined/에 이미지가 저장 됨
+        => id를 생성할 수 있게 [ ?? new ObjectId().toString() ] 추가
+      */
+      const userId =
+        (req.body.userId || req.query.userId) ?? new ObjectId().toString()
+      const uploadDirectory = req.query.directory ?? userId
       const extension = path.extname(file.originalname)
       if (!allowedExtensions.includes(extension)) {
         return callback(new Error('wrong extension'))
       }
-      const userId = req.query.userId
-      callback(null, `${uploadDirectory}/${userId}/userimage${extension}`)
+
+      const imageType = req.query.type
+      let imageFilename
+
+      if (imageType === 'body') {
+        imageFilename = `userimage${extension}`
+      } else if (imageType === 'clothing') {
+        imageFilename = file.originalname
+      } else {
+        return callback(new Error('Unknown image type'))
+      }
+
+      callback(null, `${uploadDirectory}/${imageFilename}`)
     },
     acl: 'public-read-write',
   }),
@@ -42,4 +63,20 @@ const ImageUploader = multer({
   },
 })
 
-module.exports = ImageUploader
+// 이미지 삭제
+const deleteImageFromS3 = async (imageUrl) => {
+  const imageKey = decodeURIComponent(new URL(imageUrl).pathname.substring(1))
+  const params = {
+    Bucket: 'bigprogect-bucket',
+    Key: imageKey,
+  }
+
+  try {
+    const deleteObject = new DeleteObjectCommand(params)
+    await s3.send(deleteObject)
+  } catch (error) {
+    console.error('Error deleting image from S3:', error)
+  }
+}
+
+module.exports = { ImageUploader, deleteImageFromS3 }