package net.minecraft.world.entity.schedule;

import com.google.common.collect.Maps;
import java.util.Comparator;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.stream.Collectors;
import net.minecraft.core.Registry;
import net.minecraft.core.registries.BuiltInRegistries;

public class Schedule {
	public static final int WORK_START_TIME = 2000;
	public static final int TOTAL_WORK_TIME = 7000;
	public static final Schedule EMPTY = register("empty").changeActivityAt(0, Activity.IDLE).build();
	public static final Schedule SIMPLE = register("simple").changeActivityAt(5000, Activity.WORK).changeActivityAt(11000, Activity.REST).build();
	public static final Schedule VILLAGER_BABY = register("villager_baby")
		.changeActivityAt(10, Activity.IDLE)
		.changeActivityAt(3000, Activity.PLAY)
		.changeActivityAt(6000, Activity.IDLE)
		.changeActivityAt(10000, Activity.PLAY)
		.changeActivityAt(12000, Activity.REST)
		.build();
	public static final Schedule VILLAGER_DEFAULT = register("villager_default")
		.changeActivityAt(10, Activity.IDLE)
		.changeActivityAt(2000, Activity.WORK)
		.changeActivityAt(9000, Activity.MEET)
		.changeActivityAt(11000, Activity.IDLE)
		.changeActivityAt(12000, Activity.REST)
		.build();
	private final Map<Activity, Timeline> timelines = Maps.<Activity, Timeline>newHashMap();

	protected static ScheduleBuilder register(String key) {
		Schedule schedule = Registry.register(BuiltInRegistries.SCHEDULE, key, new Schedule());
		return new ScheduleBuilder(schedule);
	}

	protected void ensureTimelineExistsFor(Activity activity) {
		if (!this.timelines.containsKey(activity)) {
			this.timelines.put(activity, new Timeline());
		}
	}

	protected Timeline getTimelineFor(Activity activity) {
		return (Timeline)this.timelines.get(activity);
	}

	protected List<Timeline> getAllTimelinesExceptFor(Activity activity) {
		return (List<Timeline>)this.timelines.entrySet().stream().filter(entry -> entry.getKey() != activity).map(Entry::getValue).collect(Collectors.toList());
	}

	public Activity getActivityAt(int dayTime) {
		return (Activity)this.timelines
			.entrySet()
			.stream()
			.max(Comparator.comparingDouble(entry -> ((Timeline)entry.getValue()).getValueAt(dayTime)))
			.map(Entry::getKey)
			.orElse(Activity.IDLE);
	}
}