ncmpcpp/src/utility/string.cpp

/***************************************************************************
 *   Copyright (C) 2008-2012 by Andrzej Rybczak                            *
 *   electricityispower@gmail.com                                          *
 *                                                                         *
 *   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, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.              *
 ***************************************************************************/

#include <cassert>
#include <cwctype>
#include <algorithm>
#include "utility/string.h"

int stringToInt(const std::string &s)
{
	return atoi(s.c_str());
}

long stringToLongInt(const std::string &s)
{
	return atol(s.c_str());
}

bool isInteger(const char *s)
{
	assert(s);
	if (*s == '\0')
		return false;
	for (const char *it = s; *it != '\0'; ++it)
		if (!isdigit(*it) && (it != s || *it != '-'))
			return false;
		return true;
}

std::string ToString(const std::wstring &ws)
{
	std::string result;
	char s[MB_CUR_MAX];
	for (size_t i = 0; i < ws.length(); ++i)
	{
		int n = wcrtomb(s, ws[i], 0);
		if (n > 0)
			result.append(s, n);
	}
	return result;
}

std::wstring ToWString(const std::string &s)
{
	std::wstring result;
	wchar_t *ws = new wchar_t[s.length()];
	const char *c_s = s.c_str();
	int n = mbsrtowcs(ws, &c_s, s.length(), 0);
	if (n > 0)
		result.append(ws, n);
	delete [] ws;
	return result;
}

std::vector<std::string> split(const std::string &s, const std::string &delimiter)
{
	if (delimiter.empty())
		return { s };
	std::vector<std::string> result;
	size_t i = 0, j = 0;
	while (true)
	{
		i = j;
		j = s.find(delimiter, i);
		if (j == std::string::npos)
			break;
		else
			result.push_back(s.substr(i, j-i));
		j += delimiter.length();
	}
	result.push_back(s.substr(i));
	return result;
}

void replace(std::string &s, const std::string &from, const std::string &to)
{
	for (size_t i = 0; (i = s.find(from, i)) != std::string::npos; i += to.length())
		s.replace(i, from.length(), to);
}

void lowercase(std::string &s)
{
	std::transform(s.begin(), s.end(), s.begin(), tolower);
}

void lowercase(std::wstring &ws)
{
	std::transform(ws.begin(), ws.end(), ws.begin(), towlower);
}

void trim(std::string &s)
{
	if (s.empty())
		return;
	
	size_t b = 0;
	size_t e = s.length()-1;
	
	while (s[e] == ' ' || s[e] == '\n' || s[e] == '\t')
		--e;
	++e;
	if (e != s.length())
		s.resize(e);
	
	while (s[b] == ' ' || s[b] == '\n' || s[e] == '\t')
		++b;
	if (b != 0)
		s = s.substr(b);
}

std::string getBasename(const std::string &path)
{
	size_t slash = path.rfind("/");
	if (slash == std::string::npos)
		return path;
	else
		return path.substr(slash+1);
}

std::string getParentDirectory(const std::string &path)
{
	size_t slash = path.rfind('/');
	if (slash == std::string::npos)
		return "/";
	else
		return path.substr(0, slash);
}

std::string getSharedDirectory(const std::string &dir1, const std::string &dir2)
{
	size_t i = 0;
	size_t min_len = std::min(dir1.length(), dir2.length());
	while (i < min_len && !dir1.compare(i, 1, dir2, i, 1))
		++i;
	i = dir1.rfind("/", i);
	if (i == std::string::npos)
		return "/";
	else
		return dir1.substr(0, i);
}

std::string getEnclosedString(const std::string &s, char a, char b, size_t *pos)
{
	std::string result;
	size_t i = s.find(a, pos ? *pos : 0);
	if (i != std::string::npos)
	{
		++i;
		while (i < s.length() && s[i] != b)
		{
			if (s[i] == '\\' && i+1 < s.length() && s[i+1] == b)
				result += s[++i];
			else
				result += s[i];
			++i;
		}
		// we want to set pos to char after b if possible
		if (i < s.length())
			++i;
	}
	if (pos != 0)
		*pos = i;
	return result;
}

void removeInvalidCharsFromFilename(std::string &filename)
{
	const char *unallowed_chars = "\"*/:<>?\\|";
	for (const char *c = unallowed_chars; *c; ++c)
	{
		for (size_t i = 0; i < filename.length(); ++i)
		{
			if (filename[i] == *c)
			{
				filename.erase(filename.begin()+i);
				--i;
			}
		}
	}
}