bustub/src/include/common/util/string_util.h at 141c00a9001864335071b8c48c14a07a8b75f79b · cmu-db/bustub · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
//===----------------------------------------------------------------------===//
//
//                         BusTub
//
// string_util.h
//
// Identification: src/include/common/util/string_util.h
//
// Copyright (c) 2015-2025, Carnegie Mellon University Database Group
//
//===----------------------------------------------------------------------===//

#pragma once

#include <cstdint>
#include <string>
#include <vector>

namespace bustub {

/**
 * StringUtil provides INEFFICIENT utility functions for working with strings. They should only be used for debugging.
 */
class StringUtil {
 public:
  static auto Contains(const std::string &haystack, const std::string &needle) -> bool;

  static auto ContainsAfter(const std::string &keyword, const std::string &haystack, const std::string &needle) -> bool;

  static auto StartsWith(const std::string &str, const std::string &prefix) -> bool;

  static auto EndsWith(const std::string &str, const std::string &suffix) -> bool;

  static auto Repeat(const std::string &str, std::size_t n) -> std::string;

  static auto Split(const std::string &str, char delimiter) -> std::vector<std::string>;

  static auto Join(const std::vector<std::string> &input, const std::string &separator) -> std::string;

  /** @return join multiple items of container with given size, transformed to string
  using function, into one string using the given separator. */
  template <typename C, typename S, typename Func>
  static auto Join(const C &input, S count, const std::string &separator, Func f) -> std::string {
    // The result
    std::string result;

    // If the input isn't empty, append the first element. We do this so we
    // don't need to introduce an if into the loop.
    if (count > 0) {
      result += f(input[0]);
    }

    // Append the remaining input components, after the first
    for (size_t i = 1; i < count; i++) {
      result += separator + f(input[i]);
    }

    return result;
  }

  static auto Prefix(const std::string &str, const std::string &prefix) -> std::string;

  static auto FormatSize(uint64_t bytes) -> std::string;

  static auto Bold(const std::string &str) -> std::string;

  static auto Upper(const std::string &str) -> std::string;

  static auto Lower(const std::string &str) -> std::string;

  static auto Format(std::string fmt_str, ...) -> std::string;

  static auto Split(const std::string &input, const std::string &split) -> std::vector<std::string>;

  static auto Count(const std::string &input, const std::string &str) -> size_t;

  static void RTrim(std::string *str);

  static void LTrim(std::string *str);

  static auto Indent(int num_indent) -> std::string;

  static auto Strip(const std::string &str, char c) -> std::string;

  static auto Replace(std::string source, const std::string &from, const std::string &to) -> std::string;

  static auto IndentAllLines(const std::string &lines, size_t num_indent, bool except_first_line = false)
      -> std::string;
};

}  // namespace bustub