use ruma::ServerName;

use super::{OutgoingKind, SendingEventType};
use crate::Result;

type OutgoingSendingIter<'a> = Box<dyn Iterator<Item = Result<(Vec<u8>, OutgoingKind, SendingEventType)>> + 'a>;
type SendingEventTypeIter<'a> = Box<dyn Iterator<Item = Result<(Vec<u8>, SendingEventType)>> + 'a>;

pub trait Data: Send + Sync {
	fn active_requests(&self) -> OutgoingSendingIter<'_>;
	fn active_requests_for(&self, outgoing_kind: &OutgoingKind) -> SendingEventTypeIter<'_>;
	fn delete_active_request(&self, key: Vec<u8>) -> Result<()>;
	fn delete_all_active_requests_for(&self, outgoing_kind: &OutgoingKind) -> Result<()>;
	fn delete_all_requests_for(&self, outgoing_kind: &OutgoingKind) -> Result<()>;
	fn queue_requests(&self, requests: &[(&OutgoingKind, SendingEventType)]) -> Result<Vec<Vec<u8>>>;
	fn queued_requests<'a>(
		&'a self, outgoing_kind: &OutgoingKind,
	) -> Box<dyn Iterator<Item = Result<(SendingEventType, Vec<u8>)>> + 'a>;
	fn mark_as_active(&self, events: &[(SendingEventType, Vec<u8>)]) -> Result<()>;
	fn set_latest_educount(&self, server_name: &ServerName, educount: u64) -> Result<()>;
	fn get_latest_educount(&self, server_name: &ServerName) -> Result<u64>;
}