haiku/src/apps/diskusage/Snapshot.cpp

/*
 * Copyright (c) 2008 Stephan Aßmus <superstippi@gmx.de>. All rights reserved.
 * Distributed under the terms of the MIT/X11 license.
 *
 * Copyright (c) 1999 Mike Steed. You are free to use and distribute this software
 * as long as it is accompanied by it's documentation and this copyright notice.
 * The software comes with no warranty, etc.
 */


#include "Snapshot.h"

#include <stdio.h>
#include <string.h>

#include <Directory.h>
#include <Mime.h>
#include <NodeInfo.h>
#include <Path.h>
#include <Volume.h>


static const char* kFileType = B_FILE_MIME_TYPE;
static const char* kDirType = "application/x-vnd.Be-directory";
static const char* kVolumeType = "application/x-vnd.Be-volume";


FileInfo::FileInfo()
	:
	pseudo(false),
	size(0),
	count(0),
	parent(NULL),
	children()
{
}


FileInfo::~FileInfo()
{
	while (children.size() != 0) {
		FileInfo* child = *children.begin();
		delete child;
		children.erase(children.begin());
	}
}


void
FileInfo::GetPath(string& path) const
{
	if (pseudo) {
		path.assign(ref.name);
	} else {
		BEntry entry(&ref, true);
		BPath pathObj(&entry);
		path.assign(pathObj.Path());
	}
}


FileInfo*
FileInfo::FindChild(const char* name) const
{
	vector<FileInfo*>::const_iterator i = children.begin();
	while (i != children.end()) {
		if (strcmp((*i)->ref.name, name) == 0)
			return *i;
		i++;
	}

	return NULL;
}


BMimeType*
FileInfo::Type() const
{
	char mimeStr[B_MIME_TYPE_LENGTH] = { '\0' };
	if (parent == NULL) {
		// This is the volume's root directory; treat it as a volume type.
		strlcpy(mimeStr, kVolumeType, sizeof(mimeStr));
	} else {
		// Get the MIME type from the registrar.
		BNode node(&ref);
		if (node.InitCheck() == B_OK) {
			BNodeInfo nodeInfo(&node);
			if (nodeInfo.InitCheck() == B_OK) {
				status_t s = nodeInfo.GetType(mimeStr);
				if (s != B_OK && children.size() > 0) {
					if (s == B_ENTRY_NOT_FOUND) {
						// This status appears to be returned only for files on
						// BFS volumes (e.g., CDFS volumes return B_BAD_VALUE).
						//nodeInfo.SetType(kDirType);
					}
					strlcpy(mimeStr, kDirType, sizeof(mimeStr));
				}
			}
		}
	}

	if (strlen(mimeStr) == 0)
		strlcpy(mimeStr, kFileType, sizeof(mimeStr));

	return new BMimeType(mimeStr);
}


// #pragma mark -


VolumeSnapshot::VolumeSnapshot(const BVolume* volume)
{
	char nameBuffer[B_FILE_NAME_LENGTH];
	volume->GetName(nameBuffer);
	name = nameBuffer;

	capacity = volume->Capacity();
	freeBytes = volume->FreeBytes();
	rootDir = NULL;
	freeSpace = NULL;
}


VolumeSnapshot::~VolumeSnapshot()
{
	delete rootDir;
	delete freeSpace;
}