use arc_bytes::serde::Bytes;

use serde::{Deserialize, Serialize};

use crate::api::{Api, ApiName};

use crate::connection::{

    AccessPolicy, Database, IdentityReference, Range, SerializedQueryKey, Session, SessionId, Sort,

};

use crate::document::{DocumentId, Header, OwnedDocument};

use crate::keyvalue::{KeyOperation, Output};

use crate::schema::view::map::{self, MappedSerializedDocuments};

use crate::schema::{CollectionName, NamedReference, Qualified, SchemaSummary, ViewName};

use crate::transaction::{Executed, OperationResult, Transaction};

/// The current protocol version.

pub const CURRENT_PROTOCOL_VERSION: &str = "bonsai-pre-1";

/// A payload with an associated id.

pub struct Payload {

    /// The authentication session id for this payload.

    pub session_id: Option<SessionId>,

    /// The unique id for this payload.

    pub id: Option<u32>,

    /// The unique name of the api

    pub name: ApiName,

    /// The payload

    pub value: Result<Bytes, crate::Error>,

}

/// Creates a database.

pub struct CreateDatabase {

    /// The database to create.

    pub database: Database,

    /// Only attempts to create the database if it doesn't already exist.

    pub only_if_needed: bool,

}

impl Api for CreateDatabase {

    type Error = crate::Error;

    type Response = ();

}

/// Deletes the database named `name`

pub struct DeleteDatabase {

    /// The name of the database to delete.

    pub name: String,

}

impl Api for DeleteDatabase {

    type Error = crate::Error;

    type Response = ();

}

/// Lists all databases.

pub struct ListDatabases;

impl Api for ListDatabases {

    type Error = crate::Error;

    type Response = Vec<Database>;

}

/// Lists available schemas.

pub struct ListAvailableSchemas;

impl Api for ListAvailableSchemas {

    type Error = crate::Error;

    type Response = Vec<SchemaSummary>;

}

/// Creates a user.

pub struct CreateUser {

    /// The unique username of the user to create.

    pub username: String,

}

impl Api for CreateUser {

    type Error = crate::Error;

    type Response = u64;

}

/// Deletes a user.

pub struct DeleteUser {

    /// The unique primary key of the user to be deleted.

    pub user: NamedReference<'static, u64>,

}

impl Api for DeleteUser {

    type Error = crate::Error;

    type Response = ();

}

/// Set's a user's password.

#[cfg(feature = "password-hashing")]

pub struct SetUserPassword {

    /// The username or id of the user.

    pub user: NamedReference<'static, u64>,

    /// The user's new password.

    pub password: crate::connection::SensitiveString,

}

#[cfg(feature = "password-hashing")]

impl Api for SetUserPassword {

    type Error = crate::Error;

    type Response = ();

}

/// Authenticate the current connection.

#[cfg(any(feature = "password-hashing", feature = "token-authentication"))]

pub struct Authenticate {

    /// The method of authentication.

    pub authentication: crate::connection::Authentication,

}

#[cfg(any(feature = "password-hashing", feature = "token-authentication"))]

impl Api for Authenticate {

    type Error = crate::Error;

    type Response = Session;

}

/// Assume an identity.

