/*
 * 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.juneau.http;

import static org.apache.juneau.commons.utils.ThrowableUtils.*;
import static org.apache.juneau.commons.utils.Utils.*;

import java.net.*;
import java.time.*;
import java.util.*;
import java.util.function.*;

import org.apache.http.*;
import org.apache.juneau.*;
import org.apache.juneau.http.header.*;
import org.apache.juneau.http.part.*;
import org.apache.juneau.httppart.*;
import org.apache.juneau.commons.lang.*;
import org.apache.juneau.commons.reflect.*;

/**
 * Standard predefined HTTP parts.
 *
 * <h5 class='section'>See Also:</h5><ul>
 *    <li class='link'><a class="doclink" href="https://juneau.apache.org/docs/topics/JuneauRestCommonBasics">juneau-rest-common Basics</a>
 * </ul>
 */
public class HttpParts {

   private static final Function<ClassMeta<?>,String> HEADER_NAME_FUNCTION = x -> {
      var n = Value.<String>empty();
      x.forEachAnnotation(org.apache.juneau.http.annotation.Header.class, y -> ne(y.value()), y -> n.set(y.value()));
      x.forEachAnnotation(org.apache.juneau.http.annotation.Header.class, y -> ne(y.name()), y -> n.set(y.name()));
      return n.orElse(null);
   };

   private static final Function<ClassMeta<?>,String> QUERY_NAME_FUNCTION = x -> {
      var n = Value.<String>empty();
      x.forEachAnnotation(org.apache.juneau.http.annotation.Query.class, y -> ne(y.value()), y -> n.set(y.value()));
      x.forEachAnnotation(org.apache.juneau.http.annotation.Query.class, y -> ne(y.name()), y -> n.set(y.name()));
      return n.orElse(null);
   };

   private static final Function<ClassMeta<?>,String> FORMDATA_NAME_FUNCTION = x -> {
      var n = Value.<String>empty();
      x.forEachAnnotation(org.apache.juneau.http.annotation.FormData.class, y -> ne(y.value()), y -> n.set(y.value()));
      x.forEachAnnotation(org.apache.juneau.http.annotation.FormData.class, y -> ne(y.name()), y -> n.set(y.name()));
      return n.orElse(null);
   };

   private static final Function<ClassMeta<?>,String> PATH_NAME_FUNCTION = x -> {
      var n = Value.<String>empty();
      x.forEachAnnotation(org.apache.juneau.http.annotation.Path.class, y -> ne(y.value()), y -> n.set(y.value()));
      x.forEachAnnotation(org.apache.juneau.http.annotation.Path.class, y -> ne(y.name()), y -> n.set(y.name()));
      return n.orElse(null);
   };

   private static final Function<ClassMeta<?>,ConstructorInfo> CONSTRUCTOR_FUNCTION = x -> {
      return x.getPublicConstructor(y -> y.hasParameterTypes(String.class)).orElseGet(() -> x.getPublicConstructor(y -> y.hasParameterTypes(String.class, String.class)).orElse(null));
   };

   /**
    * Creates a {@link BasicPart} from a name/value pair string (e.g. <js>"Foo: bar"</js>)
    *
    * @param pair The pair string.
    * @return A new {@link BasicPart} object.
    */
   public static final BasicPart basicPart(String pair) {
      return BasicPart.ofPair(pair);
   }

   /**
    * Creates a new {@link BasicPart} part.
    *
    * @param name The part name.
    * @param value The part value.
    * @return A new {@link BasicPart} object.
    */
   public static final BasicPart basicPart(String name, Object value) {
      return BasicPart.of(name, value);
   }

   /**
    * Creates a new {@link BasicPart} part with a delayed value.
    *
    * <p>
    * Part value is re-evaluated on each call to {@link NameValuePair#getValue()}.
    *
    * @param name The part name.
    * @param value The part value supplier.
    * @return A new {@link BasicPart} object.
    */
   public static final BasicPart basicPart(String name, Supplier<?> value) {
      return BasicPart.of(name, value);
   }

