libbrillo/brillo/strings/string_utils.cc

// Copyright 2014 The ChromiumOS Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <brillo/strings/string_utils.h>

#include <algorithm>
#include <string.h>
#include <utility>

#include <base/strings/string_util.h>
#include <base/strings/stringprintf.h>

namespace brillo {
namespace string_utils {

std::vector<std::string> Split(const std::string& str,
                               const std::string& delimiter,
                               bool trim_whitespaces,
                               bool purge_empty_strings) {
  std::vector<std::string> tokens;
  if (str.empty())
    return tokens;

  for (std::string::size_type i = 0;;) {
    const std::string::size_type pos =
        delimiter.empty() ? (i + 1) : str.find(delimiter, i);
    std::string tmp_str{str.substr(i, pos - i)};
    if (trim_whitespaces)
      base::TrimWhitespaceASCII(tmp_str, base::TRIM_ALL, &tmp_str);
    if (!tmp_str.empty() || !purge_empty_strings)
      tokens.emplace_back(std::move(tmp_str));
    if (pos >= str.size())
      break;
    i = pos + delimiter.size();
  }
  return tokens;
}

bool SplitAtFirst(const std::string& str,
                  const std::string& delimiter,
                  std::string* left_part,
                  std::string* right_part,
                  bool trim_whitespaces) {
  bool delimiter_found = false;
  std::string::size_type pos = str.find(delimiter);
  if (pos != std::string::npos) {
    *left_part = str.substr(0, pos);
    *right_part = str.substr(pos + delimiter.size());
    delimiter_found = true;
  } else {
    *left_part = str;
    right_part->clear();
  }

  if (trim_whitespaces) {
    base::TrimWhitespaceASCII(*left_part, base::TRIM_ALL, left_part);
    base::TrimWhitespaceASCII(*right_part, base::TRIM_ALL, right_part);
  }

  return delimiter_found;
}

std::optional<std::pair<std::string_view, std::string_view>> SplitAtFirst(
    std::string_view str,
    std::string_view delimiter,
    base::WhitespaceHandling whitespace_action) {
  std::string::size_type pos = str.find(delimiter);
  if (pos == std::string::npos) {
    return std::nullopt;
  }
  std::string_view left_part = str.substr(0, pos);
  std::string_view right_part = str.substr(pos + delimiter.size());

  if (whitespace_action == base::TRIM_WHITESPACE) {
    left_part = base::TrimWhitespaceASCII(left_part, base::TRIM_ALL);
    right_part = base::TrimWhitespaceASCII(right_part, base::TRIM_ALL);
  }

  return std::make_pair(left_part, right_part);
}

std::pair<std::string, std::string> SplitAtFirst(const std::string& str,
                                                 const std::string& delimiter,
                                                 bool trim_whitespaces) {
  std::pair<std::string, std::string> pair;
  SplitAtFirst(str, delimiter, &pair.first, &pair.second, trim_whitespaces);
  return pair;
}

std::string ToString(double value) {
  return base::StringPrintf("%g", value);
}

std::string ToString(bool value) {
  return value ? "true" : "false";
}

std::string GetBytesAsString(const std::vector<uint8_t>& buffer) {
  return std::string(buffer.begin(), buffer.end());
}

std::vector<uint8_t> GetStringAsBytes(const std::string& str) {
  return std::vector<uint8_t>(str.begin(), str.end());
}

}  // namespace string_utils
}  // namespace brillo