1 files changed, 18 insertions, 4 deletions

diff --git a/c++-libbu++/src/gatsc/main.cpp b/c++-libbu++/src/gatsc/main.cpp
index e07fe82..339e9f4 100644
--- a/c++-libbu++/src/gatsc/main.cpp
+++ b/c++-libbu++/src/gatsc/main.cpp
@@ -9,6 +9,8 @@
 #include <bu/string.h>
 #include <bu/file.h>
 #include <bu/sio.h>
+#include <bu/streamstack.h>
+#include <bu/deflate.h>
 
 #include "gats/types.h"
 #include "gats/gatsstream.h"
@@ -19,7 +21,8 @@ class Options : public OptParser
 {
 public:
     Options( int argc, char *argv[] ) :
-        bCompile( true )
+        bCompile( true ),
+        bCompress( false )
     {
         addHelpBanner("Gats Compiler\nUsage: gatsc [options] [input]\n");
 
@@ -29,11 +32,14 @@ public:
         addOption( bCompile, 'd', "decompile",
             "Convert binary gats to text gats.");
 
+        addOption( bCompress, 'z', "compress", "Compress with deflate.");
+
         addHelpOption('h', "help", "This Help");
 
         setNonOption( slot( this, &Options::setInput ) );
 
         setOverride("decompile", false );
+        setOverride("compress", true );
 
         parse( argc, argv );
     }
@@ -45,6 +51,7 @@ public:
     }
 
     bool bCompile;
+    bool bCompress;
     String sInput;
     String sOutput;
 };
@@ -62,14 +69,21 @@ int main( int argc, char *argv[] )
     if( opt.sOutput.isEmpty() )
     {
         opt.sOutput.set( opt.sInput.begin(), opt.sInput.find('.') );
-        opt.sOutput += ".gats";
+        if( opt.bCompress )
+            opt.sOutput += ".gatz";
+        else
+            opt.sOutput += ".gats";
     }
 
     if( opt.bCompile )
     {
         File fIn( opt.sInput, File::Read );
-        File fOut( opt.sOutput, File::WriteNew );
-        Gats::GatsStream gs( fOut );
+        StreamStack ssOut( new File( opt.sOutput, File::WriteNew ) );
+        if( opt.bCompress )
+        {
+            ssOut.pushFilter<Deflate>();
+        }
+        Gats::GatsStream gs( ssOut );
         Gats::Object *pObj = Gats::Object::strToGats( fIn.readAll() );
         gs.writeObject( pObj );
         delete pObj;