/wuffy-notification/src/main/java/net/wuffy/notification/types/TwitchAnnouncement.java

package net.wuffy.notification.types;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import org.bson.Document;

import com.mongodb.client.MongoCollection;
import com.mongodb.client.model.UpdateOneModel;
import com.mongodb.client.model.WriteModel;

import net.wuffy.common.logger.Logger;
import net.wuffy.common.util.GsonUtil;
import net.wuffy.core.twitch.TwitchAPI;
import net.wuffy.core.twitch.response.TwitchResponse;
import net.wuffy.core.twitch.response.TwitchResponseGame;
import net.wuffy.core.twitch.response.TwitchResponseStream;
import net.wuffy.core.twitch.response.TwitchResponseUser;
import net.wuffy.notification.WebhookQueue;
import net.wuffy.notification.Wuffy;

public class TwitchAnnouncement extends Announcement {

	private static final String DEFAULT_EMBED_MESSAGE = "{\"username\":\"Twitch\",\"avatar_url\":\"https://wuffy.eu/pictures/example_avatar_300x300.png\",\"content\":\"@here **%n** ist jetzt live auf Twitch\",\"embeds\":[{\"title\":\"%t\",\"url\":\"%urll\",\"color\":6570405,\"timestamp\":\"%sa\",\"thumbnail\":{\"url\":\"%urlg300x300\"},\"image\":{\"url\":\"%urlt580x900\"},\"author\":{\"name\":\"%n\",\"url\":\"%urll\",\"icon_url\":\"%urlpi\"},\"footer\":{\"icon_url\":\"%urlpi\",\"text\":\"%n\"},\"fields\":[{\"name\":\"Game\",\"value\":\"%g\",\"inline\":true},{\"name\":\"Viewers\",\"value\":\"%vc\",\"inline\":true}]}]}";

	private Map<String, List<Message>> queueNames = new HashMap<String, List<Message>>();

	private final TwitchAPI twitchAPI;

	public TwitchAnnouncement(MongoCollection<Document> guildCollection, TwitchAPI twitchAPI) {
		super(guildCollection, 6000, "Twitch", "TWITCH");

		this.twitchAPI = twitchAPI;
	}

	@Override
	protected void update() {
		try {
			List<String> fetchNames = new ArrayList<String>();

			for(String name : this.queueNames.keySet())
				if(fetchNames.size() > 100)
					break;
				else
					fetchNames.add(name);

			Map<String, List<Message>> messages = new HashMap<String, List<Message>>();

			fetchNames.forEach(name -> messages.put(name, this.queueNames.remove(name)));

			if(fetchNames.isEmpty()) {
				this.running.set(false);
				return;
			}

			Map<String, TwitchResponseUser> users = null;
			TwitchResponse<TwitchResponseStream> streams = null;
			Map<String, TwitchResponseGame> games = null;

			try {
				users = this.twitchAPI.getUserHandler().getByName(fetchNames).data.stream().collect(Collectors.toMap(key -> key.id, value -> value));

				streams = this.twitchAPI.getStreamHandler().getById(users.keySet().stream().collect(Collectors.toList()));

				games = this.twitchAPI.getGameHandler().getById(streams.data.stream()
						.filter(data -> data != null && data.game_id != null && data.type != null && data.type.equals("live"))
						.map(data -> data.game_id)
						.distinct()
						.collect(Collectors.toList())).data.stream()
							.collect(Collectors.toMap(key -> key.id, value -> value));
			} catch(Exception e) {
				Logger.fatal("Announcement twitch", "Error by requesting data. waiting 10 seconds and adding all names to the end of the queue again!", e);

				this.queueNames.putAll(messages);

				Thread.sleep(10000); //Waiting 10 seconds while error
				return;
			}

			List<WriteModel<Document>> writeModels = new ArrayList<WriteModel<Document>>();

			for(TwitchResponseStream stream : streams.data) {
				if(stream == null || stream.type == null || !stream.type.equals("live"))
					continue;

				TwitchResponseUser user = users.get(stream.user_id);

				for(Message message : messages.get(user.display_name.toLowerCase())) {
					if(message.lastUpdate != null && message.lastUpdate.equals(stream.started_at))
						continue;

					//Update started_at to mongodb
					writeModels.add(new UpdateOneModel<Document>(
							new Document("_guildId", message.guildId)
								.append("notification.TWITCH.name", message.name),
							new Document("$set", new Document("notification.TWITCH.$.lastUpdate", stream.started_at))));

					WebhookQueue.add(message.webhook, (message.message == null || message.message.isEmpty() ? DEFAULT_EMBED_MESSAGE : message.message)
						.replace("%vc", Integer.toString(stream.viewer_count))
						.replace("%n", user.display_name)
						.replace("%urll", String.format("https://www.twitch.tv/%s", user.display_name))
						.replace("%t", stream.title)
						.replace("%urlt300x300", stream.thumbnail_url.replace("{width}", "300").replace("{height}", "300"))
						.replace("%urlt580x900", stream.thumbnail_url.replace("{width}", "320").replace("{height}", "180"))
						.replace("%urlpi", user.profile_image_url)
						.replace("%urloi", user.offline_image_url)
						.replace("%sa", stream.started_at)
						.replace("%g", games.get(stream.game_id).name)
						.replace("%urlg300x300", games.get(stream.game_id).box_art_url.replace("{width}", "300").replace("{height}", "300"))
						.replace("%urlg580x900", games.get(stream.game_id).box_art_url.replace("{width}", "320").replace("{height}", "180")));
				}
			}

			if(!writeModels.isEmpty())
				Wuffy.getInstance().getExtensionGuild().getPrintBatchResult().onResult(Wuffy.getInstance().getGuildCollection().bulkWrite(writeModels), null);

			if(this.queueNames.isEmpty())
				this.running.set(false);
		} catch(Exception e) {
			Logger.fatal("Announcement twitch", "Failed to execute. waiting 10 seconds!", e);

			this.queueNames.clear();
			this.running.set(false);

			try {
				Thread.sleep(10000);
			} catch (InterruptedException e2) {
				e2.printStackTrace();
			}
		}
	}

	@Override
	public void _add(String guildId, List<Document> documents) {
		for(Document document : documents) {
			Message message = GsonUtil.GSON.fromJson(document.toJson(), Message.class);

			if(message != null && message.webhook != null && !message.webhook.isEmpty()) {
				message.guildId = guildId;

				if(!this.queueNames.containsKey(message.name.toLowerCase()))
					this.queueNames.put(message.name.toLowerCase(), new ArrayList<Message>());
				this.queueNames.get(message.name.toLowerCase()).add(message);
			}
		}
	}

	@Override
	protected boolean isQueueEmpty() {
		return this.queueNames.isEmpty();
	}

	class Message {
		public String guildId;
		public String name;
		public String webhook;
		public String message;

		public String lastUpdate;

		public Message(String guildId, String name, String webhook, String message, String lastUpdate) {
			this.guildId = guildId;
			this.name = name;
			this.webhook = webhook;
			this.message = message;
			this.lastUpdate = lastUpdate;
		}
	}
}