/*
 * Copyright 2020, Shubham Bhagat, shubhambhagat111@yahoo.com
 * All rights reserved. Distributed under the terms of the MIT License.
 */


#include "Directory.h"


DirectoryIterator::DirectoryIterator(Inode* inode)
{
	fInode = inode;
	fShortDir = NULL;
}


DirectoryIterator::~DirectoryIterator()
{
	delete fShortDir;
}


status_t
DirectoryIterator::Init()
{
	if (fInode->Format() == XFS_DINODE_FMT_LOCAL)
	{
		fShortDir = new(std::nothrow) ShortDirectory(fInode);
		if (fShortDir == NULL)
			return B_NO_MEMORY;
		return B_OK;
	}
	if (fInode->Format() == XFS_DINODE_FMT_EXTENTS) {
		// TODO: Only working with Block directories, not leaf.
		fExtentDir = new(std::nothrow) Extent(fInode);
		if (fExtentDir == NULL)
			return B_NO_MEMORY;
		status_t status = fExtentDir->Init();
		return status;
	}

	/* Return B_OK so even if the shortform directory has an extent directory
	 * we can atleast still list the shortform directory
	 */

	//TODO: Reading from extent based directories
	if (fInode->Format() == XFS_DINODE_FMT_EXTENTS) {
		TRACE("Iterator:GetNext: EXTENTS");
		return B_OK;
	}

	//TODO: Reading from B+Trees based directories
	if (fInode->Format() == XFS_DINODE_FMT_BTREE) {
		TRACE("Iterator:GetNext: B+TREE");
		return B_OK;
	}

	return B_BAD_VALUE;
}


status_t
DirectoryIterator::GetNext(char* name, size_t* length, xfs_ino_t* ino)
{
	if (fInode->Format() == XFS_DINODE_FMT_LOCAL) {
		status_t status = fShortDir->GetNext(name, length, ino);
		return status;
	}

	//TODO: Reading from extent based directories
	if (fInode->Format() == XFS_DINODE_FMT_EXTENTS) {
		TRACE("Iterator:GetNext: EXTENTS");
		return B_NOT_SUPPORTED;
	}

	//TODO: Reading from B+Trees based directories
	if (fInode->Format() == XFS_DINODE_FMT_BTREE) {
		TRACE("Iterator:GetNext: B+TREE");
		return B_NOT_SUPPORTED;
	}

	// Only reaches here if Inode is a device or is corrupt.
	return B_BAD_VALUE;
}


status_t
DirectoryIterator::Lookup(const char* name, size_t length, xfs_ino_t* ino)
{
	if (fInode->Format() == XFS_DINODE_FMT_LOCAL) {
		status_t status = fShortDir->Lookup(name, length, ino);
		return status;
	}

	//TODO: Reading from extent based dirs
	if (fInode->Format() == XFS_DINODE_FMT_EXTENTS) {
		TRACE("Iterator:Lookup: EXTENTS");
		#if 0
			status_t status = fExtentDir->Lookup(name, length, ino);
			return status;
		#endif
		return B_NOT_SUPPORTED;
	}

	//TODO: Reading from B+Tree based dirs
	if (fInode->Format() == XFS_DINODE_FMT_BTREE) {
		TRACE("Iterator:Lookup: B+TREE");
		return B_NOT_SUPPORTED;
	}

	return B_BAD_VALUE;
}