pub struct AssumeIdentity(pub IdentityReference<'static>);

impl Api for AssumeIdentity {

    type Error = crate::Error;

    type Response = Session;

}

/// Logs out from a session.

pub struct LogOutSession(pub SessionId);

impl Api for LogOutSession {

    type Error = crate::Error;

    type Response = ();

}

/// Alter's a user's membership in a permission group.

pub struct AlterUserPermissionGroupMembership {

    /// The username or id of the user.

    pub user: NamedReference<'static, u64>,

    /// The name or id of the group.

    pub group: NamedReference<'static, u64>,

    /// Whether the user should be in the group.

    pub should_be_member: bool,

}

impl Api for AlterUserPermissionGroupMembership {

    type Error = crate::Error;

    type Response = ();

}

/// Alter's a user's role

pub struct AlterUserRoleMembership {

    /// The username or id of the user.

    pub user: NamedReference<'static, u64>,

    /// The name or id of the role.

    pub role: NamedReference<'static, u64>,

    /// Whether the user should have the role.

    pub should_be_member: bool,

}

impl Api for AlterUserRoleMembership {

    type Error = crate::Error;

    type Response = ();

}

/// Retrieve a single document.

pub struct Get {

    /// The name of the database.

    pub database: String,

    /// The collection of the document.

    pub collection: CollectionName,

    /// The id of the document.

    pub id: DocumentId,

}

impl Api for Get {

    type Error = crate::Error;

    type Response = Option<OwnedDocument>;

}

/// Retrieve multiple documents.

pub struct GetMultiple {

    /// The name of the database.

    pub database: String,

    /// The collection of the documents.

    pub collection: CollectionName,

    /// The ids of the documents.

    pub ids: Vec<DocumentId>,

}

impl Api for GetMultiple {

    type Error = crate::Error;

    type Response = Vec<OwnedDocument>;

}

/// Retrieve multiple documents.

pub struct List {

    /// The name of the database.

    pub database: String,

    /// The collection of the documents.

    pub collection: CollectionName,

    /// The range of ids to list.

    pub ids: Range<DocumentId>,

    /// The order for the query into the collection.

    pub order: Sort,

    /// The maximum number of results to return.

    pub limit: Option<u32>,

}

impl Api for List {

    type Error = crate::Error;

    type Response = Vec<OwnedDocument>;

}

/// Retrieve multiple document headers.

pub struct ListHeaders(pub List);

impl Api for ListHeaders {

    type Error = crate::Error;

    type Response = Vec<Header>;

}

/// Counts the number of documents in the specified range.

pub struct Count {

    /// The name of the database.

    pub database: String,

    /// The collection of the documents.

    pub collection: CollectionName,

    /// The range of ids to count.

    pub ids: Range<DocumentId>,

}

impl Api for Count {

    type Error = crate::Error;

    type Response = u64;

}

/// Queries a view.

pub struct Query {

    /// The name of the database.

    pub database: String,

    /// The name of the view.

    pub view: ViewName,

    /// The filter for the view.

    pub key: Option<SerializedQueryKey>,

    /// The order for the query into the view.

    pub order: Sort,

    /// The maximum number of results to return.

    pub limit: Option<u32>,

    /// The access policy for the query.

    pub access_policy: AccessPolicy,

}

impl Api for Query {

    type Error = crate::Error;

    type Response = Vec<map::Serialized>;

}

/// Queries a view with the associated documents.

pub struct QueryWithDocs(pub Query);

impl Api for QueryWithDocs {

    type Error = crate::Error;

    type Response = MappedSerializedDocuments;

}

/// Reduces a view.

pub struct Reduce {

    /// The name of the database.

    pub database: String,

    /// The name of the view.

    pub view: ViewName,

    /// The filter for the view.

    pub key: Option<SerializedQueryKey>,

    /// The access policy for the query.

    pub access_policy: AccessPolicy,

}

impl Api for Reduce {

    type Error = crate::Error;

    type Response = Bytes;

}

/// Reduces a view, grouping the reduced values by key.

pub struct ReduceGrouped(pub Reduce);

impl Api for ReduceGrouped {

    type Error = crate::Error;

    type Response = Vec<map::MappedSerializedValue>;

}

/// Deletes the associated documents resulting from the view query.

pub struct DeleteDocs {

    /// The name of the database.

    pub database: String,

    /// The name of the view.

    pub view: ViewName,

    /// The filter for the view.

    pub key: Option<SerializedQueryKey>,

    /// The access policy for the query.

    pub access_policy: AccessPolicy,

}

impl Api for DeleteDocs {

    type Error = crate::Error;

    type Response = u64;

}

/// Applies a transaction.

pub struct ApplyTransaction {

    /// The name of the database.

    pub database: String,

    /// The trasnaction to apply.

    pub transaction: Transaction,

}

impl Api for ApplyTransaction {

    type Error = crate::Error;

    type Response = Vec<OperationResult>;

}

/// Lists executed transactions.

pub struct ListExecutedTransactions {

    /// The name of the database.

    pub database: String,

    /// The starting transaction id.

    pub starting_id: Option<u64>,

    /// The maximum number of results.

    pub result_limit: Option<u32>,

}

impl Api for ListExecutedTransactions {

    type Error = crate::Error;

    type Response = Vec<Executed>;

}

/// Queries the last transaction id.

pub struct LastTransactionId {

    /// The name of the database.

    pub database: String,

}

impl Api for LastTransactionId {

    type Error = crate::Error;

    type Response = Option<u64>;

}

/// Creates a `PubSub` [`Subscriber`](crate::pubsub::Subscriber)

pub struct CreateSubscriber {

    /// The name of the database.

    pub database: String,

}

impl Api for CreateSubscriber {

    type Error = crate::Error;

    type Response = u64;

}

/// Publishes `payload` to all subscribers of `topic`.

pub struct Publish {

    /// The name of the database.

    pub database: String,

    /// The topics to publish to.

    pub topic: Bytes,

    /// The payload to publish.

    pub payload: Bytes,

}

impl Api for Publish {

    type Error = crate::Error;

    type Response = ();

}

/// Publishes `payload` to all subscribers of all `topics`.

pub struct PublishToAll {

    /// The name of the database.

    pub database: String,

    /// The topics to publish to.

    pub topics: Vec<Bytes>,

    /// The payload to publish.

    pub payload: Bytes,

}

impl Api for PublishToAll {

    type Error = crate::Error;

    type Response = ();

}

/// Subscribes `subscriber_id` to messages for `topic`.

pub struct SubscribeTo {

    /// The name of the database.

    pub database: String,

    /// The id of the [`Subscriber`](crate::pubsub::Subscriber).

    pub subscriber_id: u64,

    /// The topic to subscribe to.

    pub topic: Bytes,

}

impl Api for SubscribeTo {

    type Error = crate::Error;

    type Response = ();

}

/// A PubSub message was received.

pub struct MessageReceived {

    /// The ID of the subscriber receiving the message.

    pub subscriber_id: u64,

    /// The topic the payload was received on.

    pub topic: Bytes,

    /// The message payload.

    pub payload: Bytes,

}

impl Api for MessageReceived {

    type Error = crate::Error;

    type Response = Self;

}

/// Unsubscribes `subscriber_id` from messages for `topic`.

pub struct UnsubscribeFrom {

    /// The name of the database.

    pub database: String,

    /// The id of the [`Subscriber`](crate::pubsub::Subscriber).

    pub subscriber_id: u64,

    /// The topic to unsubscribe from.

    pub topic: Bytes,

}

impl Api for UnsubscribeFrom {

    type Error = crate::Error;

    type Response = ();

}

/// Unregisters the subscriber.

pub struct UnregisterSubscriber {

    /// The name of the database.

    pub database: String,

    /// The id of the [`Subscriber`](crate::pubsub::Subscriber).

    pub subscriber_id: u64,

}

impl Api for UnregisterSubscriber {

    type Error = crate::Error;

    type Response = ();

}

/// Excutes a key-value store operation.

pub struct ExecuteKeyOperation {

    /// The name of the database.

    pub database: String,

    /// The operation to execute.

    pub op: KeyOperation,

}

impl Api for ExecuteKeyOperation {

    type Error = crate::Error;

    type Response = Output;

}

/// Compacts the collection.

pub struct CompactCollection {

    /// The name of the database.

    pub database: String,

    /// The name of the collection to compact.

    pub name: CollectionName,

}

impl Api for CompactCollection {

    type Error = crate::Error;

    type Response = ();

}

/// Compacts the key-value store.

pub struct CompactKeyValueStore {

    /// The name of the database.

    pub database: String,

}

impl Api for CompactKeyValueStore {

    type Error = crate::Error;

    type Response = ();

}

/// Compacts the entire database.

pub struct Compact {

    /// The name of the database.

    pub database: String,

}

impl Api for Compact {

    type Error = crate::Error;

    type Response = ();

}

/// A networking error.

pub enum Error {

    /// The server responded with a message that wasn't expected for the request

    /// sent.

    #[error("unexpected response: {0}")]

    UnexpectedResponse(String),

    /// A timeout occurred while connecting to the server.

    #[error("connection timeout")]

    ConnectTimeout,

    /// A timeout occurred waiting on a request to complete.

    #[error("request timeout")]

    RequestTimeout,

    /// The connection was interrupted.

    #[error("unexpected disconnection")]

    Disconnected,

}