FittingImage.js

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