Added liveness probe

Cargo.lock

@@ -1684,16 +1684,6 @@ dependencies = [
  "url",
 ]
 
-[[package]]
-name = "leon"
-version = "3.0.2"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "42a865ffec5587961f5afc6d365bccb304f4feaa1928f4fe94c91c9d210d7310"
-dependencies = [
- "miette",
- "thiserror 2.0.12",
-]
-
 [[package]]
 name = "libc"
 version = "0.2.172"
@@ -1782,29 +1772,6 @@ version = "2.7.4"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "78ca9ab1a0babb1e7d5695e3530886289c18cf2f87ec19a575a0abdce112e3a3"
 
-[[package]]
-name = "miette"
-version = "7.5.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "1a955165f87b37fd1862df2a59547ac542c77ef6d17c666f619d1ad22dd89484"
-dependencies = [
- "cfg-if",
- "miette-derive",
- "thiserror 1.0.69",
- "unicode-width 0.1.14",
-]
-
-[[package]]
-name = "miette-derive"
-version = "7.5.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "bf45bf44ab49be92fd1227a3be6fc6f617f1a337c06af54981048574d8783147"
-dependencies = [
- "proc-macro2",
- "quote",
- "syn",
-]
-
 [[package]]
 name = "mime"
 version = "0.3.17"
@@ -2978,7 +2945,6 @@ dependencies = [
  "hyper",
  "hyper-util",
  "ldap3",
- "leon",
  "pin-project-lite",
  "rand 0.8.5",
  "ratatui",

Cargo.toml

@@ -18,7 +18,6 @@ http-body-util = { version = "0.1.3", features = ["full"] }
 hyper = { version = "1.6.0", features = ["full"] }
 hyper-util = { version = "0.1.11", features = ["full"] }
 ldap3 = "0.11.5"
-leon = "3.0.2"
 pin-project-lite = "0.2.16"
 rand = "0.8.5"
 ratatui = { version = "0.29.0", features = ["unstable-backend-writer"] }

manifests/deployment.yaml

@@ -55,20 +55,15 @@ spec:
               value: ldap://lldap.lldap.svc.cluster.local:3890
             - name: LDAP_BASE
               value: ou=people,dc=huizinga,dc=dev
-            - name: LDAP_SEARCH_FILTER
-              value: (uid={username})
             - name: LDAP_BIND_DN
-              valueFrom:
-                secretKeyRef:
-                  name: siranga-lldap-credentials
-                  key: bind_dn
+              value: uid=siranga.siranga,ou=people,dc=huizinga,dc=dev
             - name: LDAP_PASSWORD_FILE
               value: /secrets/credentials/password
             - name: PRIVATE_KEY_FILE
               value: /secrets/key/private.pem
           livenessProbe:
             httpGet:
-              path: /health
+              path: /healthz
               port: metrics
             initialDelaySeconds: 3
             periodSeconds: 3

src/ldap.rs

@@ -1,5 +1,4 @@
 use ldap3::{LdapConnAsync, SearchEntry};
-use leon::{Template, vals};
 use russh::keys::PublicKey;
 use tokio::select;
 use tokio::task::JoinHandle;
@@ -10,7 +9,6 @@ use tracing::{debug, error};
 pub struct Ldap {
     base: String,
     ldap: ldap3::Ldap,
-    search_filter: String,
 }
 
 #[derive(Debug, thiserror::Error)]
@@ -23,10 +21,6 @@ pub enum LdapError {
     MissingEnvironmentVariable(&'static str),
     #[error("Could not read password file: {0}")]
     CouldNotReadPasswordFile(#[from] std::io::Error),
-    #[error("Failed to parse search filter: {0}")]
-    FailedToParseSearchFilter(#[from] leon::ParseError),
-    #[error("Failed to render search filter: {0}")]
-    FailedToRenderSearchFilter(#[from] leon::RenderError),
 }
 
 impl Ldap {
@@ -39,8 +33,6 @@ impl Ldap {
             .map_err(|_| LdapError::MissingEnvironmentVariable("LDAP_BASE"))?;
         let bind_dn = std::env::var("LDAP_BIND_DN")
             .map_err(|_| LdapError::MissingEnvironmentVariable("LDAP_BIND_DN"))?;
-        let search_filter = std::env::var("LDAP_SEARCH_FILTER")
-            .map_err(|_| LdapError::MissingEnvironmentVariable("LDAP_SEARCH_FILTER"))?;
 
         let password = std::env::var("LDAP_PASSWORD_FILE").map_or_else(
             |_| {
@@ -73,39 +65,20 @@ impl Ldap {
 
         ldap.simple_bind(&bind_dn, &password).await?.success()?;
 
-        Ok((
-            Self {
-                base,
-                ldap,
-                search_filter,
-            },
-            handle,
-        ))
+        Ok((Self { base, ldap }, handle))
     }
 
     pub async fn get_ssh_keys(
         &mut self,
         user: impl AsRef<str>,
     ) -> Result<Vec<PublicKey>, LdapError> {
-        let search_filter = Template::parse(&self.search_filter)?;
-
-        let search_filter = search_filter.render(&&vals(|key| {
-            if key == "username" {
-                Some(user.as_ref().to_string().into())
-            } else {
-                None
-            }
-        }))?;
-
-        debug!("search_filter = {search_filter}");
-
         Ok(self
             .ldap
             .search(
                 &self.base,
                 ldap3::Scope::Subtree,
                 // TODO: Make this not hardcoded
-                &search_filter,
+                &format!("(uid={})", user.as_ref()),
                 vec!["sshkeys"],
             )
             .await?

src/main.rs

@@ -4,8 +4,8 @@ use std::net::SocketAddr;
 use std::path::Path;
 use std::time::Duration;
 
+use axum::Router;
 use axum::routing::get;
-use axum::{Json, Router};
 use color_eyre::eyre::Context;
 use dotenvy::dotenv;
 use rand::rngs::OsRng;
@@ -126,7 +126,7 @@ async fn main() -> color_eyre::Result<()> {
     let http_task = service.serve(http_listener, token.clone());
     info!("HTTP is available on {http_addr}");
 
-    let metrics_app = Router::new().route("/health", get(async || Json("healthy")));
+    let metrics_app = Router::new().route("/health", get(async || "UP"));
     let metrics_addr = SocketAddr::from(([0, 0, 0, 0], metrics_port));
     let metrics_listener = TcpListener::bind(metrics_addr).await?;
     let metrics = axum::serve(metrics_listener, metrics_app)
@@ -140,6 +140,10 @@ async fn main() -> color_eyre::Result<()> {
         _ = shutdown_task(token.clone()) => {
             error!("Failed to shutdown gracefully");
         }
+        _ = tokio::time::sleep(std::time::Duration::from_secs(30)) => {
+            debug!("Simulating issue");
+            std::future::pending::<()>().await;
+        }
     };
 
     Ok(())