runtime_probe/functions/edid.cc

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

#include "runtime_probe/functions/edid.h"

#include <pcrecpp.h>

#include <utility>

#include <base/files/file_util.h>
#include <base/logging.h>
#include <base/strings/stringprintf.h>
#include <base/values.h>

#include "runtime_probe/system/context.h"
#include "runtime_probe/utils/edid.h"
#include "runtime_probe/utils/file_utils.h"

namespace runtime_probe {

namespace {

base::Value::Dict ProbeEdidPath(const base::FilePath& edid_path) {
  VLOG(2) << "Processing the node \"" << edid_path.value() << "\"";

  std::string raw_bytes;
  if (!base::ReadFileToString(edid_path, &raw_bytes))
    return {};
  if (raw_bytes.length() == 0)
    return {};

  auto edid =
      Edid::From(std::vector<uint8_t>(raw_bytes.begin(), raw_bytes.end()));
  if (!edid) {
    return {};
  }
  base::Value::Dict res;
  res.Set("vendor", edid->vendor);
  res.Set("product_id", base::StringPrintf("%04x", edid->product_id));
  res.Set("width", std::to_string(edid->width));
  res.Set("height", std::to_string(edid->height));
  res.Set("path", edid_path.value());
  return res;
}

}  // namespace

EdidFunction::DataType EdidFunction::EvalImpl() const {
  DataType result{};

  for (const auto& edid_pattern : edid_patterns_) {
    const auto rooted_edid_pattern =
        Context::Get()->root_dir().Append(edid_pattern);
    for (const auto& edid_path : Glob(rooted_edid_pattern)) {
      auto node_res = ProbeEdidPath(edid_path);
      if (node_res.empty())
        continue;

      result.Append(std::move(node_res));
    }
  }

  return result;
}

}  // namespace runtime_probe