use axum::extract::{Path, Query, State};
use sea_orm::{ColumnTrait, Order, PaginatorTrait, QueryFilter, QueryOrder, QuerySelect};

use crate::{model::{self, addressing::WrappedObject}, routes::activitypub::{jsonld::LD, JsonLD, Pagination}, server::{auth::AuthIdentity, Context}, url};

pub async fn get(
	State(ctx): State<Context>,
	Path(id): Path<String>,
	AuthIdentity(auth): AuthIdentity,
) -> crate::Result<JsonLD<serde_json::Value>> {
	let context = if id.starts_with('+') {
		format!("https://{}", id.replacen('+', "", 1).replace('@', "/"))
	} else {
		url!(ctx, "/context/{id}")
	};

	let count = model::addressing::Entity::find_objects()
		.filter(auth.filter_condition())
		.filter(model::object::Column::Context.eq(context))
		.count(ctx.db())
		.await?;

	Ok(JsonLD(ctx.ap_collection(&url!(ctx, "/context/{id}"), Some(count)).ld_context()))
}

pub async fn page(
	State(ctx): State<Context>,
	Path(id): Path<String>,
	Query(page): Query<Pagination>,
	AuthIdentity(auth): AuthIdentity,
) -> crate::Result<JsonLD<serde_json::Value>> {
	let limit = page.batch.unwrap_or(20).min(50);
	let offset = page.offset.unwrap_or(0);

	let context = if id.starts_with('+') {
		id.replacen('+', "https://", 1).replace('@', "/")
	} else if id.starts_with("tag:") {
		id.clone()
	} else {
		url!(ctx, "/context/{id}") // TODO need a better way to figure out which ones are our contexts
	};

	let items = model::addressing::Entity::find_objects()
		.filter(auth.filter_condition())
		.filter(model::object::Column::Context.eq(context))
		.limit(limit)
		.offset(offset)
		.into_model::<WrappedObject>()
		.all(ctx.db())
		.await?;

	let mut out = Vec::new();
	for item in items {
		out.push(item.ap_filled(ctx.db()).await?);
	}

	Ok(JsonLD(
		ctx.ap_collection_page(
			&url!(ctx, "/context/{id}/page"),
			offset, limit, out,
		).ld_context()
	))
}