From 8f381b77f4329f1ec14793cc2c25bff5cc5e21ba Mon Sep 17 00:00:00 2001
From: harshabose <hbvardhan@gmail.com>
Date: Wed, 30 Apr 2025 16:26:36 +0530
Subject: [PATCH] first commit

---
 pkg/middleware/encrypt/interceptor.go   | 15 +++++++++++++++
 pkg/middleware/encrypt/messages/init.go |  7 +------
 2 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/pkg/middleware/encrypt/interceptor.go b/pkg/middleware/encrypt/interceptor.go
index 4e4829f..20f22ba 100644
--- a/pkg/middleware/encrypt/interceptor.go
+++ b/pkg/middleware/encrypt/interceptor.go
@@ -5,6 +5,7 @@ import (
 
 	"github.com/harshabose/socket-comm/pkg/interceptor"
 	"github.com/harshabose/socket-comm/pkg/middleware/encrypt/config"
+	"github.com/harshabose/socket-comm/pkg/middleware/encrypt/encryptionerr"
 	"github.com/harshabose/socket-comm/pkg/middleware/encrypt/keyexchange"
 	"github.com/harshabose/socket-comm/pkg/middleware/encrypt/keyprovider"
 	"github.com/harshabose/socket-comm/pkg/middleware/encrypt/state"
@@ -60,3 +61,17 @@ func (i *Interceptor) UnBindSocketConnection(connection interceptor.Connection)
 func (i *Interceptor) Close() error {
 
 }
+
+func GetState(_i interceptor.Interceptor, connection interceptor.Connection) (*state.State, error) {
+	i, ok := _i.(*Interceptor)
+	if !ok {
+		return nil, encryptionerr.ErrInvalidInterceptor
+	}
+
+	s, err := i.stateManager.GetState(connection)
+	if err != nil {
+		return nil, err
+	}
+
+	return s, nil
+}
diff --git a/pkg/middleware/encrypt/messages/init.go b/pkg/middleware/encrypt/messages/init.go
index 95960a2..6f17e81 100644
--- a/pkg/middleware/encrypt/messages/init.go
+++ b/pkg/middleware/encrypt/messages/init.go
@@ -45,12 +45,7 @@ func (m *Init) WriteProcess(_ interceptor.Interceptor, _ interceptor.Connection)
 }
 
 func (m *Init) ReadProcess(_interceptor interceptor.Interceptor, connection interceptor.Connection) error {
-	i, ok := _interceptor.(*encrypt.Interceptor)
-	if !ok {
-		return encryptionerr.ErrInvalidInterceptor
-	}
-
-	s, err := i.stateManager.GetState(connection)
+	s, err := encrypt.GetState(_interceptor, connection)
 	if err != nil {
 		return err
 	}