prebuffering of output is controllable now.

1 files changed, 25 insertions, 4 deletions

diff --git a/include/sitecing/sitecing_interface_cgi.h b/include/sitecing/sitecing_interface_cgi.h
index cab947c..da538d6 100644
--- a/include/sitecing/sitecing_interface_cgi.h
+++ b/include/sitecing/sitecing_interface_cgi.h
@@ -36,7 +36,15 @@ namespace sitecing {
 	    /**
 	     * Here is where we prebuffer output.
 	     */
-	    ostringstream prebuffer;
+	    stringbuf prebuffer;
+	    /**
+	     * Output stream, initially going to prebuffer.
+	     */
+	    ostream outs;
+	    /**
+	     * Have headers been sent yet?
+	     */
+	    bool headers_flushed;
 	    /**
 	     * Pointer to the sitespace object.
 	     */
@@ -48,13 +56,26 @@ namespace sitecing {
 	    sitecing_interface_cgi(sitespace *s);
 
 	    /**
-	     * @todo TODO: wish I could remember -- document me.
+	     * Set up interface for the (possibly, new) cgi gateway object,
+	     * reset headers, empty buffer, etc.
 	     */
 	    void prepare(kingate::cgi_gateway *cg);
 	    /**
-	     * @todo TODO: wish I could remember -- document me.
+	     * Flush output stream.
+	     */
+	    void flush(bool keep_buffering=false);
+	    /**
+	     * Send headers to the output stream, if we haven't yet.
+	     */
+	    void flush_headers();
+	    /**
+	     * Are we buffering now?
+	     */
+	    bool is_buffering();
+	    /**
+	     * Control output buffering.
 	     */
-	    void flush();
+	    void set_buffering(bool do_buffer);
 
     };
 }