fix imports and add udp example

Author: saviorand

README.md

@@ -227,16 +227,50 @@ fn main() -> None:
 
 Pure Mojo-based client is available by default. This client is also used internally for testing the server.
 
-## Switching between pure Mojo and Python implementations
-
-By default, Lightbug uses the pure Mojo implementation for networking. To use Python's `socket` library instead, just import the `PythonServer` instead of the `Server` with the following line:
+### UDP Support
+To get started with UDP, just use the `listen_udp` and `dial_udp` functions, along with `write_to` and `read_from` methods, like below.
 
+On the client:
 ```mojo
-from lightbug_http.python.server import PythonServer
+from lightbug_http.connection import dial_udp
+from lightbug_http.address import UDPAddr
+from utils import StringSlice
+
+alias test_string = "Hello, lightbug!"
+
+fn main() raises:
+    print("Dialing UDP server...")
+    alias host = "127.0.0.1"
+    alias port = 12000
+    var udp = dial_udp(host, port)
+
+    print("Sending " + str(len(test_string)) + " messages to the server...")
+    for i in range(len(test_string)):
+        _ = udp.write_to(str(test_string[i]).as_bytes(), host, port)
+
+        try:
+            response, _, _ = udp.read_from(16)
+            print("Response received:", StringSlice(unsafe_from_utf8=response))
+        except e:
+            if str(e) != str("EOF"):
+                raise e
+
 ```
 
-You can then use all the regular server commands in the same way as with the default server.
-Note: as of September, 2024, `PythonServer` and `PythonClient` throw a compilation error when starting. There's an open [issue](https://github.com/saviorand/lightbug_http/issues/41) to fix this - contributions welcome!
+On the server:
+```mojo
+fn main() raises:
+    var listener = listen_udp("127.0.0.1", 12000)
+
+    while True:
+        response, host, port = listener.read_from(16)
+        var message = StringSlice(unsafe_from_utf8=response)
+        print("Message received:", message)
+
+        # Response with the same message in uppercase
+        _ = listener.write_to(String.upper(message).as_bytes(), host, port)
+
+```
 
 <!-- ROADMAP -->
 ## Roadmap

lightbug_http/client.mojo

@@ -1,10 +1,9 @@
 from collections import Dict
 from utils import StringSlice
 from memory import UnsafePointer
-from lightbug_http.net import default_buffer_size
+from lightbug_http.connection import TCPConnection, default_buffer_size, create_connection
 from lightbug_http.http import HTTPRequest, HTTPResponse, encode
 from lightbug_http.header import Headers, HeaderKey
-from lightbug_http.net import create_connection, TCPConnection
 from lightbug_http.io.bytes import Bytes, ByteReader
 from lightbug_http._logger import logger
 from lightbug_http.pool_manager import PoolManager, PoolKey

lightbug_http/pool_manager.mojo

@@ -4,7 +4,7 @@ from builtin.value import StringableCollectionElement
 from memory import UnsafePointer, bitcast, memcpy
 from collections import Dict, Optional
 from collections.dict import RepresentableKeyElement
-from lightbug_http.net import create_connection, TCPConnection, Connection
+from lightbug_http.connection import create_connection, TCPConnection, Connection
 from lightbug_http._logger import logger
 from lightbug_http._owning_list import OwningList
 from lightbug_http.uri import Scheme

tests/integration/udp/udp_client.mojo

@@ -1,4 +1,5 @@
-from lightbug_http.net import dial_udp, UDPAddr
+from lightbug_http.connection import dial_udp
+from lightbug_http.address import UDPAddr
 from utils import StringSlice
 
 alias test_string = "Hello, lightbug!"

tests/integration/udp/udp_server.mojo

@@ -1,4 +1,5 @@
-from lightbug_http.net import listen_udp, UDPAddr
+from lightbug_http.connection import listen_udp
+from lightbug_http.address import UDPAddr
 from utils import StringSlice
 
 

testutils/utils.mojo

@@ -3,7 +3,8 @@ from lightbug_http.io.bytes import Bytes
 from lightbug_http.error import ErrorHandler
 from lightbug_http.uri import URI
 from lightbug_http.http import HTTPRequest, HTTPResponse
-from lightbug_http.net import Listener, Addr, Connection, TCPAddr
+from lightbug_http.connection import Listener, Connection
+from lightbug_http.address import Addr, TCPAddr
 from lightbug_http.service import HTTPService, OK
 from lightbug_http.server import ServerTrait
 from lightbug_http.client import Client