using System.Text.Json.Serialization;

using AnthropicClient.Utils;

namespace AnthropicClient.Models;

/// <summary>
/// Represents a message request.
/// </summary>
public abstract class BaseMessageRequest
{
  /// <summary>
  /// Gets the model ID to use for the request.
  /// </summary>
  public string Model { get; init; } = string.Empty;

  /// <summary>
  /// Gets the system message that will be used as the system prompt if no system messages are provided.
  /// </summary>
  [JsonIgnore]
  public string? System { get; init; } = null;

  /// <summary>
  /// Gets the system messages to send to the model to be used as the system prompt.
  /// </summary>
  [JsonIgnore]
  public List<TextContent>? SystemMessages { get; init; } = null;

  /// <summary>
  /// Gets the system prompt that will be used for the request. 
  /// If will return the system messages if they are provided, otherwise it will return the system message.
  /// If neither are provided, it will return null.
  /// </summary>
  [JsonPropertyName("system")]
  public List<TextContent>? SystemPrompt => GetSystemPrompt();

  private List<TextContent>? GetSystemPrompt()
  {
    if (SystemMessages is not null)
    {
      return SystemMessages;
    }

    if (System is not null)
    {
      return [new TextContent(System)];
    }

    return null;
  }

  /// <summary>
  /// Gets the messages to send to the model.
  /// </summary>
  public List<Message> Messages { get; init; } = [];

  /// <summary>
  /// Gets the maximum number of tokens to generate.
  /// </summary>
  [JsonPropertyName("max_tokens")]
  public int MaxTokens { get; init; } = 1024;

  /// <summary>
  /// Gets the metadata to include with the request.
  /// </summary>
  public Dictionary<string, object>? Metadata { get; init; } = null;

  /// <summary>
  /// Gets the prompt stop sequences.
  /// </summary>
  [JsonPropertyName("stop_sequences")]
  public List<string> StopSequences { get; init; } = [];

  /// <summary>
  /// Gets the temperature to use for the request.
  /// </summary>
  public decimal Temperature { get; init; } = 0.0m;

  /// <summary>
  /// Gets the top-K value to use for the request.
  /// </summary>
  public int? TopK { get; init; } = null;

  /// <summary>
  /// Gets the top-P value to use for the request.
  /// </summary>
  public decimal? TopP { get; init; } = null;

  /// <summary>
  /// Gets the tool choice mode to use for the request.
  /// </summary>
  [JsonPropertyName("tool_choice")]
  public ToolChoice? ToolChoice { get; init; } = null;

  /// <summary>
  /// Gets the tools to use for the request.
  /// </summary>
  public List<Tool>? Tools { get; init; } = null;

  /// <summary>
  /// Gets a value indicating whether the message should be streamed.
  /// </summary>
  public bool Stream { get; init; }

  [JsonConstructor]
  internal BaseMessageRequest() { }
  
  private BaseMessageRequest(
    string model,
    List<Message> messages,
    int maxTokens,
    string? system,
    List<TextContent>? systemMessages,
    Dictionary<string, object>? metadata,
    decimal temperature,
    int? topK,
    decimal? topP,
    ToolChoice? toolChoice,
    List<Tool>? tools,
    bool stream,
    List<string>? stopSequences
  )
  {
    ArgumentValidator.ThrowIfNull(model, nameof(model));
    ArgumentValidator.ThrowIfNull(messages, nameof(messages));

    if (AnthropicModels.IsValidModel(model) is false)
    {
      throw new ArgumentException($"Invalid model ID: {model}");
    }

    if (messages.Count < 1)
    {
      throw new ArgumentException("Messages must contain at least one message");
    }

    if (maxTokens < 1)
    {
      throw new ArgumentException($"Invalid max tokens: {maxTokens}");
    }

    if (temperature < 0.0m || temperature > 1.0m)
    {
      throw new ArgumentException($"Invalid temperature: {temperature}");
    }

    Model = model;
    Messages = messages;
    MaxTokens = maxTokens;
    System = system;
    SystemMessages = systemMessages;
    Metadata = metadata;
    Temperature = temperature;
    TopK = topK;
    TopP = topP;
    ToolChoice = toolChoice;
    Tools = tools;
    Stream = stream;
    StopSequences = stopSequences ?? [];
  }

