Subversion Repositories SmartDukaan

	@Autowired

	private com.spice.profitmandi.dao.repository.dtr.LeadActivityRepository leadActivityRepositoryAuto;

	// ====================== EDIT BEAT ======================

	// Update an existing beat — name + partner stops (routes).

	// Schedules are NOT touched here; manage them via calendar drag-drop.

	@PostMapping(value = "/beatPlan/updateBeat")

	public ResponseEntity<?> updateBeat(

			HttpServletRequest request,

			@RequestParam int beatId,

			@RequestParam String planData) throws Exception {

		Beat beat = beatRepository.selectById(beatId);

		if (beat == null) return responseSender.badRequest("Beat not found");

		Gson gson = new Gson();

		Type type = new TypeToken<Map<String, Object>>() {

		}.getType();

		Map<String, Object> plan = gson.fromJson(planData, type);

		List<Map<String, Object>> days = (List<Map<String, Object>>) plan.get("days");

		if (days == null || days.isEmpty()) return responseSender.badRequest("No days provided");

		// Update name if changed (and not colliding with another beat)

		String newName = plan.get("beatName") != null ? ((String) plan.get("beatName")).trim() : beat.getName();

		if (newName != null && !newName.equalsIgnoreCase(beat.getName())) {

			// Make sure no other beat for this user already uses this name

			boolean collides = beatRepository.selectByAuthUserId(beat.getAuthUserId()).stream()

					.anyMatch(b -> b.getId() != beat.getId()

							&& b.getName() != null

							&& newName.equalsIgnoreCase(b.getName().trim()));

			if (collides) return responseSender.badRequest("Another beat with this name already exists");

			beat.setName(newName);

		}

		// Update start location from first day if present

		Map<String, Object> firstDay = days.get(0);

		if (firstDay.get("startLocationName") != null)

			beat.setStartLocationName((String) firstDay.get("startLocationName"));

		if (firstDay.get("startLatitude") != null) beat.setStartLatitude((String) firstDay.get("startLatitude"));

		if (firstDay.get("startLongitude") != null) beat.setStartLongitude((String) firstDay.get("startLongitude"));

		beat.setTotalDays(days.size());

		// Replace routes (partner stops). Schedules stay intact.

		beatRouteRepository.deleteByBeatId(beatId);

		// Collect lead IDs the user kept on the plan (for date-aware edit below)

		Set<Integer> keptLeadIds = new HashSet<>();

		for (int d = 0; d < days.size(); d++) {

			Map<String, Object> day = days.get(d);

			int dayNumber = d + 1;

			List<Map<String, Object>> visits = (List<Map<String, Object>>) day.get("visits");

			if (visits == null) continue;

			int partnerSeq = 0;

			for (int i = 0; i < visits.size(); i++) {

				Map<String, Object> v = visits.get(i);

				if ("lead".equals(v.get("type"))) {

					keptLeadIds.add(((Number) v.get("id")).intValue());

					continue; // leads live in lead_route, handled below

				}

				BeatRoute route = new BeatRoute();

				route.setBeatId(beatId);

				route.setFofoId(((Number) v.get("id")).intValue());

				route.setSequenceOrder(partnerSeq++);

				route.setDayNumber(dayNumber);

				route.setActive(true);

				beatRouteRepository.persist(route);

			}

		}

		// Date-aware lead handling: if planDate is provided, remove lead_route rows

		// for that (beat, date) that the user removed in the editor.

		String planDateStr = (String) plan.get("planDate");

		if (planDateStr != null && !planDateStr.isEmpty()) {

			try {

				LocalDate planDate = LocalDate.parse(planDateStr);

				List<LeadRoute> existing = leadRouteRepository.selectByBeatId(beatId);

				for (LeadRoute lr : existing) {

					if (!"APPROVED".equals(lr.getStatus())) continue;

					if (lr.getScheduleDate() == null || !lr.getScheduleDate().equals(planDate)) continue;

					if (keptLeadIds.contains(lr.getLeadId())) continue;

					// User removed this lead from the date — mark cancelled

					lr.setStatus("CANCELLED");

					lr.setUpdatedTimestamp(LocalDateTime.now());

					// activity log

					LeadActivity a = new LeadActivity();

					a.setLeadId(lr.getLeadId());

					a.setRemark("Removed from beat '" + beat.getName() + "' on " + planDate);

					a.setAuthId(0);

					a.setCreatedTimestamp(LocalDateTime.now());

					leadActivityRepositoryAuto.persist(a);

				}

			} catch (Exception e) {

				LOGGER.warn("Could not parse planDate '{}' — leads not adjusted", planDateStr);

			}

		}

		Map<String, Object> response = new HashMap<>();

		response.put("status", true);

		response.put("planGroupId", String.valueOf(beat.getId()));

		response.put("message", "Beat updated successfully");

		return responseSender.ok(response);

	}