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.google.mail;
  
  import java.io.ByteArrayInputStream;
  import java.io.ByteArrayOutputStream;
  import java.io.IOException;
  import java.util.HashMap;
  import java.util.Map;
  import java.util.Properties;
  
  import jakarta.mail.MessagingException;
  import jakarta.mail.Session;
  import jakarta.mail.internet.MimeMessage;
  
  import com.google.api.services.gmail.model.ListMessagesResponse;
  import com.google.api.services.gmail.model.Message;
  import org.apache.camel.builder.RouteBuilder;
  import org.apache.camel.component.google.mail.internal.GmailUsersMessagesApiMethod;
  import org.apache.camel.component.google.mail.internal.GoogleMailApiCollection;
  import org.apache.commons.codec.binary.Base64;
  import org.junit.jupiter.api.Test;
  import org.junit.jupiter.api.condition.EnabledIf;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  
  import static org.junit.jupiter.api.Assertions.assertFalse;
  import static org.junit.jupiter.api.Assertions.assertNotNull;
  import static org.junit.jupiter.api.Assertions.assertTrue;
  
  /**
   * Test class for {@link com.google.api.services.gmail.Gmail$Users$Messages} APIs.
   */
  @EnabledIf(value = "org.apache.camel.component.google.mail.AbstractGoogleMailTestSupport#hasCredentials",
             disabledReason = "Google Mail credentials were not provided")
  public class GmailUsersMessagesIT extends AbstractGoogleMailTestSupport {
  
      // userid of the currently authenticated user
      public static final String CURRENT_USERID = "me";
      private static final Logger LOG = LoggerFactory.getLogger(GmailUsersMessagesIT.class);
      private static final String PATH_PREFIX
              = GoogleMailApiCollection.getCollection().getApiName(GmailUsersMessagesApiMethod.class).getName();
  
      @Test
      public void testMessages() throws Exception {
  
          // ==== Send test email ====
          Message testEmail = createTestEmail();
          Map<String, Object> headers = new HashMap<>();
          // parameter type is String
          headers.put("CamelGoogleMail.userId", CURRENT_USERID);
          // parameter type is com.google.api.services.gmail.model.Message
          headers.put("CamelGoogleMail.content", testEmail);
  
          com.google.api.services.gmail.model.Message result = requestBodyAndHeaders("direct://SEND", null, headers);
          assertNotNull(result, "send result");
          String testEmailId = result.getId();
  
          // ==== Search for message we just sent ====
          headers = new HashMap<>();
          headers.put("CamelGoogleMail.q", "subject:\"Hello from camel-google-mail\"");
          // using String message body for single parameter "userId"
          ListMessagesResponse listOfMessages = requestBody("direct://LIST", CURRENT_USERID);
          assertTrue(idInList(testEmailId, listOfMessages));
  
          // ===== trash it ====
          headers = new HashMap<>();
          // parameter type is String
          headers.put("CamelGoogleMail.userId", CURRENT_USERID);
          // parameter type is String
          headers.put("CamelGoogleMail.id", testEmailId);
          requestBodyAndHeaders("direct://TRASH", null, headers);
  
          // ==== Search for message we just trashed ====
          headers = new HashMap<>();
          headers.put("CamelGoogleMail.q", "subject:\"Hello from camel-google-mail\"");
          // using String message body for single parameter "userId"
          listOfMessages = requestBody("direct://LIST", CURRENT_USERID);
          assertFalse(idInList(testEmailId, listOfMessages));
  
          // ===== untrash it ====
          headers = new HashMap<>();
          // parameter type is String
          headers.put("CamelGoogleMail.userId", CURRENT_USERID);
          // parameter type is String
          headers.put("CamelGoogleMail.id", testEmailId);
          requestBodyAndHeaders("direct://UNTRASH", null, headers);
  
          // ==== Search for message we just trashed ====
          headers = new HashMap<>();
          headers.put("CamelGoogleMail.q", "subject:\"Hello from camel-google-mail\"");
          // using String message body for single parameter "userId"
          listOfMessages = requestBody("direct://LIST", CURRENT_USERID);
          assertTrue(idInList(testEmailId, listOfMessages));
  
          // ===== permanently delete it ====
          headers = new HashMap<>();
          // parameter type is String
          headers.put("CamelGoogleMail.userId", CURRENT_USERID);
          // parameter type is String
          headers.put("CamelGoogleMail.id", testEmailId);
          requestBodyAndHeaders("direct://DELETE", null, headers);
  
          // ==== Search for message we just deleted ====
          headers = new HashMap<>();
          headers.put("CamelGoogleMail.q", "subject:\"Hello from camel-google-mail\"");
          // using String message body for single parameter "userId"
          listOfMessages = requestBody("direct://LIST", CURRENT_USERID);
          assertFalse(idInList(testEmailId, listOfMessages));
      }
  
      private boolean idInList(String testEmailId, ListMessagesResponse listOfMessages) {
          assertNotNull(listOfMessages, "list result");
          assertFalse(listOfMessages.getMessages().isEmpty());
          boolean foundMessage = false;
          for (Message m : listOfMessages.getMessages()) {
              if (testEmailId.equals(m.getId())) {
                  return true;
              }
          }
          return false;
      }
  
      private Message createTestEmail() throws MessagingException, IOException {
          com.google.api.services.gmail.model.Profile profile
                  = requestBody("google-mail://users/getProfile?inBody=userId", CURRENT_USERID);
          Properties props = new Properties();
          Session session = Session.getDefaultInstance(props, null);
          MimeMessage mm = new MimeMessage(session);
          mm.addRecipients(jakarta.mail.Message.RecipientType.TO, profile.getEmailAddress());
          mm.setSubject("Hello from camel-google-mail");
          mm.setContent("Camel rocks!", "text/plain");
          Message createMessageWithEmail = createMessageWithEmail(mm);
          return createMessageWithEmail;
      }
  
      private MimeMessage toMimeMessage(Message message) throws MessagingException {
          byte[] emailBytes = Base64.decodeBase64(message.getRaw());
  
          Properties props = new Properties();
          Session session = Session.getDefaultInstance(props, null);
  
          return new MimeMessage(session, new ByteArrayInputStream(emailBytes));
      }
  
      private Message createMessageWithEmail(MimeMessage email) throws MessagingException, IOException {
          ByteArrayOutputStream baos = new ByteArrayOutputStream();
          email.writeTo(baos);
          String encodedEmail = Base64.encodeBase64URLSafeString(baos.toByteArray());
          Message message = new Message();
          message.setRaw(encodedEmail);
          return message;
      }
  
      @Override
      protected RouteBuilder createRouteBuilder() {
          return new RouteBuilder() {
              @Override
              public void configure() {
                  // test route for attachments
                  from("direct://ATTACHMENTS").to("google-mail://" + PATH_PREFIX + "/attachments");
  
                  // test route for delete
                  from("direct://DELETE").to("google-mail://" + PATH_PREFIX + "/delete");
  
                  // test route for get
                  from("direct://GET").to("google-mail://" + PATH_PREFIX + "/get");
  
                  // test route for gmailImport
                  from("direct://GMAILIMPORT").to("google-mail://" + PATH_PREFIX + "/gmailImport");
  
                  // test route for gmailImport
                  from("direct://GMAILIMPORT_1").to("google-mail://" + PATH_PREFIX + "/gmailImport");
  
                  // test route for insert
                  from("direct://INSERT").to("google-mail://" + PATH_PREFIX + "/insert");
  
                  // test route for insert
                  from("direct://INSERT_1").to("google-mail://" + PATH_PREFIX + "/insert");
  
                  // test route for list
                  from("direct://LIST").to("google-mail://" + PATH_PREFIX + "/list?inBody=userId");
  
                  // test route for modify
                  from("direct://MODIFY").to("google-mail://" + PATH_PREFIX + "/modify");
  
                  // test route for send
                  from("direct://SEND").to("google-mail://" + PATH_PREFIX + "/send");
  
                  // test route for send
                  from("direct://SEND_1").to("google-mail://" + PATH_PREFIX + "/send");
  
                  // test route for trash
                  from("direct://TRASH").to("google-mail://" + PATH_PREFIX + "/trash");
  
                  // test route for untrash
                  from("direct://UNTRASH").to("google-mail://" + PATH_PREFIX + "/untrash");
  
              }
          };
      }
  }
  