  /// <summary>
  /// Initializes a new instance of the <see cref="BaseMessageRequest"/> class.
  /// </summary>
  /// <param name="model">The model ID to use for the request.</param>
  /// <param name="messages">The messages to send to the model.</param>
  /// <param name="maxTokens">The maximum number of tokens to generate.</param>
  /// <param name="system">The system prompt to use for the request.</param>
  /// <param name="metadata">The metadata to include with the request.</param>
  /// <param name="temperature">The temperature to use for the request.</param>
  /// <param name="topK">The top-K value to use for the request.</param>
  /// <param name="topP">The top-P value to use for the request.</param>
  /// <param name="toolChoice">The tool choice mode to use for the request.</param>
  /// <param name="tools">The tools to use for the request.</param>
  /// <param name="stream">A value indicating whether the message should be streamed.</param>
  /// <param name="stopSequences">The prompt stop sequences.</param>
  /// <exception cref="ArgumentException">Thrown when the model ID is invalid.</exception>
  /// <exception cref="ArgumentNullException">Thrown when the model or messages is null.</exception>
  /// <exception cref="ArgumentException">Thrown when the messages contain no messages.</exception>
  /// <exception cref="ArgumentException">Thrown when the max tokens is less than one.</exception>
  /// <exception cref="ArgumentException">Thrown when the temperature is less than zero or greater than one.</exception>
  /// <returns>A new instance of the <see cref="BaseMessageRequest"/> class.</returns>
  protected BaseMessageRequest(
    string model,
    List<Message> messages,
    int maxTokens = 1024,
    string? system = null,
    Dictionary<string, object>? metadata = null,
    decimal temperature = 0.0m,
    int? topK = null,
    decimal? topP = null,
    ToolChoice? toolChoice = null,
    List<Tool>? tools = null,
    bool stream = false,
    List<string>? stopSequences = null
  ) : this(
    model,
    messages,
    maxTokens,
    system,
    null,
    metadata,
    temperature,
    topK,
    topP,
    toolChoice,
    tools,
    stream,
    stopSequences
  )
  {
  }

  /// <summary>
  /// Initializes a new instance of the <see cref="BaseMessageRequest"/> class.
  /// </summary>
  /// <param name="model">The model ID to use for the request.</param>
  /// <param name="messages">The messages to send to the model.</param>
  /// <param name="maxTokens">The maximum number of tokens to generate.</param>
  /// <param name="systemMessages">The system messages to send to the model to be used as the system prompt.</param>
  /// <param name="metadata">The metadata to include with the request.</param>
  /// <param name="temperature">The temperature to use for the request.</param>
  /// <param name="topK">The top-K value to use for the request.</param>
  /// <param name="topP">The top-P value to use for the request.</param>
  /// <param name="toolChoice">The tool choice mode to use for the request.</param>
  /// <param name="tools">The tools to use for the request.</param>
  /// <param name="stream">A value indicating whether the message should be streamed.</param>
  /// <param name="stopSequences">The prompt stop sequences.</param>
  /// <exception cref="ArgumentException">Thrown when the model ID is invalid.</exception>
  /// <exception cref="ArgumentNullException">Thrown when the model or messages is null.</exception>
  /// <exception cref="ArgumentException">Thrown when the messages contain no messages.</exception>
  /// <exception cref="ArgumentException">Thrown when the max tokens is less than one.</exception>
  /// <exception cref="ArgumentException">Thrown when the temperature is less than zero or greater than one.</exception>
  /// <returns>A new instance of the <see cref="BaseMessageRequest"/> class.</returns>
  protected BaseMessageRequest(
    string model,
    List<Message> messages,
    int maxTokens = 1024,
    List<TextContent>? systemMessages = null,
    Dictionary<string, object>? metadata = null,
    decimal temperature = 0.0m,
    int? topK = null,
    decimal? topP = null,
    ToolChoice? toolChoice = null,
    List<Tool>? tools = null,
    bool stream = false,
    List<string>? stopSequences = null
  ) : this(
    model,
    messages,
    maxTokens,
    null,
    systemMessages,
    metadata,
    temperature,
    topK,
    topP,
    toolChoice,
    tools,
    stream,
    stopSequences
  )
  {
  }
}