src/db.rs

@@ -1,7 +1,8 @@
 use std::{error::Error, net::IpAddr};
 
+use json::JsonValue;
 use log::warn;
-use tokio_postgres::{Client, Row};
+use tokio_postgres::{Client, Row, Statement};
 
 const ENDPOINT_TABLE: &str = "endpoints";
 const HOSTS_RELATION_TABLE: &str = "hosts";

src/health.rs

@@ -26,7 +26,7 @@ pub async fn check(db: &mut BoxyDatabase) {
                 continue;
             }
         };
-
+        
         let io = TokioIo::new(stream);
 
         let (mut sender, conn) = hyper::client::conn::http1::handshake(io).await.unwrap();

src/main.rs

@@ -16,11 +16,11 @@ use matchers::api::ApiMatcher;
 use server::Server;
 use services::{
     controller::ControllerService,
-    matcher::Matcher,
+    matcher::{Matcher, MatcherService},
 };
 use tokio::{
     sync::Mutex,
-    time::{self},
+    time::{self, interval},
 };
 use tokio_postgres::NoTls;
 

src/matchers/api.rs

@@ -1,11 +1,13 @@
-use std::{net::IpAddr, sync::Arc};
+use std::{net::IpAddr, pin::Pin, sync::Arc};
 
 use async_trait::async_trait;
 use base64::{Engine, prelude::BASE64_STANDARD};
+use http::request::Parts;
 use http_body_util::BodyExt;
 use hyper::{
-    Request, StatusCode,
+    Method, Request, StatusCode,
-    body::Incoming,
+    body::{self, Incoming},
+    service::Service,
 };
 use json::JsonValue;
 use log::{debug, error, warn};
@@ -13,10 +15,10 @@ use tokio::{net::TcpStream, sync::Mutex};
 
 use crate::{
     config::{Client, Config},
-    db::BoxyDatabase,
+    db::{BoxyDatabase, Endpoint},
     routes::api::{AddHost, RegisterEndpoint, RemoveHost},
-    server::{GeneralResponse, custom_resp},
+    server::{GeneralResponse, TcpIntercept, custom_resp, default_response, json_to_vec},
-    services::matcher::Matcher,
+    services::matcher::{Matcher, MatcherService},
 };
 
 #[derive(Debug, Clone)]

src/routes/api.rs

@@ -1,9 +1,12 @@
 use async_trait::async_trait;
+use base64::{Engine, prelude::BASE64_STANDARD};
 use http::request::Parts;
+use http_body_util::BodyExt;
 use hyper::{Method, StatusCode};
-use log::error;
+use log::{debug, error, warn};
 
 use crate::{
+    config::Client,
     db::Endpoint,
     matchers::api::ApiMatcher,
     server::{custom_resp, json_to_vec},

src/services.rs

@@ -1,3 +1,3 @@
 pub mod controller;
-pub mod matcher;
 pub mod proxy;
+pub mod matcher;

src/services/matcher.rs

@@ -1,11 +1,22 @@
-use std::{pin::Pin, sync::Arc};
+use std::{net::IpAddr, pin::Pin, sync::Arc};
 
 use async_trait::async_trait;
+use base64::{Engine, prelude::BASE64_STANDARD};
 use http::request::Parts;
-use hyper::{Request, StatusCode, body::Incoming, service::Service};
+use http_body_util::BodyExt;
-use tokio::net::TcpStream;
+use hyper::{
+    Method, Request, StatusCode,
+    body::{self, Incoming},
+    service::Service,
+};
+use log::{debug, error, warn};
+use tokio::{net::TcpStream, sync::Mutex};
 
-use crate::server::{GeneralResponse, TcpIntercept, custom_resp, default_response};
+use crate::{
+    config::{Client, Config},
+    db::{BoxyDatabase, Endpoint},
+    server::{GeneralResponse, TcpIntercept, custom_resp, default_response, json_to_vec},
+};
 
 // The routes itself
 #[async_trait]
@@ -36,11 +47,9 @@ pub trait Matcher: Clone + Send + Sync + 'static {
     }
 
     // Do something with TCP stream
-    #[allow(unused_variables)]
     fn stream(&mut self, stream: &TcpStream) {}
 
     // Body parser - made universal for api server cause lazy
-    #[allow(unused_variables)]
     async fn body(&mut self, body: Incoming) -> Option<Result<GeneralResponse, hyper::Error>> {
         None
     }