nis-util-afpvol-gen/main.cc

Go to the documentation of this file.

//
// nis-util - NIS Administration Utilities
// Copyright (C) 2012 Peter Miller
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or (at
// your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// General Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with this program. If not, see <http://www.gnu.org/licenses/>.
//

#include <libexplain/output.h>

#include <lib/space/automap.h>
#include <lib/space/hosts/slurp.h>
#include <lib/space/automap/functor/afpvol_generator.h>
#include <lib/space/automap/functor/direct.h>
#include <lib/space/automap/functor/indirect.h>

#include <nis-util-afpvol-gen/arglex/afpvol.h>


static void
both_whine(void)
{
    explain_output_error_and_die
    (
        "you must use exactly one -Director or -Indirect option"
    );
}


int
main(int argc, char **argv)
{
    arglex_afpvol cmdline(argc, argv);
    cmdline.token_next();
    space_automap::pointer stp;
    rcstring outfile;
    space_hosts_slurp::pointer hosts;
    while (cmdline.token_cur() != arglex::token_eoln)
    {
        switch (cmdline.token_cur())
        {
        default:
            cmdline.generic_argument();
            break;

        case arglex_afpvol::token_direct:
            {
                rcstring filename = cmdline.get_filename_or_stdin();
                if (stp)
                    both_whine();
                stp = space_automap::create(filename);
                space_automap_functor_direct checker(stp.get());
                stp->read_and_process(checker, hosts.get());
            }
            break;

        case arglex_afpvol::token_indirect:
            {
                rcstring filename = cmdline.get_filename_or_stdin();
                if (stp)
                    both_whine();
                stp = space_automap::create(cmdline.get_string_value());
                space_automap_functor_indirect checker(stp.get());
                stp->read_and_process(checker, hosts.get());
            }
            break;

        case arglex_afpvol::token_output:
            if (!outfile.empty())
                explain_output_error_and_die("too many output files specified");
            outfile = cmdline.get_filename_or_stdout();
            break;

        case arglex_afpvol::token_debug:
            space_automap::grammar_debug();
            break;

        case arglex_afpvol::token_hosts:
            if (hosts)
                explain_output_error_and_die("too many --hosts maps specified");
            hosts = space_hosts_slurp::create(cmdline.get_filename_or_stdin());
            hosts->read_and_process();
            break;
        }
        cmdline.token_next();
    }
    if (!stp)
        both_whine();
    space_automap_functor_afpvol_generator generator(outfile);
    stp->walk(generator);
    return 0;
}


// vim: set ts=8 sw=4 et :