1 //===--------------------- StringLexer.cpp -----------------------*- C++
2 //-*-===//
3 //
4 //                     The LLVM Compiler Infrastructure
5 //
6 // This file is distributed under the University of Illinois Open Source
7 // License. See LICENSE.TXT for details.
8 //
9 //===----------------------------------------------------------------------===//
10 
11 #include "lldb/Utility/StringLexer.h"
12 
13 #include <algorithm>
14 #include <assert.h>
15 
16 using namespace lldb_utility;
17 
18 StringLexer::StringLexer(std::string s) : m_data(s), m_position(0) {}
19 
20 StringLexer::StringLexer(const StringLexer &rhs)
21     : m_data(rhs.m_data), m_position(rhs.m_position) {}
22 
23 StringLexer::Character StringLexer::Peek() { return m_data[m_position]; }
24 
25 bool StringLexer::NextIf(Character c) {
26   auto val = Peek();
27   if (val == c) {
28     Next();
29     return true;
30   }
31   return false;
32 }
33 
34 std::pair<bool, StringLexer::Character>
35 StringLexer::NextIf(std::initializer_list<Character> cs) {
36   auto val = Peek();
37   for (auto c : cs) {
38     if (val == c) {
39       Next();
40       return {true, c};
41     }
42   }
43   return {false, 0};
44 }
45 
46 bool StringLexer::AdvanceIf(const std::string &token) {
47   auto pos = m_position;
48   bool matches = true;
49   for (auto c : token) {
50     if (!NextIf(c)) {
51       matches = false;
52       break;
53     }
54   }
55   if (!matches) {
56     m_position = pos;
57     return false;
58   }
59   return true;
60 }
61 
62 StringLexer::Character StringLexer::Next() {
63   auto val = Peek();
64   Consume();
65   return val;
66 }
67 
68 bool StringLexer::HasAtLeast(Size s) {
69   return (m_data.size() - m_position) >= s;
70 }
71 
72 void StringLexer::PutBack(Size s) {
73   assert(m_position >= s);
74   m_position -= s;
75 }
76 
77 bool StringLexer::HasAny(Character c) {
78   return m_data.find(c, m_position) != std::string::npos;
79 }
80 
81 std::string StringLexer::GetUnlexed() {
82   return std::string(m_data, m_position);
83 }
84 
85 void StringLexer::Consume() { m_position++; }
86 
87 StringLexer &StringLexer::operator=(const StringLexer &rhs) {
88   if (this != &rhs) {
89     m_data = rhs.m_data;
90     m_position = rhs.m_position;
91   }
92   return *this;
93 }
94