grpc-swift/Plugin/Sources/io.swift

// Copyright 2016 Google Inc. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

import Foundation

// The I/O code below is derived from Apple's swift-protobuf project.
// https://github.com/apple/swift-protobuf
// BEGIN swift-protobuf derivation

#if os(Linux)
  import Glibc
#else
  import Darwin.C
#endif

enum PluginError: Error {
  /// Raised for any errors reading the input
  case readFailure
}

// Alias clib's write() so Stdout.write(bytes:) can call it.
private let _write = write

class Stdin {
  static func readall() throws -> Data {
    let fd: Int32 = 0
    let buffSize = 32
    var buff = [UInt8]()
    while true {
      var fragment = [UInt8](repeating: 0, count: buffSize)
      let count = read(fd, &fragment, buffSize)
      if count < 0 {
        throw PluginError.readFailure
      }
      if count < buffSize {
        buff += fragment[0..<count]
        return Data(bytes: buff)
      }
      buff += fragment
    }
  }
}

class Stdout {
  static func write(bytes: Data) {
    bytes.withUnsafeBytes { (p: UnsafePointer<UInt8>) -> () in
      _ = _write(1, p, bytes.count)
    }
  }
}

struct CodePrinter {
  private(set) var content = ""
  private var currentIndentDepth = 0
  private var currentIndent = ""
  private var atLineStart = true

  mutating func print(_ text: String...) {
    for t in text {
      for c in t.characters {
        if c == "\n" {
          content.append(c)
          atLineStart = true
        } else {
          if atLineStart {
            content.append(currentIndent)
            atLineStart = false
          }
          content.append(c)
        }
      }
    }
  }

  mutating private func resetIndent() {
    currentIndent = (0..<currentIndentDepth).map { Int -> String in return "  " } .joined(separator:"")
  }

  mutating func indent() {
    currentIndentDepth += 1
    resetIndent()
  }
  mutating func outdent() {
    currentIndentDepth -= 1
    resetIndent()
  }
}

// END swift-protobuf derivation