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,
 ]
 
@@ -35,7 +37,7 @@ mongoose
   .catch((err) => console.log(err))
 
 app.use(express.json())
-app.use(express.urlencoded({ extended: false }))
+app.use(express.urlencoded({ extended: true }))
 app.use(
   cors({
     origin: function (origin, callback) {
@@ -49,6 +51,7 @@ app.use(
     credentials: true,
   })
 )
+
 app.use(passport.initialize())
 require('./config/passport')(passport)
 
@@ -60,13 +63,34 @@ 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')
+const clothSizeRouter = require('./routes/ClothSize')
+app.use('/cloth', clothRouter)
+app.use('/cloth', clothSizeRouter)
+
+const forumRouter = require('./routes/OutfitForum')
+app.use('/api/outfitforums', forumRouter)
+
+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/Forum.js

@@ -0,0 +1,27 @@
+const mongoose = require('mongoose')
+const { Schema } = mongoose
+
+const forumSchema = new mongoose.Schema({
+  user: {
+    type: Schema.Types.ObjectId,
+    ref: 'User',
+    required: true,
+  },
+  title: {
+    type: String,
+    required: true,
+  },
+  content: {
+    type: String,
+    required: true,
+  },
+  image: {
+    type: String,
+  },
+  createdAt: {
+    type: Date,
+    default: Date.now,
+  },
+})
+
+module.exports = mongoose.model('Forum', forumSchema)

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 = 'noreply@model-fit.kro.kr'
+
+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/ClothSize.js

@@ -0,0 +1,40 @@
+const express = require('express')
+const router = express.Router()
+const User = require('../models/User')
+const SizeProfile = require('../models/SizeProfile')
+const Closet = require('../models/Closet')
+const axios = require('axios')
+
+router.post('/cloth_size', async (req, res) => {
+  try {
+    const { userId, clothId } = req.body
+
+    const user = await User.findById(userId).populate('sizeProfile')
+    const sizeProfile = user.sizeProfile
+    const closet = await Closet.findById(clothId)
+
+    const length = sizeProfile.length
+    const shoulderWidth = sizeProfile.shoulderWidth
+    const chestWidth = sizeProfile.chestWidth
+    const imageUrl = closet.clothesUrl
+    const overfit = user.favoriteStyle.fit
+
+    const aiApiResponse = await axios.get(process.env.AI_CLOTH_SIZE_API_URL, {
+      params: {
+        length,
+        shoulderWidth,
+        chestWidth,
+        imageUrl,
+        overfit,
+      },
+    })
+
+    const recommendedSize = aiApiResponse.data.size
+    res.json({ size: recommendedSize })
+  } catch (error) {
+    console.error(error)
+    res.status(500).json({ message: 'Server error' })
+  }
+})
+
+module.exports = router

src/routes/Clothes.js

@@ -0,0 +1,69 @@
+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', code: 'DELETE_DONE' })
+  } catch (error) {
+    res.status(400).json({ error: 'Error while deleting the cloth' })
+  }
+})
+
+module.exports = router