/*
 * Copyright 2008-2009 Haiku Inc. All rights reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *		Pieter Panman
 */


#include "DeviceUSB.h"

#include <sstream>
#include <stdlib.h>

#include <Catalog.h>
#include <bus/USB.h>

#undef B_TRANSLATION_CONTEXT
#define B_TRANSLATION_CONTEXT "DeviceUSB"

extern "C" {
#include "dm_wrapper.h"
#include "usb-utils.h"
}


DeviceUSB::DeviceUSB(Device* parent)
	:
	Device(parent),
	fVendorId(0),
	fDeviceId(0)
{
}


DeviceUSB::~DeviceUSB()
{
}


static BString ToHex(uint16 num)
{
	std::stringstream ss;
	ss.flags(std::ios::hex | std::ios::showbase);
	ss << num;
	return BString(ss.str().c_str());
}


void
DeviceUSB::InitFromAttributes()
{
	// Process the attributes
	fClassBaseId = atoi(fAttributeMap[USB_DEVICE_CLASS].String());
	fClassSubId = atoi(fAttributeMap[USB_DEVICE_SUBCLASS].String());
	fClassProtoId = atoi(fAttributeMap[USB_DEVICE_PROTOCOL].String());
	fVendorId = atoi(fAttributeMap[B_DEVICE_VENDOR_ID].String());
	fDeviceId = atoi(fAttributeMap[B_DEVICE_ID].String());

	// Looks better in Hex, so rewrite
	fAttributeMap[USB_DEVICE_CLASS] = ToHex(fClassBaseId);
	fAttributeMap[USB_DEVICE_SUBCLASS] = ToHex(fClassSubId);
	fAttributeMap[USB_DEVICE_PROTOCOL] = ToHex(fClassProtoId);
	fAttributeMap[B_DEVICE_VENDOR_ID] = ToHex(fVendorId);
	fAttributeMap[B_DEVICE_ID] = ToHex(fDeviceId);

	// Fetch ClassInfo
	char classInfo[128];
	usb_get_class_info(fClassBaseId, fClassSubId, fClassProtoId, classInfo,
		sizeof(classInfo));

	// Fetch ManufacturerName
	const char* vendorName = NULL;
	const char* deviceName = NULL;
	BString deviceLabel;
	BString manufacturerLabel;
	usb_get_vendor_info(fVendorId, &vendorName);
	usb_get_device_info(fVendorId, fDeviceId, &deviceName);
	if (vendorName == NULL) {
		manufacturerLabel << B_TRANSLATE("Unknown");
	} else {
		manufacturerLabel << vendorName;
	};

	// Fetch DeviceName
	if (deviceName == NULL) {
		deviceLabel << B_TRANSLATE("Unknown");
	} else {
		deviceLabel << deviceName;
	}

	SetAttribute(B_TRANSLATE("Device name"), deviceLabel);
	SetAttribute(B_TRANSLATE("Manufacturer"), manufacturerLabel);
#if 0
	// These are a source of confusion for users, leading them to think there
	// is no driver for their device. Until we can display something useful,
	// let's not show the lines at all.
	SetAttribute(B_TRANSLATE("Driver used"), B_TRANSLATE("Not implemented"));
	SetAttribute(B_TRANSLATE("Device paths"), B_TRANSLATE("Not implemented"));
#endif
	SetAttribute(B_TRANSLATE("Class info"), classInfo);
	switch (fClassBaseId) {
		case 0x1:
			fCategory = CAT_MULTIMEDIA; break;
		case 0x2:
			fCategory = CAT_COMM; break;
		case 0x3:
			fCategory = CAT_INPUT; break;
		case 0x6:	// Imaging
			fCategory = CAT_MULTIMEDIA; break;
		case 0x7:	// Printer
			fCategory = CAT_MULTIMEDIA; break;
		case 0x8:
			fCategory = CAT_MASS; break;
		case 0x9:
			fCategory = CAT_GENERIC; break;
		case 0xa:
			fCategory = CAT_COMM; break;
		case 0xe:	// Webcam
			fCategory = CAT_MULTIMEDIA; break;
		case 0xe0:
			fCategory = CAT_WIRELESS; break;
	}
	BString outlineName;
	outlineName << manufacturerLabel << " " << deviceLabel;
	SetText(outlineName.String());
}