From 31dc23b74f2f2eecf0cda5d2e8eaa4b89216963a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jakub=20Koz=C5=82owski?= <kubukoz@gmail.com>
Date: Tue, 16 May 2023 03:21:50 +0200
Subject: [PATCH] Update protocol in Localstack proxy (#125)

---
 .../client/middleware/LocalstackProxy.scala        | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/smithy4s-client-localstack/src/main/scala/kinesis4cats/smithy4s/client/middleware/LocalstackProxy.scala b/smithy4s-client-localstack/src/main/scala/kinesis4cats/smithy4s/client/middleware/LocalstackProxy.scala
index 73e1310b..4052d10d 100644
--- a/smithy4s-client-localstack/src/main/scala/kinesis4cats/smithy4s/client/middleware/LocalstackProxy.scala
+++ b/smithy4s-client-localstack/src/main/scala/kinesis4cats/smithy4s/client/middleware/LocalstackProxy.scala
@@ -26,6 +26,8 @@ import org.typelevel.ci._
 import org.typelevel.log4cats.StructuredLogger
 
 import kinesis4cats.localstack.LocalstackConfig
+import kinesis4cats.localstack.Protocol.Http
+import kinesis4cats.localstack.Protocol.Https
 import kinesis4cats.logging.LogContext
 import kinesis4cats.smithy4s.client.localstack.LocalstackKinesisClient
 
@@ -62,13 +64,19 @@ object LocalstackProxy {
     import encoders.localstackConfigEncoders._
     val newReq = req
       .withUri(
-        req.uri.copy(authority =
-          req.uri.authority.map(x =>
+        req.uri.copy(
+          authority = req.uri.authority.map(x =>
             x.copy(
               host = Uri.RegName(config.kinesisHost),
               port = config.kinesisPort.some
             )
-          )
+          ),
+          scheme = Some {
+            config.kinesisProtocol match {
+              case Http  => Uri.Scheme.http
+              case Https => Uri.Scheme.https
+            }
+          }
         )
       )
       .putHeaders(Header.Raw(ci"host", config.kinesisHost))