root/util.cc

/*  This file is part of Alue, the multiprotocol Internet discussion daemon

    Copyright © 2009, 2010 Antti-Juhani Kaijanaho

    Alue 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 3 of the License, or
    (at your option) any later version.

    Alue 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 Alue.  If not, see <http://www.gnu.org/licenses/>.

 */

#include "util.hh"

#include <boost/date_time/posix_time/posix_time.hpp>
#include <boost/asio/streambuf.hpp>

namespace util
{
        std::string split(std::string &s, const char *sep)
        {
                if (s.length() == 0) return s;
                std::string::size_type k1 = s.find_first_not_of(sep);
                if (k1 == std::string::npos) { s = ""; return ""; }
                std::string::size_type k2 = s.find_first_of(sep, k1);
                std::string rv = s.substr(k1, k2-k1);
                s.erase(0, k2);
                return rv;
        }

        bool is_number(std::string s)
        {
                if (s.empty()) return false;
                for (size_t i = 0; i < s.length(); i++) {
                        if (!is_digit(s[i])) return false;
                }
                return true;
        }

        std::string split1R(std::string &s, const char *delim)
        {
                size_t dl = s.find_first_of(delim);
                std::string rv = s.substr(0, dl);
                if (dl+1 == 0) s.clear(); else s.erase(0, dl+1);
                return rv;
        }

        std::string split1L(std::string &s, const char *delim)
        {
                size_t dl = s.find_first_of(delim);
                if (dl == std::string::npos) return "";
                std::string rv = s.substr(0, dl);
                s.erase(0, dl+1);
                return rv;
        }

        std::string current_imf_date()
        {
                using namespace boost::posix_time;
                std::ostringstream ss;
                time_facet *tf = new time_facet
                        ("%a, %d %b %Y %H:%M:%S -0000");
                ss.imbue(std::locale(std::locale::classic(), tf));
                ss << second_clock::universal_time();
                return ss.str();
        }

        std::string streambuf_to_string(boost::asio::streambuf &sbuf, size_t n)
        {
                std::string recv;
                recv.reserve(n);

                /* Note: boost.asio 1.35's commit is buggy, it doesn't
                   update the steambuf state, so we have to do this
                   the hard way. */
                sbuf.commit(n);
                typedef boost::asio::streambuf::const_buffers_type
                        const_buffers_type;
                const_buffers_type bufs = sbuf.data();
                for (const_buffers_type::const_iterator it = bufs.begin();
                     it != bufs.end(); it++)
                {
                        using  boost::asio::buffer_cast;
                        using  boost::asio::buffer_size;
                        recv += std::string(buffer_cast<const char *>(*it),
                                            buffer_size(*it));
                }
                sbuf.consume(recv.length());
                return recv;
        }
}