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

#include "chromeos/printing/usb_printer_id.h"

#include <algorithm>
#include <utility>

#include "base/stl_util.h"
#include "base/strings/string_split.h"

namespace chromeos {

// Device ID keys pulled from IEEE Standard 1284.
const char kManufacturer[] = "MANUFACTURER";
const char kManufacturerAbbr[] = "MFG";
const char kModel[] = "MODEL";
const char kModelAbbr[] = "MDL";
const char kCommandSet[] = "COMMAND SET";
const char kCommandSetAbbr[] = "CMD";

UsbPrinterId::UsbPrinterId(const std::vector<uint8_t>& device_id_data) {
  // Build mapping.
  id_mappings_ = BuildDeviceIdMapping(device_id_data);

  // Save required mappings.
  // Save make_.
  if (base::Contains(id_mappings_, kManufacturer)) {
    make_ = id_mappings_[kManufacturer].front();
  } else if (base::Contains(id_mappings_, kManufacturerAbbr)) {
    make_ = id_mappings_[kManufacturerAbbr].front();
  }

  // Save model_.
  if (base::Contains(id_mappings_, kModel)) {
    model_ = id_mappings_[kModel].front();
  } else if (base::Contains(id_mappings_, kModelAbbr)) {
    model_ = id_mappings_[kModelAbbr].front();
  }

  // Save command_set_.
  if (base::Contains(id_mappings_, kCommandSet)) {
    command_set_ = id_mappings_[kCommandSet];
  } else if (base::Contains(id_mappings_, kCommandSetAbbr)) {
    command_set_ = id_mappings_[kCommandSetAbbr];
  }
}

UsbPrinterId::UsbPrinterId() = default;
UsbPrinterId::UsbPrinterId(const UsbPrinterId& other) = default;
UsbPrinterId::~UsbPrinterId() = default;

std::map<std::string, std::vector<std::string>> BuildDeviceIdMapping(
    const std::vector<uint8_t>& data) {
  // Must contain at least the length information.
  if (data.size() < 2) {
    return {};
  }

  std::map<std::string, std::vector<std::string>> ret;

  // Convert to string to work on.
  // Note: First two bytes contain the length, so we skip those.
  std::string printer_id;
  std::copy(data.begin() + 2, data.end(), std::back_inserter(printer_id));

  // We filter out terms with empty keys or values.
  base::StringPairs terms;
  base::SplitStringIntoKeyValuePairs(printer_id, ':', ';', &terms);
  for (const auto& term : terms) {
    if (term.first.empty()) {
      continue;
    }

    std::vector<std::string> values = base::SplitString(
        term.second, ",", base::TRIM_WHITESPACE, base::SPLIT_WANT_NONEMPTY);
    if (values.empty()) {
      continue;
    }

    ret[term.first] = values;
  }

  return ret;
}

}  // namespace chromeos