0001    // ChannelStream.swift
0002    //
0003    // The MIT License (MIT)
0004    //
0005    // Copyright (c) 2015 Zewo
0006    //
0007    // Permission is hereby granted, free of charge, to any person obtaining a copy
0008    // of this software and associated documentation files (the "Software"), to deal
0009    // in the Software without restriction, including without limitation the rights
0010    // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
0011    // copies of the Software, and to permit persons to whom the Software is
0012    // furnished to do so, subject to the following conditions:
0013    //
0014    // The above copyright notice and this permission notice shall be included in all
0015    // copies or substantial portions of the Software.
0016    //
0017    // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
0018    // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
0019    // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
0020    // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
0021    // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
0022    // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
0023    // SOFTWARE.
0024    
0025    @_exported import Venice
0026    @_exported import Stream
0027    
0028    public final class ChannelStream: StreamType {
0029        public let metadata: [String: Any] = [:]
0030        private let channel = FallibleChannel<Data>()
0031    
0032        public var closed: Bool {
0033            return channel.closed
0034        }
0035    
0036        public func receive() throws -> Data {
0037            return try channel.receive()!
0038        }
0039    
0040        public func send(data: Data) throws {
0041            channel.send(data)
0042        }
0043    
0044        public func flush() throws {}
0045    
0046        public func close() -> Bool {
0047            return channel.close()
0048        }
0049    
0050        public init(stream: StreamType throws -> Void) {
0051            co {
0052                do {
0053                    try stream(self)
0054                    self.close()
0055                } catch {
0056                    self.channel.sendError(error)
0057                }
0058            }
0059        }
0060    }