finish ctx

1 files changed, 16 insertions, 6 deletions

diff --git a/src/tasks/server.rs b/src/tasks/server.rs
index 0896dca..b1cfc15 100644
--- a/src/tasks/server.rs
+++ b/src/tasks/server.rs
@@ -2,8 +2,13 @@ use std::fmt::Display;
 
 use crate::tasks::{IntoTaskErr, Task as BaseTask, TaskOutput};
 use serde::Deserialize;
+use tokio::net::TcpStream;
 
-pub struct Task<T = ()>(pub BaseTask<Name, T>);
+pub struct ServerCtx {
+    stream: TcpStream,
+}
+
+pub struct Task(pub BaseTask<Name, ServerCtx>);
 
 #[derive(Debug, Deserialize, Clone, Copy)]
 pub enum Name {
@@ -16,9 +21,14 @@ impl Display for Name {
     }
 }
 
-pub const TASKS: &[Task] = &[Task::<i32>(BaseTask { name: Name::GetIp, exec: get_ip })];
+pub const TASKS: &[Task] = &[Task(BaseTask { name: Name::GetIp, exec: get_ip })];
 
-fn get_ip(cmds: &[&str], _: i32) -> TaskOutput {}
+fn get_ip(_: &[&str], ctx: &ServerCtx) -> TaskOutput {
+    match ctx.stream.peer_addr() {
+        Ok(addr) => Ok(addr.ip().to_string()),
+        Err(e) => Err(e.to_string()),
+    }
+}
 
 impl TryFrom<&str> for &'static Task {
     type Error = IntoTaskErr;
@@ -33,14 +43,14 @@ impl TryFrom<&str> for &'static Task {
     }
 }
 
-impl From<Task> for BaseTask<Name> {
+impl From<Task> for BaseTask<Name, ServerCtx> {
     fn from(value: Task) -> Self {
         value.0
     }
 }
 
-impl AsRef<BaseTask<Name>> for Task {
-    fn as_ref(&self) -> &BaseTask<Name> {
+impl AsRef<BaseTask<Name, ServerCtx>> for Task {
+    fn as_ref(&self) -> &BaseTask<Name, ServerCtx> {
         &self.0
     }
 }