mus.cxx

Go to the documentation of this file.

#include <cxxtls/muSED.h>
#include <cxxtls/file.h>
#include <cxxtls/foreach.h>

#include <iostream>
#include <fstream>
#include <stdlib.h>

using namespace std;
using namespace cxxtls;




string programName;

static void printHelp();

static void bomb(string const &msg)
{
   cerr << programName << " - fatal error:  " << msg << endl;
   exit(1);
}

static void apply(muSED::CompiledScript &script, list<string> const &container)
{
    list<string> output;

    muSED::apply(script, container, output);

    CXXTLS_FOREACH(string const &cur, output)
    {
       cout << cur << endl;
    }

}

static void readFile(string const &name, list<string> &container)
{
    if(name == "-")
    {
        string line;
        
        while(getline(cin, line))
        {
           container.push_back(line);
        }
    }
    else
    {
        string line;

        ifstream file(name.c_str());

        if(!file.good())
        {
           bomb(string(" reading input file ") + name );
        }
        
        while(getline(file, line))
        {
           container.push_back(line);
        }

    }
}



int main(int argc, char **argv)
{
   if(argc == 1)
   {
      printHelp();
      exit(1);
   }


   // consume options

   FileName  me(*argv++);

   programName = me.basename();

   list<string> script;  // -e and -f options get accumulated into this script

   bool addPrint = true;

   for(; *argv; ++argv)
   {
       char const *optionPtr = *argv; 

       if(optionPtr[0] != '-' || optionPtr[1] == 0 )
       {
           break;
       }

       string option(optionPtr);

       if(option == "-f")
       {
           ++argv;

           if(*argv == 0)
           {
              bomb("expected script name after -f");
           }

          ifstream scriptFragment(*argv);

          if(!scriptFragment.good())
          {
              bomb(string("reading file ") + *argv);
          }

          string line;

          while(getline(scriptFragment, line))
          {
              script.push_back(line);
          }


       }
       else
       if(option == "-e")
       {
           ++argv;

           if(*argv == 0)
           {
              bomb("expected script string after -e");
           }
           
           script.push_back(*argv);
       }
       else
       if(option == "-n")
       {
           addPrint = false;
       }
       else if(option == "-h" || option == "--help")
       {
          printHelp();
       }
       else
       {
          bomb(string("unknown option: ") + *argv);
       }
   }

   // when we get here, *argv will either be a null pointer or it will
   // point to the first input file.

   if(addPrint)
   {
       script.push_back("p");
   }


   // compile script

   muSED::CompiledScript executor(script);

   if(!executor.ok())
   {
       cerr << programName << " - fatal error in the following script" << endl;

       int line=1;

       CXXTLS_FOREACH(string const &cur, script)
       {
           cerr << "[" << line++ << "]=" << cur << endl;
       }

       cerr << endl;

       bomb(executor.error());
           
   }

   // execute the script on all incoming files

   if(*argv == 0)
   {
      // read from stdin and quit

      list<string> stdinContents;

      readFile("-", stdinContents);

      apply(executor, stdinContents);

   }
   else
   {
       while(*argv != 0)
       {
           FileName name(*argv++);

           list<string> fileContents;

           readFile(name, fileContents);

           apply(executor, fileContents);

       }
   }


    exit(0);
}

static void printHelp()
{
   char const *lines[]=
   {
      "mus - MICRO-SED",
      "",
      "Usage:",
      "    mus  [options] [files]",
      "",
      "Description:",
      "",
      "   mus is a micro-sed.  It provides a proper subset of sed's capabilities",
      "   but also provides additional commands and features that sed does not.",
      "   mus is not preferred to sed, it exists to allow scripts consistent",
      "   with the muSED implementation to be used to perform textual transformation.",
      "",
      "Options",
      "",
      "    -f  scriptName    The name of file containing a mus script.",
      "    -e  scriptText    A string defining a script.",
      "    -n                Suppresses automatic printing of every line.",
      "    -h                Print this help message",
      "    --help            Print this help message",
      "",
      "Online Documentation",
      "   See http://www.bordoon.com/tools/index.html",
      "       muSED",
      "       muSED.h",

      "",
      "",
      0
   };

   char const **scan = lines;

   while(*scan)
   {
      cout << *scan++ << endl;
   }

   exit(1);
}