api/resources/Inventory/InventoryController.js

import { Inventory } from "../Inventory/InventoryModel.js";
import User from "../user/userModel.js";
import { KYC } from "../KYC/KycModel.js";
import ShippingAddress from "../ShippingAddresses/ShippingAddressModel.js";
import TerritoryManager from "../TerritoryManagers/TerritoryManagerModel.js";
import SalesCoordinator from "../SalesCoOrdinators/SalesCoOrdinatorModel.js";
// Add inventory data
export const addInventory = async (req, res) => {
  try {
    const { products, addedFor, addedForId } = req.body;
    const userId = req.user._id;
    const userType = req.userType;
    // console.log("req.user", req.user);
    const newInventory = new Inventory({
      userId,
      userType,
      addedFor,
      addedForId,
      products,
    });

    await newInventory.save();
    res.status(201).json({
      success: true,
      message: "Inventory added successfully",
      data: newInventory,
    });
  } catch (error) {
    res.status(500).json({ success: false, message: "Server error", error });
  }
};

// Get all distributors (PD or RD)
export const getDistributors = async (req, res) => {
  try {
    const { type } = req.params;
    
    if (!["PrincipalDistributor", "RetailDistributor"].includes(type)) {
      return res.status(400).json({ message: "Invalid distributor type" });
    }

    let distributors;
// console.log("type",type);
    if (type === "PrincipalDistributor") {
      // Fetch all PrincipalDistributors
      const principalDistributors = await User.find({ role: "principal-Distributor" });
// console.log("principalDistributors",principalDistributors);
      // Map each PrincipalDistributor to include their ShippingAddress
      distributors = await Promise.all(principalDistributors.map(async (distributor) => {
        const shippingAddress = await ShippingAddress.findOne({ user: distributor._id });
        return {
          ...distributor.toObject(),
          shippingAddress,
        };
      }));
    } else {
      // For RetailDistributor, fetch approved KYC documents
      distributors = await KYC.find({ status: "approved" });
    }

    res.status(200).json(distributors);
  } catch (error) {
    res.status(500).json({ message: error.message });
  }
};

export const getAllInventories = async (req, res) => {
  try {
    const {
      page = 1,
      show = 10,
      startDate,
      endDate
    } = req.query;

    // Build query
    const query = {};
    
    if (startDate && endDate) {
      const start = new Date(startDate);
      const end = new Date(endDate);

      if (start.toDateString() === end.toDateString()) {
        // If startDate and endDate are the same, fetch records for that day
        query.createdAt = {
          $gte: new Date(startDate),
          $lt: new Date(startDate).setDate(new Date(startDate).getDate() + 1) // Until the end of that day
        };
      } else {
        // If startDate and endDate are different, fetch records between these dates
        query.createdAt = {
          $gte: start,
          $lte:new Date(end).setDate(new Date(end).getDate() + 1)
        };
      }
    } else if (startDate) {
      // Only startDate is provided
      query.createdAt = {
        $gte: new Date(startDate),
        $lte: new Date() // Up to today's date
      };
    } else if (endDate) {
      // Only endDate is provided
      query.createdAt = {
        $lte: new Date(endDate)
      };
    }

    // Fetch total count of documents matching the query
    const total_data = await Inventory.countDocuments(query);

    // Fetch inventory data with pagination
    const inventories = await Inventory.find(query)
      .skip((page - 1) * show)
      .limit(Number(show))
      .sort({ createdAt: -1 }); // Optional: Sort by createdAt date in descending order

    // Populate additional details
    const populatedInventories = await Promise.all(inventories.map(async (inventory) => {
      // Populate user details based on userType
      let user = null;
      if (inventory.userType === 'TerritoryManager') {
        user = await TerritoryManager.findById(inventory.userId);
      } else if (inventory.userType === 'SalesCoordinator') {
        user = await SalesCoordinator.findById(inventory.userId);
      }

      // Populate addedFor details based on addedFor
      let addedForData = null;
      if (inventory.addedFor === 'PrincipalDistributor') {
        addedForData = await User.findById(inventory.addedForId);
        const shippingAddress = await ShippingAddress.findOne({ user: addedForData._id });
        addedForData = {
          ...addedForData.toObject(),
          shippingAddress,
        };
      } else if (inventory.addedFor === 'RetailDistributor') {
        addedForData = await KYC.findById(inventory.addedForId);
      }
      return {
        ...inventory.toObject(),
        user,
        addedForData,
      };
    }));

    // Send response with pagination info
    res.status(200).json({
      total_data,
      inventories: populatedInventories
    });
  } catch (error) {
    res.status(500).json({ message: error.message });
  }
};

export const getSingleInventory = async (req, res) => {
  try {
    const { id } = req.params;

    const inventory = await Inventory.findById(id);
    if (!inventory) {
      return res.status(404).json({ message: "Inventory not found" });
    }

    // Populate user details based on userType
    let user = null;
    if (inventory.userType === 'TerritoryManager') {
      user = await TerritoryManager.findById(inventory.userId);
    } else if (inventory.userType === 'SalesCoOrdinator') {
      user = await SalesCoordinator.findById(inventory.userId);
    }

    // Populate addedFor details based on addedFor
    let addedForData = null;
    if (inventory.addedFor === 'PrincipalDistributor') {
      addedForData = await User.findById(inventory.addedForId);
      const shippingAddress = await ShippingAddress.findOne({ user: addedForData._id });
      addedForData = {
        ...addedForData.toObject(),
        shippingAddress,
      };
    } else if (inventory.addedFor === 'RetailDistributor') {
      addedForData = await KYC.findById(inventory.addedForId);
    }

    res.status(200).json({
      ...inventory.toObject(),
      user,
      addedForData,
    });
  } catch (error) {
    res.status(500).json({ message: error.message });
  }
};