gistfile1.scala

/*
 * Quick sketch of wrapper to aid in raw stream processing.
 * Source is nice, but it blocks in some cases (ie, client
 * socket input streams). While loops suck, so ....
 * 
 * import MIO._
 * val in: InputStream = ....
 * consume(in) until("\r\n\r\n")
 *
 * chris@thegodcode.net
 */

import java.io._

object StreamConsumer {
  
  def read(in: InputStream) = {
    val buf = new scala.collection.mutable.ArrayBuffer[Byte]
    var b = 0
    while({b = in.read; b} != -1) {
      buf += b.toByte
    }
    buf.toArray
  }
  
  def read(in: InputStream, byteSeq: Seq[Byte]) = {
    val buf = new scala.collection.mutable.ArrayBuffer[Byte]
    
    var stillSearching = true
    var b = 0
    while({b = in.read; b} != -1 && stillSearching) {
      buf += b.toByte
      stillSearching = ! (buf.size >= byteSeq.size && byteSeq.equalsWith(buf.drop(buf.size - byteSeq.size).toArray) { _ == _ })
    }
    
    buf.toArray
  }
  
}

class StreamConsumer(in: InputStream) {
  
  def fully = StreamConsumer.read(in)
  
  def until(byteSeq: Seq[Byte]) = StreamConsumer.read(in, byteSeq)
  
  def until(seq: String): String = new String(until(seq.getBytes).toArray)
  
}

object MIO {
  def consume(in: InputStream) = new StreamConsumer(in)
}