/* Copyright 2017 - 2026 R. Thomas
 * Copyright 2017 - 2026 Quarkslab
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#include <sstream>

#include "LIEF/utils.hpp"
#include "LIEF/Visitor.hpp"

#include "LIEF/PE/resources/ResourceVarFileInfo.hpp"
#include "LIEF/PE/resources/ResourceVar.hpp"
#include "LIEF/BinaryStream/BinaryStream.hpp"

#include "internal_utils.hpp"
#include "logging.hpp"


namespace LIEF::PE {

result<ResourceVarFileInfo> ResourceVarFileInfo::parse(BinaryStream& stream) {
  ResourceVarFileInfo info;
  auto wLength = stream.read<uint16_t>();
  if (!wLength) {
    return make_error_code(wLength.error());
  }

  auto wValueLength = stream.read<uint16_t>();
  if (!wValueLength) {
    return make_error_code(wValueLength.error());
  }

  auto wType = stream.read<uint16_t>();
  if (!wType) {
    return make_error_code(wType.error());
  }

  if (*wType != 0 && wType != 1) {
    return make_error_code(lief_errors::corrupted);
  }

  auto szKey = stream.read_u16string();
  if (!szKey) {
    return make_error_code(wType.error());
  }

  if (u16tou8(*szKey) != "VarFileInfo") {
    return make_error_code(lief_errors::corrupted);
  }

  info.type(*wType).key(std::move(*szKey));


  while (stream) {
    stream.align(sizeof(uint32_t));
    auto var = ResourceVar::parse(stream);
    if (!var) {
      LIEF_WARN("Failed to parse resource var #{}", info.vars_.size());
      return info;
    }
    info.add_var(std::move(*var));
  }

  return info;
}

void ResourceVarFileInfo::accept(Visitor& visitor) const {
  visitor.visit(*this);
}


std::string ResourceVarFileInfo::key_u8() const {
  return u16tou8(key());
}

std::ostream& operator<<(std::ostream& os, const ResourceVarFileInfo& info) {
  os << fmt::format("BLOCK '{}' {{\n", info.key_u8());
  for (const ResourceVar& var : info.vars()) {
    std::ostringstream oss;
    oss << var;
    os << indent(oss.str(), 2);
  }
  os << '}';
  return os;
}


}