use chrono::{DateTime, Utc};
use serde::{Deserialize, Serialize};
use serde_repr::{Deserialize_repr, Serialize_repr};

/// Parameters used to customize the collab vector search query.
/// In response, a list of [SearchDocumentResponseItem] is returned.
#[derive(Clone, Debug, Deserialize)]
pub struct SearchDocumentRequest {
  /// Query statement to search for.
  pub query: String,
  /// Maximum number of results to return. Default: 10.
  #[serde(skip_serializing_if = "Option::is_none")]
  pub limit: Option<u32>,
  /// Maximum length of the content string preview to return. Default: 180.
  #[serde(skip_serializing_if = "Option::is_none")]
  pub preview_size: Option<u32>,
}

/// Response array element for the collab vector search query.
/// See: [SearchDocumentRequest].
#[derive(Clone, Debug, Serialize, Deserialize)]
pub struct SearchDocumentResponseItem {
  /// Unique object identifier.
  pub object_id: String,
  /// Workspace, result object belongs to.
  pub workspace_id: String,
  /// Match score of this search result to an original query.
  /// The lower the better. List of results is sorted by this value by default.
  pub score: f64,
  /// Type of the content to be presented in preview field. This is a hint what
  /// kind of content was used to match the user query ie. document plain text, pdf attachment etc.
  pub content_type: Option<SearchContentType>,
  /// First N characters of the indexed content matching the user query. It doesn't have to contain
  /// the user query itself.
  pub preview: Option<String>,
  /// Name of the user who created/own the document.
  pub created_by: String,
  /// Date when the document was created.
  pub created_at: DateTime<Utc>,
}

/// Type of the document content to be presented in the search results.
/// See: [SearchDocumentResponseItem].
#[repr(i32)]
#[derive(Clone, Copy, Debug, Serialize_repr, Deserialize_repr)]
pub enum SearchContentType {
  /// Document block contents displayed as plain text.
  PlainText = 0,
}

impl SearchContentType {
  /// Converts the database content type (an integer) into current [SearchContentType].
  pub fn from_record(content_type: i32) -> Option<Self> {
    match content_type {
      0 => Some(SearchContentType::PlainText),
      _ => None,
    }
  }
}