const Person = require('../../../models/person');
import dbConnect from '../../../utils/helpers/dbHelpers';

async function handler(req, res) {
  const { method } = req;

  await dbConnect();

  switch (method) {
    case 'GET': {
      try {
        const dataCount = await Person.countDocuments();

        if (dataCount === 0) {
          res.status(200).json({
            message: 'No data.',
            dataIds: [],
          });
          break;
        }

        const dataFromDB = await Person.find({}).limit(4);

        if (!dataFromDB) {
          throw new Error('No data!');
        }

        const dataIds = dataFromDB.map((item) => item.customID);

        res.status(200).json({
          message: 'Fetch ids successfull!',
          dataIds: dataIds,
        });
      } catch (error) {
        res.status(400).json({ message: error.message });
      }
      break;
    }
    default:
      res.status(405).json({ message: 'Method not allowed' });
      break;
  }
}

export default handler;