lib/rcstring/replace.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 <lib/ac/string.h>

#include <lib/rcstring.h>
#include <lib/rcstring/accumulator.h>


rcstring
rcstring::replace(const rcstring &from, const rcstring &to, size_t ntimes)
    const
{
    if (from.empty() || ntimes == 0)
        return rcstring(c_str(), size());
    rcstring_accumulator acc;
    const char *cp = c_str();
    const char *end = cp + size();
    while (ntimes > 0)
    {
        const char *s =
            (const char *)memmem(cp, end - cp, from.c_str(), from.size());
        if (!s)
            break;
        acc.push_back(cp, s - cp);
        acc.push_back(to);
        cp = s + from.size();
        --ntimes;
    }
    acc.push_back(cp, end - cp);
    return acc.mkstr();
}


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