   /**
    * Creates a new {@link BasicBooleanPart} part.
    *
    * @param name The part name.
    * @param value
    *    The part value.
    *    <br>Can be any of the following:
    *    <ul>
    *       <li>{@link Boolean} - As-is.
    *       <li>{@link String} - Parsed using {@link Boolean#parseBoolean(String)}.
    *       <li>Anything else - Converted to <c>String</c> and then parsed.
    *    </ul>
    * @return A new {@link BasicBooleanPart} object, or <jk>null</jk> if the name or value is <jk>null</jk>.
    */
   public static final BasicBooleanPart booleanPart(String name, Boolean value) {
      return BasicBooleanPart.of(name, value);
   }

   /**
    * Creates a new {@link BasicBooleanPart} part with a delayed value.
    *
    * <p>
    * Part value is re-evaluated on each call to {@link NameValuePair#getValue()}.
    *
    * @param name The part name.
    * @param value
    *    The part value supplier.
    *    <br>Can be any of the following:
    *    <ul>
    *       <li>{@link Boolean} - As-is.
    *       <li>{@link String} - Parsed using {@link Boolean#parseBoolean(String)}.
    *       <li>Anything else - Converted to <c>String</c> and then parsed.
    *    </ul>
    * @return A new {@link BasicBooleanPart} object, or <jk>null</jk> if the name or value is <jk>null</jk>.
    */
   public static final BasicBooleanPart booleanPart(String name, Supplier<Boolean> value) {
      return BasicBooleanPart.of(name, value);
   }

   /**
    * Returns <jk>true</jk> if the {@link #cast(Object)} method can be used on the specified object.
    *
    * @param o The object to check.
    * @return <jk>true</jk> if the {@link #cast(Object)} method can be used on the specified object.
    */
   public static boolean canCast(Object o) {
      if (o == null)
         return false;
      var ci = ClassInfo.of(o);
      return nn(ci) && ci.isChildOfAny(Headerable.class, NameValuePair.class, NameValuePairable.class, Map.Entry.class);
   }

   /**
    * Utility method for converting an arbitrary object to a {@link NameValuePair}.
    *
    * @param o
    *    The object to cast or convert to a {@link NameValuePair}.
    * @return Either the same object cast as a {@link NameValuePair} or converted to a {@link NameValuePair}.
    */
   public static NameValuePair cast(Object o) {
      if (o instanceof NameValuePair o2)
         return o2;
      if (o instanceof Headerable o3) {
         Header x = o3.asHeader();
         return BasicPart.of(x.getName(), x.getValue());
      }
      if (o instanceof Map.Entry o4) {
         return BasicPart.of(s(o4.getKey()), o4.getValue());
      }
      throw rex("Object of type {0} could not be converted to a Part.", cn(o));
   }

   /**
    * Creates a new {@link BasicCsvArrayPart} part.
    *
    * @param name The part name.
    * @param value
    *    The part value.
    *    <br>Can be any of the following:
    *    <ul>
    *       <li><c>String</c> - A comma-delimited string.
    *       <li><c>String[]</c> - A pre-parsed value.
    *       <li>Any other array type - Converted to <c>String[]</c>.
    *       <li>Any {@link Collection} - Converted to <c>String[]</c>.
    *       <li>Anything else - Converted to <c>String</c>.
    *    </ul>
    * @return A new {@link BasicCsvArrayPart} object, or <jk>null</jk> if the name or value is <jk>null</jk>.
    */
   public static final BasicCsvArrayPart csvArrayPart(String name, String...value) {
      return BasicCsvArrayPart.of(name, value);
   }

   /**
    * Creates a new {@link BasicCsvArrayPart} part with a delayed value.
    *
    * <p>
    * Part value is re-evaluated on each call to {@link NameValuePair#getValue()}.
    *
    * @param name The part name.
    * @param value
    *    The part value supplier.
    *    <br>Can be any of the following:
    *    <ul>
    *       <li><c>String</c> - A comma-delimited string.
    *       <li><c>String[]</c> - A pre-parsed value.
    *       <li>Any other array type - Converted to <c>String[]</c>.
    *       <li>Any {@link Collection} - Converted to <c>String[]</c>.
    *       <li>Anything else - Converted to <c>String</c>.
    *    </ul>
    * @return A new {@link BasicCsvArrayPart} object, or <jk>null</jk> if the name or value is <jk>null</jk>.
    */
   public static final BasicCsvArrayPart csvArrayPart(String name, Supplier<String[]> value) {
      return BasicCsvArrayPart.of(name, value);
   }

