improve uri handling

Author: saviorand

lightbug_http/http.mojo

@@ -232,7 +232,7 @@ fn OK(body: Bytes, content_type: String) -> HTTPResponse:
     )
 
 
-fn encode(req: HTTPRequest) raises -> Bytes:
+fn encode(req: HTTPRequest, uri: URI) raises -> Bytes:
     var res_str = String()
     var protocol = strHttp11
     var current_time = String()
@@ -241,7 +241,7 @@ fn encode(req: HTTPRequest) raises -> Bytes:
 
     _ = builder.write(req.header.method())
     _ = builder.write_string(String(" "))
-    _ = builder.write(req.header.request_uri())
+    _ = builder.write(uri.request_uri())
     _ = builder.write_string(String(" "))
     _ = builder.write(protocol)
     _ = builder.write_string(String("\r\n"))

lightbug_http/sys/client.mojo

@@ -93,7 +93,7 @@ struct MojoClient(Client):
 
         var conn = create_connection(self.fd, host_str, port)
 
-        var req_encoded = encode(req)
+        var req_encoded = encode(req, uri)
         var bytes_sent = conn.write(req_encoded)
         if bytes_sent == -1:
             raise Error("Failed to send message")

lightbug_http/sys/net.mojo

@@ -358,7 +358,7 @@ struct addrinfo_unix(AnAddrInfo):
         Returns:
             UInt32 - The IP address.
         """
-        var host_ptr = to_char_ptr(host)
+        var host_ptr = to_char_ptr(String(host))
         var servinfo = Pointer[Self]().alloc(1)
         servinfo.store(Self())
 
@@ -420,8 +420,6 @@ fn create_connection(sock: c_int, host: String, port: UInt16) raises -> SysConne
         _ = shutdown(sock, SHUT_RDWR)
         raise Error("Failed to connect to server")
 
-    print("Connected to server at " + host + ":" + port.__str__())
-
     var laddr = TCPAddr()
     var raddr = TCPAddr(host, int(port))
     var conn = SysConnection(sock, laddr, raddr)

lightbug_http/uri.mojo

@@ -186,54 +186,39 @@ struct URI:
 
         # Defaults to HTTP/1.1
         var proto_str = String(strHttp11)
-
-        # Parse requestURI
-        var n = raw_uri.rfind(" ")
-        if n < 0:
-            n = len(raw_uri)
-        elif n == 0:
-            raise Error("Request URI cannot be empty")
-        else:
-            var proto = raw_uri[n + 1 :]
-            if proto != strHttp11:
-                proto_str = proto
-
-        var request_uri = raw_uri[:n]
-
-        # Parse host from requestURI
-        n = request_uri.find("://")
-
         var is_https = False
 
-        if n >= 0:
-            var host_and_port = request_uri[n + 3 :]
-
-            if request_uri[:n] == https:
+        # Parse the protocol
+        var proto_end = raw_uri.find("://")
+        var remainder_uri: String
+        if proto_end >= 0:
+            proto_str = raw_uri[:proto_end]
+            if proto_str == https:
                 is_https = True
-
-            n = host_and_port.find("/")
-            if n >= 0:
-                self.__host = host_and_port[:n]._buffer
-                request_uri = request_uri[n + 3 :]
-            else:
-                self.__host = host_and_port._buffer
-                request_uri = strSlash
+            remainder_uri = raw_uri[proto_end + 3:]
+        else:
+            raise Error("Invalid URI: Missing protocol")
+
+        # Parse the host and optional port
+        var path_start = remainder_uri.find("/")
+        var host_and_port: String
+        var request_uri: String
+        if path_start >= 0:
+            host_and_port = remainder_uri[:path_start]
+            request_uri = remainder_uri[path_start:]
         else:
-            n = request_uri.find("/")
-            if n >= 0:
-                self.__host = request_uri[:n]._buffer
-                request_uri = request_uri[n:]
-            else:
-                self.__host = request_uri._buffer
-                request_uri = strSlash
+            host_and_port = remainder_uri
+            request_uri = strSlash  # Assume root if no path is provided
+
+        self.__host = host_and_port[:path_start]._buffer
 
         if is_https:
             _ = self.set_scheme(https)
         else:
             _ = self.set_scheme(http)
-
+        
         # Parse path
-        n = request_uri.find("?")
+        var n = request_uri.find("?")
         if n >= 0:
             self.__path_original = request_uri[:n]._buffer
             self.__query_string = request_uri[n + 1 :]._buffer