use super::*;

impl SpacetimeClient {
    pub async fn export_auth_store_snapshot_from_tables(
        &self,
    ) -> Result<AuthStoreSnapshotRecord, SpacetimeClientError> {
        self.call_after_connect(move |connection, sender| {
            connection
                .procedures()
                .export_auth_store_snapshot_from_tables_then(move |_, result| {
                    let mapped = result
                        .map_err(SpacetimeClientError::from_sdk_error)
                        .and_then(map_auth_store_snapshot_procedure_result);
                    send_once(&sender, mapped);
                });
        })
        .await
    }

    pub async fn get_auth_store_snapshot(
        &self,
    ) -> Result<AuthStoreSnapshotRecord, SpacetimeClientError> {
        self.call_after_connect(move |connection, sender| {
            connection
                .procedures()
                .get_auth_store_snapshot_then(move |_, result| {
                    let mapped = result
                        .map_err(SpacetimeClientError::from_sdk_error)
                        .and_then(map_auth_store_snapshot_procedure_result);
                    send_once(&sender, mapped);
                });
        })
        .await
    }

    pub async fn upsert_auth_store_snapshot(
        &self,
        snapshot_json: String,
        updated_at_micros: i64,
    ) -> Result<AuthStoreSnapshotRecord, SpacetimeClientError> {
        let procedure_input = AuthStoreSnapshotUpsertInput {
            snapshot_json,
            updated_at_micros,
        };

        self.call_after_connect(move |connection, sender| {
            connection.procedures().upsert_auth_store_snapshot_then(
                procedure_input,
                move |_, result| {
                    let mapped = result
                        .map_err(SpacetimeClientError::from_sdk_error)
                        .and_then(map_auth_store_snapshot_procedure_result);
                    send_once(&sender, mapped);
                },
            );
        })
        .await
    }

    pub async fn import_auth_store_snapshot_json(
        &self,
        snapshot_json: String,
        updated_at_micros: i64,
    ) -> Result<AuthStoreSnapshotImportRecord, SpacetimeClientError> {
        let procedure_input = AuthStoreSnapshotUpsertInput {
            snapshot_json,
            updated_at_micros,
        };

        self.call_after_connect(move |connection, sender| {
            connection
                .procedures()
                .import_auth_store_snapshot_json_then(procedure_input, move |_, result| {
                    let mapped = result
                        .map_err(SpacetimeClientError::from_sdk_error)
                        .and_then(map_auth_store_snapshot_import_procedure_result);
                    send_once(&sender, mapped);
                });
        })
        .await
    }

    pub async fn import_auth_store_snapshot(
        &self,
    ) -> Result<AuthStoreSnapshotImportRecord, SpacetimeClientError> {
        self.call_after_connect(move |connection, sender| {
            connection
                .procedures()
                .import_auth_store_snapshot_then(move |_, result| {
                    let mapped = result
                        .map_err(SpacetimeClientError::from_sdk_error)
                        .and_then(map_auth_store_snapshot_import_procedure_result);
                    send_once(&sender, mapped);
                });
        })
        .await
    }
}