Implement getLocalAddressGen on JVM

Author: mpilquist

io/jvm-native/src/main/scala/fs2/io/net/SocketAddressHelpers.scala

@@ -0,0 +1,22 @@
+package fs2
+package io
+package net
+
+import com.comcast.ip4s.{GenSocketAddress, SocketAddress, UnixSocketAddress}
+import java.net.{InetSocketAddress, UnixDomainSocketAddress}
+import jnr.unixsocket.{UnixSocketAddress => JnrUnixSocketAddress}
+
+private[net] object SocketAddressHelpers {
+
+  def toGenSocketAddress(address: java.net.SocketAddress): GenSocketAddress = {
+    address match {
+      case addr: InetSocketAddress => SocketAddress.fromInetSocketAddress(addr)
+      case _ =>
+        if (JdkUnixSocketsProvider.supported && address.isInstanceOf[UnixDomainSocketAddress]) {
+          UnixSocketAddress(address.asInstanceOf[UnixDomainSocketAddress].getPath.toString)
+        } else if (JnrUnixSocketsProvider.supported && address.isInstanceOf[JnrUnixSocketAddress]) {
+          UnixSocketAddress(address.asInstanceOf[JnrUnixSocketAddress].path)
+        } else throw new IllegalArgumentException("Unsupported address type: " + address)
+    }
+  }
+}

io/jvm-native/src/main/scala/fs2/io/net/SocketInfoPlatform.scala

@@ -23,10 +23,9 @@ package fs2
 package io
 package net
 
-import com.comcast.ip4s.{GenSocketAddress, SocketAddress}
+import com.comcast.ip4s.GenSocketAddress
 import cats.effect.Async
 
-import java.net.InetSocketAddress
 import java.nio.channels.NetworkChannel
 
 import scala.jdk.CollectionConverters.*
@@ -44,12 +43,7 @@ private[net] trait SocketInfoCompanionPlatform {
     protected def channel: NetworkChannel
 
     override def localAddressGen: F[GenSocketAddress] =
-      asyncInstance.delay(
-        channel.getLocalAddress match {
-          case addr: InetSocketAddress => SocketAddress.fromInetSocketAddress(addr)
-          // TODO handle unix sockets
-        }
-      )
+      asyncInstance.delay(SocketAddressHelpers.toGenSocketAddress(channel.getLocalAddress))
 
     override def supportedOptions: F[Set[SocketOption.Key[?]]] =
       asyncInstance.delay {

io/jvm-native/src/main/scala/fs2/io/net/SocketPlatform.scala

@@ -124,7 +124,7 @@ private[net] trait SocketCompanionPlatform {
         F.delay(Some(endOfInput.voidError))
       }
 
-    def write(bytes: Chunk[Byte]): F[Unit] = {
+    override def write(bytes: Chunk[Byte]): F[Unit] = {
       def go(buff: ByteBuffer): F[Unit] =
         F.async[Int] { cb =>
           ch.write(
@@ -150,23 +150,24 @@ private[net] trait SocketCompanionPlatform {
         )
       )
 
-    def remoteAddress: F[SocketAddress[IpAddress]] =
+    override def remoteAddress: F[SocketAddress[IpAddress]] =
       F.delay(
         SocketAddress.fromInetSocketAddress(
           ch.getRemoteAddress.asInstanceOf[InetSocketAddress]
         )
       )
 
-    override def remoteAddressGen: F[GenSocketAddress] = ???
+    override def remoteAddressGen: F[GenSocketAddress] =
+      F.delay(SocketAddressHelpers.toGenSocketAddress(ch.getRemoteAddress))
 
-    def isOpen: F[Boolean] = F.delay(ch.isOpen)
+    override def isOpen: F[Boolean] = F.delay(ch.isOpen)
 
-    def endOfOutput: F[Unit] =
+    override def endOfOutput: F[Unit] =
       F.delay {
         ch.shutdownOutput(); ()
       }
 
-    def endOfInput: F[Unit] =
+    override def endOfInput: F[Unit] =
       F.delay {
         ch.shutdownInput(); ()
       }