content: /*
   * Licensed to the Apache Software Foundation (ASF) under one or more
   * contributor license agreements.  See the NOTICE file distributed with
   * this work for additional information regarding copyright ownership.
   * The ASF licenses this file to You under the Apache License, Version 2.0
   * (the "License"); you may not use this file except in compliance with
   * the License.  You may obtain a copy of the License at
   *
   *      http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package org.apache.camel.component.telegram.springboot;
  
  import java.io.IOException;
  import java.nio.charset.StandardCharsets;
  import java.util.Collections;
  
  import org.apache.camel.Endpoint;
  import org.apache.camel.EndpointInject;
  import org.apache.camel.Exchange;
  import org.apache.camel.builder.RouteBuilder;
  import org.apache.camel.component.telegram.TelegramConstants;
  import org.apache.camel.component.telegram.TelegramMediaType;
  import org.apache.camel.component.telegram.TelegramParseMode;
  import org.apache.camel.component.telegram.model.ForceReply;
  import org.apache.camel.component.telegram.model.InlineKeyboardButton;
  import org.apache.camel.component.telegram.model.InlineKeyboardMarkup;
  import org.apache.camel.component.telegram.model.OutgoingGameMessage;
  import org.apache.camel.component.telegram.model.OutgoingTextMessage;
  import org.apache.camel.component.telegram.model.ReplyKeyboardMarkup;
  import org.apache.camel.component.telegram.model.ReplyKeyboardRemove;
  import org.apache.camel.component.telegram.springboot.TelegramMockRoutes.MockProcessor;
  import org.apache.camel.spring.boot.CamelAutoConfiguration;
  
  import org.junit.jupiter.api.Disabled;
  import org.junit.jupiter.api.Test;
  
  import static org.junit.jupiter.api.Assertions.assertEquals;
  import static org.junit.jupiter.api.Assertions.assertNotNull;
  import static org.junit.jupiter.api.Assertions.assertNull;
  import static org.junit.jupiter.api.Assertions.assertTrue;
  
  import org.springframework.boot.test.context.SpringBootTest;
  import org.springframework.context.annotation.Bean;
  import org.springframework.context.annotation.Configuration;
  import org.springframework.test.annotation.DirtiesContext;
  import org.springframework.test.annotation.DirtiesContext.ClassMode;
  import org.apache.camel.test.spring.junit5.CamelSpringBootTest;
  
  @DirtiesContext(classMode = ClassMode.AFTER_EACH_TEST_METHOD)
  @CamelSpringBootTest
  @SpringBootTest(classes = { CamelAutoConfiguration.class, TelegramProducerMediaTest.class,
          TelegramProducerMediaTest.TestConfiguration.class })
  public class TelegramProducerMediaTest extends TelegramTestSupport {
  
      static TelegramMockRoutes mockRoutes;
  
      @EndpointInject("direct:telegram")
      private Endpoint endpoint;
  
      @Test
      public void testRouteWithPngImage() throws Exception {
  
          final MockProcessor<byte[]> mockProcessor = mockRoutes.getMock("sendPhoto");
          mockProcessor.clearRecordedMessages();
  
          Exchange ex = endpoint.createExchange();
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TITLE_CAPTION, "Photo");
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TYPE, TelegramMediaType.PHOTO_PNG.name());
          byte[] image = TelegramTestUtil.createSampleImage("PNG");
          ex.getIn().setBody(image);
  
          template.send(endpoint, ex);
  
          /* message contains a multipart body */
          final byte[] message = mockProcessor.awaitRecordedMessages(1, 5000).get(0);
          assertMultipartText(message, "chat_id", "my-id");
          assertTrue(contains(message, image));
          assertMultipartFilename(message, "photo", "photo.png");
          assertMultipartText(message, "caption", "Photo");
      }
  
      @Test
      public void testRouteWithJpgImage() throws Exception {
  
          final MockProcessor<byte[]> mockProcessor = mockRoutes.getMock("sendPhoto");
          mockProcessor.clearRecordedMessages();
  
          Exchange ex = endpoint.createExchange();
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TITLE_CAPTION, "Photo");
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TYPE, TelegramMediaType.PHOTO_JPG); // without using
                                                                                                    // .name()
          byte[] image = TelegramTestUtil.createSampleImage("JPG");
          ex.getIn().setBody(image);
  
          template.send(endpoint, ex);
  
          final byte[] message = mockProcessor.awaitRecordedMessages(1, 5000).get(0);
          assertMultipartText(message, "chat_id", "my-id");
          assertTrue(contains(message, image));
          assertMultipartFilename(message, "photo", "photo.jpg");
          assertMultipartText(message, "caption", "Photo");
      }
  
      @Test
      public void testRouteWithJpgImageAndForceReply() throws IOException {
          final MockProcessor<byte[]> mockProcessor = mockRoutes.getMock("sendPhoto");
          mockProcessor.clearRecordedMessages();
  
          Exchange ex = endpoint.createExchange();
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TITLE_CAPTION, "Photo");
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TYPE, TelegramMediaType.PHOTO_JPG);
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_MARKUP, new ForceReply(true));
  
          byte[] image = TelegramTestUtil.createSampleImage("JPG");
          ex.getIn().setBody(image);
  
          template.send(endpoint, ex);
  
          final byte[] message = mockProcessor.awaitRecordedMessages(1, 5000).get(0);
  
          assertMultipartText(message, "chat_id", "my-id");
          assertTrue(contains(message, image));
          assertMultipartFilename(message, "photo", "photo.jpg");
          assertMultipartText(message, "caption", "Photo");
          assertMultipartText(message, "reply_markup", new ForceReply(true).toJson());
      }
  
      @Test
      public void testRouteWithJpgAndReplyKeyboardRemove() throws IOException {
          final MockProcessor<byte[]> mockProcessor = mockRoutes.getMock("sendPhoto");
          mockProcessor.clearRecordedMessages();
  
          Exchange ex = endpoint.createExchange();
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TITLE_CAPTION, "Photo");
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TYPE, TelegramMediaType.PHOTO_JPG);
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_MARKUP, new ReplyKeyboardRemove(true));
  
          byte[] image = TelegramTestUtil.createSampleImage("JPG");
          ex.getIn().setBody(image);
  
          template.send(endpoint, ex);
  
          final byte[] message = mockProcessor.awaitRecordedMessages(1, 5000).get(0);
  
          assertMultipartText(message, "chat_id", "my-id");
          assertTrue(contains(message, image));
          assertMultipartFilename(message, "photo", "photo.jpg");
          assertMultipartText(message, "caption", "Photo");
          assertMultipartText(message, "reply_markup", new ReplyKeyboardRemove(true).toJson());
      }
  
      @Test
      public void testRouteWithJpgAndInlineKeyboardMarkup() throws IOException {
          final MockProcessor<byte[]> mockProcessor = mockRoutes.getMock("sendPhoto");
          mockProcessor.clearRecordedMessages();
  
          InlineKeyboardMarkup ik = InlineKeyboardMarkup.builder().addRow(Collections
                  .singletonList(InlineKeyboardButton.builder().text("test").url("https://camel.apache.org").build()))
                  .build();
  
          Exchange ex = endpoint.createExchange();
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TITLE_CAPTION, "Photo");
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TYPE, TelegramMediaType.PHOTO_JPG);
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_MARKUP, ik);
  
          byte[] image = TelegramTestUtil.createSampleImage("JPG");
          ex.getIn().setBody(image);
  
          template.send(endpoint, ex);
  
          final byte[] message = mockProcessor.awaitRecordedMessages(1, 5000).get(0);
  
          assertMultipartText(message, "chat_id", "my-id");
          assertTrue(contains(message, image));
          assertMultipartFilename(message, "photo", "photo.jpg");
          assertMultipartText(message, "caption", "Photo");
          assertMultipartText(message, "reply_markup", ik.toJson());
      }
  
      @Test
      public void testRouteWithAudio() throws Exception {
          final MockProcessor<byte[]> mockProcessor = mockRoutes.getMock("sendAudio");
          mockProcessor.clearRecordedMessages();
  
          Exchange ex = endpoint.createExchange();
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TITLE_CAPTION, "Audio");
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TYPE, TelegramMediaType.AUDIO);
          byte[] audio = TelegramTestUtil.createSampleAudio();
          ex.getIn().setBody(audio);
  
          template.send(endpoint, ex);
  
          final byte[] message = mockProcessor.awaitRecordedMessages(1, 5000).get(0);
  
          assertMultipartText(message, "chat_id", "my-id");
          assertTrue(contains(message, audio));
          assertMultipartFilename(message, "audio", "audio.mp3");
          assertMultipartText(message, "title", "Audio");
      }
  
      @Test
      public void testRouteWithAudioAndReplyMarkup() throws Exception {
          final MockProcessor<byte[]> mockProcessor = mockRoutes.getMock("sendAudio");
          mockProcessor.clearRecordedMessages();
  
          Exchange ex = endpoint.createExchange();
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TITLE_CAPTION, "Audio");
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TYPE, TelegramMediaType.AUDIO);
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_MARKUP, new ForceReply(true));
          byte[] audio = TelegramTestUtil.createSampleAudio();
          ex.getIn().setBody(audio);
  
          template.send(endpoint, ex);
  
          final byte[] message = mockProcessor.awaitRecordedMessages(1, 5000).get(0);
  
          assertMultipartText(message, "chat_id", "my-id");
          assertTrue(contains(message, audio));
          assertMultipartFilename(message, "audio", "audio.mp3");
          assertMultipartText(message, "title", "Audio");
          assertMultipartText(message, "reply_markup", new ForceReply(true).toJson());
      }
  
      @Test
      public void testRouteWithVideo() throws Exception {
          final MockProcessor<byte[]> mockProcessor = mockRoutes.getMock("sendVideo");
          mockProcessor.clearRecordedMessages();
  
          Exchange ex = endpoint.createExchange();
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TITLE_CAPTION, "Video");
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TYPE, TelegramMediaType.VIDEO.name());
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_MARKUP, new ForceReply(true));
          byte[] video = TelegramTestUtil.createSampleVideo();
          ex.getIn().setBody(video);
  
          template.send(endpoint, ex);
  
          final byte[] message = mockProcessor.awaitRecordedMessages(1, 5000).get(0);
          assertMultipartText(message, "chat_id", "my-id");
          assertTrue(contains(message, video));
          assertMultipartFilename(message, "video", "video.mp4");
          assertMultipartText(message, "caption", "Video");
          assertMultipartText(message, "reply_markup", new ForceReply(true).toJson());
      }
  
      @Test
      public void testRouteWithDocument() throws Exception {
          final MockProcessor<byte[]> mockProcessor = mockRoutes.getMock("sendDocument");
          mockProcessor.clearRecordedMessages();
  
          Exchange ex = endpoint.createExchange();
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TITLE_CAPTION, "Document");
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TYPE, TelegramMediaType.DOCUMENT);
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_MARKUP, new ForceReply(true));
          byte[] document = TelegramTestUtil.createSampleDocument();
          ex.getIn().setBody(document);
  
          template.send(endpoint, ex);
  
          final byte[] message = mockProcessor.awaitRecordedMessages(1, 5000).get(0);
          assertMultipartText(message, "chat_id", "my-id");
          assertTrue(contains(message, document));
          assertMultipartFilename(message, "document", "file");
          assertMultipartText(message, "caption", "Document");
          assertMultipartText(message, "reply_markup", new ForceReply(true).toJson());
      }
  
      @Test
      public void testRouteWithText() {
          final MockProcessor<OutgoingTextMessage> mockProcessor = mockRoutes.getMock("sendMessage");
          mockProcessor.clearRecordedMessages();
  
          Exchange ex = endpoint.createExchange();
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TYPE, TelegramMediaType.TEXT.name());
          ex.getIn().setBody("Hello");
  
          template.send(endpoint, ex);
  
          final OutgoingTextMessage message = mockProcessor.awaitRecordedMessages(1, 5000).get(0);
          assertEquals("my-id", message.getChatId());
          assertEquals("Hello", message.getText());
          assertNull(message.getParseMode());
      }
  
      @Test
      @Disabled
      public void testRouteWithTextAndCustomKeyBoard() {
          final MockProcessor<OutgoingTextMessage> mockProcessor = mockRoutes.getMock("sendMessage");
          mockProcessor.clearRecordedMessages();
  
          Exchange ex = endpoint.createExchange();
  
          OutgoingTextMessage msg = new OutgoingTextMessage.Builder().text("Hello").build();
          withInlineKeyboardContainingTwoRows(msg);
  
          ex.getIn().setBody(msg);
  
          template.send(endpoint, ex);
  
          final OutgoingTextMessage message = mockProcessor.awaitRecordedMessages(1, 5000).get(0);
          assertEquals("my-id", message.getChatId());
          assertEquals("Hello", message.getText());
          assertEquals(2, ((ReplyKeyboardMarkup) message.getReplyMarkup()).getKeyboard().size());
          assertEquals(true, ((ReplyKeyboardMarkup) message.getReplyMarkup()).getOneTimeKeyboard());
          assertNull(message.getParseMode());
      }
  
      @Test
      public void testRouteWithTextHtml() {
          final MockProcessor<OutgoingTextMessage> mockProcessor = mockRoutes.getMock("sendMessage");
          mockProcessor.clearRecordedMessages();
  
          Exchange ex = endpoint.createExchange();
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TYPE, TelegramMediaType.TEXT.name());
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_PARSE_MODE, TelegramParseMode.HTML.name());
          ex.getIn().setBody("Hello");
  
          template.send(endpoint, ex);
  
          final OutgoingTextMessage message = mockProcessor.awaitRecordedMessages(1, 5000).get(0);
          assertEquals("my-id", message.getChatId());
          assertEquals("Hello", message.getText());
          assertEquals("HTML", message.getParseMode());
      }
  
      @Test
      public void testRouteWithTextMarkdown() {
          final MockProcessor<OutgoingTextMessage> mockProcessor = mockRoutes.getMock("sendMessage");
          mockProcessor.clearRecordedMessages();
  
          Exchange ex = endpoint.createExchange();
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_MEDIA_TYPE, TelegramMediaType.TEXT.name());
          ex.getIn().setHeader(TelegramConstants.TELEGRAM_PARSE_MODE, TelegramParseMode.MARKDOWN);
          ex.getIn().setBody("Hello");
  
          template.send(endpoint, ex);
  
          final OutgoingTextMessage message = mockProcessor.awaitRecordedMessages(1, 5000).get(0);
          assertEquals("my-id", message.getChatId());
          assertEquals("Hello", message.getText());
          assertEquals("Markdown", message.getParseMode());
      }
  
      @Test
      public void testRouteWithGame() {
          final MockProcessor<OutgoingGameMessage> mockProcessor = mockRoutes.getMock("sendGame");
          mockProcessor.clearRecordedMessages();
  
          Exchange ex = endpoint.createExchange();
  
          OutgoingGameMessage msg = new OutgoingGameMessage();
          msg.setGameShortName("shortName");
  
          ex.getIn().setBody(msg);
  
          template.send(endpoint, ex);
  
          final OutgoingGameMessage message = mockProcessor.awaitRecordedMessages(1, 5000).get(0);
          assertNotNull("my-id", message.getChatId());
          assertNotNull("shortName", message.getGameShortName());
      }
  
      // *************************************
      // Config
      // *************************************
  
      @Configuration
      public class TestConfiguration {
          @Bean
          public RouteBuilder routeBuilder() {
              return new RouteBuilder() {
                  @Override
                  public void configure() {
                      from("direct:telegram").to("telegram:bots?authorizationToken=mock-token&chatId=my-id");
                  }
              };
          }
  
      }
  
      @Override
      @Bean
      protected TelegramMockRoutes createMockRoutes() {
          mockRoutes = new TelegramMockRoutes(port)
                  .addEndpoint("sendPhoto", "POST", byte[].class,
                          TelegramTestUtil.stringResource("messages/send-photo.json"))
                  .addEndpoint("sendAudio", "POST", byte[].class,
                          TelegramTestUtil.stringResource("messages/send-audio.json"))
                  .addEndpoint("sendVideo", "POST", byte[].class,
                          TelegramTestUtil.stringResource("messages/send-video.json"))
                  .addEndpoint("sendDocument", "POST", byte[].class,
                          TelegramTestUtil.stringResource("messages/send-document.json"))
                  .addEndpoint("sendMessage", "POST", OutgoingTextMessage.class,
                          TelegramTestUtil.stringResource("messages/send-message.json"))
                  .addEndpoint("sendGame", "POST", OutgoingGameMessage.class,
                          TelegramTestUtil.stringResource("messages/send-game.json"));
          return mockRoutes;
  
      }
  
      static void assertMultipartFilename(byte[] message, String name, String filename) {
          assertTrue(contains(message,
                  ("name=\"" + name + "\"; filename=\"" + filename + "\"").getBytes(StandardCharsets.UTF_8)));
      }
  
      static void assertMultipartText(byte[] message, String key, String value) {
          assertTrue(contains(message, ("name=\"" + key + "\"\r\nContent-Type: text/plain; charset=UTF-8\r\n\r\n" + value)
                  .getBytes(StandardCharsets.UTF_8)));
      }
  
      static boolean contains(byte[] array, byte[] target) {
          if (target.length == 0) {
              return true;
          }
          OUTER_FOR: for (int i = 0; i < array.length - target.length + 1; i++) {
              for (int j = 0; j < target.length; j++) {
                  if (array[i + j] != target[j]) {
                      continue OUTER_FOR;
                  }
              }
              return true;
          }
          return false;
      }
  }
  