116 lines
3.0 KiB
JavaScript
116 lines
3.0 KiB
JavaScript
const express = require("express");
|
|
const Schedule = require("../models/Schedule");
|
|
|
|
const router = express.Router();
|
|
|
|
const startOfWeek = (date = new Date()) => {
|
|
const copy = new Date(date);
|
|
const day = copy.getDay();
|
|
const diff = day === 0 ? -6 : 1 - day;
|
|
copy.setDate(copy.getDate() + diff);
|
|
copy.setHours(0, 0, 0, 0);
|
|
return copy;
|
|
};
|
|
|
|
const endOfWeek = (date = new Date()) => {
|
|
const start = startOfWeek(date);
|
|
const end = new Date(start);
|
|
end.setDate(end.getDate() + 6);
|
|
end.setHours(23, 59, 59, 999);
|
|
return end;
|
|
};
|
|
|
|
const startOfMonth = (date = new Date()) => {
|
|
return new Date(date.getFullYear(), date.getMonth(), 1, 0, 0, 0, 0);
|
|
};
|
|
|
|
const endOfMonth = (date = new Date()) => {
|
|
return new Date(date.getFullYear(), date.getMonth() + 1, 0, 23, 59, 59, 999);
|
|
};
|
|
|
|
router.get("/", async (req, res) => {
|
|
try {
|
|
const schedules = await Schedule.find().sort({ startDate: 1 });
|
|
res.json(schedules);
|
|
} catch (error) {
|
|
res.status(500).json({ message: "Failed to fetch schedules" });
|
|
}
|
|
});
|
|
|
|
router.get("/week", async (req, res) => {
|
|
try {
|
|
const start = startOfWeek();
|
|
const end = endOfWeek();
|
|
const schedules = await Schedule.find({
|
|
startDate: { $lte: end },
|
|
endDate: { $gte: start },
|
|
}).sort({ startDate: 1 });
|
|
res.json(schedules);
|
|
} catch (error) {
|
|
res.status(500).json({ message: "Failed to fetch weekly schedules" });
|
|
}
|
|
});
|
|
|
|
router.get("/month", async (req, res) => {
|
|
try {
|
|
const start = startOfMonth();
|
|
const end = endOfMonth();
|
|
const schedules = await Schedule.find({
|
|
startDate: { $lte: end },
|
|
endDate: { $gte: start },
|
|
}).sort({ startDate: 1 });
|
|
res.json(schedules);
|
|
} catch (error) {
|
|
res.status(500).json({ message: "Failed to fetch monthly schedules" });
|
|
}
|
|
});
|
|
|
|
router.post("/", async (req, res) => {
|
|
try {
|
|
const schedule = await Schedule.create(req.body);
|
|
res.status(201).json(schedule);
|
|
} catch (error) {
|
|
res.status(400).json({ message: "Failed to create schedule" });
|
|
}
|
|
});
|
|
|
|
router.get("/:id", async (req, res) => {
|
|
try {
|
|
const schedule = await Schedule.findById(req.params.id);
|
|
if (!schedule) {
|
|
return res.status(404).json({ message: "Schedule not found" });
|
|
}
|
|
res.json(schedule);
|
|
} catch (error) {
|
|
res.status(400).json({ message: "Failed to fetch schedule" });
|
|
}
|
|
});
|
|
|
|
router.put("/:id", async (req, res) => {
|
|
try {
|
|
const schedule = await Schedule.findByIdAndUpdate(req.params.id, req.body, {
|
|
new: true,
|
|
});
|
|
if (!schedule) {
|
|
return res.status(404).json({ message: "Schedule not found" });
|
|
}
|
|
res.json(schedule);
|
|
} catch (error) {
|
|
res.status(400).json({ message: "Failed to update schedule" });
|
|
}
|
|
});
|
|
|
|
router.delete("/:id", async (req, res) => {
|
|
try {
|
|
const schedule = await Schedule.findByIdAndDelete(req.params.id);
|
|
if (!schedule) {
|
|
return res.status(404).json({ message: "Schedule not found" });
|
|
}
|
|
res.json({ ok: true });
|
|
} catch (error) {
|
|
res.status(400).json({ message: "Failed to delete schedule" });
|
|
}
|
|
});
|
|
|
|
module.exports = router;
|