Compare commits
34 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
4cd44367a8
|
|||
|
d76f55e267
|
|||
|
4b91ab7d5c
|
|||
|
a8702f4849
|
|||
|
f3ea830379
|
|||
|
4242330bef
|
|||
|
1b6e6423b1
|
|||
|
657af14bb6
|
|||
|
33d42034fc
|
|||
|
c8822aeffe
|
|||
|
7076b13df4
|
|||
|
3f4bc12d36
|
|||
|
fd53b6520f
|
|||
|
0ef7240d61
|
|||
|
de22f44f86
|
|||
|
f9c0e20777
|
|||
|
edc7d40dbc
|
|||
|
ba8d90038d
|
|||
|
dfdca9f5e9
|
|||
|
67ae171532
|
|||
|
a347c05062
|
|||
|
f9446c97a0
|
|||
|
d3b8d1792d
|
|||
|
e05c74f42f
|
|||
|
e1c2481e70
|
|||
|
820351f800
|
|||
|
9d8b500b8d
|
|||
|
b7bead090e
|
|||
|
3db5cc44a9
|
|||
|
1a5ca66e83
|
|||
|
54d9d57927
|
|||
|
a5c535fb1e
|
|||
|
7e6c6af3dd
|
|||
|
faf49fafd5
|
139
README.md
139
README.md
@@ -2,41 +2,114 @@
|
|||||||
|
|
||||||
Interact with node streams in object mode using [`Pipes`]!
|
Interact with node streams in object mode using [`Pipes`]!
|
||||||
|
|
||||||
## Example
|
## Install
|
||||||
```purescript
|
|
||||||
import Prelude
|
|
||||||
|
|
||||||
import Effect.Aff (launchAff_)
|
|
||||||
import Effect (Effect)
|
|
||||||
import Effect.Class (liftEffect)
|
|
||||||
import Effect.Console (log)
|
|
||||||
import Pipes.Prelude ((>->))
|
|
||||||
import Pipes.Prelude as Pipes
|
|
||||||
import Pipes.Core as Pipes.Core
|
|
||||||
import Pipes.Node.FS.Stream as FS
|
|
||||||
import Pipes.Node.Zlib as Zlib
|
|
||||||
import Pipes.CSV.Parse as CSV.Parse
|
|
||||||
|
|
||||||
-- == my-zipped-data.csv ==
|
|
||||||
-- id,foo,is_deleted
|
|
||||||
-- 1,hello,f
|
|
||||||
-- 2,goodbye,t
|
|
||||||
|
|
||||||
-- Logs:
|
|
||||||
-- {id: 1, foo: "hello", is_deleted: false}
|
|
||||||
-- {id: 2, foo: "goodbye", is_deleted: true}
|
|
||||||
main :: Effect Unit
|
|
||||||
main =
|
|
||||||
Pipes.Core.runEffect
|
|
||||||
$ FS.createReadStream "my-zipped-data.csv.gz"
|
|
||||||
>-> Zlib.gunzip
|
|
||||||
>-> CSV.Parse.parse @{id :: Int, foo :: String, is_deleted :: Boolean}
|
|
||||||
>-> Pipes.mapM (liftEffect <<< log)
|
|
||||||
```
|
|
||||||
|
|
||||||
## Installing
|
|
||||||
```bash
|
```bash
|
||||||
spago install node-stream-pipes
|
spago install node-stream-pipes
|
||||||
```
|
```
|
||||||
|
|
||||||
|
## Usage
|
||||||
|
### Node Streams
|
||||||
|
#### Raw Streams
|
||||||
|
Raw `objectMode` Node streams are represented in `Node.Stream.Object`:
|
||||||
|
- `Writable a` accepts chunks of type `a`
|
||||||
|
- `Readable a` emits chunks of type `a`
|
||||||
|
- `Transform a b` transforms chunks from `a` to `b`
|
||||||
|
|
||||||
|
Non-Object streams can also be represented with these types; for example an `fs.WriteStream`
|
||||||
|
can be coerced to `Writable Buffer` without issue.
|
||||||
|
|
||||||
|
Interop between these types and `Node.Stream` are provided in `Node.Stream.Object`:
|
||||||
|
- `unsafeFrom{String,Buffer}{Writable,Readable,Transform}`
|
||||||
|
- `unsafeCoerce{Writable,Readable,Transform}`
|
||||||
|
|
||||||
|
#### Pipes
|
||||||
|
Streams in `Node.Stream.Object` can be converted to `Producer`s, `Consumer`s and `Pipe`s with `Pipes.Node.Stream`:
|
||||||
|
- `fromReadable :: forall a. <Readable a> -> Producer (Maybe a) <Aff> Unit`
|
||||||
|
- `fromWritable :: forall a. <Writable a> -> Consumer (Maybe a) <Aff> Unit`
|
||||||
|
- `fromTransform :: forall a b. <Transform a b> -> Pipe (Maybe a) (Maybe b) <Aff> Unit`
|
||||||
|
|
||||||
|
#### EOS Marker
|
||||||
|
Normally, pipe computations will not be executed once any computation in a pipeline exits.
|
||||||
|
|
||||||
|
To allow for resource cleanup and awareness that the stream is about to close,
|
||||||
|
`Maybe a` is used occasionally in this package as an End-of-Stream marker:
|
||||||
|
|
||||||
|
```purescript
|
||||||
|
-- foo.txt is "hello, world!\n"
|
||||||
|
chunks <- Pipes.Collect.toArray $ Pipes.FS.read "foo.txt" >-> Pipes.Node.Stream.inEOS (Pipes.Buffer.toString UTF8)
|
||||||
|
chunks `shouldEqual` [Just "hello, world!\n", Nothing]
|
||||||
|
```
|
||||||
|
|
||||||
|
Pipes from `a -> b` unaware of EOS can be lifted to `Maybe a -> Maybe b` with `Pipes.Node.Stream.inEOS`.
|
||||||
|
|
||||||
|
Producers of `Maybe a` can drop the EOS marker and emit `a` with `Pipes.Node.Stream.unEOS`.
|
||||||
|
|
||||||
|
Producers of `a` can have an EOS marker added with `Pipes.Node.Stream.withEOS`.
|
||||||
|
|
||||||
|
#### Example
|
||||||
|
`Pipes.PassThrough.js`
|
||||||
|
```javascript
|
||||||
|
import {PassThrough} from 'stream'
|
||||||
|
|
||||||
|
export const makePassThrough = () => new PassThrough()
|
||||||
|
```
|
||||||
|
|
||||||
|
`Pipes.PassThrough.purs`
|
||||||
|
```purescript
|
||||||
|
module Pipes.PassThrough where
|
||||||
|
|
||||||
|
import Prelude
|
||||||
|
|
||||||
|
import Effect (Effect)
|
||||||
|
import Effect.Class (liftEffect)
|
||||||
|
import Effect.Aff (Aff)
|
||||||
|
import Pipes.Core (Pipe)
|
||||||
|
import Node.Stream.Object as ObjectStream
|
||||||
|
import Pipes.Node.Stream as Pipes.Node.Stream
|
||||||
|
|
||||||
|
type PassThroughStream a = ObjectStream.Transform a a
|
||||||
|
|
||||||
|
foreign import makeRaw :: Effect PassThroughStream
|
||||||
|
|
||||||
|
passThrough :: forall a. Pipe a a Aff Unit
|
||||||
|
passThrough = do
|
||||||
|
raw <- liftEffect $ makeRaw
|
||||||
|
Pipes.Node.Stream.fromTransform raw
|
||||||
|
```
|
||||||
|
|
||||||
|
### Utilities
|
||||||
|
This package provides utilities that explicitly use `MonadRec` to ensure stack-safety
|
||||||
|
when dealing with producers of large amounts of data.
|
||||||
|
|
||||||
|
- `Pipes.Collect` provides stack-safe utilities for executing a pipeline and collecting results into a collection, `Buffer`, `Monoid` etc.
|
||||||
|
- `Pipes.Construct` provides stack-safe utilities for creating producers from in-memory collections.
|
||||||
|
- `Pipes.Util` provides some miscellaneous utilities missing from `pipes`.
|
||||||
|
|
||||||
|
### Zlib
|
||||||
|
Pipes for compression & decompression using `zlib` are provided in `Pipes.Node.Zlib`.
|
||||||
|
|
||||||
|
### FS
|
||||||
|
Read files with:
|
||||||
|
- `Pipes.Node.FS.read <path>`
|
||||||
|
- `Pipes.Node.FS.read' <WriteStreamOptions> <path>`
|
||||||
|
|
||||||
|
```purescript
|
||||||
|
Pipes.Collect.toStringWith UTF8 $ Pipes.Node.FS.read "foo.txt" >-> Pipes.Stream.unEOS
|
||||||
|
```
|
||||||
|
|
||||||
|
Write files with:
|
||||||
|
- `Pipes.Node.FS.write' <WriteStreamOptions> <path>`
|
||||||
|
- `Pipes.Node.FS.trunc <path>`
|
||||||
|
- `Pipes.Node.FS.create <path>`
|
||||||
|
- `Pipes.Node.FS.append <path>`
|
||||||
|
|
||||||
|
```purescript
|
||||||
|
Pipes.Stream.withEOS (
|
||||||
|
Pipes.Construct.eachArray ["id,name", "1,henry", "2,suzie"]
|
||||||
|
>-> Pipes.Util.intersperse "\n"
|
||||||
|
>-> Pipes.Buffer.fromString UTF8
|
||||||
|
)
|
||||||
|
>-> Pipes.Node.FS.create "foo.csv"
|
||||||
|
```
|
||||||
|
|
||||||
[`Pipes`]: https://pursuit.purescript.org/packages/purescript-pipes/8.0.0
|
[`Pipes`]: https://pursuit.purescript.org/packages/purescript-pipes/8.0.0
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "purescript-node-stream-pipes",
|
"name": "purescript-node-stream-pipes",
|
||||||
"version": "v1.2.2",
|
"version": "v1.6.1",
|
||||||
"type": "module",
|
"type": "module",
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"csv-parse": "^5.5.5",
|
"csv-parse": "^5.5.5",
|
||||||
|
|||||||
11
spago.lock
11
spago.lock
@@ -9,18 +9,17 @@ workspace:
|
|||||||
- either: ">=6.1.0 <7.0.0"
|
- either: ">=6.1.0 <7.0.0"
|
||||||
- exceptions: ">=6.0.0 <7.0.0"
|
- exceptions: ">=6.0.0 <7.0.0"
|
||||||
- foldable-traversable: ">=6.0.0 <7.0.0"
|
- foldable-traversable: ">=6.0.0 <7.0.0"
|
||||||
- foreign-object
|
- foreign-object: ">=4.1.0 <5.0.0"
|
||||||
- lists
|
- lists: ">=7.0.0 <8.0.0"
|
||||||
- maybe: ">=6.0.0 <7.0.0"
|
- maybe: ">=6.0.0 <7.0.0"
|
||||||
- mmorph: ">=7.0.0 <8.0.0"
|
- mmorph: ">=7.0.0 <8.0.0"
|
||||||
- newtype: ">=5.0.0 <6.0.0"
|
|
||||||
- node-buffer: ">=9.0.0 <10.0.0"
|
- node-buffer: ">=9.0.0 <10.0.0"
|
||||||
- node-event-emitter: ">=3.0.0 <4.0.0"
|
- node-event-emitter: ">=3.0.0 <4.0.0"
|
||||||
- node-fs: ">=9.1.0 <10.0.0"
|
- node-fs: ">=9.1.0 <10.0.0"
|
||||||
- node-path: ">=5.0.0 <6.0.0"
|
- node-path: ">=5.0.0 <6.0.0"
|
||||||
- node-streams: ">=9.0.0 <10.0.0"
|
- node-streams: ">=9.0.0 <10.0.0"
|
||||||
- node-zlib: ">=0.4.0 <0.5.0"
|
- node-zlib: ">=0.4.0 <0.5.0"
|
||||||
- ordered-collections
|
- ordered-collections: ">=3.2.0 <4.0.0"
|
||||||
- parallel: ">=6.0.0 <7.0.0"
|
- parallel: ">=6.0.0 <7.0.0"
|
||||||
- pipes: ">=8.0.0 <9.0.0"
|
- pipes: ">=8.0.0 <9.0.0"
|
||||||
- prelude: ">=6.0.1 <7.0.0"
|
- prelude: ">=6.0.1 <7.0.0"
|
||||||
@@ -28,8 +27,8 @@ workspace:
|
|||||||
- strings: ">=6.0.1 <7.0.0"
|
- strings: ">=6.0.1 <7.0.0"
|
||||||
- tailrec: ">=6.1.0 <7.0.0"
|
- tailrec: ">=6.1.0 <7.0.0"
|
||||||
- transformers: ">=6.0.0 <7.0.0"
|
- transformers: ">=6.0.0 <7.0.0"
|
||||||
- tuples
|
- tuples: ">=7.0.0 <8.0.0"
|
||||||
- unordered-collections
|
- unordered-collections: ">=3.1.0 <4.0.0"
|
||||||
- unsafe-coerce: ">=6.0.0 <7.0.0"
|
- unsafe-coerce: ">=6.0.0 <7.0.0"
|
||||||
test_dependencies:
|
test_dependencies:
|
||||||
- console
|
- console
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
package:
|
package:
|
||||||
name: node-stream-pipes
|
name: node-stream-pipes
|
||||||
publish:
|
publish:
|
||||||
version: '1.2.2'
|
version: '1.6.1'
|
||||||
license: 'GPL-3.0-or-later'
|
license: 'GPL-3.0-or-later'
|
||||||
location:
|
location:
|
||||||
githubOwner: 'cakekindel'
|
githubOwner: 'cakekindel'
|
||||||
@@ -20,7 +20,6 @@ package:
|
|||||||
- lists: ">=7.0.0 <8.0.0"
|
- lists: ">=7.0.0 <8.0.0"
|
||||||
- maybe: ">=6.0.0 <7.0.0"
|
- maybe: ">=6.0.0 <7.0.0"
|
||||||
- mmorph: ">=7.0.0 <8.0.0"
|
- mmorph: ">=7.0.0 <8.0.0"
|
||||||
- newtype: ">=5.0.0 <6.0.0"
|
|
||||||
- node-buffer: ">=9.0.0 <10.0.0"
|
- node-buffer: ">=9.0.0 <10.0.0"
|
||||||
- node-event-emitter: ">=3.0.0 <4.0.0"
|
- node-event-emitter: ">=3.0.0 <4.0.0"
|
||||||
- node-fs: ">=9.1.0 <10.0.0"
|
- node-fs: ">=9.1.0 <10.0.0"
|
||||||
@@ -40,6 +39,8 @@ package:
|
|||||||
- unsafe-coerce: ">=6.0.0 <7.0.0"
|
- unsafe-coerce: ">=6.0.0 <7.0.0"
|
||||||
test:
|
test:
|
||||||
main: Test.Main
|
main: Test.Main
|
||||||
|
build:
|
||||||
|
strict: true
|
||||||
dependencies:
|
dependencies:
|
||||||
- console
|
- console
|
||||||
- gen
|
- gen
|
||||||
|
|||||||
@@ -3,31 +3,36 @@ import Stream from "stream";
|
|||||||
/** @type {(s: Stream.Readable | Stream.Transform) => () => boolean} */
|
/** @type {(s: Stream.Readable | Stream.Transform) => () => boolean} */
|
||||||
export const isReadableImpl = (s) => () => s.readable;
|
export const isReadableImpl = (s) => () => s.readable;
|
||||||
|
|
||||||
|
/** @type {(s: Stream.Readable | Stream.Transform) => () => number} */
|
||||||
|
export const readableLengthImpl = (s) => () => s.readableLength;
|
||||||
|
|
||||||
/** @type {(s: Stream.Writable | Stream.Readable) => () => boolean} */
|
/** @type {(s: Stream.Writable | Stream.Readable) => () => boolean} */
|
||||||
export const isClosedImpl = (s) => () => s.closed;
|
export const isClosedImpl = (s) => () => s.closed;
|
||||||
|
|
||||||
/** @type {(s: Stream.Writable | Stream.Transform) => () => boolean} */
|
/** @type {(s: Stream.Writable | Stream.Transform) => () => boolean} */
|
||||||
export const isWritableImpl = (s) => () => s.writable;
|
export const isWritableImpl = (s) => () => s.writable;
|
||||||
|
|
||||||
|
/** @type {(s: Stream.Writable | Stream.Transform) => () => boolean} */
|
||||||
|
export const needsDrainImpl = (s) => () => s.writableNeedDrain;
|
||||||
|
|
||||||
/** @type {(s: Stream.Readable | Stream.Transform) => () => boolean} */
|
/** @type {(s: Stream.Readable | Stream.Transform) => () => boolean} */
|
||||||
export const isReadableEndedImpl = (s) => () => s.readableEnded;
|
export const isReadableEndedImpl = (s) => () => s.readableEnded;
|
||||||
|
|
||||||
/** @type {(s: Stream.Writable | Stream.Transform) => () => boolean} */
|
/** @type {(s: Stream.Writable | Stream.Transform) => () => boolean} */
|
||||||
export const isWritableEndedImpl = (s) => () => s.writableEnded;
|
export const isWritableEndedImpl = (s) => () => s.writableEnded;
|
||||||
|
|
||||||
|
/** @type {(s: Stream.Writable | Stream.Transform) => () => boolean} */
|
||||||
|
export const isWritableFinishedImpl = (s) => () => s.writableFinished;
|
||||||
|
|
||||||
/** @type {(s: Stream.Writable | Stream.Transform) => () => void} */
|
/** @type {(s: Stream.Writable | Stream.Transform) => () => void} */
|
||||||
export const endImpl = (s) => () => s.end();
|
export const endImpl = (s) => () => s.end();
|
||||||
|
|
||||||
/** @type {<WriteResult>(o: {ok: WriteResult, wouldBlock: WriteResult, closed: WriteResult}) => (s: Stream.Writable | Stream.Transform) => (a: unknown) => () => WriteResult} */
|
/** @type {<WriteResult>(o: {ok: WriteResult, wouldBlock: WriteResult}) => (s: Stream.Writable | Stream.Transform) => (a: unknown) => () => WriteResult} */
|
||||||
export const writeImpl =
|
export const writeImpl =
|
||||||
({ ok, wouldBlock, closed }) =>
|
({ ok, wouldBlock }) =>
|
||||||
(s) =>
|
(s) =>
|
||||||
(a) =>
|
(a) =>
|
||||||
() => {
|
() => {
|
||||||
if (s.closed || s.writableEnded) {
|
|
||||||
return closed;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (s.write(a)) {
|
if (s.write(a)) {
|
||||||
return ok;
|
return ok;
|
||||||
} else {
|
} else {
|
||||||
@@ -35,15 +40,11 @@ export const writeImpl =
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
/** @type {<ReadResult>(o: {just: (_a: unknown) => ReadResult, wouldBlock: ReadResult, closed: ReadResult}) => (s: Stream.Readable | Stream.Transform) => () => ReadResult} */
|
/** @type {<ReadResult>(o: {just: (_a: unknown) => ReadResult, wouldBlock: ReadResult}) => (s: Stream.Readable | Stream.Transform) => () => ReadResult} */
|
||||||
export const readImpl =
|
export const readImpl =
|
||||||
({ just, closed, wouldBlock }) =>
|
({ just, wouldBlock }) =>
|
||||||
(s) =>
|
(s) =>
|
||||||
() => {
|
() => {
|
||||||
if (s.closed || s.readableEnded) {
|
|
||||||
return closed;
|
|
||||||
}
|
|
||||||
|
|
||||||
const a = s.read();
|
const a = s.read();
|
||||||
if (a === null) {
|
if (a === null) {
|
||||||
return wouldBlock;
|
return wouldBlock;
|
||||||
|
|||||||
@@ -14,8 +14,9 @@ import Data.Maybe (Maybe(..))
|
|||||||
import Data.Show.Generic (genericShow)
|
import Data.Show.Generic (genericShow)
|
||||||
import Effect (Effect)
|
import Effect (Effect)
|
||||||
import Effect.Aff (Aff, effectCanceler, makeAff)
|
import Effect.Aff (Aff, effectCanceler, makeAff)
|
||||||
|
import Effect.Aff as Aff
|
||||||
import Effect.Class (liftEffect)
|
import Effect.Class (liftEffect)
|
||||||
import Effect.Exception (Error)
|
import Effect.Exception (Error, error)
|
||||||
import Effect.Uncurried (mkEffectFn1)
|
import Effect.Uncurried (mkEffectFn1)
|
||||||
import Node.Buffer (Buffer)
|
import Node.Buffer (Buffer)
|
||||||
import Node.EventEmitter (EventHandle(..))
|
import Node.EventEmitter (EventHandle(..))
|
||||||
@@ -26,7 +27,6 @@ import Unsafe.Coerce (unsafeCoerce)
|
|||||||
|
|
||||||
data ReadResult a
|
data ReadResult a
|
||||||
= ReadWouldBlock
|
= ReadWouldBlock
|
||||||
| ReadClosed
|
|
||||||
| ReadJust a
|
| ReadJust a
|
||||||
|
|
||||||
derive instance Generic (ReadResult a) _
|
derive instance Generic (ReadResult a) _
|
||||||
@@ -35,9 +35,12 @@ derive instance Eq a => Eq (ReadResult a)
|
|||||||
instance Show (ReadResult a) where
|
instance Show (ReadResult a) where
|
||||||
show = genericShow <<< map (const "..")
|
show = genericShow <<< map (const "..")
|
||||||
|
|
||||||
|
maybeReadResult :: forall a. ReadResult a -> Maybe a
|
||||||
|
maybeReadResult (ReadWouldBlock) = Nothing
|
||||||
|
maybeReadResult (ReadJust a) = Just a
|
||||||
|
|
||||||
data WriteResult
|
data WriteResult
|
||||||
= WriteWouldBlock
|
= WriteWouldBlock
|
||||||
| WriteClosed
|
|
||||||
| WriteOk
|
| WriteOk
|
||||||
|
|
||||||
derive instance Generic WriteResult _
|
derive instance Generic WriteResult _
|
||||||
@@ -45,8 +48,8 @@ derive instance Eq WriteResult
|
|||||||
instance Show WriteResult where
|
instance Show WriteResult where
|
||||||
show = genericShow
|
show = genericShow
|
||||||
|
|
||||||
type ReadResultFFI a = { closed :: ReadResult a, wouldBlock :: ReadResult a, just :: a -> ReadResult a }
|
type ReadResultFFI a = { wouldBlock :: ReadResult a, just :: a -> ReadResult a }
|
||||||
type WriteResultFFI = { closed :: WriteResult, wouldBlock :: WriteResult, ok :: WriteResult }
|
type WriteResultFFI = { wouldBlock :: WriteResult, ok :: WriteResult }
|
||||||
|
|
||||||
foreign import data Writable :: Type -> Type
|
foreign import data Writable :: Type -> Type
|
||||||
foreign import data Readable :: Type -> Type
|
foreign import data Readable :: Type -> Type
|
||||||
@@ -59,13 +62,16 @@ foreign import isReadableImpl :: forall s. s -> Effect Boolean
|
|||||||
foreign import isWritableImpl :: forall s. s -> Effect Boolean
|
foreign import isWritableImpl :: forall s. s -> Effect Boolean
|
||||||
foreign import isReadableEndedImpl :: forall s. s -> Effect Boolean
|
foreign import isReadableEndedImpl :: forall s. s -> Effect Boolean
|
||||||
foreign import isWritableEndedImpl :: forall s. s -> Effect Boolean
|
foreign import isWritableEndedImpl :: forall s. s -> Effect Boolean
|
||||||
|
foreign import isWritableFinishedImpl :: forall s. s -> Effect Boolean
|
||||||
foreign import isClosedImpl :: forall s. s -> Effect Boolean
|
foreign import isClosedImpl :: forall s. s -> Effect Boolean
|
||||||
|
foreign import needsDrainImpl :: forall s. s -> Effect Boolean
|
||||||
|
foreign import readableLengthImpl :: forall s. s -> Effect Int
|
||||||
|
|
||||||
readResultFFI :: forall a. ReadResultFFI a
|
readResultFFI :: forall a. ReadResultFFI a
|
||||||
readResultFFI = { closed: ReadClosed, wouldBlock: ReadWouldBlock, just: ReadJust }
|
readResultFFI = { wouldBlock: ReadWouldBlock, just: ReadJust }
|
||||||
|
|
||||||
writeResultFFI :: WriteResultFFI
|
writeResultFFI :: WriteResultFFI
|
||||||
writeResultFFI = { closed: WriteClosed, wouldBlock: WriteWouldBlock, ok: WriteOk }
|
writeResultFFI = { wouldBlock: WriteWouldBlock, ok: WriteOk }
|
||||||
|
|
||||||
class Stream :: Type -> Constraint
|
class Stream :: Type -> Constraint
|
||||||
class Stream s where
|
class Stream s where
|
||||||
@@ -81,25 +87,31 @@ else instance Stream s => Stream s where
|
|||||||
isClosed s = isClosed s
|
isClosed s = isClosed s
|
||||||
|
|
||||||
class Stream s <= Read s a | s -> a where
|
class Stream s <= Read s a | s -> a where
|
||||||
|
readableLength :: s -> Effect Int
|
||||||
isReadable :: s -> Effect Boolean
|
isReadable :: s -> Effect Boolean
|
||||||
isReadableEnded :: s -> Effect Boolean
|
isReadableEnded :: s -> Effect Boolean
|
||||||
read :: s -> Effect (ReadResult a)
|
read :: s -> Effect (ReadResult a)
|
||||||
|
|
||||||
class Stream s <= Write s a | s -> a where
|
class Stream s <= Write s a | s -> a where
|
||||||
isWritable :: s -> Effect Boolean
|
isWritable :: s -> Effect Boolean
|
||||||
|
needsDrain :: s -> Effect Boolean
|
||||||
isWritableEnded :: s -> Effect Boolean
|
isWritableEnded :: s -> Effect Boolean
|
||||||
|
isWritableFinished :: s -> Effect Boolean
|
||||||
write :: s -> a -> Effect WriteResult
|
write :: s -> a -> Effect WriteResult
|
||||||
end :: s -> Effect Unit
|
end :: s -> Effect Unit
|
||||||
|
|
||||||
instance Read (Readable a) a where
|
instance Read (Readable a) a where
|
||||||
|
readableLength = readableLengthImpl
|
||||||
isReadable = isReadableImpl
|
isReadable = isReadableImpl
|
||||||
isReadableEnded = isReadableEndedImpl
|
isReadableEnded = isReadableEndedImpl
|
||||||
read = readImpl readResultFFI
|
read = readImpl readResultFFI
|
||||||
else instance Read (Transform a b) b where
|
else instance Read (Transform a b) b where
|
||||||
|
readableLength = readableLengthImpl
|
||||||
isReadable = isReadableImpl
|
isReadable = isReadableImpl
|
||||||
isReadableEnded = isReadableEndedImpl
|
isReadableEnded = isReadableEndedImpl
|
||||||
read = readImpl readResultFFI
|
read = readImpl readResultFFI
|
||||||
else instance (Read s a) => Read s a where
|
else instance (Read s a) => Read s a where
|
||||||
|
readableLength = readableLengthImpl
|
||||||
isReadable = isReadableImpl
|
isReadable = isReadableImpl
|
||||||
isReadableEnded = isReadableEndedImpl
|
isReadableEnded = isReadableEndedImpl
|
||||||
read s = read s
|
read s = read s
|
||||||
@@ -107,18 +119,24 @@ else instance (Read s a) => Read s a where
|
|||||||
instance Write (Writable a) a where
|
instance Write (Writable a) a where
|
||||||
isWritable = isWritableImpl
|
isWritable = isWritableImpl
|
||||||
isWritableEnded = isWritableEndedImpl
|
isWritableEnded = isWritableEndedImpl
|
||||||
|
isWritableFinished = isWritableFinishedImpl
|
||||||
write s = writeImpl writeResultFFI s
|
write s = writeImpl writeResultFFI s
|
||||||
end = endImpl
|
end = endImpl
|
||||||
|
needsDrain = needsDrainImpl
|
||||||
else instance Write (Transform a b) a where
|
else instance Write (Transform a b) a where
|
||||||
isWritable = isWritableImpl
|
isWritable = isWritableImpl
|
||||||
isWritableEnded = isWritableEndedImpl
|
isWritableEnded = isWritableEndedImpl
|
||||||
|
isWritableFinished = isWritableFinishedImpl
|
||||||
write s = writeImpl writeResultFFI s
|
write s = writeImpl writeResultFFI s
|
||||||
end = endImpl
|
end = endImpl
|
||||||
|
needsDrain = needsDrainImpl
|
||||||
else instance (Write s a) => Write s a where
|
else instance (Write s a) => Write s a where
|
||||||
isWritable = isWritableImpl
|
isWritable = isWritableImpl
|
||||||
isWritableEnded = isWritableEndedImpl
|
isWritableEnded = isWritableEndedImpl
|
||||||
|
isWritableFinished = isWritableFinishedImpl
|
||||||
write s a = write s a
|
write s a = write s a
|
||||||
end s = end s
|
end s = end s
|
||||||
|
needsDrain = needsDrainImpl
|
||||||
|
|
||||||
withErrorST :: forall s. Stream s => s -> Effect { cancel :: Effect Unit, error :: STRef Global (Maybe Error) }
|
withErrorST :: forall s. Stream s => s -> Effect { cancel :: Effect Unit, error :: STRef Global (Maybe Error) }
|
||||||
withErrorST s = do
|
withErrorST s = do
|
||||||
@@ -126,39 +144,74 @@ withErrorST s = do
|
|||||||
cancel <- flip (Event.once errorH) s \e -> void $ liftST $ STRef.write (Just e) error
|
cancel <- flip (Event.once errorH) s \e -> void $ liftST $ STRef.write (Just e) error
|
||||||
pure { error, cancel }
|
pure { error, cancel }
|
||||||
|
|
||||||
fromBufferReadable :: forall r. Stream.Readable r -> Readable Buffer
|
unsafeCoerceWritable :: forall r a. Stream.Writable r -> Writable a
|
||||||
fromBufferReadable = unsafeCoerce
|
unsafeCoerceWritable = unsafeCoerce
|
||||||
|
|
||||||
fromBufferTransform :: Stream.Duplex -> Transform Buffer Buffer
|
unsafeCoerceReadable :: forall r a. Stream.Readable r -> Readable a
|
||||||
fromBufferTransform = unsafeCoerce
|
unsafeCoerceReadable = unsafeCoerce
|
||||||
|
|
||||||
fromBufferWritable :: forall r. Stream.Writable r -> Writable Buffer
|
unsafeCoerceTransform :: forall a b. Stream.Duplex -> Transform a b
|
||||||
fromBufferWritable = unsafeCoerce
|
unsafeCoerceTransform = unsafeCoerce
|
||||||
|
|
||||||
fromStringReadable :: forall r. Stream.Readable r -> Readable String
|
unsafeFromBufferReadable :: forall r. Stream.Readable r -> Readable Buffer
|
||||||
fromStringReadable = unsafeCoerce
|
unsafeFromBufferReadable = unsafeCoerce
|
||||||
|
|
||||||
fromStringTransform :: Stream.Duplex -> Transform String String
|
unsafeFromBufferTransform :: forall a. Stream.Duplex -> Transform Buffer a
|
||||||
fromStringTransform = unsafeCoerce
|
unsafeFromBufferTransform = unsafeCoerce
|
||||||
|
|
||||||
fromStringWritable :: forall r. Stream.Writable r -> Writable String
|
unsafeFromBufferWritable :: forall r. Stream.Writable r -> Writable Buffer
|
||||||
fromStringWritable = unsafeCoerce
|
unsafeFromBufferWritable = unsafeCoerce
|
||||||
|
|
||||||
|
unsafeFromStringReadable :: forall r. Stream.Readable r -> Readable String
|
||||||
|
unsafeFromStringReadable = unsafeCoerce
|
||||||
|
|
||||||
|
unsafeFromStringTransform :: forall a. Stream.Duplex -> Transform String a
|
||||||
|
unsafeFromStringTransform = unsafeCoerce
|
||||||
|
|
||||||
|
unsafeFromStringWritable :: forall r. Stream.Writable r -> Writable String
|
||||||
|
unsafeFromStringWritable = unsafeCoerce
|
||||||
|
|
||||||
awaitReadableOrClosed :: forall s a. Read s a => s -> Aff Unit
|
awaitReadableOrClosed :: forall s a. Read s a => s -> Aff Unit
|
||||||
awaitReadableOrClosed s = do
|
awaitReadableOrClosed s = Aff.supervise do
|
||||||
|
fiber <-
|
||||||
|
Aff.forkAff
|
||||||
|
$ parOneOf
|
||||||
|
[ onceAff0 readableH s $> Right unit
|
||||||
|
, onceAff0 closeH s $> Right unit
|
||||||
|
, Left <$> onceAff1 errorH s
|
||||||
|
]
|
||||||
closed <- liftEffect $ isClosed s
|
closed <- liftEffect $ isClosed s
|
||||||
ended <- liftEffect $ isReadableEnded s
|
readEnded <- liftEffect $ isReadableEnded s
|
||||||
readable <- liftEffect $ isReadable s
|
readable <- liftEffect $ isReadable s
|
||||||
when (not ended && not closed && not readable)
|
length <- liftEffect $ readableLength s
|
||||||
$ liftEither =<< parOneOf [ onceAff0 readableH s $> Right unit, onceAff0 closeH s $> Right unit, Left <$> onceAff1 errorH s ]
|
if (not closed && not readEnded && readable && length == 0) then
|
||||||
|
liftEither =<< Aff.joinFiber fiber
|
||||||
|
else
|
||||||
|
Aff.killFiber (error "") fiber
|
||||||
|
|
||||||
|
awaitFinished :: forall s a. Write s a => s -> Aff Unit
|
||||||
|
awaitFinished s = Aff.supervise do
|
||||||
|
fiber <- Aff.forkAff $ onceAff0 finishH s
|
||||||
|
finished <- liftEffect $ isWritableFinished s
|
||||||
|
if not finished then Aff.joinFiber fiber else Aff.killFiber (error "") fiber
|
||||||
|
|
||||||
awaitWritableOrClosed :: forall s a. Write s a => s -> Aff Unit
|
awaitWritableOrClosed :: forall s a. Write s a => s -> Aff Unit
|
||||||
awaitWritableOrClosed s = do
|
awaitWritableOrClosed s = Aff.supervise do
|
||||||
|
fiber <-
|
||||||
|
Aff.forkAff
|
||||||
|
$ parOneOf
|
||||||
|
[ onceAff0 drainH s $> Right unit
|
||||||
|
, onceAff0 closeH s $> Right unit
|
||||||
|
, Left <$> onceAff1 errorH s
|
||||||
|
]
|
||||||
closed <- liftEffect $ isClosed s
|
closed <- liftEffect $ isClosed s
|
||||||
ended <- liftEffect $ isWritableEnded s
|
writeEnded <- liftEffect $ isWritableEnded s
|
||||||
writable <- liftEffect $ isWritable s
|
writable <- liftEffect $ isWritable s
|
||||||
when (not ended && not closed && not writable)
|
needsDrain <- liftEffect $ needsDrain s
|
||||||
$ liftEither =<< parOneOf [ onceAff0 drainH s $> Right unit, onceAff0 closeH s $> Right unit, Left <$> onceAff1 errorH s ]
|
if not closed && not writeEnded && writable && needsDrain then
|
||||||
|
liftEither =<< Aff.joinFiber fiber
|
||||||
|
else
|
||||||
|
Aff.killFiber (error "") fiber
|
||||||
|
|
||||||
onceAff0 :: forall e. EventHandle0 e -> e -> Aff Unit
|
onceAff0 :: forall e. EventHandle0 e -> e -> Aff Unit
|
||||||
onceAff0 h emitter = makeAff \res -> do
|
onceAff0 h emitter = makeAff \res -> do
|
||||||
@@ -184,3 +237,6 @@ errorH = EventHandle "error" mkEffectFn1
|
|||||||
|
|
||||||
endH :: forall s a. Write s a => EventHandle0 s
|
endH :: forall s a. Write s a => EventHandle0 s
|
||||||
endH = EventHandle "end" identity
|
endH = EventHandle "end" identity
|
||||||
|
|
||||||
|
finishH :: forall s a. Write s a => EventHandle0 s
|
||||||
|
finishH = EventHandle "finish" identity
|
||||||
|
|||||||
@@ -1 +0,0 @@
|
|||||||
module Pipes.CSV.Parse where
|
|
||||||
@@ -2,12 +2,9 @@ module Pipes.Collect where
|
|||||||
|
|
||||||
import Prelude
|
import Prelude
|
||||||
|
|
||||||
import Control.Monad.Maybe.Trans (MaybeT(..), runMaybeT)
|
|
||||||
import Control.Monad.Rec.Class (class MonadRec, Step(..), tailRecM)
|
import Control.Monad.Rec.Class (class MonadRec, Step(..), tailRecM)
|
||||||
import Control.Monad.ST.Class (liftST)
|
import Control.Monad.ST.Class (liftST)
|
||||||
import Control.Monad.Trans.Class (lift)
|
|
||||||
import Data.Array.ST as Array.ST
|
import Data.Array.ST as Array.ST
|
||||||
import Data.Either (hush)
|
|
||||||
import Data.HashMap (HashMap)
|
import Data.HashMap (HashMap)
|
||||||
import Data.HashMap as HashMap
|
import Data.HashMap as HashMap
|
||||||
import Data.Hashable (class Hashable)
|
import Data.Hashable (class Hashable)
|
||||||
@@ -15,45 +12,69 @@ import Data.List (List)
|
|||||||
import Data.List as List
|
import Data.List as List
|
||||||
import Data.Map (Map)
|
import Data.Map (Map)
|
||||||
import Data.Map as Map
|
import Data.Map as Map
|
||||||
import Data.Maybe (fromMaybe)
|
import Data.Maybe (Maybe(..), maybe)
|
||||||
import Data.Tuple.Nested (type (/\), (/\))
|
import Data.Tuple.Nested (type (/\), (/\))
|
||||||
import Effect.Class (class MonadEffect, liftEffect)
|
import Effect.Class (class MonadEffect, liftEffect)
|
||||||
import Foreign.Object (Object)
|
import Foreign.Object (Object)
|
||||||
import Foreign.Object.ST as Object.ST
|
import Foreign.Object.ST as Object.ST
|
||||||
import Foreign.Object.ST.Unsafe as Object.ST.Unsafe
|
import Foreign.Object.ST.Unsafe as Object.ST.Unsafe
|
||||||
import Pipes (next) as Pipes
|
import Node.Buffer (Buffer)
|
||||||
|
import Node.Buffer as Buffer
|
||||||
|
import Node.Encoding (Encoding)
|
||||||
import Pipes.Core (Producer)
|
import Pipes.Core (Producer)
|
||||||
|
import Pipes.Internal (Proxy(..))
|
||||||
-- | Fold every value produced
|
|
||||||
-- |
|
|
||||||
-- | Uses `MonadRec`, supporting producers of arbitrary length.
|
|
||||||
fold :: forall a b m. MonadRec m => (b -> a -> b) -> b -> Producer a m Unit -> m b
|
|
||||||
fold f b p =
|
|
||||||
let
|
|
||||||
insertNext b' p' = runMaybeT do
|
|
||||||
a /\ p'' <- MaybeT $ hush <$> Pipes.next p'
|
|
||||||
pure $ Loop $ f b' a /\ p''
|
|
||||||
in
|
|
||||||
flip tailRecM (b /\ p) \(b' /\ p') -> fromMaybe (Done b') <$> insertNext b' p'
|
|
||||||
|
|
||||||
-- | Fold every value produced with a monadic action
|
-- | Fold every value produced with a monadic action
|
||||||
-- |
|
-- |
|
||||||
-- | Uses `MonadRec`, supporting producers of arbitrary length.
|
-- | Uses `MonadRec`, supporting producers of arbitrary length.
|
||||||
traverse :: forall a b m. MonadRec m => (b -> a -> m b) -> b -> Producer a m Unit -> m b
|
traverse :: forall a b m. MonadRec m => (b -> a -> m b) -> b -> Producer a m Unit -> m b
|
||||||
traverse f b p =
|
traverse f b0 p0 =
|
||||||
let
|
flip tailRecM (p0 /\ b0) \(p /\ b) ->
|
||||||
insertNext b' p' = runMaybeT do
|
case p of
|
||||||
a /\ p'' <- MaybeT $ hush <$> Pipes.next p'
|
Respond a m -> do
|
||||||
b'' <- lift $ f b' a
|
b' <- f b a
|
||||||
pure $ Loop $ b'' /\ p''
|
pure $ Loop $ m unit /\ b'
|
||||||
in
|
M m -> do
|
||||||
flip tailRecM (b /\ p) \(b' /\ p') -> fromMaybe (Done b') <$> insertNext b' p'
|
n <- m
|
||||||
|
pure $ Loop $ (n /\ b)
|
||||||
|
Request _ _ -> pure $ Done b
|
||||||
|
Pure _ -> pure $ Done b
|
||||||
|
|
||||||
|
-- | Fold every value produced
|
||||||
|
-- |
|
||||||
|
-- | Uses `MonadRec`, supporting producers of arbitrary length.
|
||||||
|
fold :: forall a b m. MonadRec m => (b -> a -> b) -> b -> Producer a m Unit -> m b
|
||||||
|
fold f b0 p0 = traverse (\b a -> pure $ f b a) b0 p0
|
||||||
|
|
||||||
-- | Execute a monadic action on every item in a producer.
|
-- | Execute a monadic action on every item in a producer.
|
||||||
-- |
|
-- |
|
||||||
-- | Uses `MonadRec`, supporting producers of arbitrary length.
|
-- | Uses `MonadRec`, supporting producers of arbitrary length.
|
||||||
foreach :: forall a m. MonadRec m => (a -> m Unit) -> Producer a m Unit -> m Unit
|
foreach :: forall a m. MonadRec m => (a -> m Unit) -> Producer a m Unit -> m Unit
|
||||||
foreach f = traverse (const f) unit
|
foreach f p0 = traverse (\_ a -> f a) unit p0
|
||||||
|
|
||||||
|
-- | `append` all emitted values to `mempty`
|
||||||
|
toMonoid :: forall a m. Monoid a => MonadRec m => MonadEffect m => Producer a m Unit -> m a
|
||||||
|
toMonoid = fold (<>) mempty
|
||||||
|
|
||||||
|
-- | Concatenate all buffers to a single buffer, then decode with the
|
||||||
|
-- | provided encoding.
|
||||||
|
toStringWith :: forall m. MonadRec m => MonadEffect m => Encoding -> Producer Buffer m Unit -> m String
|
||||||
|
toStringWith enc = (liftEffect <<< Buffer.toString enc) <=< toBuffer
|
||||||
|
|
||||||
|
-- | Concatenate all produced buffers to a single buffer
|
||||||
|
toBuffer :: forall m. MonadRec m => MonadEffect m => Producer Buffer m Unit -> m Buffer
|
||||||
|
toBuffer p =
|
||||||
|
(liftEffect <<< maybe (Buffer.alloc 0) pure)
|
||||||
|
=<< traverse
|
||||||
|
( flip \b ->
|
||||||
|
case _ of
|
||||||
|
Just acc -> do
|
||||||
|
new <- liftEffect $ Buffer.concat [ acc, b ]
|
||||||
|
pure $ Just new
|
||||||
|
_ -> pure $ Just b
|
||||||
|
)
|
||||||
|
Nothing
|
||||||
|
p
|
||||||
|
|
||||||
-- | Collect all values from a `Producer` into an array.
|
-- | Collect all values from a `Producer` into an array.
|
||||||
toArray :: forall a m. MonadRec m => MonadEffect m => Producer a m Unit -> m (Array a)
|
toArray :: forall a m. MonadRec m => MonadEffect m => Producer a m Unit -> m (Array a)
|
||||||
@@ -63,9 +84,18 @@ toArray p = do
|
|||||||
liftEffect $ liftST $ Array.ST.unsafeFreeze st
|
liftEffect $ liftST $ Array.ST.unsafeFreeze st
|
||||||
|
|
||||||
-- | Collect all values from a `Producer` into a list.
|
-- | Collect all values from a `Producer` into a list.
|
||||||
|
-- |
|
||||||
|
-- | Reverses the list after collecting, so that values will be
|
||||||
|
-- | in the order they were emitted.
|
||||||
toList :: forall a m. MonadRec m => MonadEffect m => Producer a m Unit -> m (List a)
|
toList :: forall a m. MonadRec m => MonadEffect m => Producer a m Unit -> m (List a)
|
||||||
toList = map List.reverse <<< fold (flip List.Cons) List.Nil
|
toList = map List.reverse <<< fold (flip List.Cons) List.Nil
|
||||||
|
|
||||||
|
-- | Collect all values from a `Producer` into a list.
|
||||||
|
-- |
|
||||||
|
-- | Does not reverse the list after collecting.
|
||||||
|
toListRev :: forall a m. MonadRec m => MonadEffect m => Producer a m Unit -> m (List a)
|
||||||
|
toListRev = map List.reverse <<< fold (flip List.Cons) List.Nil
|
||||||
|
|
||||||
-- | Collect all values from a `Producer` into a Javascript Object.
|
-- | Collect all values from a `Producer` into a Javascript Object.
|
||||||
toObject :: forall a m. MonadRec m => MonadEffect m => Producer (String /\ a) m Unit -> m (Object a)
|
toObject :: forall a m. MonadRec m => MonadEffect m => Producer (String /\ a) m Unit -> m (Object a)
|
||||||
toObject p = do
|
toObject p = do
|
||||||
|
|||||||
@@ -1 +1,64 @@
|
|||||||
module Pipes.Construct where
|
module Pipes.Construct where
|
||||||
|
|
||||||
|
import Prelude
|
||||||
|
|
||||||
|
import Control.Monad.Maybe.Trans (MaybeT(..), runMaybeT)
|
||||||
|
import Control.Monad.Rec.Class (class MonadRec, Step(..), tailRecM)
|
||||||
|
import Control.Monad.ST.Class (liftST)
|
||||||
|
import Control.Monad.Trans.Class (lift)
|
||||||
|
import Data.Array as Array
|
||||||
|
import Data.Array.ST as Array.ST
|
||||||
|
import Data.List (List)
|
||||||
|
import Data.List as List
|
||||||
|
import Data.Map (Map)
|
||||||
|
import Data.Map.Internal as Map.Internal
|
||||||
|
import Data.Maybe (fromMaybe)
|
||||||
|
import Data.Tuple.Nested (type (/\), (/\))
|
||||||
|
import Effect.Class (class MonadEffect, liftEffect)
|
||||||
|
import Pipes (yield, (>->))
|
||||||
|
import Pipes.Core (Producer)
|
||||||
|
import Pipes.Prelude as Pipe
|
||||||
|
import Pipes.Util as Pipe.Util
|
||||||
|
|
||||||
|
-- Producer that will emit monotonically increasing integers
|
||||||
|
-- ex `monotonic 0 -> 0 1 2 3 4 5 6 7 ..`
|
||||||
|
monotonic :: forall m. MonadRec m => Int -> Producer Int m Unit
|
||||||
|
monotonic start = flip tailRecM start \n -> yield n $> Loop (n + 1)
|
||||||
|
|
||||||
|
-- Producer that will emit integers from `start` (inclusive) to `end` (exclusive)
|
||||||
|
range :: forall m. MonadRec m => Int -> Int -> Producer Int m Unit
|
||||||
|
range start end = monotonic start >-> Pipe.take end
|
||||||
|
|
||||||
|
-- | Stack-safe producer that yields every value in an Array
|
||||||
|
eachArray :: forall a m. MonadRec m => Array a -> Producer a m Unit
|
||||||
|
eachArray as = monotonic 0 >-> Pipe.map (Array.index as) >-> Pipe.Util.whileJust
|
||||||
|
|
||||||
|
-- | Stack-safe producer that yields every value in a List
|
||||||
|
eachList :: forall a m. MonadRec m => List a -> Producer a m Unit
|
||||||
|
eachList init =
|
||||||
|
flip tailRecM init \as -> fromMaybe (Done unit) <$> runMaybeT do
|
||||||
|
head <- MaybeT $ pure $ List.head as
|
||||||
|
tail <- MaybeT $ pure $ List.tail as
|
||||||
|
lift $ yield head
|
||||||
|
pure $ Loop tail
|
||||||
|
|
||||||
|
-- | Stack-safe producer that yields every value in a Map
|
||||||
|
eachMap :: forall k v m. MonadEffect m => MonadRec m => Map k v -> Producer (k /\ v) m Unit
|
||||||
|
eachMap init = do
|
||||||
|
stack <- liftEffect $ liftST $ Array.ST.new
|
||||||
|
let
|
||||||
|
push a = void $ liftEffect $ liftST $ Array.ST.push a stack
|
||||||
|
pop = liftEffect $ liftST $ Array.ST.pop stack
|
||||||
|
flip tailRecM init case _ of
|
||||||
|
Map.Internal.Leaf -> fromMaybe (Done unit) <$> runMaybeT do
|
||||||
|
a <- MaybeT pop
|
||||||
|
pure $ Loop a
|
||||||
|
Map.Internal.Node _ _ k v Map.Internal.Leaf Map.Internal.Leaf -> do
|
||||||
|
yield $ k /\ v
|
||||||
|
pure $ Loop Map.Internal.Leaf
|
||||||
|
Map.Internal.Node _ _ k v Map.Internal.Leaf r -> do
|
||||||
|
yield $ k /\ v
|
||||||
|
pure $ Loop r
|
||||||
|
Map.Internal.Node a b k v l r -> do
|
||||||
|
push $ Map.Internal.Node a b k v Map.Internal.Leaf r
|
||||||
|
pure $ Loop l
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ import Effect.Aff.Class (class MonadAff)
|
|||||||
import Effect.Class (liftEffect)
|
import Effect.Class (liftEffect)
|
||||||
import Effect.Exception (Error)
|
import Effect.Exception (Error)
|
||||||
import Node.Buffer (Buffer)
|
import Node.Buffer (Buffer)
|
||||||
import Node.FS.Stream (WriteStreamOptions)
|
import Node.FS.Stream (WriteStreamOptions, ReadStreamOptions)
|
||||||
import Node.FS.Stream as FS.Stream
|
import Node.FS.Stream as FS.Stream
|
||||||
import Node.Path (FilePath)
|
import Node.Path (FilePath)
|
||||||
import Node.Stream.Object as O
|
import Node.Stream.Object as O
|
||||||
@@ -23,7 +23,7 @@ import Prim.Row (class Union)
|
|||||||
-- |
|
-- |
|
||||||
-- | See `Pipes.Node.Stream.withEOS` for converting `Producer a`
|
-- | See `Pipes.Node.Stream.withEOS` for converting `Producer a`
|
||||||
-- | into `Producer (Maybe a)`, emitting `Nothing` before exiting.
|
-- | into `Producer (Maybe a)`, emitting `Nothing` before exiting.
|
||||||
write
|
write'
|
||||||
:: forall r trash m
|
:: forall r trash m
|
||||||
. Union r trash WriteStreamOptions
|
. Union r trash WriteStreamOptions
|
||||||
=> MonadAff m
|
=> MonadAff m
|
||||||
@@ -31,27 +31,27 @@ write
|
|||||||
=> Record r
|
=> Record r
|
||||||
-> FilePath
|
-> FilePath
|
||||||
-> Consumer (Maybe Buffer) m Unit
|
-> Consumer (Maybe Buffer) m Unit
|
||||||
write o p = do
|
write' o p = do
|
||||||
w <- liftEffect $ FS.Stream.createWriteStream' p o
|
w <- liftEffect $ FS.Stream.createWriteStream' p o
|
||||||
fromWritable $ O.fromBufferWritable w
|
fromWritable $ O.unsafeCoerceWritable w
|
||||||
|
|
||||||
-- | Open a file in write mode, failing if the file already exists.
|
-- | Open a file in write mode, failing if the file already exists.
|
||||||
-- |
|
-- |
|
||||||
-- | `write {flags: "wx"}`
|
-- | `write' {flags: "wx"}`
|
||||||
create :: forall m. MonadAff m => MonadThrow Error m => FilePath -> Consumer (Maybe Buffer) m Unit
|
create :: forall m. MonadAff m => MonadThrow Error m => FilePath -> Consumer (Maybe Buffer) m Unit
|
||||||
create = write { flags: "wx" }
|
create = write' { flags: "wx" }
|
||||||
|
|
||||||
-- | Open a file in write mode, truncating it if the file already exists.
|
-- | Open a file in write mode, truncating it if the file already exists.
|
||||||
-- |
|
-- |
|
||||||
-- | `write {flags: "w"}`
|
-- | `write' {flags: "w"}`
|
||||||
truncate :: forall m. MonadAff m => MonadThrow Error m => FilePath -> Consumer (Maybe Buffer) m Unit
|
trunc :: forall m. MonadAff m => MonadThrow Error m => FilePath -> Consumer (Maybe Buffer) m Unit
|
||||||
truncate = write { flags: "w" }
|
trunc = write' { flags: "w" }
|
||||||
|
|
||||||
-- | Open a file in write mode, appending written contents if the file already exists.
|
-- | Open a file in write mode, appending written contents if the file already exists.
|
||||||
-- |
|
-- |
|
||||||
-- | `write {flags: "a"}`
|
-- | `write' {flags: "a"}`
|
||||||
append :: forall m. MonadAff m => MonadThrow Error m => FilePath -> Consumer (Maybe Buffer) m Unit
|
append :: forall m. MonadAff m => MonadThrow Error m => FilePath -> Consumer (Maybe Buffer) m Unit
|
||||||
append = write { flags: "a" }
|
append = write' { flags: "a" }
|
||||||
|
|
||||||
-- | Creates a `fs.Readable` stream for the file at the given path.
|
-- | Creates a `fs.Readable` stream for the file at the given path.
|
||||||
-- |
|
-- |
|
||||||
@@ -60,4 +60,20 @@ append = write { flags: "a" }
|
|||||||
read :: forall m. MonadAff m => MonadThrow Error m => FilePath -> Producer (Maybe Buffer) m Unit
|
read :: forall m. MonadAff m => MonadThrow Error m => FilePath -> Producer (Maybe Buffer) m Unit
|
||||||
read p = do
|
read p = do
|
||||||
r <- liftEffect $ FS.Stream.createReadStream p
|
r <- liftEffect $ FS.Stream.createReadStream p
|
||||||
fromReadable $ O.fromBufferReadable r
|
fromReadable $ O.unsafeCoerceReadable r
|
||||||
|
|
||||||
|
-- | Creates a `fs.Readable` stream for the file at the given path.
|
||||||
|
-- |
|
||||||
|
-- | Emits `Nothing` before closing. To opt out of this behavior,
|
||||||
|
-- | use `Pipes.Node.Stream.withoutEOS` or `Pipes.Node.Stream.unEOS`.
|
||||||
|
read'
|
||||||
|
:: forall r trash m
|
||||||
|
. Union r trash ReadStreamOptions
|
||||||
|
=> MonadAff m
|
||||||
|
=> MonadThrow Error m
|
||||||
|
=> Record r
|
||||||
|
-> FilePath
|
||||||
|
-> Producer (Maybe Buffer) m Unit
|
||||||
|
read' opts p = do
|
||||||
|
r <- liftEffect $ FS.Stream.createReadStream' p opts
|
||||||
|
fromReadable $ O.unsafeCoerceReadable r
|
||||||
|
|||||||
@@ -1,17 +1,16 @@
|
|||||||
module Pipes.Node.Stream where
|
module Pipes.Node.Stream where
|
||||||
|
|
||||||
import Prelude
|
import Prelude hiding (join)
|
||||||
|
|
||||||
import Control.Monad.Error.Class (class MonadThrow, throwError)
|
import Control.Monad.Error.Class (class MonadThrow, throwError)
|
||||||
import Control.Monad.Rec.Class (class MonadRec, Step(..), tailRecM)
|
import Control.Monad.Rec.Class (class MonadRec, Step(..), tailRecM, whileJust)
|
||||||
import Control.Monad.ST.Class (liftST)
|
import Control.Monad.ST.Class (liftST)
|
||||||
import Control.Monad.ST.Ref as STRef
|
import Control.Monad.ST.Ref as STRef
|
||||||
import Control.Monad.Trans.Class (lift)
|
import Control.Monad.Trans.Class (lift)
|
||||||
import Data.Maybe (Maybe(..))
|
import Control.Parallel (parOneOf)
|
||||||
import Data.Newtype (wrap)
|
import Data.Maybe (Maybe(..), maybe)
|
||||||
import Data.Traversable (for_)
|
import Data.Traversable (for_, traverse, traverse_)
|
||||||
import Data.Tuple.Nested ((/\))
|
import Data.Tuple.Nested ((/\))
|
||||||
import Effect.Aff (delay)
|
|
||||||
import Effect.Aff.Class (class MonadAff, liftAff)
|
import Effect.Aff.Class (class MonadAff, liftAff)
|
||||||
import Effect.Class (liftEffect)
|
import Effect.Class (liftEffect)
|
||||||
import Effect.Exception (Error)
|
import Effect.Exception (Error)
|
||||||
@@ -19,7 +18,6 @@ import Node.Stream.Object as O
|
|||||||
import Pipes (await, yield)
|
import Pipes (await, yield)
|
||||||
import Pipes (for) as P
|
import Pipes (for) as P
|
||||||
import Pipes.Core (Consumer, Pipe, Producer, Producer_)
|
import Pipes.Core (Consumer, Pipe, Producer, Producer_)
|
||||||
import Pipes.Prelude (mapFoldable) as P
|
|
||||||
import Pipes.Util (InvokeResult(..), invoke)
|
import Pipes.Util (InvokeResult(..), invoke)
|
||||||
|
|
||||||
-- | Convert a `Readable` stream to a `Pipe`.
|
-- | Convert a `Readable` stream to a `Pipe`.
|
||||||
@@ -34,15 +32,19 @@ fromReadable r =
|
|||||||
pure $ Done unit
|
pure $ Done unit
|
||||||
|
|
||||||
go { error, cancel } = do
|
go { error, cancel } = do
|
||||||
liftAff $ delay $ wrap 0.0
|
|
||||||
err <- liftEffect $ liftST $ STRef.read error
|
err <- liftEffect $ liftST $ STRef.read error
|
||||||
for_ err throwError
|
for_ err throwError
|
||||||
|
|
||||||
res <- liftEffect $ O.read r
|
res <- liftEffect $ O.read r
|
||||||
case res of
|
case res of
|
||||||
O.ReadJust a -> yield (Just a) $> Loop { error, cancel }
|
O.ReadJust a -> yield (Just a) $> Loop { error, cancel }
|
||||||
O.ReadWouldBlock -> liftAff (O.awaitReadableOrClosed r) $> Loop { error, cancel }
|
O.ReadWouldBlock -> do
|
||||||
O.ReadClosed -> yield Nothing *> cleanup cancel
|
ended <- liftEffect $ O.isReadableEnded r
|
||||||
|
if ended then do
|
||||||
|
yield Nothing
|
||||||
|
cleanup cancel
|
||||||
|
else
|
||||||
|
liftAff (O.awaitReadableOrClosed r) $> Loop { error, cancel }
|
||||||
in
|
in
|
||||||
do
|
do
|
||||||
e <- liftEffect $ O.withErrorST r
|
e <- liftEffect $ O.withErrorST r
|
||||||
@@ -53,73 +55,78 @@ fromReadable r =
|
|||||||
-- | When `Nothing` is piped to this, the stream will
|
-- | When `Nothing` is piped to this, the stream will
|
||||||
-- | be `end`ed, and the pipe will noop if invoked again.
|
-- | be `end`ed, and the pipe will noop if invoked again.
|
||||||
fromWritable :: forall s a m. MonadThrow Error m => MonadAff m => O.Write s a => s -> Consumer (Maybe a) m Unit
|
fromWritable :: forall s a m. MonadThrow Error m => MonadAff m => O.Write s a => s -> Consumer (Maybe a) m Unit
|
||||||
fromWritable w =
|
fromWritable w = do
|
||||||
|
{ error: errorST, cancel: removeErrorListener } <- liftEffect $ O.withErrorST w
|
||||||
|
|
||||||
let
|
let
|
||||||
cleanup rmErrorListener = do
|
maybeThrow = traverse_ throwError =<< liftEffect (liftST $ STRef.read errorST)
|
||||||
liftEffect rmErrorListener
|
|
||||||
liftEffect $ O.end w
|
|
||||||
pure $ Done unit
|
|
||||||
|
|
||||||
go { error, cancel } = do
|
waitCanWrite = do
|
||||||
liftAff $ delay $ wrap 0.0
|
shouldWait <- liftEffect $ O.needsDrain w
|
||||||
err <- liftEffect $ liftST $ STRef.read error
|
when shouldWait $ liftAff $ O.awaitWritableOrClosed w
|
||||||
for_ err throwError
|
|
||||||
|
|
||||||
ma <- await
|
cleanup = do
|
||||||
case ma of
|
liftAff $ O.awaitFinished w
|
||||||
Nothing -> cleanup cancel
|
maybeThrow
|
||||||
Just a -> do
|
liftEffect removeErrorListener
|
||||||
res <- liftEffect $ O.write w a
|
|
||||||
case res of
|
onEOS = liftEffect (O.end w) *> cleanup $> Done unit
|
||||||
O.WriteOk -> pure $ Loop { error, cancel }
|
onChunk a = liftEffect (O.write w a) $> Loop unit
|
||||||
O.WriteWouldBlock -> do
|
|
||||||
liftAff (O.awaitWritableOrClosed w)
|
go _ = do
|
||||||
pure $ Loop { error, cancel }
|
maybeThrow
|
||||||
O.WriteClosed -> cleanup cancel
|
waitCanWrite
|
||||||
in
|
ended <- liftEffect $ O.isWritableEnded w
|
||||||
do
|
if ended then
|
||||||
r <- liftEffect $ O.withErrorST w
|
cleanup $> Done unit
|
||||||
tailRecM go r
|
else
|
||||||
|
await >>= maybe onEOS onChunk
|
||||||
|
|
||||||
|
tailRecM go unit
|
||||||
|
|
||||||
-- | Convert a `Transform` stream to a `Pipe`.
|
-- | Convert a `Transform` stream to a `Pipe`.
|
||||||
-- |
|
-- |
|
||||||
-- | When `Nothing` is piped to this, the `Transform` stream will
|
-- | When `Nothing` is piped to this, the `Transform` stream will
|
||||||
-- | be `end`ed, and the pipe will noop if invoked again.
|
-- | be `end`ed, and the pipe will noop if invoked again.
|
||||||
fromTransform :: forall a b m. MonadThrow Error m => MonadAff m => O.Transform a b -> Pipe (Maybe a) (Maybe b) m Unit
|
fromTransform :: forall a b m. MonadThrow Error m => MonadAff m => O.Transform a b -> Pipe (Maybe a) (Maybe b) m Unit
|
||||||
fromTransform t =
|
fromTransform t = do
|
||||||
|
{ error: errorST, cancel: removeErrorListener } <- liftEffect $ O.withErrorST t
|
||||||
let
|
let
|
||||||
cleanup removeErrorListener = do
|
maybeThrow = traverse_ throwError =<< liftEffect (liftST $ STRef.read errorST)
|
||||||
liftEffect $ O.end t
|
|
||||||
liftEffect $ removeErrorListener
|
|
||||||
fromReadable t
|
|
||||||
pure $ Done unit
|
|
||||||
yieldFromReadableHalf = do
|
|
||||||
res <- liftEffect (O.read t)
|
|
||||||
case res of
|
|
||||||
O.ReadJust a -> yield (Just a)
|
|
||||||
O.ReadWouldBlock -> pure unit
|
|
||||||
O.ReadClosed -> yield Nothing *> pure unit
|
|
||||||
go { error, cancel } = do
|
|
||||||
liftAff $ delay $ wrap 0.0
|
|
||||||
err <- liftEffect $ liftST $ STRef.read error
|
|
||||||
for_ err throwError
|
|
||||||
|
|
||||||
ma <- await
|
cleanup = do
|
||||||
case ma of
|
flip tailRecM unit $ const do
|
||||||
Nothing -> cleanup cancel
|
liftAff $ O.awaitReadableOrClosed t
|
||||||
Just a' -> do
|
readEnded <- liftEffect $ O.isReadableEnded t
|
||||||
res <- liftEffect $ O.write t a'
|
yieldWhileReadable
|
||||||
yieldFromReadableHalf
|
pure $ (if readEnded then Done else Loop) unit
|
||||||
case res of
|
|
||||||
O.WriteClosed -> cleanup cancel
|
liftAff $ O.awaitFinished t
|
||||||
O.WriteOk -> pure $ Loop { error, cancel }
|
maybeThrow
|
||||||
O.WriteWouldBlock -> do
|
liftEffect $ removeErrorListener
|
||||||
liftAff $ O.awaitWritableOrClosed t
|
yield Nothing
|
||||||
pure $ Loop { error, cancel }
|
|
||||||
in
|
yieldWhileReadable = void $ whileJust $ maybeYield1
|
||||||
do
|
|
||||||
r <- liftEffect $ O.withErrorST t
|
maybeYield1 = traverse (\a -> yield (Just a) $> Just unit) =<< O.maybeReadResult <$> liftEffect (O.read t)
|
||||||
tailRecM go r
|
|
||||||
|
onEOS = liftEffect (O.end t) *> cleanup $> Done unit
|
||||||
|
onChunk a = liftEffect (O.write t a) $> Loop unit
|
||||||
|
|
||||||
|
go _ = do
|
||||||
|
maybeThrow
|
||||||
|
needsDrain <- liftEffect $ O.needsDrain t
|
||||||
|
ended <- liftEffect $ O.isWritableEnded t
|
||||||
|
if needsDrain then do
|
||||||
|
yieldWhileReadable
|
||||||
|
liftAff $ parOneOf [O.awaitWritableOrClosed t, O.awaitReadableOrClosed t]
|
||||||
|
pure $ Loop unit
|
||||||
|
else if ended then
|
||||||
|
cleanup $> Done unit
|
||||||
|
else
|
||||||
|
await >>= maybe onEOS onChunk
|
||||||
|
|
||||||
|
tailRecM go unit
|
||||||
|
|
||||||
-- | Given a `Producer` of values, wrap them in `Just`.
|
-- | Given a `Producer` of values, wrap them in `Just`.
|
||||||
-- |
|
-- |
|
||||||
@@ -131,7 +138,7 @@ withEOS a = do
|
|||||||
|
|
||||||
-- | Strip a pipeline of the EOS signal
|
-- | Strip a pipeline of the EOS signal
|
||||||
unEOS :: forall a m. Monad m => Pipe (Maybe a) a m Unit
|
unEOS :: forall a m. Monad m => Pipe (Maybe a) a m Unit
|
||||||
unEOS = P.mapFoldable identity
|
unEOS = tailRecM (const $ maybe (pure $ Done unit) (\a -> yield a $> Loop unit) =<< await) unit
|
||||||
|
|
||||||
-- | Lift a `Pipe a a` to `Pipe (Maybe a) (Maybe a)`.
|
-- | Lift a `Pipe a a` to `Pipe (Maybe a) (Maybe a)`.
|
||||||
-- |
|
-- |
|
||||||
|
|||||||
@@ -18,7 +18,7 @@ import Pipes.Node.Stream (fromTransform)
|
|||||||
fromZlib :: forall r m. MonadAff m => MonadThrow Error m => Effect (ZlibStream r) -> Pipe (Maybe Buffer) (Maybe Buffer) m Unit
|
fromZlib :: forall r m. MonadAff m => MonadThrow Error m => Effect (ZlibStream r) -> Pipe (Maybe Buffer) (Maybe Buffer) m Unit
|
||||||
fromZlib z = do
|
fromZlib z = do
|
||||||
raw <- liftEffect $ Zlib.toDuplex <$> z
|
raw <- liftEffect $ Zlib.toDuplex <$> z
|
||||||
fromTransform $ O.fromBufferTransform raw
|
fromTransform $ O.unsafeCoerceTransform raw
|
||||||
|
|
||||||
gzip :: forall m. MonadAff m => MonadThrow Error m => Pipe (Maybe Buffer) (Maybe Buffer) m Unit
|
gzip :: forall m. MonadAff m => MonadThrow Error m => Pipe (Maybe Buffer) (Maybe Buffer) m Unit
|
||||||
gzip = fromZlib Zlib.createGzip
|
gzip = fromZlib Zlib.createGzip
|
||||||
|
|||||||
@@ -3,23 +3,35 @@ module Pipes.Util where
|
|||||||
import Prelude
|
import Prelude
|
||||||
|
|
||||||
import Control.Monad.Maybe.Trans (MaybeT(..), runMaybeT)
|
import Control.Monad.Maybe.Trans (MaybeT(..), runMaybeT)
|
||||||
import Control.Monad.Rec.Class (class MonadRec, forever, whileJust)
|
import Control.Monad.Rec.Class (class MonadRec, Step(..), forever, tailRecM)
|
||||||
|
import Control.Monad.Rec.Class as Rec
|
||||||
import Control.Monad.ST.Class (liftST)
|
import Control.Monad.ST.Class (liftST)
|
||||||
import Control.Monad.ST.Ref (STRef)
|
import Control.Monad.ST.Ref (STRef)
|
||||||
import Control.Monad.ST.Ref as STRef
|
import Control.Monad.ST.Ref as STRef
|
||||||
import Control.Monad.Trans.Class (lift)
|
import Control.Monad.Trans.Class (lift)
|
||||||
import Data.Array.ST (STArray)
|
import Data.Array.ST (STArray)
|
||||||
import Data.Array.ST as Array.ST
|
import Data.Array.ST as Array.ST
|
||||||
|
import Data.Either (hush)
|
||||||
import Data.HashSet as HashSet
|
import Data.HashSet as HashSet
|
||||||
import Data.Hashable (class Hashable, hash)
|
import Data.Hashable (class Hashable, hash)
|
||||||
import Data.List.NonEmpty (NonEmptyList)
|
import Data.List.NonEmpty (NonEmptyList)
|
||||||
import Data.Maybe (Maybe(..))
|
import Data.Maybe (Maybe(..), fromMaybe)
|
||||||
import Data.Tuple.Nested (type (/\), (/\))
|
import Data.Tuple.Nested (type (/\), (/\))
|
||||||
import Effect.Class (class MonadEffect, liftEffect)
|
import Effect.Class (class MonadEffect, liftEffect)
|
||||||
import Pipes (await, yield)
|
import Pipes (await, yield)
|
||||||
import Pipes.Core (Pipe)
|
import Pipes as Pipes
|
||||||
|
import Pipes.Core (Pipe, Producer)
|
||||||
import Pipes.Internal (Proxy(..))
|
import Pipes.Internal (Proxy(..))
|
||||||
|
|
||||||
|
-- | Re-yield all `Just`s, and close when `Nothing` is encountered
|
||||||
|
whileJust :: forall m a. MonadRec m => Pipe (Maybe a) a m Unit
|
||||||
|
whileJust = do
|
||||||
|
first <- await
|
||||||
|
flip tailRecM first $ \ma -> fromMaybe (Done unit) <$> runMaybeT do
|
||||||
|
a <- MaybeT $ pure ma
|
||||||
|
lift $ yield a
|
||||||
|
lift $ Loop <$> await
|
||||||
|
|
||||||
-- | Yields a separator value `sep` between received values
|
-- | Yields a separator value `sep` between received values
|
||||||
-- |
|
-- |
|
||||||
-- | ```purescript
|
-- | ```purescript
|
||||||
@@ -33,7 +45,7 @@ intersperse sep = do
|
|||||||
getIsFirst = liftEffect $ liftST $ STRef.read isFirstST
|
getIsFirst = liftEffect $ liftST $ STRef.read isFirstST
|
||||||
markNotFirst = void $ liftEffect $ liftST $ STRef.write false isFirstST
|
markNotFirst = void $ liftEffect $ liftST $ STRef.write false isFirstST
|
||||||
|
|
||||||
whileJust $ runMaybeT do
|
Rec.whileJust $ runMaybeT do
|
||||||
a <- MaybeT await
|
a <- MaybeT await
|
||||||
isFirst <- getIsFirst
|
isFirst <- getIsFirst
|
||||||
if isFirst then markNotFirst else lift $ yield $ Just sep
|
if isFirst then markNotFirst else lift $ yield $ Just sep
|
||||||
@@ -41,6 +53,16 @@ intersperse sep = do
|
|||||||
|
|
||||||
yield Nothing
|
yield Nothing
|
||||||
|
|
||||||
|
-- Pair every emitted value from 2 producers together, exiting when either exits.
|
||||||
|
zip :: forall a b m. MonadRec m => Producer a m Unit -> Producer b m Unit -> Producer (a /\ b) m Unit
|
||||||
|
zip as bs =
|
||||||
|
flip tailRecM (as /\ bs) \(as' /\ bs') ->
|
||||||
|
fromMaybe (Done unit) <$> runMaybeT do
|
||||||
|
a /\ as'' <- MaybeT $ lift $ hush <$> Pipes.next as'
|
||||||
|
b /\ bs'' <- MaybeT $ lift $ hush <$> Pipes.next bs'
|
||||||
|
lift $ yield $ a /\ b
|
||||||
|
pure $ Loop $ as'' /\ bs''
|
||||||
|
|
||||||
-- | Accumulate values in chunks of a given size.
|
-- | Accumulate values in chunks of a given size.
|
||||||
-- |
|
-- |
|
||||||
-- | If the pipe closes without yielding a multiple of `size` elements,
|
-- | If the pipe closes without yielding a multiple of `size` elements,
|
||||||
@@ -60,12 +82,18 @@ chunked size = do
|
|||||||
void $ flip STRef.write chunkST =<< Array.ST.new
|
void $ flip STRef.write chunkST =<< Array.ST.new
|
||||||
Array.ST.unsafeFreeze chunkArray
|
Array.ST.unsafeFreeze chunkArray
|
||||||
|
|
||||||
whileJust $ runMaybeT do
|
Rec.whileJust $ runMaybeT do
|
||||||
a <- MaybeT await
|
a <- MaybeT await
|
||||||
chunkPut a
|
chunkPut a
|
||||||
len <- chunkLength
|
len <- lift chunkLength
|
||||||
when (len >= size) $ lift $ yield =<< Just <$> chunkTake
|
when (len >= size) do
|
||||||
yield =<< Just <$> chunkTake
|
chunk <- lift chunkTake
|
||||||
|
lift $ yield $ Just chunk
|
||||||
|
len <- chunkLength
|
||||||
|
when (len > 0) do
|
||||||
|
chunk <- chunkTake
|
||||||
|
yield $ Just chunk
|
||||||
|
|
||||||
yield Nothing
|
yield Nothing
|
||||||
|
|
||||||
-- | Equivalent of unix `uniq`, filtering out duplicate values passed to it.
|
-- | Equivalent of unix `uniq`, filtering out duplicate values passed to it.
|
||||||
|
|||||||
@@ -9,12 +9,14 @@ import Test.Pipes.Node.Stream as Test.Pipes.Node.Stream
|
|||||||
import Test.Pipes.Node.Buffer as Test.Pipes.Node.Buffer
|
import Test.Pipes.Node.Buffer as Test.Pipes.Node.Buffer
|
||||||
import Test.Pipes.Node.FS as Test.Pipes.Node.FS
|
import Test.Pipes.Node.FS as Test.Pipes.Node.FS
|
||||||
import Test.Pipes.Collect as Test.Pipes.Collect
|
import Test.Pipes.Collect as Test.Pipes.Collect
|
||||||
|
import Test.Pipes.Construct as Test.Pipes.Construct
|
||||||
import Test.Spec.Reporter (specReporter)
|
import Test.Spec.Reporter (specReporter)
|
||||||
import Test.Spec.Runner (defaultConfig, runSpec')
|
import Test.Spec.Runner (defaultConfig, runSpec')
|
||||||
|
|
||||||
main :: Effect Unit
|
main :: Effect Unit
|
||||||
main = launchAff_ $ runSpec' (defaultConfig { failFast = true, timeout = Nothing }) [ specReporter ] do
|
main = launchAff_ $ runSpec' (defaultConfig { exit = false, timeout = Nothing }) [ specReporter ] do
|
||||||
Test.Pipes.Node.Stream.spec
|
Test.Pipes.Node.Stream.spec
|
||||||
Test.Pipes.Node.Buffer.spec
|
Test.Pipes.Node.Buffer.spec
|
||||||
Test.Pipes.Node.FS.spec
|
Test.Pipes.Node.FS.spec
|
||||||
Test.Pipes.Collect.spec
|
Test.Pipes.Collect.spec
|
||||||
|
Test.Pipes.Construct.spec
|
||||||
|
|||||||
58
test/Test/Pipes.Construct.purs
Normal file
58
test/Test/Pipes.Construct.purs
Normal file
@@ -0,0 +1,58 @@
|
|||||||
|
module Test.Pipes.Construct where
|
||||||
|
|
||||||
|
import Prelude
|
||||||
|
|
||||||
|
import Data.Array as Array
|
||||||
|
import Data.List as List
|
||||||
|
import Data.Map as Map
|
||||||
|
import Data.Tuple.Nested (type (/\), (/\))
|
||||||
|
import Effect.Class (liftEffect)
|
||||||
|
import Pipes.Collect as Pipes.Collect
|
||||||
|
import Pipes.Construct as Pipes.Construct
|
||||||
|
import Test.Spec (Spec, describe, it)
|
||||||
|
import Test.Spec.Assertions (shouldEqual)
|
||||||
|
|
||||||
|
spec :: Spec Unit
|
||||||
|
spec =
|
||||||
|
describe "Test.Pipes.Construct" do
|
||||||
|
describe "eachMap" do
|
||||||
|
it "empty map" do
|
||||||
|
kvs <- Pipes.Collect.toArray $ Pipes.Construct.eachMap Map.empty
|
||||||
|
kvs `shouldEqual` ([] :: Array (Int /\ Int))
|
||||||
|
it "nonempty map" do
|
||||||
|
let
|
||||||
|
exp = (\n -> n /\ n) <$> Array.range 0 99999
|
||||||
|
map = Map.fromFoldable exp
|
||||||
|
kvs <-
|
||||||
|
liftEffect
|
||||||
|
$ Pipes.Collect.toArray
|
||||||
|
$ Pipes.Construct.eachMap
|
||||||
|
$ map
|
||||||
|
kvs `shouldEqual` exp
|
||||||
|
describe "eachArray" do
|
||||||
|
it "empty array" do
|
||||||
|
kvs <- Pipes.Collect.toArray $ Pipes.Construct.eachArray []
|
||||||
|
kvs `shouldEqual` ([] :: Array Int)
|
||||||
|
it "nonempty array" do
|
||||||
|
let
|
||||||
|
inp = (\n -> n /\ n) <$> Array.range 0 99999
|
||||||
|
kvs <-
|
||||||
|
liftEffect
|
||||||
|
$ Pipes.Collect.toArray
|
||||||
|
$ Pipes.Construct.eachArray
|
||||||
|
$ inp
|
||||||
|
kvs `shouldEqual` inp
|
||||||
|
describe "eachList" do
|
||||||
|
it "empty list" do
|
||||||
|
kvs <- Pipes.Collect.toArray $ Pipes.Construct.eachList List.Nil
|
||||||
|
kvs `shouldEqual` ([] :: Array Int)
|
||||||
|
it "nonempty list" do
|
||||||
|
let
|
||||||
|
inp = (\n -> n /\ n) <$> Array.range 0 99999
|
||||||
|
kvs <-
|
||||||
|
liftEffect
|
||||||
|
$ Pipes.Collect.toArray
|
||||||
|
$ Pipes.Construct.eachList
|
||||||
|
$ List.fromFoldable
|
||||||
|
$ inp
|
||||||
|
kvs `shouldEqual` inp
|
||||||
@@ -33,18 +33,13 @@ spec = describe "Pipes.Node.FS" do
|
|||||||
liftEffect $ FS.writeTextFile UTF8 p "foo"
|
liftEffect $ FS.writeTextFile UTF8 p "foo"
|
||||||
s <- fold <$> Pipes.toListM (Pipes.Node.FS.read p >-> unEOS >-> Pipes.Node.Buffer.toString UTF8)
|
s <- fold <$> Pipes.toListM (Pipes.Node.FS.read p >-> unEOS >-> Pipes.Node.Buffer.toString UTF8)
|
||||||
s `shouldEqual` "foo"
|
s `shouldEqual` "foo"
|
||||||
around tmpFile $ it "fails if the file already exists" \p -> do
|
|
||||||
liftEffect $ FS.writeTextFile UTF8 "foo" p
|
|
||||||
flip catchError (const $ pure unit) do
|
|
||||||
Pipes.runEffect $ withEOS (yield "foo" >-> Pipes.Node.Buffer.fromString UTF8) >-> Pipes.Node.FS.create p
|
|
||||||
fail "should have thrown"
|
|
||||||
describe "create" do
|
describe "create" do
|
||||||
around tmpFile $ it "creates the file when not exists" \p -> do
|
around tmpFile $ it "creates the file when not exists" \p -> do
|
||||||
Pipes.runEffect $ withEOS (yield "foo" >-> Pipes.Node.Buffer.fromString UTF8) >-> Pipes.Node.FS.create p
|
Pipes.runEffect $ withEOS (yield "foo" >-> Pipes.Node.Buffer.fromString UTF8) >-> Pipes.Node.FS.create p
|
||||||
contents <- liftEffect $ FS.readTextFile UTF8 p
|
contents <- liftEffect $ FS.readTextFile UTF8 p
|
||||||
contents `shouldEqual` "foo"
|
contents `shouldEqual` "foo"
|
||||||
around tmpFile $ it "fails if the file already exists" \p -> do
|
around tmpFile $ it "fails if the file already exists" \p -> do
|
||||||
liftEffect $ FS.writeTextFile UTF8 "foo" p
|
liftEffect $ FS.writeTextFile UTF8 p "foo"
|
||||||
flip catchError (const $ pure unit) do
|
flip catchError (const $ pure unit) do
|
||||||
Pipes.runEffect $ withEOS (yield "foo" >-> Pipes.Node.Buffer.fromString UTF8) >-> Pipes.Node.FS.create p
|
Pipes.runEffect $ withEOS (yield "foo" >-> Pipes.Node.Buffer.fromString UTF8) >-> Pipes.Node.FS.create p
|
||||||
fail "should have thrown"
|
fail "should have thrown"
|
||||||
@@ -59,14 +54,14 @@ spec = describe "Pipes.Node.FS" do
|
|||||||
Pipes.runEffect $ withEOS (yield "bar" >-> Pipes.Node.Buffer.fromString UTF8) >-> Pipes.Node.FS.append p
|
Pipes.runEffect $ withEOS (yield "bar" >-> Pipes.Node.Buffer.fromString UTF8) >-> Pipes.Node.FS.append p
|
||||||
contents <- liftEffect $ FS.readTextFile UTF8 p
|
contents <- liftEffect $ FS.readTextFile UTF8 p
|
||||||
contents `shouldEqual` "foo\nbar"
|
contents `shouldEqual` "foo\nbar"
|
||||||
describe "truncate" do
|
describe "trunc" do
|
||||||
around tmpFile $ it "creates the file when not exists" \p -> do
|
around tmpFile $ it "creates the file when not exists" \p -> do
|
||||||
Pipes.runEffect $ withEOS (yield "foo" >-> Pipes.Node.Buffer.fromString UTF8) >-> Pipes.Node.FS.truncate p
|
Pipes.runEffect $ withEOS (yield "foo" >-> Pipes.Node.Buffer.fromString UTF8) >-> Pipes.Node.FS.trunc p
|
||||||
contents <- liftEffect $ FS.readTextFile UTF8 p
|
contents <- liftEffect $ FS.readTextFile UTF8 p
|
||||||
contents `shouldEqual` "foo"
|
contents `shouldEqual` "foo"
|
||||||
around tmpFile $ it "overwrites contents" \p -> do
|
around tmpFile $ it "overwrites contents" \p -> do
|
||||||
Pipes.runEffect $ withEOS (yield "foo" >-> Pipes.Node.Buffer.fromString UTF8) >-> Pipes.Node.FS.truncate p
|
Pipes.runEffect $ withEOS (yield "foo" >-> Pipes.Node.Buffer.fromString UTF8) >-> Pipes.Node.FS.trunc p
|
||||||
Pipes.runEffect $ withEOS (yield "bar" >-> Pipes.Node.Buffer.fromString UTF8) >-> Pipes.Node.FS.truncate p
|
Pipes.runEffect $ withEOS (yield "bar" >-> Pipes.Node.Buffer.fromString UTF8) >-> Pipes.Node.FS.trunc p
|
||||||
contents <- liftEffect $ FS.readTextFile UTF8 p
|
contents <- liftEffect $ FS.readTextFile UTF8 p
|
||||||
contents `shouldEqual` "bar"
|
contents `shouldEqual` "bar"
|
||||||
around tmpFiles $ it "json lines >-> parse >-> _.foo >-> write" \(a /\ b) -> do
|
around tmpFiles $ it "json lines >-> parse >-> _.foo >-> write" \(a /\ b) -> do
|
||||||
|
|||||||
@@ -40,11 +40,11 @@ foreign import charsTransform :: Effect (O.Transform String String)
|
|||||||
|
|
||||||
writer :: forall m. MonadEffect m => String -> m (O.Writable Buffer /\ Consumer (Maybe Buffer) Aff Unit)
|
writer :: forall m. MonadEffect m => String -> m (O.Writable Buffer /\ Consumer (Maybe Buffer) Aff Unit)
|
||||||
writer a = do
|
writer a = do
|
||||||
stream <- liftEffect $ O.fromBufferWritable <$> FS.Stream.createWriteStream a
|
stream <- liftEffect $ O.unsafeCoerceWritable <$> FS.Stream.createWriteStream a
|
||||||
pure $ stream /\ S.fromWritable stream
|
pure $ stream /\ S.fromWritable stream
|
||||||
|
|
||||||
reader :: forall m. MonadEffect m => String -> m (Producer (Maybe Buffer) Aff Unit)
|
reader :: forall m. MonadEffect m => String -> m (Producer (Maybe Buffer) Aff Unit)
|
||||||
reader a = liftEffect $ S.fromReadable <$> O.fromBufferReadable <$> FS.Stream.createReadStream a
|
reader a = liftEffect $ S.fromReadable <$> O.unsafeCoerceReadable <$> FS.Stream.createReadStream a
|
||||||
|
|
||||||
spec :: Spec Unit
|
spec :: Spec Unit
|
||||||
spec =
|
spec =
|
||||||
@@ -64,7 +64,7 @@ spec =
|
|||||||
describe "Writable" $ around tmpFile do
|
describe "Writable" $ around tmpFile do
|
||||||
describe "fs.WriteStream" do
|
describe "fs.WriteStream" do
|
||||||
it "pipe to file" \p -> do
|
it "pipe to file" \p -> do
|
||||||
stream <- O.fromBufferWritable <$> liftEffect (FS.Stream.createWriteStream p)
|
stream <- O.unsafeCoerceWritable <$> liftEffect (FS.Stream.createWriteStream p)
|
||||||
let
|
let
|
||||||
w = S.fromWritable stream
|
w = S.fromWritable stream
|
||||||
source = do
|
source = do
|
||||||
@@ -75,7 +75,7 @@ spec =
|
|||||||
contents `shouldEqual` "hello"
|
contents `shouldEqual` "hello"
|
||||||
shouldEqual true =<< liftEffect (O.isWritableEnded stream)
|
shouldEqual true =<< liftEffect (O.isWritableEnded stream)
|
||||||
it "async pipe to file" \p -> do
|
it "async pipe to file" \p -> do
|
||||||
w <- S.fromWritable <$> O.fromBufferWritable <$> liftEffect (FS.Stream.createWriteStream p)
|
w <- S.fromWritable <$> O.unsafeCoerceWritable <$> liftEffect (FS.Stream.createWriteStream p)
|
||||||
let
|
let
|
||||||
source = do
|
source = do
|
||||||
yield "hello, "
|
yield "hello, "
|
||||||
@@ -110,7 +110,7 @@ spec =
|
|||||||
let
|
let
|
||||||
json = yield $ writeJSON { foo: "bar" }
|
json = yield $ writeJSON { foo: "bar" }
|
||||||
exp = "1f8b0800000000000003ab564acbcf57b2524a4a2c52aa0500eff52bfe0d000000"
|
exp = "1f8b0800000000000003ab564acbcf57b2524a4a2c52aa0500eff52bfe0d000000"
|
||||||
gzip <- S.fromTransform <$> O.fromBufferTransform <$> liftEffect (Zlib.toDuplex <$> Zlib.createGzip)
|
gzip <- S.fromTransform <$> O.unsafeCoerceTransform <$> liftEffect (Zlib.toDuplex <$> Zlib.createGzip)
|
||||||
outs :: List.List String <- Pipes.toListM (S.withEOS (json >-> Pipes.Buffer.fromString UTF8) >-> gzip >-> S.unEOS >-> Pipes.Buffer.toString Hex)
|
outs :: List.List String <- Pipes.toListM (S.withEOS (json >-> Pipes.Buffer.fromString UTF8) >-> gzip >-> S.unEOS >-> Pipes.Buffer.toString Hex)
|
||||||
fold outs `shouldEqual` exp
|
fold outs `shouldEqual` exp
|
||||||
around tmpFiles
|
around tmpFiles
|
||||||
@@ -118,11 +118,11 @@ spec =
|
|||||||
liftEffect $ FS.writeTextFile UTF8 a $ writeJSON [ 1, 2, 3, 4 ]
|
liftEffect $ FS.writeTextFile UTF8 a $ writeJSON [ 1, 2, 3, 4 ]
|
||||||
areader <- liftEffect $ reader a
|
areader <- liftEffect $ reader a
|
||||||
bwritestream /\ bwriter <- liftEffect $ writer b
|
bwritestream /\ bwriter <- liftEffect $ writer b
|
||||||
gzip <- S.fromTransform <$> O.fromBufferTransform <$> liftEffect (Zlib.toDuplex <$> Zlib.createGzip)
|
gzip <- S.fromTransform <$> O.unsafeCoerceTransform <$> liftEffect (Zlib.toDuplex <$> Zlib.createGzip)
|
||||||
runEffect $ areader >-> gzip >-> bwriter
|
runEffect $ areader >-> gzip >-> bwriter
|
||||||
shouldEqual true =<< liftEffect (O.isWritableEnded bwritestream)
|
shouldEqual true =<< liftEffect (O.isWritableEnded bwritestream)
|
||||||
|
|
||||||
gunzip <- S.fromTransform <$> O.fromBufferTransform <$> liftEffect (Zlib.toDuplex <$> Zlib.createGunzip)
|
gunzip <- S.fromTransform <$> O.unsafeCoerceTransform <$> liftEffect (Zlib.toDuplex <$> Zlib.createGunzip)
|
||||||
breader <- liftEffect $ reader b
|
breader <- liftEffect $ reader b
|
||||||
nums <- Pipes.toListM (breader >-> gunzip >-> S.unEOS >-> Pipes.Buffer.toString UTF8 >-> jsonParse @(Array Int) >-> Pipes.mapFoldable identity)
|
nums <- Pipes.toListM (breader >-> gunzip >-> S.unEOS >-> Pipes.Buffer.toString UTF8 >-> jsonParse @(Array Int) >-> Pipes.mapFoldable identity)
|
||||||
Array.fromFoldable nums `shouldEqual` [ 1, 2, 3, 4 ]
|
Array.fromFoldable nums `shouldEqual` [ 1, 2, 3, 4 ]
|
||||||
|
|||||||
Reference in New Issue
Block a user