use pile_config::Label;
use std::sync::Arc;

mod epub_meta;
pub use epub_meta::*;

mod epub_text;
pub use epub_text::*;

use crate::{
	extract::traits::{ExtractState, ObjectExtractor},
	value::{Item, PileValue},
};

pub struct EpubExtractor {
	text: Arc<EpubTextExtractor>,
	meta: Arc<EpubMetaExtractor>,
}

impl EpubExtractor {
	pub fn new(item: &Item) -> Self {
		Self {
			text: Arc::new(EpubTextExtractor::new(item)),
			meta: Arc::new(EpubMetaExtractor::new(item)),
		}
	}
}

#[async_trait::async_trait]
impl ObjectExtractor for EpubExtractor {
	async fn field(
		&self,
		state: &ExtractState,
		name: &pile_config::Label,
		args: Option<&str>,
	) -> Result<Option<PileValue>, std::io::Error> {
		match (name.as_str(), args) {
			("text", args) => Ok(Some(
				self.text
					.field(state, name, args)
					.await
					.map(|x| x.unwrap_or(PileValue::Null))?,
			)),

			("meta", None) => Ok(Some(PileValue::ObjectExtractor(self.meta.clone()))),
			_ => Ok(None),
		}
	}

	#[expect(clippy::unwrap_used)]
	async fn fields(&self) -> Result<Vec<Label>, std::io::Error> {
		Ok(vec![
			Label::new("text").unwrap(),
			Label::new("meta").unwrap(),
		])
	}
}