Allow disabling grpc at the build level

This commit is contained in:
benthecarman 2023-09-20 13:07:38 -05:00
parent 26f296f76f
commit c83ba7db39
No known key found for this signature in database
GPG Key ID: D7CC770B81FD22A8
5 changed files with 48 additions and 35 deletions

View File

@ -11,14 +11,18 @@ license = "MIT"
keywords = ["nostr", "server"]
categories = ["network-programming", "web-programming"]
[features]
default = ["grpc"]
grpc = ["tonic", "prost", "tonic-build"]
[dependencies]
clap = { version = "4.0.32", features = ["env", "default", "derive"]}
tracing = "0.1.37"
tracing-appender = "0.2.2"
tracing-subscriber = "0.3.16"
tokio = { version = "1", features = ["full", "tracing", "signal"] }
prost = "0.11"
tonic = "0.8.3"
prost = { version = "0.11", optional = true }
tonic = { version = "0.8.3", optional = true }
console-subscriber = "0.1.8"
futures = "0.3"
futures-util = "0.3"
@ -63,4 +67,4 @@ log = "0.4"
anyhow = "1"
[build-dependencies]
tonic-build = { version="0.8.3", features = ["prost"] }
tonic-build = { version="0.8.3", features = ["prost"], optional = true }

View File

@ -1,4 +1,5 @@
fn main() -> Result<(), Box<dyn std::error::Error>> {
#[cfg(feature = "grpc")]
tonic_build::configure()
.build_server(false)
.protoc_arg("--experimental_allow_proto3_optional")

View File

@ -2,6 +2,7 @@
use crate::config::Settings;
use crate::error::{Error, Result};
use crate::event::Event;
#[cfg(feature = "grpc")]
use crate::nauthz;
use crate::notice::Notice;
use crate::payment::PaymentMessage;
@ -136,6 +137,7 @@ pub async fn db_writer(
}
// create a client if GRPC is enabled.
// Check with externalized event admitter service, if one is defined.
#[cfg(feature = "grpc")]
let mut grpc_client = if let Some(svr) = settings.grpc.event_admission_server {
Some(nauthz::EventAuthzService::connect(&svr).await)
} else {
@ -345,29 +347,31 @@ pub async fn db_writer(
}
}
// nip05 address
let nip05_address: Option<crate::nip05::Nip05Name> =
validation.and_then(|x| x.ok().map(|y| y.name));
// GRPC check
if let Some(ref mut c) = grpc_client {
trace!("checking if grpc permits");
let grpc_start = Instant::now();
let decision_res = c
.admit_event(
&event,
&subm_event.source_ip,
subm_event.origin,
subm_event.user_agent,
nip05_address,
subm_event.auth_pubkey,
)
.await;
match decision_res {
Ok(decision) => {
if !decision.permitted() {
// GPRC returned a decision to reject this event
info!(
#[cfg(feature = "grpc")]
{
// nip05 address
let nip05_address: Option<crate::nip05::Nip05Name> =
validation.and_then(|x| x.ok().map(|y| y.name));
if let Some(ref mut c) = grpc_client {
trace!("checking if grpc permits");
let grpc_start = Instant::now();
let decision_res = c
.admit_event(
&event,
&subm_event.source_ip,
subm_event.origin,
subm_event.user_agent,
nip05_address,
subm_event.auth_pubkey,
)
.await;
match decision_res {
Ok(decision) => {
if !decision.permitted() {
// GPRC returned a decision to reject this event
info!(
"GRPC rejected event: {:?} (kind: {}) from: {:?} in: {:?} (IP: {:?})",
event.get_event_id_prefix(),
event.kind,
@ -375,17 +379,18 @@ pub async fn db_writer(
grpc_start.elapsed(),
subm_event.source_ip
);
notice_tx
.try_send(Notice::blocked(
event.id,
&decision.message().unwrap_or_default(),
))
.ok();
continue;
notice_tx
.try_send(Notice::blocked(
event.id,
&decision.message().unwrap_or_default(),
))
.ok();
continue;
}
}
Err(e) => {
warn!("GRPC server error: {:?}", e);
}
}
Err(e) => {
warn!("GRPC server error: {:?}", e);
}
}
}

View File

@ -66,6 +66,7 @@ pub enum Error {
ChannelClosed,
#[error("Authz error")]
AuthzError,
#[cfg(feature = "grpc")]
#[error("Tonic GRPC error")]
TonicError(tonic::Status),
#[error("Invalid AUTH message")]
@ -151,6 +152,7 @@ impl From<config::ConfigError> for Error {
}
}
#[cfg(feature = "grpc")]
impl From<tonic::Status> for Error {
/// Wrap Config error
fn from(r: tonic::Status) -> Self {

View File

@ -8,6 +8,7 @@ pub mod error;
pub mod event;
pub mod hexrange;
pub mod info;
#[cfg(feature = "grpc")]
pub mod nauthz;
pub mod nip05;
pub mod notice;