   /**
    * Creates a new {@link BasicDatePart} part with a delayed value.
    *
    * <p>
    * Part value is re-evaluated on each call to {@link NameValuePair#getValue()}.
    *
    * @param name The part name.
    * @param value
    *    The part value supplier.
    *    <br>Can be any of the following:
    *    <ul>
    *       <li><c>String</c> - An ISO-8601 formated string (e.g. <js>"1994-10-29T19:43:31Z"</js>).
    *       <li>{@link ZonedDateTime}
    *       <li>{@link Calendar}
    *       <li>Anything else - Converted to <c>String</c>.
    *    </ul>
    * @return A new {@link BasicDatePart} object, or <jk>null</jk> if the name or value is <jk>null</jk>.
    */
   public static final BasicDatePart datePart(String name, Supplier<ZonedDateTime> value) {
      return BasicDatePart.of(name, value);
   }

   /**
    * Creates a new {@link BasicDatePart} part.
    *
    * @param name The part name.
    * @param value
    *    The part value.
    *    <br>Can be any of the following:
    *    <ul>
    *       <li><c>String</c> - An ISO-8601 formated string (e.g. <js>"1994-10-29T19:43:31Z"</js>).
    *       <li>{@link ZonedDateTime}
    *       <li>{@link Calendar}
    *       <li>Anything else - Converted to <c>String</c>.
    *    </ul>
    * @return A new {@link BasicDatePart} object, or <jk>null</jk> if the name or value is <jk>null</jk>.
    */
   public static final BasicDatePart datePart(String name, ZonedDateTime value) {
      return BasicDatePart.of(name, value);
   }

   /**
    * Returns the constructor for the specified type.
    *
    * <p>
    * Looks for one of the following constructors:
    * <ul class='javatree'>
    *    <li class='jm><c><jk>public</jk> T(String <jv>value</jv>);</c>
    *    <li class='jm><c><jk>public</jk> T(String <jv>name</jv>, String <jv>value</jv>);</c>
    * </ul>
    *
    * @param type The header type to find the constructor on.
    * @return The constructor.  Never <jk>null</jk>.
    */
   public static Optional<ConstructorInfo> getConstructor(ClassMeta<?> type) {
      return type.getProperty("HttpPart.Constructor", CONSTRUCTOR_FUNCTION);
   }

   /**
    * Returns the name of the specified part type.
    *
    * <p>
    * Gets the name from one of the following annotations:
    * <ul class='javatreec'>
    *    <li class='ja'>{@link org.apache.juneau.http.annotation.Header}
    *    <li class='ja'>{@link org.apache.juneau.http.annotation.Query}
    *    <li class='ja'>{@link org.apache.juneau.http.annotation.FormData}
    *    <li class='ja'>{@link org.apache.juneau.http.annotation.Path}
    * </ul>
    *
    * @param partType The part type.
    * @param type The type to check.
    * @return The part name.  Never <jk>null</jk>.
    */
   public static Optional<String> getName(HttpPartType partType, ClassMeta<?> type) {
      return switch (partType) {
         case FORMDATA -> type.getProperty("HttpPart.formData.name", FORMDATA_NAME_FUNCTION);
         case HEADER -> type.getProperty("HttpPart.header.name", HEADER_NAME_FUNCTION);
         case PATH -> type.getProperty("HttpPart.path.name", PATH_NAME_FUNCTION);
         case QUERY -> type.getProperty("HttpPart.query.name", QUERY_NAME_FUNCTION);
         default -> opte();
      };
   }

