blob: b433e7f31036f29cd1b15608a2a5c5b27f38b924 [file] [log] [blame]
// Copyright 2019 The Pigweed Authors
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not
// use this file except in compliance with the License. You may obtain a copy of
// the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
// License for the specific language governing permissions and limitations under
// the License.
#include "pw_string/string_builder.h"
#include <cstdio>
#include "pw_string/format.h"
#include "pw_string/util.h"
namespace pw {
void StringBuilder::clear() {
size_ = 0;
NullTerminate();
status_ = Status::OK;
last_status_ = Status::OK;
}
StringBuilder& StringBuilder::append(size_t count, char ch) {
char* const append_destination = &buffer_[size_];
std::memset(append_destination, ch, ResizeAndTerminate(count));
return *this;
}
StringBuilder& StringBuilder::append(const char* str, size_t count) {
char* const append_destination = &buffer_[size_];
std::memcpy(append_destination, str, ResizeAndTerminate(count));
return *this;
}
StringBuilder& StringBuilder::append(const char* str) {
// Use buffer_.size() - size() as the maximum length so that strings too long
// to fit in the buffer will request one character too many, which sets the
// status to RESOURCE_EXHAUSTED.
return append(str, string::Length(str, buffer_.size() - size()));
}
StringBuilder& StringBuilder::append(const std::string_view& str) {
return append(str.data(), str.size());
}
StringBuilder& StringBuilder::append(const std::string_view& str,
size_t pos,
size_t count) {
if (pos > str.size()) {
SetErrorStatus(Status::OUT_OF_RANGE);
return *this;
}
return append(str.data() + pos, std::min(str.size() - pos, count));
}
size_t StringBuilder::ResizeAndTerminate(size_t chars_to_append) {
const size_t copied = std::min(chars_to_append, max_size() - size());
size_ += copied;
NullTerminate();
if (buffer_.empty() || chars_to_append != copied) {
SetErrorStatus(Status::RESOURCE_EXHAUSTED);
} else {
last_status_ = Status::OK;
}
return copied;
}
void StringBuilder::resize(size_t new_size) {
if (new_size <= size_) {
size_ = new_size;
NullTerminate();
last_status_ = Status::OK;
} else {
SetErrorStatus(Status::OUT_OF_RANGE);
}
}
StringBuilder& StringBuilder::Format(const char* format, ...) {
va_list args;
va_start(args, format);
FormatVaList(format, args);
va_end(args);
return *this;
}
StringBuilder& StringBuilder::FormatVaList(const char* format, va_list args) {
HandleStatusWithSize(
string::FormatVaList(buffer_.subspan(size_), format, args));
return *this;
}
void StringBuilder::CopySizeAndStatus(const StringBuilder& other) {
size_ = other.size_;
status_ = other.status_;
last_status_ = other.last_status_;
}
void StringBuilder::HandleStatusWithSize(StatusWithSize written) {
const Status status = written.status();
last_status_ = status;
if (!status.ok()) {
status_ = status;
}
size_ += written.size();
}
void StringBuilder::SetErrorStatus(Status status) {
last_status_ = status;
status_ = status;
}
} // namespace pw