pro.zackpollard.telegrambot.api.chat.message.send.InputFile Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jtelegram-botapi Show documentation

The Java Telegram Bot API is a wrapper for the Telegram HTTP Bot API

There is a newer version: 1.6.5

package pro.zackpollard.telegrambot.api.chat.message.send;

import com.mashape.unirest.http.HttpResponse;
import com.mashape.unirest.http.Unirest;
import com.mashape.unirest.http.exceptions.UnirestException;
import lombok.Getter;
import org.apache.commons.io.FilenameUtils;
import pro.zackpollard.telegrambot.api.TelegramBot;
import pro.zackpollard.telegrambot.api.internal.FileExtension;
import pro.zackpollard.telegrambot.api.internal.managers.FileManager;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.nio.file.CopyOption;
import java.nio.file.Files;
import java.nio.file.StandardCopyOption;

/**
 * @author Zack Pollard
 */
public class InputFile {

	@Getter
	private final String fileID;
	@Getter
	private final File file;
    @Getter
    private final String fileName;

	public InputFile(URL url) {

		File file = TelegramBot.getFileManager().getFile(url);
		if (file == null) {
			try {
				String stringifiedUrl = url.toExternalForm();
				HttpResponse response = Unirest.get(stringifiedUrl).asBinary();
				String extension = FileExtension.getByMimeType(response.getHeaders().getFirst("content-type"));
				if (extension == null) {
					extension = stringifiedUrl.substring(stringifiedUrl.lastIndexOf('.') + 1);
					if (extension.length() > 4) {
						extension = null; // Default to .tmp if there was no valid extension
					}
				}
				file = File.createTempFile("jtb-"+System.currentTimeMillis(), "." + extension, FileManager.getTemporaryFolder());
				file.deleteOnExit();
				TelegramBot.getFileManager().cacheUrl(url, file);
				Files.copy(response.getRawBody(), file.toPath(), StandardCopyOption.REPLACE_EXISTING);
			} catch (UnirestException | IOException ex) {
				ex.printStackTrace();
			}
		}
        this.fileName = FilenameUtils.getBaseName(url.toString()) + "." + FilenameUtils.getExtension(url.toString());
        this.file = file;
		this.fileID = TelegramBot.getFileManager().getFileID(file);
	}

	public InputFile(File file) {

		this.file = file;
        this.fileName = file.getName();
		this.fileID = TelegramBot.getFileManager().getFileID(file);
	}

	public InputFile(String fileID) {

		this.file = null;
        this.fileName = null;
		this.fileID = fileID;
	}
}