Public Access
fix: format and lint
This commit is contained in:
@@ -308,7 +308,10 @@ fn extract_metadata(entity: &Entity) -> Result<Value, Box<dyn std::error::Error>
|
||||
}
|
||||
let mut annotations = serde_json::Map::new();
|
||||
if !m.resource_type.is_empty() {
|
||||
annotations.insert("keycloak.org/resource-type".to_string(), json!(m.resource_type));
|
||||
annotations.insert(
|
||||
"keycloak.org/resource-type".to_string(),
|
||||
json!(m.resource_type),
|
||||
);
|
||||
}
|
||||
if !m.realm.is_empty() {
|
||||
annotations.insert("keycloak.org/realm".to_string(), json!(m.realm));
|
||||
@@ -465,10 +468,10 @@ fn extract_spec(entity: &Entity) -> Option<Value> {
|
||||
Some(Spec::DefectdojoSpec(s)) => {
|
||||
// DefectDojo plugin spec - parse config_json if present
|
||||
let mut spec = json!({});
|
||||
if !s.config_json.is_empty() {
|
||||
if let Ok(config) = serde_json::from_str::<Value>(&s.config_json) {
|
||||
spec = config;
|
||||
}
|
||||
if !s.config_json.is_empty()
|
||||
&& let Ok(config) = serde_json::from_str::<Value>(&s.config_json)
|
||||
{
|
||||
spec = config;
|
||||
}
|
||||
if !s.sync_status.is_empty() {
|
||||
spec["sync_status"] = json!(s.sync_status);
|
||||
@@ -478,10 +481,10 @@ fn extract_spec(entity: &Entity) -> Option<Value> {
|
||||
Some(Spec::KeycloakSpec(s)) => {
|
||||
// Keycloak plugin spec - parse config_json if present
|
||||
let mut spec = json!({});
|
||||
if !s.config_json.is_empty() {
|
||||
if let Ok(config) = serde_json::from_str::<Value>(&s.config_json) {
|
||||
spec = config;
|
||||
}
|
||||
if !s.config_json.is_empty()
|
||||
&& let Ok(config) = serde_json::from_str::<Value>(&s.config_json)
|
||||
{
|
||||
spec = config;
|
||||
}
|
||||
if !s.sync_status.is_empty() {
|
||||
spec["sync_status"] = json!(s.sync_status);
|
||||
@@ -497,10 +500,10 @@ fn extract_spec(entity: &Entity) -> Option<Value> {
|
||||
Some(Spec::DependencytrackSpec(s)) => {
|
||||
// Dependency-Track plugin spec - parse config_json if present
|
||||
let mut spec = json!({});
|
||||
if !s.config_json.is_empty() {
|
||||
if let Ok(config) = serde_json::from_str::<Value>(&s.config_json) {
|
||||
spec = config;
|
||||
}
|
||||
if !s.config_json.is_empty()
|
||||
&& let Ok(config) = serde_json::from_str::<Value>(&s.config_json)
|
||||
{
|
||||
spec = config;
|
||||
}
|
||||
if !s.sync_status.is_empty() {
|
||||
spec["sync_status"] = json!(s.sync_status);
|
||||
|
||||
+10
-10
@@ -71,16 +71,16 @@ async fn get_locations(State(state): State<YamlAdapterState>) -> Response {
|
||||
// Check cache first
|
||||
{
|
||||
let cache = state.cache.read().await;
|
||||
if let Some(ref cached) = *cache {
|
||||
if cached.cached_at.elapsed() < state.cache_ttl {
|
||||
debug!("Serving /yaml/locations from cache");
|
||||
return (
|
||||
StatusCode::OK,
|
||||
[("content-type", "text/yaml; charset=utf-8")],
|
||||
cached.data.clone(),
|
||||
)
|
||||
.into_response();
|
||||
}
|
||||
if let Some(ref cached) = *cache
|
||||
&& cached.cached_at.elapsed() < state.cache_ttl
|
||||
{
|
||||
debug!("Serving /yaml/locations from cache");
|
||||
return (
|
||||
StatusCode::OK,
|
||||
[("content-type", "text/yaml; charset=utf-8")],
|
||||
cached.data.clone(),
|
||||
)
|
||||
.into_response();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
+1
-2
@@ -231,8 +231,7 @@ impl Config {
|
||||
fn substitute_env_vars(content: &str) -> Result<String> {
|
||||
let mut result = content.to_string();
|
||||
// Match ${VAR_NAME} or ${VAR_NAME:-default_value}
|
||||
let var_pattern =
|
||||
regex::Regex::new(r"\$\{([A-Z_][A-Z0-9_]*)(?::-((?:[^}])*))?\}").unwrap();
|
||||
let var_pattern = regex::Regex::new(r"\$\{([A-Z_][A-Z0-9_]*)(?::-((?:[^}])*))?\}").unwrap();
|
||||
|
||||
for capture in var_pattern.captures_iter(content) {
|
||||
let full_match = &capture[0];
|
||||
|
||||
+200
-99
@@ -403,10 +403,14 @@ impl EntityRepository {
|
||||
// Decode cursor: "timestamp|uuid" (pipe separator since RFC3339 contains colons)
|
||||
let cursor = page_token.and_then(|token| {
|
||||
use base64::Engine;
|
||||
let decoded = base64::engine::general_purpose::STANDARD.decode(token).ok()?;
|
||||
let decoded = base64::engine::general_purpose::STANDARD
|
||||
.decode(token)
|
||||
.ok()?;
|
||||
let s = String::from_utf8(decoded).ok()?;
|
||||
let (ts_str, id_str) = s.rsplit_once('|')?;
|
||||
let ts = DateTime::parse_from_rfc3339(ts_str).ok()?.with_timezone(&Utc);
|
||||
let ts = DateTime::parse_from_rfc3339(ts_str)
|
||||
.ok()?
|
||||
.with_timezone(&Utc);
|
||||
let id = Uuid::parse_str(id_str).ok()?;
|
||||
Some((ts, id))
|
||||
});
|
||||
@@ -673,7 +677,10 @@ impl EntityRepository {
|
||||
Ok(Some((entity_data_bytes, merged_annotations_json))) => {
|
||||
// Rebuild protobuf blob with merged annotations
|
||||
let mut entity = Entity::decode(entity_data_bytes.as_slice())?;
|
||||
if let Ok(merged) = serde_json::from_value::<std::collections::HashMap<String, String>>(merged_annotations_json) {
|
||||
if let Ok(merged) = serde_json::from_value::<
|
||||
std::collections::HashMap<String, String>,
|
||||
>(merged_annotations_json)
|
||||
{
|
||||
entity.annotations = merged;
|
||||
}
|
||||
entity.updated_at = Some(prost_types::Timestamp {
|
||||
@@ -872,56 +879,112 @@ fn apply_field_mask(
|
||||
|
||||
// Nested fields - annotations
|
||||
path if path.starts_with("annotations.") => {
|
||||
if let Some(key) = path.strip_prefix("annotations.") {
|
||||
if let Some(value) = partial.annotations.get(key) {
|
||||
existing.annotations.insert(key.to_string(), value.clone());
|
||||
}
|
||||
if let Some(key) = path.strip_prefix("annotations.")
|
||||
&& let Some(value) = partial.annotations.get(key)
|
||||
{
|
||||
existing.annotations.insert(key.to_string(), value.clone());
|
||||
}
|
||||
}
|
||||
|
||||
// Component metadata fields
|
||||
"component_metadata.name" => apply_metadata_field!(partial, existing, ComponentMetadata, name),
|
||||
"component_metadata.namespace" => apply_metadata_field!(partial, existing, ComponentMetadata, namespace),
|
||||
"component_metadata.description" => apply_metadata_field!(partial, existing, ComponentMetadata, description),
|
||||
"component_metadata.labels" => apply_metadata_field!(partial, existing, ComponentMetadata, labels),
|
||||
"component_metadata.tags" => apply_metadata_field!(partial, existing, ComponentMetadata, tags),
|
||||
"component_metadata.links" => apply_metadata_field!(partial, existing, ComponentMetadata, links),
|
||||
"component_metadata.name" => {
|
||||
apply_metadata_field!(partial, existing, ComponentMetadata, name)
|
||||
}
|
||||
"component_metadata.namespace" => {
|
||||
apply_metadata_field!(partial, existing, ComponentMetadata, namespace)
|
||||
}
|
||||
"component_metadata.description" => {
|
||||
apply_metadata_field!(partial, existing, ComponentMetadata, description)
|
||||
}
|
||||
"component_metadata.labels" => {
|
||||
apply_metadata_field!(partial, existing, ComponentMetadata, labels)
|
||||
}
|
||||
"component_metadata.tags" => {
|
||||
apply_metadata_field!(partial, existing, ComponentMetadata, tags)
|
||||
}
|
||||
"component_metadata.links" => {
|
||||
apply_metadata_field!(partial, existing, ComponentMetadata, links)
|
||||
}
|
||||
|
||||
// Component spec fields
|
||||
"component_spec.type" => apply_spec_field!(partial, existing, ComponentSpec, r#type),
|
||||
"component_spec.lifecycle" => apply_spec_field!(partial, existing, ComponentSpec, lifecycle),
|
||||
"component_spec.lifecycle" => {
|
||||
apply_spec_field!(partial, existing, ComponentSpec, lifecycle)
|
||||
}
|
||||
"component_spec.owner" => apply_spec_field!(partial, existing, ComponentSpec, owner),
|
||||
"component_spec.system" => apply_spec_field!(partial, existing, ComponentSpec, system),
|
||||
"component_spec.subcomponent_of" => apply_spec_field!(partial, existing, ComponentSpec, subcomponent_of),
|
||||
"component_spec.depends_on" => apply_spec_field!(partial, existing, ComponentSpec, depends_on),
|
||||
"component_spec.provides_apis" => apply_spec_field!(partial, existing, ComponentSpec, provides_apis),
|
||||
"component_spec.consumes_apis" => apply_spec_field!(partial, existing, ComponentSpec, consumes_apis),
|
||||
"component_spec.subcomponent_of" => {
|
||||
apply_spec_field!(partial, existing, ComponentSpec, subcomponent_of)
|
||||
}
|
||||
"component_spec.depends_on" => {
|
||||
apply_spec_field!(partial, existing, ComponentSpec, depends_on)
|
||||
}
|
||||
"component_spec.provides_apis" => {
|
||||
apply_spec_field!(partial, existing, ComponentSpec, provides_apis)
|
||||
}
|
||||
"component_spec.consumes_apis" => {
|
||||
apply_spec_field!(partial, existing, ComponentSpec, consumes_apis)
|
||||
}
|
||||
|
||||
// Service metadata fields
|
||||
"service_metadata.name" => apply_metadata_field!(partial, existing, ServiceMetadata, name),
|
||||
"service_metadata.namespace" => apply_metadata_field!(partial, existing, ServiceMetadata, namespace),
|
||||
"service_metadata.description" => apply_metadata_field!(partial, existing, ServiceMetadata, description),
|
||||
"service_metadata.labels" => apply_metadata_field!(partial, existing, ServiceMetadata, labels),
|
||||
"service_metadata.tags" => apply_metadata_field!(partial, existing, ServiceMetadata, tags),
|
||||
"service_metadata.links" => apply_metadata_field!(partial, existing, ServiceMetadata, links),
|
||||
"service_metadata.name" => {
|
||||
apply_metadata_field!(partial, existing, ServiceMetadata, name)
|
||||
}
|
||||
"service_metadata.namespace" => {
|
||||
apply_metadata_field!(partial, existing, ServiceMetadata, namespace)
|
||||
}
|
||||
"service_metadata.description" => {
|
||||
apply_metadata_field!(partial, existing, ServiceMetadata, description)
|
||||
}
|
||||
"service_metadata.labels" => {
|
||||
apply_metadata_field!(partial, existing, ServiceMetadata, labels)
|
||||
}
|
||||
"service_metadata.tags" => {
|
||||
apply_metadata_field!(partial, existing, ServiceMetadata, tags)
|
||||
}
|
||||
"service_metadata.links" => {
|
||||
apply_metadata_field!(partial, existing, ServiceMetadata, links)
|
||||
}
|
||||
|
||||
// Service spec fields
|
||||
"service_spec.type" => apply_spec_field!(partial, existing, ServiceSpec, r#type),
|
||||
"service_spec.lifecycle" => apply_spec_field!(partial, existing, ServiceSpec, lifecycle),
|
||||
"service_spec.lifecycle" => {
|
||||
apply_spec_field!(partial, existing, ServiceSpec, lifecycle)
|
||||
}
|
||||
"service_spec.owner" => apply_spec_field!(partial, existing, ServiceSpec, owner),
|
||||
"service_spec.system" => apply_spec_field!(partial, existing, ServiceSpec, system),
|
||||
"service_spec.subcomponent_of" => apply_spec_field!(partial, existing, ServiceSpec, subcomponent_of),
|
||||
"service_spec.depends_on" => apply_spec_field!(partial, existing, ServiceSpec, depends_on),
|
||||
"service_spec.consumes_apis" => apply_spec_field!(partial, existing, ServiceSpec, consumes_apis),
|
||||
"service_spec.provides_apis" => apply_spec_field!(partial, existing, ServiceSpec, provides_apis),
|
||||
"service_spec.subcomponent_of" => {
|
||||
apply_spec_field!(partial, existing, ServiceSpec, subcomponent_of)
|
||||
}
|
||||
"service_spec.depends_on" => {
|
||||
apply_spec_field!(partial, existing, ServiceSpec, depends_on)
|
||||
}
|
||||
"service_spec.consumes_apis" => {
|
||||
apply_spec_field!(partial, existing, ServiceSpec, consumes_apis)
|
||||
}
|
||||
"service_spec.provides_apis" => {
|
||||
apply_spec_field!(partial, existing, ServiceSpec, provides_apis)
|
||||
}
|
||||
|
||||
// System metadata fields
|
||||
"system_metadata.name" => apply_metadata_field!(partial, existing, SystemMetadata, name),
|
||||
"system_metadata.namespace" => apply_metadata_field!(partial, existing, SystemMetadata, namespace),
|
||||
"system_metadata.description" => apply_metadata_field!(partial, existing, SystemMetadata, description),
|
||||
"system_metadata.labels" => apply_metadata_field!(partial, existing, SystemMetadata, labels),
|
||||
"system_metadata.tags" => apply_metadata_field!(partial, existing, SystemMetadata, tags),
|
||||
"system_metadata.links" => apply_metadata_field!(partial, existing, SystemMetadata, links),
|
||||
"system_metadata.name" => {
|
||||
apply_metadata_field!(partial, existing, SystemMetadata, name)
|
||||
}
|
||||
"system_metadata.namespace" => {
|
||||
apply_metadata_field!(partial, existing, SystemMetadata, namespace)
|
||||
}
|
||||
"system_metadata.description" => {
|
||||
apply_metadata_field!(partial, existing, SystemMetadata, description)
|
||||
}
|
||||
"system_metadata.labels" => {
|
||||
apply_metadata_field!(partial, existing, SystemMetadata, labels)
|
||||
}
|
||||
"system_metadata.tags" => {
|
||||
apply_metadata_field!(partial, existing, SystemMetadata, tags)
|
||||
}
|
||||
"system_metadata.links" => {
|
||||
apply_metadata_field!(partial, existing, SystemMetadata, links)
|
||||
}
|
||||
|
||||
// System spec fields
|
||||
"system_spec.owner" => apply_spec_field!(partial, existing, SystemSpec, owner),
|
||||
@@ -929,8 +992,12 @@ fn apply_field_mask(
|
||||
|
||||
// API metadata fields
|
||||
"api_metadata.name" => apply_metadata_field!(partial, existing, ApiMetadata, name),
|
||||
"api_metadata.namespace" => apply_metadata_field!(partial, existing, ApiMetadata, namespace),
|
||||
"api_metadata.description" => apply_metadata_field!(partial, existing, ApiMetadata, description),
|
||||
"api_metadata.namespace" => {
|
||||
apply_metadata_field!(partial, existing, ApiMetadata, namespace)
|
||||
}
|
||||
"api_metadata.description" => {
|
||||
apply_metadata_field!(partial, existing, ApiMetadata, description)
|
||||
}
|
||||
"api_metadata.labels" => apply_metadata_field!(partial, existing, ApiMetadata, labels),
|
||||
"api_metadata.tags" => apply_metadata_field!(partial, existing, ApiMetadata, tags),
|
||||
"api_metadata.links" => apply_metadata_field!(partial, existing, ApiMetadata, links),
|
||||
@@ -944,9 +1011,15 @@ fn apply_field_mask(
|
||||
|
||||
// User metadata fields
|
||||
"user_metadata.name" => apply_metadata_field!(partial, existing, UserMetadata, name),
|
||||
"user_metadata.namespace" => apply_metadata_field!(partial, existing, UserMetadata, namespace),
|
||||
"user_metadata.description" => apply_metadata_field!(partial, existing, UserMetadata, description),
|
||||
"user_metadata.labels" => apply_metadata_field!(partial, existing, UserMetadata, labels),
|
||||
"user_metadata.namespace" => {
|
||||
apply_metadata_field!(partial, existing, UserMetadata, namespace)
|
||||
}
|
||||
"user_metadata.description" => {
|
||||
apply_metadata_field!(partial, existing, UserMetadata, description)
|
||||
}
|
||||
"user_metadata.labels" => {
|
||||
apply_metadata_field!(partial, existing, UserMetadata, labels)
|
||||
}
|
||||
"user_metadata.tags" => apply_metadata_field!(partial, existing, UserMetadata, tags),
|
||||
"user_metadata.links" => apply_metadata_field!(partial, existing, UserMetadata, links),
|
||||
|
||||
@@ -956,43 +1029,48 @@ fn apply_field_mask(
|
||||
|
||||
// User spec nested profile fields
|
||||
"user_spec.profile.display_name" => {
|
||||
if let Some(Spec::UserSpec(ref partial_spec)) = partial.spec {
|
||||
if let Some(Spec::UserSpec(ref mut existing_spec)) = existing.spec {
|
||||
if let Some(ref partial_profile) = partial_spec.profile {
|
||||
let profile = existing_spec.profile.get_or_insert_default();
|
||||
profile.display_name = partial_profile.display_name.clone();
|
||||
}
|
||||
}
|
||||
if let Some(Spec::UserSpec(ref partial_spec)) = partial.spec
|
||||
&& let Some(Spec::UserSpec(ref mut existing_spec)) = existing.spec
|
||||
&& let Some(ref partial_profile) = partial_spec.profile
|
||||
{
|
||||
let profile = existing_spec.profile.get_or_insert_default();
|
||||
profile.display_name = partial_profile.display_name.clone();
|
||||
}
|
||||
}
|
||||
"user_spec.profile.email" => {
|
||||
if let Some(Spec::UserSpec(ref partial_spec)) = partial.spec {
|
||||
if let Some(Spec::UserSpec(ref mut existing_spec)) = existing.spec {
|
||||
if let Some(ref partial_profile) = partial_spec.profile {
|
||||
let profile = existing_spec.profile.get_or_insert_default();
|
||||
profile.email = partial_profile.email.clone();
|
||||
}
|
||||
}
|
||||
if let Some(Spec::UserSpec(ref partial_spec)) = partial.spec
|
||||
&& let Some(Spec::UserSpec(ref mut existing_spec)) = existing.spec
|
||||
&& let Some(ref partial_profile) = partial_spec.profile
|
||||
{
|
||||
let profile = existing_spec.profile.get_or_insert_default();
|
||||
profile.email = partial_profile.email.clone();
|
||||
}
|
||||
}
|
||||
"user_spec.profile.picture" => {
|
||||
if let Some(Spec::UserSpec(ref partial_spec)) = partial.spec {
|
||||
if let Some(Spec::UserSpec(ref mut existing_spec)) = existing.spec {
|
||||
if let Some(ref partial_profile) = partial_spec.profile {
|
||||
let profile = existing_spec.profile.get_or_insert_default();
|
||||
profile.picture = partial_profile.picture.clone();
|
||||
}
|
||||
}
|
||||
if let Some(Spec::UserSpec(ref partial_spec)) = partial.spec
|
||||
&& let Some(Spec::UserSpec(ref mut existing_spec)) = existing.spec
|
||||
&& let Some(ref partial_profile) = partial_spec.profile
|
||||
{
|
||||
let profile = existing_spec.profile.get_or_insert_default();
|
||||
profile.picture = partial_profile.picture.clone();
|
||||
}
|
||||
}
|
||||
|
||||
// Group metadata fields
|
||||
"group_metadata.name" => apply_metadata_field!(partial, existing, GroupMetadata, name),
|
||||
"group_metadata.namespace" => apply_metadata_field!(partial, existing, GroupMetadata, namespace),
|
||||
"group_metadata.description" => apply_metadata_field!(partial, existing, GroupMetadata, description),
|
||||
"group_metadata.labels" => apply_metadata_field!(partial, existing, GroupMetadata, labels),
|
||||
"group_metadata.namespace" => {
|
||||
apply_metadata_field!(partial, existing, GroupMetadata, namespace)
|
||||
}
|
||||
"group_metadata.description" => {
|
||||
apply_metadata_field!(partial, existing, GroupMetadata, description)
|
||||
}
|
||||
"group_metadata.labels" => {
|
||||
apply_metadata_field!(partial, existing, GroupMetadata, labels)
|
||||
}
|
||||
"group_metadata.tags" => apply_metadata_field!(partial, existing, GroupMetadata, tags),
|
||||
"group_metadata.links" => apply_metadata_field!(partial, existing, GroupMetadata, links),
|
||||
"group_metadata.links" => {
|
||||
apply_metadata_field!(partial, existing, GroupMetadata, links)
|
||||
}
|
||||
|
||||
// Group spec fields
|
||||
"group_spec.type" => apply_spec_field!(partial, existing, GroupSpec, r#type),
|
||||
@@ -1003,60 +1081,83 @@ fn apply_field_mask(
|
||||
|
||||
// Group spec nested profile fields
|
||||
"group_spec.profile.display_name" => {
|
||||
if let Some(Spec::GroupSpec(ref partial_spec)) = partial.spec {
|
||||
if let Some(Spec::GroupSpec(ref mut existing_spec)) = existing.spec {
|
||||
if let Some(ref partial_profile) = partial_spec.profile {
|
||||
let profile = existing_spec.profile.get_or_insert_default();
|
||||
profile.display_name = partial_profile.display_name.clone();
|
||||
}
|
||||
}
|
||||
if let Some(Spec::GroupSpec(ref partial_spec)) = partial.spec
|
||||
&& let Some(Spec::GroupSpec(ref mut existing_spec)) = existing.spec
|
||||
&& let Some(ref partial_profile) = partial_spec.profile
|
||||
{
|
||||
let profile = existing_spec.profile.get_or_insert_default();
|
||||
profile.display_name = partial_profile.display_name.clone();
|
||||
}
|
||||
}
|
||||
"group_spec.profile.email" => {
|
||||
if let Some(Spec::GroupSpec(ref partial_spec)) = partial.spec {
|
||||
if let Some(Spec::GroupSpec(ref mut existing_spec)) = existing.spec {
|
||||
if let Some(ref partial_profile) = partial_spec.profile {
|
||||
let profile = existing_spec.profile.get_or_insert_default();
|
||||
profile.email = partial_profile.email.clone();
|
||||
}
|
||||
}
|
||||
if let Some(Spec::GroupSpec(ref partial_spec)) = partial.spec
|
||||
&& let Some(Spec::GroupSpec(ref mut existing_spec)) = existing.spec
|
||||
&& let Some(ref partial_profile) = partial_spec.profile
|
||||
{
|
||||
let profile = existing_spec.profile.get_or_insert_default();
|
||||
profile.email = partial_profile.email.clone();
|
||||
}
|
||||
}
|
||||
"group_spec.profile.picture" => {
|
||||
if let Some(Spec::GroupSpec(ref partial_spec)) = partial.spec {
|
||||
if let Some(Spec::GroupSpec(ref mut existing_spec)) = existing.spec {
|
||||
if let Some(ref partial_profile) = partial_spec.profile {
|
||||
let profile = existing_spec.profile.get_or_insert_default();
|
||||
profile.picture = partial_profile.picture.clone();
|
||||
}
|
||||
}
|
||||
if let Some(Spec::GroupSpec(ref partial_spec)) = partial.spec
|
||||
&& let Some(Spec::GroupSpec(ref mut existing_spec)) = existing.spec
|
||||
&& let Some(ref partial_profile) = partial_spec.profile
|
||||
{
|
||||
let profile = existing_spec.profile.get_or_insert_default();
|
||||
profile.picture = partial_profile.picture.clone();
|
||||
}
|
||||
}
|
||||
|
||||
// Domain metadata fields
|
||||
"domain_metadata.name" => apply_metadata_field!(partial, existing, DomainMetadata, name),
|
||||
"domain_metadata.namespace" => apply_metadata_field!(partial, existing, DomainMetadata, namespace),
|
||||
"domain_metadata.description" => apply_metadata_field!(partial, existing, DomainMetadata, description),
|
||||
"domain_metadata.labels" => apply_metadata_field!(partial, existing, DomainMetadata, labels),
|
||||
"domain_metadata.tags" => apply_metadata_field!(partial, existing, DomainMetadata, tags),
|
||||
"domain_metadata.links" => apply_metadata_field!(partial, existing, DomainMetadata, links),
|
||||
"domain_metadata.name" => {
|
||||
apply_metadata_field!(partial, existing, DomainMetadata, name)
|
||||
}
|
||||
"domain_metadata.namespace" => {
|
||||
apply_metadata_field!(partial, existing, DomainMetadata, namespace)
|
||||
}
|
||||
"domain_metadata.description" => {
|
||||
apply_metadata_field!(partial, existing, DomainMetadata, description)
|
||||
}
|
||||
"domain_metadata.labels" => {
|
||||
apply_metadata_field!(partial, existing, DomainMetadata, labels)
|
||||
}
|
||||
"domain_metadata.tags" => {
|
||||
apply_metadata_field!(partial, existing, DomainMetadata, tags)
|
||||
}
|
||||
"domain_metadata.links" => {
|
||||
apply_metadata_field!(partial, existing, DomainMetadata, links)
|
||||
}
|
||||
|
||||
// Domain spec fields
|
||||
"domain_spec.owner" => apply_spec_field!(partial, existing, DomainSpec, owner),
|
||||
|
||||
// Resource metadata fields
|
||||
"resource_metadata.name" => apply_metadata_field!(partial, existing, ResourceMetadata, name),
|
||||
"resource_metadata.namespace" => apply_metadata_field!(partial, existing, ResourceMetadata, namespace),
|
||||
"resource_metadata.description" => apply_metadata_field!(partial, existing, ResourceMetadata, description),
|
||||
"resource_metadata.labels" => apply_metadata_field!(partial, existing, ResourceMetadata, labels),
|
||||
"resource_metadata.tags" => apply_metadata_field!(partial, existing, ResourceMetadata, tags),
|
||||
"resource_metadata.links" => apply_metadata_field!(partial, existing, ResourceMetadata, links),
|
||||
"resource_metadata.name" => {
|
||||
apply_metadata_field!(partial, existing, ResourceMetadata, name)
|
||||
}
|
||||
"resource_metadata.namespace" => {
|
||||
apply_metadata_field!(partial, existing, ResourceMetadata, namespace)
|
||||
}
|
||||
"resource_metadata.description" => {
|
||||
apply_metadata_field!(partial, existing, ResourceMetadata, description)
|
||||
}
|
||||
"resource_metadata.labels" => {
|
||||
apply_metadata_field!(partial, existing, ResourceMetadata, labels)
|
||||
}
|
||||
"resource_metadata.tags" => {
|
||||
apply_metadata_field!(partial, existing, ResourceMetadata, tags)
|
||||
}
|
||||
"resource_metadata.links" => {
|
||||
apply_metadata_field!(partial, existing, ResourceMetadata, links)
|
||||
}
|
||||
|
||||
// Resource spec fields
|
||||
"resource_spec.type" => apply_spec_field!(partial, existing, ResourceSpec, r#type),
|
||||
"resource_spec.owner" => apply_spec_field!(partial, existing, ResourceSpec, owner),
|
||||
"resource_spec.system" => apply_spec_field!(partial, existing, ResourceSpec, system),
|
||||
"resource_spec.depends_on" => apply_spec_field!(partial, existing, ResourceSpec, depends_on),
|
||||
"resource_spec.depends_on" => {
|
||||
apply_spec_field!(partial, existing, ResourceSpec, depends_on)
|
||||
}
|
||||
|
||||
_ => {
|
||||
tracing::warn!("Unknown field mask path: {}", path);
|
||||
|
||||
@@ -51,10 +51,10 @@ impl EventBus for MemoryEventBus {
|
||||
);
|
||||
|
||||
// Send through broadcast channel if we have one
|
||||
if let Some(sender) = &self.sender {
|
||||
if let Err(e) = sender.send(event.clone()) {
|
||||
error!("Failed to send event through broadcast channel: {}", e);
|
||||
}
|
||||
if let Some(sender) = &self.sender
|
||||
&& let Err(e) = sender.send(event.clone())
|
||||
{
|
||||
error!("Failed to send event through broadcast channel: {}", e);
|
||||
}
|
||||
|
||||
// Always dispatch directly to handlers
|
||||
|
||||
+5
-5
@@ -4,11 +4,11 @@
|
||||
//! Events are published when entities are created, updated, or deleted.
|
||||
//! Plugins and other components can subscribe to react to these events.
|
||||
|
||||
pub mod types;
|
||||
pub mod handler;
|
||||
pub mod bus;
|
||||
pub mod backends;
|
||||
pub mod bus;
|
||||
pub mod handler;
|
||||
pub mod types;
|
||||
|
||||
pub use types::*;
|
||||
pub use handler::*;
|
||||
pub use bus::*;
|
||||
pub use handler::*;
|
||||
pub use types::*;
|
||||
|
||||
@@ -10,10 +10,7 @@ use crate::scanners::NormalizedFinding;
|
||||
/// 2. Fallback: hash of (scanner + rule_id + file_path) — deliberately excludes line number
|
||||
/// to avoid ghost resolve/create when lines shift. Trade-off: if the same rule fires
|
||||
/// twice in the same file, they collapse into one finding.
|
||||
pub fn compute_fingerprint(
|
||||
scanner: &str,
|
||||
finding: &NormalizedFinding,
|
||||
) -> String {
|
||||
pub fn compute_fingerprint(scanner: &str, finding: &NormalizedFinding) -> String {
|
||||
if let Some(ref fp) = finding.scanner_fingerprint {
|
||||
return fp.clone();
|
||||
}
|
||||
@@ -79,7 +76,10 @@ mod tests {
|
||||
let fp1 = compute_fingerprint("scanner", &f1);
|
||||
let fp2 = compute_fingerprint("scanner", &f2);
|
||||
|
||||
assert_eq!(fp1, fp2, "line number should NOT affect fallback fingerprint");
|
||||
assert_eq!(
|
||||
fp1, fp2,
|
||||
"line number should NOT affect fallback fingerprint"
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
|
||||
+19
-12
@@ -10,7 +10,7 @@ use crate::charybdis::ingestion::{
|
||||
ReconciliationSummary,
|
||||
};
|
||||
use crate::database::EntityRepository;
|
||||
use crate::findings::reconciler::{ReconciliationEngine, ReconciledFinding};
|
||||
use crate::findings::reconciler::{ReconciledFinding, ReconciliationEngine};
|
||||
use crate::scanners::{ParserRegistry, Severity};
|
||||
|
||||
pub struct MyIngestionService {
|
||||
@@ -61,7 +61,6 @@ impl MyIngestionService {
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
#[tonic::async_trait]
|
||||
impl IngestionService for MyIngestionService {
|
||||
#[instrument(skip(self, request), fields(component_ref, lifecycle, format))]
|
||||
@@ -85,10 +84,9 @@ impl IngestionService for MyIngestionService {
|
||||
|
||||
let component_id = self.resolve_component_id(&req.component_ref).await?;
|
||||
|
||||
let parser = self
|
||||
.parser_registry
|
||||
.get(&req.format)
|
||||
.ok_or_else(|| Status::invalid_argument(format!("Unsupported format: {}", req.format)))?;
|
||||
let parser = self.parser_registry.get(&req.format).ok_or_else(|| {
|
||||
Status::invalid_argument(format!("Unsupported format: {}", req.format))
|
||||
})?;
|
||||
|
||||
let report = parser
|
||||
.parse(&req.data)
|
||||
@@ -108,7 +106,12 @@ impl IngestionService for MyIngestionService {
|
||||
|
||||
let result = self
|
||||
.reconciler
|
||||
.reconcile(&component_id, &req.lifecycle, &scanner_name, &report.findings)
|
||||
.reconcile(
|
||||
&component_id,
|
||||
&req.lifecycle,
|
||||
&scanner_name,
|
||||
&report.findings,
|
||||
)
|
||||
.await
|
||||
.map_err(|e| {
|
||||
error!(error = %e, "Reconciliation failed");
|
||||
@@ -175,10 +178,9 @@ impl IngestionService for MyIngestionService {
|
||||
|
||||
let component_id = self.resolve_component_id(&req.component_ref).await?;
|
||||
|
||||
let parser = self
|
||||
.parser_registry
|
||||
.get(&req.format)
|
||||
.ok_or_else(|| Status::invalid_argument(format!("Unsupported format: {}", req.format)))?;
|
||||
let parser = self.parser_registry.get(&req.format).ok_or_else(|| {
|
||||
Status::invalid_argument(format!("Unsupported format: {}", req.format))
|
||||
})?;
|
||||
|
||||
let report = parser
|
||||
.parse(&req.data)
|
||||
@@ -192,7 +194,12 @@ impl IngestionService for MyIngestionService {
|
||||
|
||||
let result = self
|
||||
.reconciler
|
||||
.reconcile(&component_id, &req.lifecycle, &scanner_name, &report.findings)
|
||||
.reconcile(
|
||||
&component_id,
|
||||
&req.lifecycle,
|
||||
&scanner_name,
|
||||
&report.findings,
|
||||
)
|
||||
.await
|
||||
.map_err(|e| {
|
||||
error!(error = %e, "Reconciliation failed");
|
||||
|
||||
+28
-18
@@ -5,9 +5,11 @@ use anyhow::Result;
|
||||
use chrono::Utc;
|
||||
use tracing::{info, instrument};
|
||||
|
||||
use crate::charybdis::core::{FindingMetadata, FindingSpec, FindingState, Severity as ProtoSeverity};
|
||||
use crate::charybdis::entities::entity::{Metadata, Spec};
|
||||
use crate::charybdis::core::{
|
||||
FindingMetadata, FindingSpec, FindingState, Severity as ProtoSeverity,
|
||||
};
|
||||
use crate::charybdis::entities::Entity;
|
||||
use crate::charybdis::entities::entity::{Metadata, Spec};
|
||||
use crate::database::EntityRepository;
|
||||
use crate::findings::fingerprint::compute_fingerprint;
|
||||
use crate::scanners::{NormalizedFinding, Severity};
|
||||
@@ -51,7 +53,9 @@ impl ReconciliationEngine {
|
||||
scanner: &str,
|
||||
incoming: &[NormalizedFinding],
|
||||
) -> Result<ReconciliationResult> {
|
||||
let existing = self.load_existing_findings(component_ref, lifecycle).await?;
|
||||
let existing = self
|
||||
.load_existing_findings(component_ref, lifecycle)
|
||||
.await?;
|
||||
|
||||
let mut existing_by_fingerprint: HashMap<String, Entity> = HashMap::new();
|
||||
for entity in existing {
|
||||
@@ -140,8 +144,16 @@ impl ReconciliationEngine {
|
||||
.unwrap_or_default(),
|
||||
proto_severity_to_domain(spec.severity),
|
||||
spec.rule_id.clone(),
|
||||
if spec.file_path.is_empty() { None } else { Some(spec.file_path.clone()) },
|
||||
if spec.line_start == 0 { None } else { Some(spec.line_start) },
|
||||
if spec.file_path.is_empty() {
|
||||
None
|
||||
} else {
|
||||
Some(spec.file_path.clone())
|
||||
},
|
||||
if spec.line_start == 0 {
|
||||
None
|
||||
} else {
|
||||
Some(spec.line_start)
|
||||
},
|
||||
),
|
||||
_ => (String::new(), Severity::Medium, String::new(), None, None),
|
||||
})
|
||||
@@ -216,29 +228,27 @@ impl ReconciliationEngine {
|
||||
for reconciled in &result.unchanged {
|
||||
if let Some(ref entity_id) = reconciled.entity_id {
|
||||
let mut annotations = std::collections::HashMap::new();
|
||||
annotations.insert(
|
||||
"charybdis.io/last-seen".to_string(),
|
||||
now.to_rfc3339(),
|
||||
);
|
||||
annotations.insert(
|
||||
"charybdis.io/scan-id".to_string(),
|
||||
scan_id.to_string(),
|
||||
);
|
||||
self.repository.update_annotations(entity_id, annotations).await?;
|
||||
annotations.insert("charybdis.io/last-seen".to_string(), now.to_rfc3339());
|
||||
annotations.insert("charybdis.io/scan-id".to_string(), scan_id.to_string());
|
||||
self.repository
|
||||
.update_annotations(entity_id, annotations)
|
||||
.await?;
|
||||
}
|
||||
}
|
||||
|
||||
// Resolve findings no longer detected
|
||||
for reconciled in &result.resolved {
|
||||
if let Some(ref entity_id) = reconciled.entity_id {
|
||||
self.mark_finding_state(entity_id, FindingState::Resolved).await?;
|
||||
self.mark_finding_state(entity_id, FindingState::Resolved)
|
||||
.await?;
|
||||
}
|
||||
}
|
||||
|
||||
// Reopen findings detected again
|
||||
for reconciled in &result.reopened {
|
||||
if let Some(ref entity_id) = reconciled.entity_id {
|
||||
self.mark_finding_state(entity_id, FindingState::Reopened).await?;
|
||||
self.mark_finding_state(entity_id, FindingState::Reopened)
|
||||
.await?;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -323,8 +333,8 @@ fn build_finding_entity(
|
||||
package_version: finding.package_version.clone().unwrap_or_default(),
|
||||
fixed_version: finding.fixed_version.clone().unwrap_or_default(),
|
||||
details_url: finding.details_url.clone().unwrap_or_default(),
|
||||
first_seen: Some(timestamp.clone()),
|
||||
last_seen: Some(timestamp.clone()),
|
||||
first_seen: Some(*timestamp),
|
||||
last_seen: Some(*timestamp),
|
||||
resolved_at: None,
|
||||
scan_id: scan_id.to_string(),
|
||||
};
|
||||
|
||||
+31
-19
@@ -16,6 +16,7 @@ pub mod security;
|
||||
pub mod telemetry;
|
||||
|
||||
// Include the generated protobuf code with serde support
|
||||
#[allow(clippy::large_enum_variant)]
|
||||
pub mod charybdis {
|
||||
pub mod core {
|
||||
tonic::include_proto!("charybdis.core");
|
||||
@@ -106,14 +107,10 @@ impl EntityService for MyEntityService {
|
||||
let entity_kind = entity_data.kind.clone();
|
||||
tracing::Span::current().record("entity.kind", &entity_kind);
|
||||
|
||||
let created_entity = self
|
||||
.repository
|
||||
.create(&entity_data)
|
||||
.await
|
||||
.map_err(|e| {
|
||||
error!(error = %e, "Failed to create entity");
|
||||
Status::from(e)
|
||||
})?;
|
||||
let created_entity = self.repository.create(&entity_data).await.map_err(|e| {
|
||||
error!(error = %e, "Failed to create entity");
|
||||
Status::from(e)
|
||||
})?;
|
||||
|
||||
let entity_id = created_entity.id.clone();
|
||||
info!(entity_id = %entity_id, entity_kind = %entity_kind, "Entity created");
|
||||
@@ -214,9 +211,7 @@ impl EntityService for MyEntityService {
|
||||
error!(entity_id = %entity_id, error = %e, "Failed to update entity");
|
||||
Status::from(e)
|
||||
})?
|
||||
.ok_or_else(|| {
|
||||
Status::not_found(format!("Entity not found: {}", entity_id))
|
||||
})?;
|
||||
.ok_or_else(|| Status::not_found(format!("Entity not found: {}", entity_id)))?;
|
||||
|
||||
info!(entity_id = %entity_id, entity_kind = %entity_kind, "Entity updated");
|
||||
|
||||
@@ -265,9 +260,7 @@ impl EntityService for MyEntityService {
|
||||
error!(entity_id = %entity_id, error = %e, "Failed to get entity for deletion");
|
||||
Status::from(e)
|
||||
})?
|
||||
.ok_or_else(|| {
|
||||
Status::not_found(format!("Entity not found: {}", entity_id))
|
||||
})?;
|
||||
.ok_or_else(|| Status::not_found(format!("Entity not found: {}", entity_id)))?;
|
||||
|
||||
let entity_kind = entity.kind.clone();
|
||||
tracing::Span::current().record("entity.kind", &entity_kind);
|
||||
@@ -278,7 +271,10 @@ impl EntityService for MyEntityService {
|
||||
})?;
|
||||
|
||||
if !deleted {
|
||||
return Err(Status::not_found(format!("Entity not found: {}", entity_id)));
|
||||
return Err(Status::not_found(format!(
|
||||
"Entity not found: {}",
|
||||
entity_id
|
||||
)));
|
||||
}
|
||||
|
||||
info!(entity_id = %entity_id, entity_kind = %entity_kind, "Entity deleted");
|
||||
@@ -314,10 +310,26 @@ impl EntityService for MyEntityService {
|
||||
};
|
||||
|
||||
let req = request.into_inner();
|
||||
let kind = if req.kind.is_empty() { None } else { Some(req.kind.as_str()) };
|
||||
let name = if req.name.is_empty() { None } else { Some(req.name.as_str()) };
|
||||
let page_size = if req.page_size == 0 { 100 } else { req.page_size };
|
||||
let page_token = if req.page_token.is_empty() { None } else { Some(req.page_token.as_str()) };
|
||||
let kind = if req.kind.is_empty() {
|
||||
None
|
||||
} else {
|
||||
Some(req.kind.as_str())
|
||||
};
|
||||
let name = if req.name.is_empty() {
|
||||
None
|
||||
} else {
|
||||
Some(req.name.as_str())
|
||||
};
|
||||
let page_size = if req.page_size == 0 {
|
||||
100
|
||||
} else {
|
||||
req.page_size
|
||||
};
|
||||
let page_token = if req.page_token.is_empty() {
|
||||
None
|
||||
} else {
|
||||
Some(req.page_token.as_str())
|
||||
};
|
||||
|
||||
let paginated = self
|
||||
.repository
|
||||
|
||||
+1
-2
@@ -135,8 +135,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
|
||||
// Initialize parser registry and ingestion service
|
||||
let parser_registry = Arc::new(ParserRegistry::with_builtins());
|
||||
let my_ingestion_service =
|
||||
MyIngestionService::new(entity_repository.clone(), parser_registry);
|
||||
let my_ingestion_service = MyIngestionService::new(entity_repository.clone(), parser_registry);
|
||||
|
||||
// Configure and build the reflection service using the embedded descriptor set.
|
||||
// In tonic-reflection 0.14+, use build_v1() instead of build()
|
||||
|
||||
@@ -275,7 +275,7 @@ impl FieldMapper {
|
||||
pub async fn map_all(&self, entity: &Entity) -> Result<HashMap<String, Value>> {
|
||||
let mut result = HashMap::new();
|
||||
|
||||
for (field_name, _) in &self.mappings {
|
||||
for field_name in self.mappings.keys() {
|
||||
match self.map(field_name, entity).await {
|
||||
Ok(value) => {
|
||||
result.insert(field_name.clone(), value);
|
||||
@@ -540,7 +540,8 @@ fn entity_to_json(entity: &Entity) -> Result<Value> {
|
||||
if let Some(profile) = &s.profile {
|
||||
let mut profile_obj = serde_json::Map::new();
|
||||
if !profile.display_name.is_empty() {
|
||||
profile_obj.insert("display_name".to_string(), json!(&profile.display_name));
|
||||
profile_obj
|
||||
.insert("display_name".to_string(), json!(&profile.display_name));
|
||||
}
|
||||
if !profile.email.is_empty() {
|
||||
profile_obj.insert("email".to_string(), json!(&profile.email));
|
||||
@@ -563,7 +564,8 @@ fn entity_to_json(entity: &Entity) -> Result<Value> {
|
||||
if let Some(profile) = &s.profile {
|
||||
let mut profile_obj = serde_json::Map::new();
|
||||
if !profile.display_name.is_empty() {
|
||||
profile_obj.insert("display_name".to_string(), json!(&profile.display_name));
|
||||
profile_obj
|
||||
.insert("display_name".to_string(), json!(&profile.display_name));
|
||||
}
|
||||
if !profile.email.is_empty() {
|
||||
profile_obj.insert("email".to_string(), json!(&profile.email));
|
||||
|
||||
+10
-11
@@ -285,20 +285,19 @@ impl PluginHttpClient {
|
||||
let request = build_request();
|
||||
match request.send().await {
|
||||
Ok(response) => {
|
||||
if response.status() == StatusCode::TOO_MANY_REQUESTS
|
||||
if (response.status() == StatusCode::TOO_MANY_REQUESTS
|
||||
|| response.status() == StatusCode::BAD_GATEWAY
|
||||
|| response.status() == StatusCode::SERVICE_UNAVAILABLE
|
||||
|| response.status() == StatusCode::GATEWAY_TIMEOUT
|
||||
|| response.status() == StatusCode::GATEWAY_TIMEOUT)
|
||||
&& attempt < MAX_RETRIES
|
||||
{
|
||||
if attempt < MAX_RETRIES {
|
||||
debug!("Retryable status {}, will retry", response.status());
|
||||
last_error = Some(anyhow!(
|
||||
"HTTP {} (attempt {})",
|
||||
response.status(),
|
||||
attempt + 1
|
||||
));
|
||||
continue;
|
||||
}
|
||||
debug!("Retryable status {}, will retry", response.status());
|
||||
last_error = Some(anyhow!(
|
||||
"HTTP {} (attempt {})",
|
||||
response.status(),
|
||||
attempt + 1
|
||||
));
|
||||
continue;
|
||||
}
|
||||
return Ok(response);
|
||||
}
|
||||
|
||||
@@ -107,10 +107,10 @@ impl SyncScheduler {
|
||||
info!("Sync scheduler started with tokio_cron_scheduler");
|
||||
self.scheduler = Some(scheduler);
|
||||
|
||||
if let Some(s) = &self.scheduler {
|
||||
if let Err(e) = s.start().await {
|
||||
error!("Failed to start scheduler: {}", e);
|
||||
}
|
||||
if let Some(s) = &self.scheduler
|
||||
&& let Err(e) = s.start().await
|
||||
{
|
||||
error!("Failed to start scheduler: {}", e);
|
||||
}
|
||||
} else {
|
||||
info!("Sync scheduler started with manual cron implementation");
|
||||
|
||||
@@ -149,10 +149,10 @@ fn extract_scanner_fingerprint(result: &SarifResult) -> Option<String> {
|
||||
return Some(v.clone());
|
||||
}
|
||||
}
|
||||
if let Some(ref fps) = result.fingerprints {
|
||||
if let Some(v) = fps.values().next() {
|
||||
return Some(v.clone());
|
||||
}
|
||||
if let Some(ref fps) = result.fingerprints
|
||||
&& let Some(v) = fps.values().next()
|
||||
{
|
||||
return Some(v.clone());
|
||||
}
|
||||
None
|
||||
}
|
||||
|
||||
+2
-22
@@ -1,7 +1,7 @@
|
||||
use serde::{Deserialize, Serialize};
|
||||
use std::collections::HashMap;
|
||||
|
||||
#[derive(Debug, Clone, Deserialize, Serialize)]
|
||||
#[derive(Debug, Clone, Deserialize, Serialize, Default)]
|
||||
pub struct SecurityConfig {
|
||||
#[serde(default)]
|
||||
pub mtls: MtlsConfig,
|
||||
@@ -10,15 +10,6 @@ pub struct SecurityConfig {
|
||||
pub rbac: RbacConfig,
|
||||
}
|
||||
|
||||
impl Default for SecurityConfig {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
mtls: MtlsConfig::default(),
|
||||
rbac: RbacConfig::default(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, Deserialize, Serialize)]
|
||||
pub struct MtlsConfig {
|
||||
#[serde(default)]
|
||||
@@ -57,7 +48,7 @@ impl Default for MtlsConfig {
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, Deserialize, Serialize)]
|
||||
#[derive(Debug, Clone, Deserialize, Serialize, Default)]
|
||||
pub struct RbacConfig {
|
||||
#[serde(default)]
|
||||
pub enabled: bool,
|
||||
@@ -72,17 +63,6 @@ pub struct RbacConfig {
|
||||
pub audit: AuditConfig,
|
||||
}
|
||||
|
||||
impl Default for RbacConfig {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
enabled: false,
|
||||
role_mappings: vec![],
|
||||
permissions: HashMap::new(),
|
||||
audit: AuditConfig::default(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, Deserialize, Serialize)]
|
||||
pub struct RoleMapping {
|
||||
pub role: String,
|
||||
|
||||
@@ -1,3 +1,5 @@
|
||||
use std::fmt;
|
||||
|
||||
use thiserror::Error;
|
||||
use tracing::debug;
|
||||
|
||||
@@ -72,9 +74,10 @@ impl ClientIdentity {
|
||||
certificate_serial: serial,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
/// Display identity for logging
|
||||
pub fn to_string(&self) -> String {
|
||||
impl fmt::Display for ClientIdentity {
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||
let mut parts = vec![format!("CN={}", self.common_name)];
|
||||
|
||||
if let Some(ref ou) = self.organizational_unit {
|
||||
@@ -85,7 +88,7 @@ impl ClientIdentity {
|
||||
parts.push(format!("O={}", o));
|
||||
}
|
||||
|
||||
parts.join(", ")
|
||||
f.write_str(&parts.join(", "))
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -45,7 +45,7 @@ impl AuthInterceptor {
|
||||
tracing::Span::current().record("client.cn", &identity.common_name);
|
||||
|
||||
debug!(
|
||||
identity = %identity.to_string(),
|
||||
identity = %identity,
|
||||
"Extracted client identity from certificate"
|
||||
);
|
||||
|
||||
@@ -62,10 +62,7 @@ impl AuthInterceptor {
|
||||
"No role assigned to identity",
|
||||
duration,
|
||||
);
|
||||
Status::permission_denied(format!(
|
||||
"No role assigned to identity: {}",
|
||||
identity.to_string()
|
||||
))
|
||||
Status::permission_denied(format!("No role assigned to identity: {}", identity))
|
||||
})?;
|
||||
|
||||
tracing::Span::current().record("role", &role);
|
||||
@@ -135,7 +132,7 @@ impl AuthInterceptor {
|
||||
}
|
||||
|
||||
debug!(
|
||||
identity = %identity.to_string(),
|
||||
identity = %identity,
|
||||
role = %role,
|
||||
"Request authorized"
|
||||
);
|
||||
@@ -150,13 +147,12 @@ impl AuthInterceptor {
|
||||
// Try 1: Extract from Tonic's TlsConnectInfo (direct mTLS)
|
||||
// This requires the tls-connect-info feature enabled in Tonic
|
||||
// Note: Must use TlsConnectInfo<TcpConnectInfo>, not TlsConnectInfo<SocketAddr>
|
||||
if let Some(connect_info) = request.extensions().get::<TlsConnectInfo<TcpConnectInfo>>() {
|
||||
if let Some(certs) = connect_info.peer_certs() {
|
||||
if let Some(cert) = certs.first() {
|
||||
debug!("Extracted certificate from TlsConnectInfo (direct mTLS)");
|
||||
return Ok(cert.as_ref().to_vec());
|
||||
}
|
||||
}
|
||||
if let Some(connect_info) = request.extensions().get::<TlsConnectInfo<TcpConnectInfo>>()
|
||||
&& let Some(certs) = connect_info.peer_certs()
|
||||
&& let Some(cert) = certs.first()
|
||||
{
|
||||
debug!("Extracted certificate from TlsConnectInfo (direct mTLS)");
|
||||
return Ok(cert.as_ref().to_vec());
|
||||
}
|
||||
|
||||
// Try 2: Extract from headers (reverse proxy injected certificate)
|
||||
|
||||
+4
-4
@@ -127,10 +127,10 @@ pub fn validate_cert_files(config: &MtlsConfig) -> Result<()> {
|
||||
}
|
||||
|
||||
// Check CRL if specified
|
||||
if let Some(ref crl_file) = config.crl_file {
|
||||
if !std::path::Path::new(crl_file).exists() {
|
||||
anyhow::bail!("CRL file not found: {}", crl_file);
|
||||
}
|
||||
if let Some(ref crl_file) = config.crl_file
|
||||
&& !std::path::Path::new(crl_file).exists()
|
||||
{
|
||||
anyhow::bail!("CRL file not found: {}", crl_file);
|
||||
}
|
||||
|
||||
debug!("All certificate files validated successfully");
|
||||
|
||||
+27
-27
@@ -76,20 +76,20 @@ fn init_tracer(
|
||||
.with_resource(resource);
|
||||
|
||||
// Add OTLP exporter if enabled
|
||||
if config.enable_otlp {
|
||||
if let Some(endpoint) = &config.otlp_endpoint {
|
||||
info!("Configuring OTLP trace exporter: {}", endpoint);
|
||||
if config.enable_otlp
|
||||
&& let Some(endpoint) = &config.otlp_endpoint
|
||||
{
|
||||
info!("Configuring OTLP trace exporter: {}", endpoint);
|
||||
|
||||
// In OpenTelemetry 0.31+, use SpanExporter::builder()
|
||||
let exporter = opentelemetry_otlp::SpanExporter::builder()
|
||||
.with_tonic()
|
||||
.with_endpoint(endpoint.clone())
|
||||
.with_timeout(Duration::from_secs(10))
|
||||
.build()?;
|
||||
// In OpenTelemetry 0.31+, use SpanExporter::builder()
|
||||
let exporter = opentelemetry_otlp::SpanExporter::builder()
|
||||
.with_tonic()
|
||||
.with_endpoint(endpoint.clone())
|
||||
.with_timeout(Duration::from_secs(10))
|
||||
.build()?;
|
||||
|
||||
// In OpenTelemetry 0.31+, runtime parameter is no longer needed
|
||||
builder = builder.with_batch_exporter(exporter);
|
||||
}
|
||||
// In OpenTelemetry 0.31+, runtime parameter is no longer needed
|
||||
builder = builder.with_batch_exporter(exporter);
|
||||
}
|
||||
|
||||
let provider = builder.build();
|
||||
@@ -108,24 +108,24 @@ fn init_metrics(
|
||||
let mut meter_provider_builder = SdkMeterProvider::builder().with_resource(resource);
|
||||
|
||||
// Add OTLP exporter if enabled
|
||||
if config.enable_otlp {
|
||||
if let Some(endpoint) = &config.otlp_endpoint {
|
||||
info!("Configuring OTLP metrics exporter: {}", endpoint);
|
||||
if config.enable_otlp
|
||||
&& let Some(endpoint) = &config.otlp_endpoint
|
||||
{
|
||||
info!("Configuring OTLP metrics exporter: {}", endpoint);
|
||||
|
||||
// In OpenTelemetry 0.31+, use MetricExporter::builder() (singular, not plural)
|
||||
let exporter = opentelemetry_otlp::MetricExporter::builder()
|
||||
.with_tonic()
|
||||
.with_endpoint(endpoint.clone())
|
||||
.with_timeout(Duration::from_secs(10))
|
||||
.build()?;
|
||||
// In OpenTelemetry 0.31+, use MetricExporter::builder() (singular, not plural)
|
||||
let exporter = opentelemetry_otlp::MetricExporter::builder()
|
||||
.with_tonic()
|
||||
.with_endpoint(endpoint.clone())
|
||||
.with_timeout(Duration::from_secs(10))
|
||||
.build()?;
|
||||
|
||||
// PeriodicReader::builder() now takes only exporter, runtime is handled internally
|
||||
let reader = PeriodicReader::builder(exporter)
|
||||
.with_interval(Duration::from_secs(30))
|
||||
.build();
|
||||
// PeriodicReader::builder() now takes only exporter, runtime is handled internally
|
||||
let reader = PeriodicReader::builder(exporter)
|
||||
.with_interval(Duration::from_secs(30))
|
||||
.build();
|
||||
|
||||
meter_provider_builder = meter_provider_builder.with_reader(reader);
|
||||
}
|
||||
meter_provider_builder = meter_provider_builder.with_reader(reader);
|
||||
}
|
||||
|
||||
let meter_provider = meter_provider_builder.build();
|
||||
|
||||
Reference in New Issue
Block a user