-
Notifications
You must be signed in to change notification settings - Fork 593
/
UnixSocketsPlatform.scala
122 lines (111 loc) · 4.24 KB
/
UnixSocketsPlatform.scala
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
/*
* Copyright (c) 2013 Functional Streams for Scala
*
* Permission is hereby granted, free of charge, to any person obtaining a copy of
* this software and associated documentation files (the "Software"), to deal in
* the Software without restriction, including without limitation the rights to
* use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
* the Software, and to permit persons to whom the Software is furnished to do so,
* subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
* FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
* COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
* IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
package fs2
package io.net.unixsocket
import cats.effect.IO
import cats.effect.LiftIO
import cats.effect.kernel.Async
import cats.effect.kernel.Resource
import cats.effect.std.Dispatcher
import cats.syntax.all._
import fs2.concurrent.Channel
import fs2.io.file.Files
import fs2.io.file.Path
import fs2.io.net.Socket
import fs2.io.internal.facade
import scala.scalajs.js
private[unixsocket] trait UnixSocketsCompanionPlatform {
def forIO: UnixSockets[IO] = forLiftIO
implicit def forLiftIO[F[_]: Async: LiftIO]: UnixSockets[F] = {
val _ = LiftIO[F]
forAsyncAndFiles
}
def forAsync[F[_]](implicit F: Async[F]): UnixSockets[F] =
forAsyncAndFiles(Files.forAsync(F), F)
def forAsyncAndFiles[F[_]: Files](implicit F: Async[F]): UnixSockets[F] =
new UnixSockets[F] {
override def client(address: UnixSocketAddress): Resource[F, Socket[F]] =
Resource
.make(
F.delay(
new facade.net.Socket(new facade.net.SocketOptions { allowHalfOpen = true })
)
)(socket =>
F.delay {
if (!socket.destroyed)
socket.destroy()
}
)
.evalTap { socket =>
F.async_[Unit] { cb =>
socket.connect(address.path, () => cb(Right(())))
()
}
}
.flatMap(Socket.forAsync[F])
override def server(
address: UnixSocketAddress,
deleteIfExists: Boolean,
deleteOnClose: Boolean
): fs2.Stream[F, Socket[F]] =
for {
dispatcher <- Stream.resource(Dispatcher.sequential[F])
channel <- Stream.eval(Channel.unbounded[F, facade.net.Socket])
errored <- Stream.eval(F.deferred[js.JavaScriptException])
server <- Stream.bracket(
F.delay {
facade.net.createServer(
new facade.net.ServerOptions {
pauseOnConnect = true
allowHalfOpen = true
},
sock => dispatcher.unsafeRunAndForget(channel.send(sock))
)
}
)(server =>
F.async_[Unit] { cb =>
if (server.listening) {
server.close(e => cb(e.toLeft(()).leftMap(js.JavaScriptException)))
()
} else
cb(Right(()))
}
)
_ <- Stream
.resource(
server.registerListener[F, js.Error]("error", dispatcher) { e =>
errored.complete(js.JavaScriptException(e)).void
}
)
.concurrently(Stream.eval(errored.get.flatMap(F.raiseError[Unit])))
_ <- Stream.bracket(
if (deleteIfExists) Files[F].deleteIfExists(Path(address.path)).void else F.unit
)(_ => if (deleteOnClose) Files[F].deleteIfExists(Path(address.path)).void else F.unit)
_ <- Stream.eval(
F.async_[Unit] { cb =>
server.listen(address.path, () => cb(Right(())))
()
}
)
socket <- channel.stream.flatMap(sock => Stream.resource(Socket.forAsync(sock)))
} yield socket
}
}