Subversion Repositories SmartDukaan

		int oldTotalDays = beat.getTotalDays();

		int newTotalDays = days.size();

		// Hard rule: you cannot grow the number of days on an existing beat.

		// If you need more days, create a new beat. (Shrinking is allowed and

		// the schedules for dropped day numbers are cleaned below.)

		if (newTotalDays > oldTotalDays) {

			return responseSender.badRequest(

					"Cannot increase the number of days on an existing beat. "

							+ "Original: " + oldTotalDays + " day(s), tried: " + newTotalDays + " day(s). "

							+ "Please create a new beat for additional days.");

		}

		beat.setTotalDays(newTotalDays);

		// dayNumber > newTotalDays cleanup below if the beat shrank).

		// If the beat shrank, drop schedule rows for day numbers that no longer exist

		if (newTotalDays < oldTotalDays) {

			List<BeatSchedule> currentSchedules = beatScheduleRepository.selectByBeatId(beatId);

			for (BeatSchedule s : currentSchedules) {

				if (s.getDayNumber() > newTotalDays) beatScheduleRepository.delete(s);

			}

		}

		// Process per-lead actions sent from the editor's removed-leads popup.

		// Each entry: {leadId, action: "cancel"|"reschedule", toDate?: "yyyy-MM-dd"}.

		// - cancel: mark the lead's current APPROVED row for this beat as CANCELLED.

		//   on toDate, the whole update fails (so the caller can prompt again).

		int leadsCancelled = 0, leadsRescheduled = 0;

		List<String> leadFailures = new ArrayList<>();

		String removedLeadActionsJson = (String) plan.get("removedLeadActions");

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

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

			}.getType();

			List<Map<String, Object>> actions = gson.fromJson(removedLeadActionsJson, listType);

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

			for (Map<String, Object> act : actions) {

				int leadId = ((Number) act.get("leadId")).intValue();

				String mode = (String) act.get("action");

				// Find this lead's most-recent APPROVED row on this beat

				LeadRoute current = beatLeads.stream()

						.filter(r -> r.getLeadId() == leadId && "APPROVED".equals(r.getStatus()))

						.findFirst().orElse(null);

				if (current == null) continue; // already removed/cancelled; nothing to do

				if ("reschedule".equalsIgnoreCase(mode)) {

						leadFailures.add("Lead " + leadId + ": reschedule date missing");

						continue;

					// Find ANY beat this user has scheduled on toDate

					Beat targetBeat = null;

					Integer targetDayNumber = null;

						List<BeatSchedule> sl = beatScheduleRepository.selectByBeatId(b.getId());

								targetBeat = b;

								targetDayNumber = s.getDayNumber();

								break;

							}

						}

					}

					if (targetBeat == null) {

						return responseSender.badRequest(

										+ ". Pick a different date for lead " + leadId

										+ ", or choose Cancel for it.");

					}

					// Cancel the current attachment to this beat

					current.setUpdatedTimestamp(LocalDateTime.now());

					// Create the new attachment on the target beat/date

					LeadRoute fresh = new LeadRoute();

					fresh.setBeatId(targetBeat.getId());

					fresh.setLeadId(leadId);

					fresh.setNearestStoreId(current.getNearestStoreId());

					fresh.setScheduleDate(toDate);

					fresh.setSequenceOrder(9999); // append; the planner can reorder

					fresh.setStatus("APPROVED");

					fresh.setRequestedBy(current.getRequestedBy());

					fresh.setApprovedBy(current.getApprovedBy());

					fresh.setApprovedTimestamp(LocalDateTime.now());

					fresh.setCreatedTimestamp(LocalDateTime.now());

					leadRouteRepository.persist(fresh);

					LeadActivity la = new LeadActivity();

					la.setLeadId(leadId);

					la.setRemark("Rescheduled from beat '" + beat.getName() + "' to '"

							+ targetBeat.getName() + "' on " + toDateStr + " (day " + targetDayNumber + ")");

					la.setAuthId(0);

					la.setCreatedTimestamp(LocalDateTime.now());

					leadActivityRepositoryAuto.persist(la);

					leadsRescheduled++;

				} else {

					current.setStatus("CANCELLED");

					current.setUpdatedTimestamp(LocalDateTime.now());

					leadsCancelled++;

		response.put("leadsCancelled", leadsCancelled);

		response.put("leadsRescheduled", leadsRescheduled);

		response.put("leadFailures", leadFailures);

				+ (leadsCancelled > 0 ? " (" + leadsCancelled + " lead(s) cancelled)" : "")

				+ (leadsRescheduled > 0 ? " (" + leadsRescheduled + " lead(s) rescheduled)" : ""));

	// Used by the edit-mode "removed leads" popup so the date picker can warn

	// upfront when the user picks a date that has no beat for them.

	@GetMapping(value = "/beatPlan/userBeatsOnDate")

	public ResponseEntity<?> userBeatsOnDate(

			@RequestParam int authUserId,

			@RequestParam String date) {

		LocalDate target;

		try {

			target = LocalDate.parse(date);

		} catch (Exception e) {

			return responseSender.badRequest("Invalid date");

		}

		List<Map<String, Object>> hits = new ArrayList<>();

		List<Beat> userBeats = beatRepository.selectActiveByAuthUserId(authUserId);

		for (Beat b : userBeats) {

			List<BeatSchedule> schedules = beatScheduleRepository.selectByBeatId(b.getId());

			for (BeatSchedule s : schedules) {

				if (s.getStartDate() != null && s.getStartDate().equals(target)) {

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

					m.put("beatId", b.getId());

					m.put("beatName", b.getName());

					m.put("dayNumber", s.getDayNumber());

					hits.add(m);

				}

			}

		}

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

		result.put("date", date);

		result.put("authUserId", authUserId);

		result.put("beats", hits);

		return responseSender.ok(result);

	}

	// Returns the user's DEFAULT base location. Falls back to most-recent for

	// legacy users who pre-date the is_default column.

		result.put("isDefault", baseLoc.isDefault());

		return responseSender.ok(result);

	}

	// Returns ALL BASE locations for a user, default first.

	@GetMapping(value = "/beatPlan/listBaseLocations")

	public ResponseEntity<?> listBaseLocations(@RequestParam int authUserId) {

		List<AuthUserLocation> all = authUserLocationRepository.selectAllByAuthUserIdAndType(authUserId, "BASE");

		// Default at the top, then by created desc (the repo already returns desc).

		all.sort((a, b) -> {

			if (a.isDefault() && !b.isDefault()) return -1;

			if (!a.isDefault() && b.isDefault()) return 1;

			return 0;

		});

		List<Map<String, Object>> rows = new ArrayList<>();

		for (AuthUserLocation l : all) {

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

			row.put("id", l.getId());

			row.put("locationName", l.getLocationName());

			row.put("latitude", l.getLatitude());

			row.put("longitude", l.getLongitude());

			row.put("address", l.getAddress());

			row.put("isDefault", l.isDefault());

			row.put("createdTimestamp", l.getCreatedTimestamp() != null ? l.getCreatedTimestamp().toString() : null);

			rows.add(row);

		}

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

		result.put("authUserId", authUserId);

		result.put("locations", rows);

		return responseSender.ok(result);

	}

	// Flip the default flag — set this id default, clear all others.

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

	public ResponseEntity<?> setDefaultBaseLocation(

			HttpServletRequest request,

			@RequestParam int id) throws ProfitMandiBusinessException {

		LoginDetails ld = cookiesProcessor.getCookiesObject(request);

		AuthUser me = authRepository.selectByEmailOrMobile(ld.getEmailId());

		if (me == null) return responseSender.unauthorized("Not logged in");

		if (!isBaseLocationManager(me)) {

			return responseSender.forbidden("You are not authorized for this action. Only Sales L3 and above can manage base locations.");

		}

		AuthUserLocation target = authUserLocationRepository.selectById(id);

		if (target == null) return responseSender.badRequest("Location not found");

		List<AuthUserLocation> all = authUserLocationRepository.selectAllByAuthUserIdAndType(target.getAuthUserId(), "BASE");

		for (AuthUserLocation l : all) {

			boolean shouldBeDefault = (l.getId() == id);

			if (l.isDefault() != shouldBeDefault) {

				l.setDefault(shouldBeDefault);

				authUserLocationRepository.persist(l); // saveOrUpdate

			}

		}

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

		result.put("status", true);

		result.put("id", id);

		result.put("message", "Default base location updated");

	// Delete a base location. The DEFAULT one cannot be deleted — user must

	// first pick another row as default.

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

	public ResponseEntity<?> deleteBaseLocation(

			HttpServletRequest request,

			@RequestParam int id) throws ProfitMandiBusinessException {

		LoginDetails ld = cookiesProcessor.getCookiesObject(request);

		AuthUser me = authRepository.selectByEmailOrMobile(ld.getEmailId());

		if (me == null) return responseSender.unauthorized("Not logged in");

		if (!isBaseLocationManager(me)) {

			return responseSender.forbidden("You are not authorized for this action. Only Sales L3 and above can manage base locations.");

		}

		AuthUserLocation target = authUserLocationRepository.selectById(id);

		if (target == null) return responseSender.badRequest("Location not found");

		if (target.isDefault()) {

			return responseSender.badRequest("Default base location cannot be removed. Set another location as default first.");

		}

		authUserLocationRepository.delete(target);

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

		result.put("status", true);

		result.put("message", "Base location removed");

		return responseSender.ok(result);

	}

	// Shared permission check for base-location admin actions.

	private boolean isBaseLocationManager(AuthUser me) {

		Set<String> baseLocationAllowList = new HashSet<>(Arrays.asList(

				"tarun.verma@smartdukaan.com"

		));

		String myEmail = me.getEmailId() != null ? me.getEmailId().toLowerCase() : "";

		if (baseLocationAllowList.contains(myEmail)) return true;

		return csService.getAuthUserIds(

						com.spice.profitmandi.common.model.ProfitMandiConstants.TICKET_CATEGORY_SALES,

						Arrays.asList(EscalationType.L3, EscalationType.L4))

				.stream().anyMatch(u -> u.getId() == me.getId());

	}

		// First BASE for this user → auto-default so every user always has one.

		List<AuthUserLocation> existing = authUserLocationRepository.selectAllByAuthUserIdAndType(authUserId, "BASE");

		boolean noExistingDefault = existing.stream().noneMatch(AuthUserLocation::isDefault);

		loc.setDefault(existing.isEmpty() || noExistingDefault);

		result.put("isDefault", loc.isDefault());

	// Adds a new base location for the user. Caller can request this new row

	// becomes the default. If the user has NO base locations yet, the new row

	// is auto-defaulted (so every user always has exactly one default).

			@RequestParam(required = false, defaultValue = "false") boolean isDefault) throws Exception {

		List<AuthUserLocation> existing = authUserLocationRepository.selectAllByAuthUserIdAndType(authUserId, "BASE");

		boolean noExistingDefault = existing.stream().noneMatch(AuthUserLocation::isDefault);

		boolean makeDefault = isDefault || existing.isEmpty() || noExistingDefault;

		// If this new row becomes the default, clear any existing default.

		if (makeDefault) {

			for (AuthUserLocation e : existing) {

				if (e.isDefault()) {

					e.setDefault(false);

					authUserLocationRepository.persist(e);

				}

			}

		}

		loc.setDefault(makeDefault);

		result.put("message", makeDefault ? "Base location added and set as default" : "Base location added");