   /**
    * Creates a new {@link BasicIntegerPart} part.
    *
    * @param name The part name.
    * @param value
    *    The part value.
    *    <br>Can be any of the following:
    *    <ul>
    *       <li>{@link Number} - Converted to an integer using {@link Number#intValue()}.
    *       <li>{@link String} - Parsed using {@link Integer#parseInt(String)}.
    *       <li>Anything else - Converted to <c>String</c>.
    *    </ul>
    * @return A new {@link BasicIntegerPart} object, or <jk>null</jk> if the name or value is <jk>null</jk>.
    */
   public static final BasicIntegerPart integerPart(String name, Integer value) {
      return BasicIntegerPart.of(name, value);
   }

   /**
    * Creates a new {@link BasicIntegerPart} part with a delayed value.
    *
    * <p>
    * Part value is re-evaluated on each call to {@link NameValuePair#getValue()}.
    *
    * @param name The part name.
    * @param value
    *    The part value supplier.
    *    <br>Can be any of the following:
    *    <ul>
    *       <li>{@link Number} - Converted to an integer using {@link Number#intValue()}.
    *       <li>{@link String} - Parsed using {@link Integer#parseInt(String)}.
    *       <li>Anything else - Converted to <c>String</c>.
    *    </ul>
    * @return A new {@link BasicIntegerPart} object, or <jk>null</jk> if the name or value is <jk>null</jk>.
    */
   public static final BasicIntegerPart integerPart(String name, Supplier<Integer> value) {
      return BasicIntegerPart.of(name, value);
   }

   /**
    * Returns <jk>true</jk> if the specified type is a part type.
    *
    * <p>
    * A part type extends from either {@link org.apache.http.Header} or {@link org.apache.http.NameValuePair}
    * or is annotated with {@link org.apache.juneau.http.annotation.Header}, {@link org.apache.juneau.http.annotation.Query},
    * {@link org.apache.juneau.http.annotation.FormData}, or {@link org.apache.juneau.http.annotation.Path}.
    *
    * @param partType The part type.
    * @param type The type to check.
    * @return <jk>true</jk> if the specified type is a part type.
    */
   public static boolean isHttpPart(HttpPartType partType, ClassMeta<?> type) {
      return switch (partType) {
         case PATH, QUERY, FORMDATA -> type.getProperty("HttpPart.isNameValuePair", x -> x.isChildOf(NameValuePair.class)).orElse(false);
         case HEADER -> type.getProperty("HttpPart.isHeader", x -> x.isChildOf(org.apache.http.Header.class)).orElse(false);
         default -> false;
      };
   }

   /**
    * Creates a new {@link BasicLongPart} part.
    *
    * @param name The part name.
    * @param value
    *    The part value.
    *    <br>Can be any of the following:
    *    <ul>
    *       <li>{@link Number} - Converted to a long using {@link Number#longValue()}.
    *       <li>{@link String} - Parsed using {@link Long#parseLong(String)}.
    *       <li>Anything else - Converted to <c>String</c>.
    *    </ul>
    * @return A new {@link BasicLongPart} object, or <jk>null</jk> if the name or value is <jk>null</jk>.
    */
   public static final BasicLongPart longPart(String name, Long value) {
      return BasicLongPart.of(name, value);
   }

   /**
    * Creates a new {@link BasicLongPart} part with a delayed value.
    *
    * <p>
    * Part value is re-evaluated on each call to {@link NameValuePair#getValue()}.
    *
    * @param name The part name.
    * @param value
    *    The part value supplier.
    *    <br>Can be any of the following:
    *    <ul>
    *       <li>{@link Number} - Converted to a long using {@link Number#longValue()}.
    *       <li>{@link String} - Parsed using {@link Long#parseLong(String)}.
    *       <li>Anything else - Converted to <c>String</c>.
    *    </ul>
    * @return A new {@link BasicLongPart} object, or <jk>null</jk> if the name or value is <jk>null</jk>.
    */
   public static final BasicLongPart longPart(String name, Supplier<Long> value) {
      return BasicLongPart.of(name, value);
   }

   /**
    * Instantiates a new {@link org.apache.juneau.http.part.PartList}.
    *
    * @return A new part list.
    */
   public static final PartList partList() {
      return PartList.create();
   }

   /**
    * Creates a new {@link PartList} initialized with the specified parts.
    *
    * @param parts The parts to add to the list.  Can be <jk>null</jk>.  <jk>null</jk> entries are ignored.
    * @return A new unmodifiable instance, never <jk>null</jk>.
    */
   public static final PartList partList(List<NameValuePair> parts) {
      return PartList.of(parts);
   }

