shaka-packager/packager/utils/absl_flag_hexbytes.cc

// Copyright 2023 Google LLC. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd

#include "packager/utils/absl_flag_hexbytes.h"
#include "packager/utils/hex_parser.h"

#include <absl/flags/flag.h>
#include <absl/flags/parse.h>
#include <absl/flags/usage.h>
#include <absl/strings/ascii.h>
#include <absl/strings/escaping.h>
#include <absl/strings/str_format.h>
#include <absl/strings/str_join.h>
#include <absl/strings/str_split.h>
#include <iostream>
#include <vector>

namespace shaka {

// Custom flag parser for HexBytes
bool AbslParseFlag(absl::string_view text, HexBytes* flag, std::string* error) {
  std::string hexString(text);
  absl::RemoveExtraAsciiWhitespace(&hexString);

  if (hexString.empty()) {
    flag->bytes = std::vector<uint8_t>(0, 0);
    return true;
  }

  std::string bytesRaw;
  if (!shaka::ValidHexStringToBytes(hexString, &bytesRaw)) {
    *error = "Invalid hex string";
    return false;
  }

  absl::string_view hex_str_view(bytesRaw);
  absl::Span<const uint8_t> span(
      reinterpret_cast<const uint8_t*>(hex_str_view.data()),
      hex_str_view.size());
  flag->bytes = std::vector<uint8_t>(span.begin(), span.end());

  return true;
}

std::string AbslUnparseFlag(const HexBytes& flag) {
  std::string result;
  for (const auto byte : flag.bytes) {
    absl::StrAppendFormat(&result, "%02x", byte);
  }
  return result;
}

}  // namespace shaka