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

package net.wuffy.notification.types;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.bson.Document;

import com.google.api.services.youtube.model.Channel;
import com.google.api.services.youtube.model.SearchResult;
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.youtube.YoutubeAPI;
import net.wuffy.core.youtube.YoutubeOptionalParameterBuilder;
import net.wuffy.core.youtube.YoutubeOptionalParameters;
import net.wuffy.core.youtube.YoutubePart;
import net.wuffy.notification.WebhookQueue;
import net.wuffy.notification.Wuffy;

public class YoutubeAnnouncement extends Announcement {

	private static final String DEFAULT_EMBED_MESSAGE = "{\"username\":\"Youtube\",\"avatar_url\":\"%tm300x300\",\"embeds\":[{\"title\":\"%ti\",\"image\":{\"url\":\"%tm580x900\"},\"url\":\"%u\",\"color\":16711680,\"timestamp\":\"%pa\",\"footer\":{\"icon_url\":\"%tm300x300\",\"text\":\"%ct\"},\"author\":{\"name\":\"%ct\",\"url\":\"%u\",\"icon_url\":\"%tm300x300\"}}]}";

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

	private final YoutubeAPI youtubeAPI;

	public YoutubeAnnouncement(MongoCollection<Document> guildCollection, YoutubeAPI youtubeAPI) {
		super(guildCollection, 4000, "Youtube", "YOUTUBE");

		this.youtubeAPI = youtubeAPI;
	}

	@Override
	protected void update() {
		try {
			if(this.queueNames.isEmpty()) {
				this.running.set(false);
				return;
			}

			String id = this.queueNames.keySet().toArray()[0].toString();
			List<Message> messages = this.queueNames.remove(id);

			Channel channel = null;
			List<SearchResult> searches = null;

			try {
				channel = this.youtubeAPI.getChannelHandler().getById(id,
						YoutubePart.ID,
						YoutubePart.SNIPPET).getItems().get(0);

				searches = this.youtubeAPI.getSearchHandler().get(new YoutubeOptionalParameterBuilder()
						.add(YoutubeOptionalParameters.CHANNELID, id)
						.add(YoutubeOptionalParameters.ORDER, "date")
						.add(YoutubeOptionalParameters.TYPE, "video")
						.add(YoutubeOptionalParameters.MAXRESULTS, 5),
						YoutubePart.SNIPPET).getItems();
			} catch(Exception e) {
				Logger.fatal("Announcement youtube", "Error by requesting data. waiting 10 seconds and adding id to the end of the queue again!", e);

				this.queueNames.put(id, messages);

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

			if(searches == null || searches.isEmpty())
				return;

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

			for(Message message : messages) {
				String last = null;
				String lastLive = null;

				List<String> webhookMessages = null;

				for(SearchResult result : searches) {
					if(result.getSnippet().getLiveBroadcastContent().equals("live")) {
						if(message.lastUpdateLiveStream != null && message.lastUpdateLiveStream.equals(result.getId().getVideoId()))
							break;

						if(lastLive == null)
							lastLive = result.getId().getVideoId();
					} else {
						if(message.lastUpdate != null && message.lastUpdate.equals(result.getId().getVideoId()))
							break;

						if(last == null)
							last = result.getId().getVideoId();
					}

					if(webhookMessages == null)
						webhookMessages = new ArrayList<String>();

					webhookMessages.add((message.message == null || message.message.isEmpty() ? DEFAULT_EMBED_MESSAGE : message.message)
							.replace("%ct", result.getSnippet().getChannelTitle())
							.replace("%ti", result.getSnippet().getTitle())
							.replace("%pa", result.getSnippet().getPublishedAt().toStringRfc3339())
							.replace("%d", result.getSnippet().getDescription())
							.replace("%u", String.format("https://www.youtube.com/watch?v=%s", result.getId().getVideoId()))
							.replace("%td300x300", result.getSnippet().getThumbnails().getDefault().setWidth(300L).setHeight(300L).getUrl())
							.replace("%tm300x300", result.getSnippet().getThumbnails().getMedium().setWidth(300L).setHeight(300L).getUrl())
							.replace("%th300x300", result.getSnippet().getThumbnails().getHigh().setWidth(300L).setHeight(300L).getUrl())
							.replace("%td580x900", result.getSnippet().getThumbnails().getDefault().setWidth(580L).setHeight(900L).getUrl())
							.replace("%tm580x900", result.getSnippet().getThumbnails().getMedium().setWidth(580L).setHeight(900L).getUrl())
							.replace("%th580x900", result.getSnippet().getThumbnails().getHigh().setWidth(580L).setHeight(900L).getUrl())
							.replace("%tcd300x300", channel.getSnippet().getThumbnails().getDefault().setWidth(300L).setHeight(300L).getUrl())
							.replace("%tcm300x300", channel.getSnippet().getThumbnails().getMedium().setWidth(300L).setHeight(300L).getUrl())
							.replace("%tch300x300", channel.getSnippet().getThumbnails().getHigh().setWidth(300L).setHeight(300L).getUrl())
							.replace("%tcd580x900", channel.getSnippet().getThumbnails().getDefault().setWidth(580L).setHeight(900L).getUrl())
							.replace("%tcm580x900", channel.getSnippet().getThumbnails().getMedium().setWidth(580L).setHeight(900L).getUrl())
							.replace("%tch580x900", channel.getSnippet().getThumbnails().getHigh().setWidth(580L).setHeight(900L).getUrl()));

					if(message.lastUpdate == null)
						break;
				}

				if(last != null)
					writeModels.add(new UpdateOneModel<Document>(
							new Document("_guildId", message.guildId)
								.append("notification.YOUTUBE.channelId", message.channelId),
							new Document("$set", new Document("notification.YOUTUBE.$.lastUpdate", last))));

				if(lastLive != null)
					writeModels.add(new UpdateOneModel<Document>(
							new Document("_guildId", message.guildId)
								.append("notification.YOUTUBE.channelId", message.channelId),
							new Document("$set", new Document("notification.YOUTUBE.$.lastUpdateLiveStream", lastLive))));

				if(webhookMessages != null && !webhookMessages.isEmpty()) {
					if(webhookMessages.size() > 1)
						Collections.reverse(webhookMessages);

					for(String webhookMessage : webhookMessages)
						WebhookQueue.add(message.webhook, webhookMessage);
				}
			}

			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 youtube", "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(message.channelId == null)
					continue;

				if(!this.queueNames.containsKey(message.channelId))
					this.queueNames.put(message.channelId, new ArrayList<Message>());
				this.queueNames.get(message.channelId).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 String lastUpdateLiveStream;
		public String channelId;

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