/*
 *    Copyright (c) 2018-2025, ztt All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 * Redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution.
 * Neither the name of the pig4cloud.com developer nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 * Author: ztt
 */
package com.chinaztt.mes.aps.service.impl;

import cn.hutool.core.collection.CollectionUtil;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.chinaztt.mes.aps.entity.TimetableException;
import com.chinaztt.mes.aps.entity.WeekTimetable;
import com.chinaztt.mes.aps.mapper.TimetableExceptionMapper;
import com.chinaztt.mes.aps.mapper.WeekTimetableMapper;
import com.chinaztt.mes.aps.service.TimetableService;
import com.chinaztt.mes.common.util.JsonUtils;
import com.chinaztt.ztt.common.core.util.R;
import lombok.AllArgsConstructor;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.time.LocalTime;
import java.util.*;
import java.util.stream.Collectors;

/**
 * 工作日历
 *
 * @author zhangxy
 * @date 2020-09-21 14:40:55
 */
@Service
@AllArgsConstructor
@Transactional(rollbackFor = Exception.class)
public class TimetableServiceImpl extends ServiceImpl<WeekTimetableMapper, WeekTimetable> implements TimetableService {

	private JdbcTemplate jdbcTemplate;

	private TimetableExceptionMapper timetableExceptionMapper;


	@Override
	public R<Boolean> save(List<WeekTimetable> list) {
		jdbcTemplate.execute("   truncate table aps_week_timetable");

		if (CollectionUtil.isNotEmpty(list)) {
			Iterator<WeekTimetable> it = list.iterator();
			while (it.hasNext()) {
				WeekTimetable ce = it.next();
				if (CollectionUtil.isEmpty(ce.getRange())) {
					it.remove();
					continue;
				}
				ce.setFromTime(ce.getRange().get(0));
				ce.setToTime(ce.getRange().get(1));
			}

			Map<String, List<WeekTimetable>> map = list.stream().collect(Collectors.groupingBy(WeekTimetable::getDayOfWeek));

			map.values().forEach(groupedList -> {
				List<WeekTimetable> mergedList = merge(groupedList);
				mergedList.forEach(w -> baseMapper.insert(w));
			});
		}
		return R.ok();
	}

	@Override
	public R<Boolean> saveEx(List<TimetableException> list) {
		if (CollectionUtil.isNotEmpty(list)) {
			Iterator<TimetableException> it = list.iterator();
			while (it.hasNext()) {
				TimetableException ce = it.next();
				if (CollectionUtil.isEmpty(ce.getRange())) {
					it.remove();
					continue;
				}
				ce.setFromDate(ce.getRange().get(0));
				ce.setToDate(ce.getRange().get(1));

				if (ce.getId() == null || ce.getId() == 0) {
					timetableExceptionMapper.insert(ce);
				} else {
					timetableExceptionMapper.updateById(ce);
				}
			}
		}
		return null;
	}

	@Override
	public JSONObject getAll() {
		List<WeekTimetable> list = baseMapper.selectList(null);
		if (CollectionUtil.isNotEmpty(list)) {
			for (WeekTimetable week : list) {
				week.setRange(Arrays.asList(week.getFromTime(), week.getToTime()));
			}
			return JsonUtils.toJsonObject(list.stream().collect(Collectors.groupingBy(WeekTimetable::getDayOfWeek)));
		}
		return null;
	}

	@Override
	public List<TimetableException> getAllEx() {
		List<TimetableException> list = timetableExceptionMapper.selectList(null);
		if (CollectionUtil.isNotEmpty(list)) {
			for (TimetableException ex : list) {
				ex.setRange(Arrays.asList(ex.getFromDate(), ex.getToDate()));
			}
			return list;
		}
		return null;
	}

	@Override
	public R<Boolean> removeExById(Long id) {
		timetableExceptionMapper.deleteById(id);
		return R.ok();
	}


	/**
	 * 合并重叠时间
	 *
	 * @param input
	 * @return
	 */
	public static List<WeekTimetable> merge(List<WeekTimetable> input) {
		ArrayList<WeekTimetable> result = new ArrayList<>();
		if (CollectionUtil.isEmpty(input)) {
			return result;
		}
		Collections.sort(input);
		WeekTimetable prev = input.get(0);
		for (int i = 1; i < input.size(); i++) {
			WeekTimetable curr = input.get(i);
			if (prev.getToTime().compareTo(curr.getFromTime()) == 1) {
				LocalTime end = prev.getToTime().compareTo(curr.getToTime()) == 1 ? prev.getToTime() : curr.getToTime();
				WeekTimetable temp = new WeekTimetable();
				temp.setFromTime(prev.getFromTime());
				temp.setToTime(end);
				temp.setDayOfWeek(prev.getDayOfWeek());
				prev = temp;
			} else {
				result.add(prev);
				prev = curr;
			}
		}
		result.add(prev);
		return result;
	}

}