   /**
    * Creates a new {@link PartList} initialized with the specified parts.
    *
    * @param parts The parts to add to the list.  <jk>null</jk> entries are ignored.
    * @return A new unmodifiable instance, never <jk>null</jk>.
    */
   public static final PartList partList(NameValuePair...parts) {
      return PartList.of(parts);
   }

   /**
    * Creates a new {@link PartList} initialized with the specified name/value pairs.
    *
    * @param pairs
    *    Initial list of pairs.
    *    <br>Must be an even number of parameters representing key/value pairs.
    * @throws RuntimeException If odd number of parameters were specified.
    * @return A new instance.
    */
   public static PartList partList(String...pairs) {
      return PartList.ofPairs(pairs);
   }

   /**
    * Creates a new {@link SerializedPart} part.
    *
    * @param name The part name.
    * @param value
    *    The part value.
    *    <br>Can be any POJO.
    * @return A new {@link SerializedPart} object, never <jk>null</jk>.
    */
   public static final SerializedPart serializedPart(String name, Object value) {
      return SerializedPart.of(name, value);
   }

   /**
    * Creates a new {@link SerializedPart} part with a delayed value.
    *
    * @param name The part name.
    * @param value
    *    The part value supplier.
    *    <br>Can be a supplier of any POJO.
    * @return A new {@link SerializedPart} object, never <jk>null</jk>.
    */
   public static final SerializedPart serializedPart(String name, Supplier<?> value) {
      return SerializedPart.of(name, value);
   }

   /**
    * Creates a new {@link BasicStringPart} part.
    *
    * @param name The part name.
    * @param value
    *    The part value.
    *    <br>Can be any of the following:
    *    <ul>
    *       <li>{@link String}
    *       <li>Anything else - Converted to <c>String</c> then parsed.
    *    </ul>
    * @return A new {@link BasicStringPart} object, or <jk>null</jk> if the name or value is <jk>null</jk>.
    */
   public static final BasicStringPart stringPart(String name, String value) {
      return BasicStringPart.of(name, value);
   }

   /**
    * Creates a new {@link BasicStringPart} part with a delayed value.
    *
    * <p>
    * Part value is re-evaluated on each call to {@link NameValuePair#getValue()}.
    *
    * @param name The part name.
    * @param value
    *    The part value supplier.
    *    <br>Can be any of the following:
    *    <ul>
    *       <li>{@link String}
    *       <li>Anything else - Converted to <c>String</c> then parsed.
    *    </ul>
    * @return A new {@link BasicStringPart} object, or <jk>null</jk> if the name or value is <jk>null</jk>.
    */
   public static final BasicStringPart stringPart(String name, Supplier<String> value) {
      return BasicStringPart.of(name, value);
   }

   /**
    * Creates a new {@link BasicUriPart} part with a delayed value.
    *
    * <p>
    * Part value is re-evaluated on each call to {@link NameValuePair#getValue()}.
    *
    * @param name The header name.
    * @param value
    *    The header value supplier.
    *    <br>Can be any of the following:
    *    <ul>
    *       <li>{@link String}
    *       <li>Anything else - Converted to <c>String</c> then parsed.
    *    </ul>
    * @return A new {@link BasicUriPart} object, or <jk>null</jk> if the name or value is <jk>null</jk>.
    */
   public static final BasicUriPart uriPart(String name, Supplier<URI> value) {
      return BasicUriPart.of(name, value);
   }

   /**
    * Creates a new {@link BasicUriPart} part.
    *
    * @param name The header name.
    * @param value
    *    The header value.
    *    <br>Can be any of the following:
    *    <ul>
    *       <li>{@link String}
    *       <li>Anything else - Converted to <c>String</c> then parsed.
    *    </ul>
    * @return A new {@link BasicUriPart} object, or <jk>null</jk> if the name or value is <jk>null</jk>.
    */
   public static final BasicUriPart uriPart(String name, URI value) {
      return BasicUriPart.of(name, value);
   }
}