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

#include "components/optimization_guide/core/model_execution/json_response_parser.h"

#include <optional>

#include "base/functional/callback.h"
#include "base/json/json_reader.h"
#include "base/types/expected.h"
#include "base/values.h"
#include "components/optimization_guide/core/model_execution/on_device_model_execution_proto_descriptors.h"
#include "components/optimization_guide/core/model_execution/response_parser.h"

namespace optimization_guide {

namespace {

ResponseParser::Result ExtractProto(
    const std::string& type_name,
    const std::optional<base::Value>& parse_result) {
  if (!parse_result.has_value()) {
    return base::unexpected(ResponseParsingError::kFailed);
  }
  auto result = ConvertToAnyWrappedProto(*parse_result, type_name);
  if (!result) {
    return base::unexpected(ResponseParsingError::kFailed);
  }
  return *result;
}

}  // namespace

JsonResponseParser::JsonResponseParser(std::string_view proto_type)
    : proto_type_(proto_type) {}

void JsonResponseParser::ParseAsync(const std::string& redacted_output,
                                    ResultCallback callback) const {
  std::move(callback).Run(ExtractProto(
      proto_type_,
      base::JSONReader::Read(redacted_output, base::JSON_PARSE_RFC)));
}

bool JsonResponseParser::SuppressParsingIncompleteResponse() const {
  // Json parser can only parse complete responses.
  return true;
}

}  // namespace optimization_guide