#![allow(clippy::pedantic, clippy::nursery, clippy::all)]
use serde::{Deserialize, Serialize};
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct Params {
#[serde(skip_serializing_if = "Option::is_none")]
pub author: Option<proto_blue_syntax::AtIdentifier>,
#[serde(skip_serializing_if = "Option::is_none")]
pub cursor: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub domain: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub lang: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub limit: Option<i64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub mentions: Option<proto_blue_syntax::AtIdentifier>,
pub q: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub since: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub sort: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub tag: Option<Vec<String>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub until: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub url: Option<String>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct Output {
#[serde(skip_serializing_if = "Option::is_none")]
pub cursor: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub hits_total: Option<i64>,
pub posts: Vec<crate::app::bsky::feed::defs::PostView>,
}
#[derive(Debug, thiserror::Error)]
pub enum CallError {
#[error("BadQueryString")]
BadQueryString,
#[error("{0}")]
Xrpc(proto_blue_xrpc::XrpcError),
#[error(transparent)]
Transport(#[from] proto_blue_xrpc::Error),
#[error(transparent)]
Json(#[from] serde_json::Error),
}
fn map_xrpc_error(err: proto_blue_xrpc::XrpcError) -> CallError {
match err.error.as_deref() {
Some("BadQueryString") => CallError::BadQueryString,
_ => CallError::Xrpc(err),
}
}
fn to_query_params(p: &Params) -> proto_blue_xrpc::QueryParams {
let mut qp = proto_blue_xrpc::QueryParams::new();
if let Some(v) = &p.author {
qp.insert(
"author".to_string(),
proto_blue_xrpc::QueryValue::String(v.to_string()),
);
}
if let Some(v) = &p.cursor {
qp.insert(
"cursor".to_string(),
proto_blue_xrpc::QueryValue::String(v.to_string()),
);
}
if let Some(v) = &p.domain {
qp.insert(
"domain".to_string(),
proto_blue_xrpc::QueryValue::String(v.to_string()),
);
}
if let Some(v) = &p.lang {
qp.insert(
"lang".to_string(),
proto_blue_xrpc::QueryValue::String(v.to_string()),
);
}
if let Some(v) = &p.limit {
qp.insert(
"limit".to_string(),
proto_blue_xrpc::QueryValue::Integer(*v),
);
}
if let Some(v) = &p.mentions {
qp.insert(
"mentions".to_string(),
proto_blue_xrpc::QueryValue::String(v.to_string()),
);
}
{
let v = &p.q;
qp.insert(
"q".to_string(),
proto_blue_xrpc::QueryValue::String(v.to_string()),
);
}
if let Some(v) = &p.since {
qp.insert(
"since".to_string(),
proto_blue_xrpc::QueryValue::String(v.to_string()),
);
}
if let Some(v) = &p.sort {
qp.insert(
"sort".to_string(),
proto_blue_xrpc::QueryValue::String(v.to_string()),
);
}
if let Some(v) = &p.tag {
qp.insert(
"tag".to_string(),
proto_blue_xrpc::QueryValue::Array(
v.iter()
.map(|x| proto_blue_xrpc::QueryValue::String(x.to_string()))
.collect(),
),
);
}
if let Some(v) = &p.until {
qp.insert(
"until".to_string(),
proto_blue_xrpc::QueryValue::String(v.to_string()),
);
}
if let Some(v) = &p.url {
qp.insert(
"url".to_string(),
proto_blue_xrpc::QueryValue::String(v.to_string()),
);
}
qp
}
pub async fn call(
client: &proto_blue_xrpc::XrpcClient,
params: Option<&Params>,
opts: Option<&proto_blue_xrpc::CallOptions>,
) -> Result<Output, CallError> {
let qp = params.map(to_query_params);
let response = match client
.query("app.bsky.feed.searchPosts", qp.as_ref(), opts)
.await
{
Ok(r) => r,
Err(proto_blue_xrpc::Error::Xrpc(x)) => return Err(map_xrpc_error(x)),
Err(e) => return Err(CallError::Transport(e)),
};
Ok(serde_json::from_value(response.data)?)
}
#[cfg(feature = "server")]
pub fn register<F, Fut>(
server: proto_blue_xrpc::XrpcServer,
handler: F,
) -> proto_blue_xrpc::XrpcServer
where
F: Fn(proto_blue_xrpc::HandlerContext, Option<Params>) -> Fut + Send + Sync + 'static,
Fut: std::future::Future<Output = Result<Output, proto_blue_xrpc::XrpcServerError>>
+ Send
+ 'static,
{
let handler = std::sync::Arc::new(handler);
server.query("app.bsky.feed.searchPosts", move |ctx| {
let handler = handler.clone();
async move {
let params = params_from_ctx(&ctx);
let out = handler(ctx, params).await?;
let value = serde_json::to_value(&out).map_err(|e| {
proto_blue_xrpc::XrpcServerError::new(
proto_blue_xrpc::ResponseType::InternalServerError,
format!("output serialize: {e}"),
)
})?;
Ok::<_, proto_blue_xrpc::XrpcServerError>(value)
}
})
}
#[cfg(feature = "server")]
fn params_from_ctx(ctx: &proto_blue_xrpc::HandlerContext) -> Option<Params> {
Some(Params {
author: ctx
.params
.get("author")
.and_then(|v| proto_blue_syntax::AtIdentifier::new(v).ok()),
cursor: ctx.params.get("cursor").cloned(),
domain: ctx.params.get("domain").cloned(),
lang: ctx.params.get("lang").cloned(),
limit: ctx.params.get("limit").and_then(|v| v.parse::<i64>().ok()),
mentions: ctx
.params
.get("mentions")
.and_then(|v| proto_blue_syntax::AtIdentifier::new(v).ok()),
q: (ctx.params.get("q").cloned())?,
since: ctx.params.get("since").cloned(),
sort: ctx.params.get("sort").cloned(),
tag: Some(
ctx.params
.get("tag")
.map(|v| v.split(',').map(String::from).collect::<Vec<_>>())
.unwrap_or_default(),
),
until: ctx.params.get("until").cloned(),
url: ctx.params.get("url").cloned(),
})
}