/**
 ******************************************************************************
 * Xenia : Xbox 360 Emulator Research Project                                 *
 ******************************************************************************
 * Copyright 2014 Ben Vanik. All rights reserved.                             *
 * Released under the BSD license - see LICENSE in the root for more details. *
 ******************************************************************************
 */

#include "xenia/vfs/devices/stfs_container_entry.h"

#include "xenia/base/math.h"
#include "xenia/vfs/devices/stfs_container_file.h"

namespace xe {
namespace vfs {

STFSContainerEntry::STFSContainerEntry(Device* device, const char* path,
                                       MappedMemory* mmap,
                                       STFSEntry* stfs_entry)
    : Entry(device, path),
      mmap_(mmap),
      stfs_entry_(stfs_entry),
      it_(stfs_entry_->children.begin()) {}

STFSContainerEntry::~STFSContainerEntry() = default;

X_STATUS STFSContainerEntry::QueryInfo(
    X_FILE_NETWORK_OPEN_INFORMATION* out_info) {
  assert_not_null(out_info);
  out_info->creation_time = stfs_entry_->update_timestamp;
  out_info->last_access_time = stfs_entry_->access_timestamp;
  out_info->last_write_time = stfs_entry_->update_timestamp;
  out_info->change_time = stfs_entry_->update_timestamp;
  out_info->allocation_size = xe::round_up(stfs_entry_->size, 4096);
  out_info->end_of_file = stfs_entry_->size;
  out_info->attributes = stfs_entry_->attributes;
  return X_STATUS_SUCCESS;
}

X_STATUS STFSContainerEntry::QueryDirectory(
    X_FILE_DIRECTORY_INFORMATION* out_info, size_t length,
    const char* file_name, bool restart) {
  assert_not_null(out_info);

  STFSEntry* entry(nullptr);

  if (file_name != nullptr) {
    // Only queries in the current directory are supported for now
    assert_true(std::strchr(file_name, '\\') == nullptr);

    find_engine_.SetRule(file_name);

    // Always restart the search?
    it_ = stfs_entry_->children.begin();
    entry = stfs_entry_->GetChild(find_engine_, it_);
    if (!entry) {
      return X_STATUS_NO_SUCH_FILE;
    }
  } else {
    if (restart) {
      it_ = stfs_entry_->children.begin();
    }

    entry = stfs_entry_->GetChild(find_engine_, it_);
    if (!entry) {
      return X_STATUS_NO_SUCH_FILE;
    }

    auto end = (uint8_t*)out_info + length;
    auto entry_name = entry->name;
    if (((uint8_t*)&out_info->file_name[0]) + entry_name.size() > end) {
      return X_STATUS_NO_SUCH_FILE;
    }
  }

  out_info->file_index = 0xCDCDCDCD;
  out_info->creation_time = entry->update_timestamp;
  out_info->last_access_time = entry->access_timestamp;
  out_info->last_write_time = entry->update_timestamp;
  out_info->change_time = entry->update_timestamp;
  out_info->end_of_file = entry->size;
  out_info->allocation_size = xe::round_up(entry->size, 4096);
  out_info->attributes = entry->attributes;
  out_info->file_name_length = static_cast<uint32_t>(entry->name.size());
  memcpy(out_info->file_name, entry->name.c_str(), entry->name.size());

  return X_STATUS_SUCCESS;
}

X_STATUS STFSContainerEntry::Open(KernelState* kernel_state, Mode mode,
                                  bool async, XFile** out_file) {
  *out_file = new STFSContainerFile(kernel_state, mode, this);
  return X_STATUS_SUCCESS;
}

}  // namespace vfs
}  // namespace xe