chromium/components/optimization_guide/core/model_execution/substitution.cc

// 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/substitution.h"

#include <sys/types.h>

#include <optional>
#include <sstream>
#include <string>
#include <vector>

#include "base/debug/dump_without_crashing.h"
#include "base/logging.h"
#include "base/strings/string_number_conversions.h"
#include "components/optimization_guide/core/model_execution/on_device_model_execution_proto_descriptors.h"
#include "components/optimization_guide/core/model_execution/on_device_model_execution_proto_value_utils.h"
#include "components/optimization_guide/proto/descriptors.pb.h"
#include "components/optimization_guide/proto/substitution.pb.h"

namespace optimization_guide {

namespace {

RepeatedPtrField;

// A context for resolving substitution expressions.
struct ResolutionContext { … };

// Returns whether `condition` applies based on `message`.
bool EvaluateCondition(const ResolutionContext& ctx,
                       const proto::Condition& condition) { … }

bool AndConditions(const ResolutionContext& ctx,
                   const RepeatedPtrField<proto::Condition>& conditions) { … }

bool OrConditions(const ResolutionContext& ctx,
                  const RepeatedPtrField<proto::Condition>& conditions) { … }

// Returns whether `conditions` apply based on `message`.
bool DoConditionsApply(const ResolutionContext& ctx,
                       const proto::ConditionList& conditions) { … }

// Resolve various expression in proto::SubstitutedString by appending
// appropriate text to an output string and updating state.
// Methods return false on error.
class StringBuilder { … };

StringBuilder::Error StringBuilder::ResolveProtoField(
    const ResolutionContext& ctx,
    const proto::ProtoField& field) { … }

StringBuilder::Error StringBuilder::ResolveRangeExpr(
    const ResolutionContext& ctx,
    const proto::RangeExpr& expr) { … }

StringBuilder::Error StringBuilder::ResolveIndexExpr(
    const ResolutionContext& ctx,
    const proto::IndexExpr& expr) { … }

StringBuilder::Error StringBuilder::ResolveStringArg(
    const ResolutionContext& ctx,
    const proto::StringArg& candidate) { … }

StringBuilder::Error StringBuilder::ResolveSubstitution(
    const ResolutionContext& ctx,
    const proto::StringSubstitution& arg) { … }

StringBuilder::Error StringBuilder::ResolveSubstitutedString(
    const ResolutionContext& ctx,
    const proto::SubstitutedString& substitution) { … }

}  // namespace

std::optional<SubstitutionResult> CreateSubstitutions(
    const google::protobuf::MessageLite& request,
    const google::protobuf::RepeatedPtrField<proto::SubstitutedString>&
        config_substitutions) { … }

}  // namespace optimization_guide