TelegramRequest.java

package com.github.stickerifier.stickerify.telegram.model;

import static com.github.stickerifier.stickerify.telegram.Answer.ABOUT;
import static com.github.stickerifier.stickerify.telegram.Answer.HELP;
import static com.github.stickerifier.stickerify.telegram.Answer.PRIVACY_POLICY;
import static java.util.Comparator.comparing;

import com.fasterxml.jackson.annotation.JsonProperty;
import com.github.stickerifier.stickerify.telegram.Answer;
import com.pengrad.telegrambot.model.Document;
import com.pengrad.telegrambot.model.LivePhoto;
import com.pengrad.telegrambot.model.Message;
import com.pengrad.telegrambot.model.PhotoSize;
import com.pengrad.telegrambot.model.Sticker;
import com.pengrad.telegrambot.model.Video;
import com.pengrad.telegrambot.model.VideoNote;
import org.jspecify.annotations.Nullable;

import java.util.Arrays;
import java.util.Objects;
import java.util.Optional;
import java.util.stream.Stream;

/**
 * Data class wrapping a {@link Message} instance to represent a Telegram request.
 *
 * @param message the message to wrap
 */
public record TelegramRequest(Message message) {

	private static final String START_COMMAND = "/start";
	private static final String HELP_COMMAND = "/help";
	private static final String PRIVACY_COMMAND = "/privacy";

	public @Nullable TelegramFile getFile() {
		return getMessageMedia()
				.map(media -> switch (media) {
					case LivePhoto livePhoto -> new TelegramFile(livePhoto.fileId(), livePhoto.fileSize());
					case PhotoSize[] photos when photos.length > 0 -> getBestPhoto(photos);
					case Document document -> new TelegramFile(document.fileId(), document.fileSize());
					case Sticker sticker -> new TelegramFile(sticker.fileId(), sticker.fileSize());
					case Video video -> new TelegramFile(video.fileId(), video.fileSize());
					case VideoNote videoNote -> new TelegramFile(videoNote.fileId(), videoNote.fileSize());
					default -> TelegramFile.NOT_SUPPORTED;
				})
				.orElse(null);
	}

	private Optional<?> getMessageMedia() {
		return Stream.of(message.livePhoto(), message.photo(), message.document(), message.sticker(),
						message.video(), message.videoNote(),
						message.audio(), message.voice())
				.filter(Objects::nonNull)
				.findFirst();
	}

	private TelegramFile getBestPhoto(PhotoSize[] photos) {
		return Arrays.stream(photos)
				.map(photo -> new TelegramFile(photo.fileId(), photo.fileSize()))
				.filter(TelegramFile::canBeDownloaded)
				.max(comparing(TelegramFile::sizeValue))
				.orElse(TelegramFile.TOO_LARGE);
	}

	public long getChatId() {
		return message.chat().id();
	}

	public Integer getMessageId() {
		return message.messageId();
	}

	private Long getUserId() {
		return message.from().id();
	}

	private boolean isNewUser() {
		return START_COMMAND.equals(message.text());
	}

	public Answer getAnswerMessage() {
		return switch (message.text()) {
			case HELP_COMMAND, START_COMMAND -> HELP;
			case PRIVACY_COMMAND -> PRIVACY_POLICY;
			case null, default -> ABOUT;
		};
	}

	public RequestDetails toRequestDetails() {
		return new RequestDetails(getUserId(), isNewUser());
	}

	@Override
	public String toString() {
		var file = Optional.ofNullable(getFile()).map(TelegramFile::id).orElse(null);
		var text = Optional.ofNullable(message.text()).orElse(message.caption());

		return "request ["
				+ "chat=" + getChatId()
				+ ", from=" + getUserId()
				+ writeIfNotEmpty("file", file)
				+ writeIfNotEmpty("text", text)
				+ "]";
	}

	private static String writeIfNotEmpty(String field, @Nullable String value) {
		return value != null && !value.isEmpty()
				? ", " + field + "=" + value
				: "";
	}

	public record RequestDetails(@JsonProperty("user_id") Long userId, @JsonProperty("new_user") boolean isNewUser) {}
}