/* ========================= Module _Launch ========================= */

#include "Python.h"

#include "pymactoolbox.h"

/* Macro to test whether a weak-loaded CFM function exists */

#define PyMac_PRECHECK(rtn) do { if ( &rtn == NULL )  {\

        PyErr_SetString(PyExc_NotImplementedError, \

        "Not available in this shared library/OS version"); \

        return NULL; \

    }} while(0)

#if PY_VERSION_HEX < 0x02040000

PyObject *PyMac_GetOSErrException(void);

#endif

#include <ApplicationServices/ApplicationServices.h>

/*

** Optional CFStringRef. None will pass NULL

*/

static int

OptCFStringRefObj_Convert(PyObject *v, CFStringRef *spec)

{

        if (v == Py_None) {

                *spec = NULL;

                return 1;

        }

        return CFStringRefObj_Convert(v, spec);

}

PyObject *

OptCFStringRefObj_New(CFStringRef it)

{

        if (it == NULL) {

                Py_INCREF(Py_None);

                return Py_None;

        }

        return CFStringRefObj_New(it);

}

/*

** Convert LSItemInfoRecord to Python.

*/

PyObject *

LSItemInfoRecord_New(LSItemInfoRecord *it)

{

#ifndef __LP64__

        return Py_BuildValue("{s:is:O&s:O&s:O&s:O&s:i}",

                "flags", it->flags,

                "filetype", PyMac_BuildOSType, it->filetype,

                "creator", PyMac_BuildOSType, it->creator,

                "extension", OptCFStringRefObj_New, it->extension,

                "iconFileName", OptCFStringRefObj_New, it->iconFileName,

                "kindID", it->kindID);

#else

        return Py_BuildValue("{s:is:O&s:O&s:O&}",

                "flags", it->flags,

                "filetype", PyMac_BuildOSType, it->filetype,

                "creator", PyMac_BuildOSType, it->creator,

                "extension", OptCFStringRefObj_New, it->extension);

#endif

}

static PyObject *Launch_Error;

static PyObject *Launch_LSCopyItemInfoForRef(PyObject *_self, PyObject *_args)

{

	PyObject *_res = NULL;

	OSStatus _err;

	FSRef inItemRef;

	LSRequestedInfo inWhichInfo;

	LSItemInfoRecord outItemInfo;

	if (!PyArg_ParseTuple(_args, "O&l",

	                      PyMac_GetFSRef, &inItemRef,

	                      &inWhichInfo))

		return NULL;

	_err = LSCopyItemInfoForRef(&inItemRef,

	                            inWhichInfo,

	                            &outItemInfo);

	if (_err != noErr) return PyMac_Error(_err);

	_res = Py_BuildValue("O&",

	                     LSItemInfoRecord_New, &outItemInfo);

	return _res;

}

static PyObject *Launch_LSCopyItemInfoForURL(PyObject *_self, PyObject *_args)

{

	PyObject *_res = NULL;

	OSStatus _err;

	CFURLRef inURL;

	LSRequestedInfo inWhichInfo;

	LSItemInfoRecord outItemInfo;

	if (!PyArg_ParseTuple(_args, "O&l",

	                      CFURLRefObj_Convert, &inURL,

	                      &inWhichInfo))

		return NULL;

	_err = LSCopyItemInfoForURL(inURL,

	                            inWhichInfo,

	                            &outItemInfo);

	if (_err != noErr) return PyMac_Error(_err);

	_res = Py_BuildValue("O&",

	                     LSItemInfoRecord_New, &outItemInfo);

	return _res;

}

static PyObject *Launch_LSGetExtensionInfo(PyObject *_self, PyObject *_args)

{

	PyObject *_res = NULL;

	OSStatus _err;

	UniChar *inNameLen__in__;

	UniCharCount inNameLen__len__;

	int inNameLen__in_len__;

	UniCharCount outExtStartIndex;

	if (!PyArg_ParseTuple(_args, "u#",

	                      &inNameLen__in__, &inNameLen__in_len__))

		return NULL;

	inNameLen__len__ = inNameLen__in_len__;

	_err = LSGetExtensionInfo(inNameLen__len__, inNameLen__in__,

	                          &outExtStartIndex);

	if (_err != noErr) return PyMac_Error(_err);

	_res = Py_BuildValue("l",

	                     outExtStartIndex);

	return _res;

}

static PyObject *Launch_LSCopyDisplayNameForRef(PyObject *_self, PyObject *_args)

{

	PyObject *_res = NULL;

	OSStatus _err;

	FSRef inRef;

	CFStringRef outDisplayName;

	if (!PyArg_ParseTuple(_args, "O&",

	                      PyMac_GetFSRef, &inRef))

		return NULL;

	_err = LSCopyDisplayNameForRef(&inRef,

	                               &outDisplayName);

	if (_err != noErr) return PyMac_Error(_err);

	_res = Py_BuildValue("O&",

	                     CFStringRefObj_New, outDisplayName);

	return _res;

}

static PyObject *Launch_LSCopyDisplayNameForURL(PyObject *_self, PyObject *_args)

{

	PyObject *_res = NULL;

	OSStatus _err;

	CFURLRef inURL;

	CFStringRef outDisplayName;

	if (!PyArg_ParseTuple(_args, "O&",

	                      CFURLRefObj_Convert, &inURL))

		return NULL;

	_err = LSCopyDisplayNameForURL(inURL,

	                               &outDisplayName);

	if (_err != noErr) return PyMac_Error(_err);

	_res = Py_BuildValue("O&",

	                     CFStringRefObj_New, outDisplayName);

	return _res;

}

static PyObject *Launch_LSSetExtensionHiddenForRef(PyObject *_self, PyObject *_args)

{

	PyObject *_res = NULL;

	OSStatus _err;

	FSRef inRef;

	Boolean inHide;

	if (!PyArg_ParseTuple(_args, "O&b",

	                      PyMac_GetFSRef, &inRef,

	                      &inHide))

		return NULL;

	_err = LSSetExtensionHiddenForRef(&inRef,

	                                  inHide);

	if (_err != noErr) return PyMac_Error(_err);

	Py_INCREF(Py_None);

	_res = Py_None;

	return _res;

}

static PyObject *Launch_LSSetExtensionHiddenForURL(PyObject *_self, PyObject *_args)

{

	PyObject *_res = NULL;

	OSStatus _err;

	CFURLRef inURL;

	Boolean inHide;

	if (!PyArg_ParseTuple(_args, "O&b",

	                      CFURLRefObj_Convert, &inURL,

	                      &inHide))

		return NULL;

	_err = LSSetExtensionHiddenForURL(inURL,

	                                  inHide);

	if (_err != noErr) return PyMac_Error(_err);

	Py_INCREF(Py_None);

	_res = Py_None;

	return _res;

}

static PyObject *Launch_LSCopyKindStringForRef(PyObject *_self, PyObject *_args)

{

	PyObject *_res = NULL;

	OSStatus _err;

	FSRef inFSRef;

	CFStringRef outKindString;

	if (!PyArg_ParseTuple(_args, "O&",

	                      PyMac_GetFSRef, &inFSRef))

		return NULL;

	_err = LSCopyKindStringForRef(&inFSRef,

	                              &outKindString);

	if (_err != noErr) return PyMac_Error(_err);

	_res = Py_BuildValue("O&",

	                     CFStringRefObj_New, outKindString);

	return _res;

}

static PyObject *Launch_LSCopyKindStringForURL(PyObject *_self, PyObject *_args)

{

	PyObject *_res = NULL;

	OSStatus _err;

	CFURLRef inURL;

	CFStringRef outKindString;

	if (!PyArg_ParseTuple(_args, "O&",

	                      CFURLRefObj_Convert, &inURL))

		return NULL;

	_err = LSCopyKindStringForURL(inURL,

	                              &outKindString);

	if (_err != noErr) return PyMac_Error(_err);

	_res = Py_BuildValue("O&",

	                     CFStringRefObj_New, outKindString);

	return _res;

}

static PyObject *Launch_LSGetApplicationForItem(PyObject *_self, PyObject *_args)

{

	PyObject *_res = NULL;

	OSStatus _err;

	FSRef inItemRef;

	LSRolesMask inRoleMask;

	FSRef outAppRef;

	CFURLRef outAppURL;

	if (!PyArg_ParseTuple(_args, "O&l",

	                      PyMac_GetFSRef, &inItemRef,

	                      &inRoleMask))

		return NULL;

	_err = LSGetApplicationForItem(&inItemRef,

	                               inRoleMask,

	                               &outAppRef,

	                               &outAppURL);

	if (_err != noErr) return PyMac_Error(_err);

	_res = Py_BuildValue("O&O&",

	                     PyMac_BuildFSRef, &outAppRef,

	                     CFURLRefObj_New, outAppURL);

	return _res;

}

static PyObject *Launch_LSGetApplicationForInfo(PyObject *_self, PyObject *_args)

{

	PyObject *_res = NULL;

	OSStatus _err;

	OSType inType;

	OSType inCreator;

	CFStringRef inExtension;

	LSRolesMask inRoleMask;

	FSRef outAppRef;

	CFURLRef outAppURL;

	if (!PyArg_ParseTuple(_args, "O&O&O&l",

	                      PyMac_GetOSType, &inType,

	                      PyMac_GetOSType, &inCreator,

	                      OptCFStringRefObj_Convert, &inExtension,

	                      &inRoleMask))

		return NULL;

	_err = LSGetApplicationForInfo(inType,

	                               inCreator,

	                               inExtension,

	                               inRoleMask,

	                               &outAppRef,

	                               &outAppURL);

	if (_err != noErr) return PyMac_Error(_err);

	_res = Py_BuildValue("O&O&",

	                     PyMac_BuildFSRef, &outAppRef,

	                     CFURLRefObj_New, outAppURL);

	return _res;

}

static PyObject *Launch_LSGetApplicationForURL(PyObject *_self, PyObject *_args)

{

	PyObject *_res = NULL;

	OSStatus _err;

	CFURLRef inURL;

	LSRolesMask inRoleMask;

	FSRef outAppRef;

	CFURLRef outAppURL;

	if (!PyArg_ParseTuple(_args, "O&l",

	                      CFURLRefObj_Convert, &inURL,

	                      &inRoleMask))

		return NULL;

	_err = LSGetApplicationForURL(inURL,

	                              inRoleMask,

	                              &outAppRef,

	                              &outAppURL);

	if (_err != noErr) return PyMac_Error(_err);

	_res = Py_BuildValue("O&O&",

	                     PyMac_BuildFSRef, &outAppRef,

	                     CFURLRefObj_New, outAppURL);

	return _res;

}

static PyObject *Launch_LSFindApplicationForInfo(PyObject *_self, PyObject *_args)

{

	PyObject *_res = NULL;

	OSStatus _err;

	OSType inCreator;

	CFStringRef inBundleID;

	CFStringRef inName;

	FSRef outAppRef;

	CFURLRef outAppURL;

	if (!PyArg_ParseTuple(_args, "O&O&O&",

	                      PyMac_GetOSType, &inCreator,

	                      OptCFStringRefObj_Convert, &inBundleID,

	                      OptCFStringRefObj_Convert, &inName))

		return NULL;

	_err = LSFindApplicationForInfo(inCreator,

	                                inBundleID,

	                                inName,

	                                &outAppRef,

	                                &outAppURL);

	if (_err != noErr) return PyMac_Error(_err);

	_res = Py_BuildValue("O&O&",

	                     PyMac_BuildFSRef, &outAppRef,

	                     CFURLRefObj_New, outAppURL);

	return _res;

}

static PyObject *Launch_LSCanRefAcceptItem(PyObject *_self, PyObject *_args)

{

	PyObject *_res = NULL;

	OSStatus _err;

	FSRef inItemFSRef;

	FSRef inTargetRef;

	LSRolesMask inRoleMask;

	LSAcceptanceFlags inFlags;

	Boolean outAcceptsItem;

	if (!PyArg_ParseTuple(_args, "O&O&ll",

	                      PyMac_GetFSRef, &inItemFSRef,

	                      PyMac_GetFSRef, &inTargetRef,

	                      &inRoleMask,

	                      &inFlags))

		return NULL;

	_err = LSCanRefAcceptItem(&inItemFSRef,

	                          &inTargetRef,

	                          inRoleMask,

	                          inFlags,

	                          &outAcceptsItem);

	if (_err != noErr) return PyMac_Error(_err);

	_res = Py_BuildValue("b",

	                     outAcceptsItem);

	return _res;

}

static PyObject *Launch_LSCanURLAcceptURL(PyObject *_self, PyObject *_args)

{

	PyObject *_res = NULL;

	OSStatus _err;

	CFURLRef inItemURL;

	CFURLRef inTargetURL;

	LSRolesMask inRoleMask;

	LSAcceptanceFlags inFlags;

	Boolean outAcceptsItem;

	if (!PyArg_ParseTuple(_args, "O&O&ll",

	                      CFURLRefObj_Convert, &inItemURL,

	                      CFURLRefObj_Convert, &inTargetURL,

	                      &inRoleMask,

	                      &inFlags))

		return NULL;

	_err = LSCanURLAcceptURL(inItemURL,

	                         inTargetURL,

	                         inRoleMask,

	                         inFlags,

	                         &outAcceptsItem);

	if (_err != noErr) return PyMac_Error(_err);

	_res = Py_BuildValue("b",

	                     outAcceptsItem);

	return _res;

}

static PyObject *Launch_LSOpenFSRef(PyObject *_self, PyObject *_args)

{

	PyObject *_res = NULL;

	OSStatus _err;

	FSRef inRef;

	FSRef outLaunchedRef;

	if (!PyArg_ParseTuple(_args, "O&",

	                      PyMac_GetFSRef, &inRef))

		return NULL;

	_err = LSOpenFSRef(&inRef,

	                   &outLaunchedRef);

	if (_err != noErr) return PyMac_Error(_err);

	_res = Py_BuildValue("O&",

	                     PyMac_BuildFSRef, &outLaunchedRef);

	return _res;

}

static PyObject *Launch_LSOpenCFURLRef(PyObject *_self, PyObject *_args)

{

	PyObject *_res = NULL;

	OSStatus _err;

	CFURLRef inURL;

	CFURLRef outLaunchedURL;

	if (!PyArg_ParseTuple(_args, "O&",

	                      CFURLRefObj_Convert, &inURL))

		return NULL;

	_err = LSOpenCFURLRef(inURL,

	                      &outLaunchedURL);

	if (_err != noErr) return PyMac_Error(_err);

	_res = Py_BuildValue("O&",

	                     CFURLRefObj_New, outLaunchedURL);

	return _res;

}

static PyMethodDef Launch_methods[] = {

	{"LSCopyItemInfoForRef", (PyCFunction)Launch_LSCopyItemInfoForRef, 1,

	 PyDoc_STR("(FSRef inItemRef, LSRequestedInfo inWhichInfo) -> (LSItemInfoRecord outItemInfo)")},

	{"LSCopyItemInfoForURL", (PyCFunction)Launch_LSCopyItemInfoForURL, 1,

	 PyDoc_STR("(CFURLRef inURL, LSRequestedInfo inWhichInfo) -> (LSItemInfoRecord outItemInfo)")},

	{"LSGetExtensionInfo", (PyCFunction)Launch_LSGetExtensionInfo, 1,

	 PyDoc_STR("(Buffer inNameLen) -> (UniCharCount outExtStartIndex)")},

	{"LSCopyDisplayNameForRef", (PyCFunction)Launch_LSCopyDisplayNameForRef, 1,

	 PyDoc_STR("(FSRef inRef) -> (CFStringRef outDisplayName)")},

	{"LSCopyDisplayNameForURL", (PyCFunction)Launch_LSCopyDisplayNameForURL, 1,

	 PyDoc_STR("(CFURLRef inURL) -> (CFStringRef outDisplayName)")},

	{"LSSetExtensionHiddenForRef", (PyCFunction)Launch_LSSetExtensionHiddenForRef, 1,

	 PyDoc_STR("(FSRef inRef, Boolean inHide) -> None")},

	{"LSSetExtensionHiddenForURL", (PyCFunction)Launch_LSSetExtensionHiddenForURL, 1,

	 PyDoc_STR("(CFURLRef inURL, Boolean inHide) -> None")},

	{"LSCopyKindStringForRef", (PyCFunction)Launch_LSCopyKindStringForRef, 1,

	 PyDoc_STR("(FSRef inFSRef) -> (CFStringRef outKindString)")},

	{"LSCopyKindStringForURL", (PyCFunction)Launch_LSCopyKindStringForURL, 1,

	 PyDoc_STR("(CFURLRef inURL) -> (CFStringRef outKindString)")},

	{"LSGetApplicationForItem", (PyCFunction)Launch_LSGetApplicationForItem, 1,

	 PyDoc_STR("(FSRef inItemRef, LSRolesMask inRoleMask) -> (FSRef outAppRef, CFURLRef outAppURL)")},

	{"LSGetApplicationForInfo", (PyCFunction)Launch_LSGetApplicationForInfo, 1,

	 PyDoc_STR("(OSType inType, OSType inCreator, CFStringRef inExtension, LSRolesMask inRoleMask) -> (FSRef outAppRef, CFURLRef outAppURL)")},

	{"LSGetApplicationForURL", (PyCFunction)Launch_LSGetApplicationForURL, 1,

	 PyDoc_STR("(CFURLRef inURL, LSRolesMask inRoleMask) -> (FSRef outAppRef, CFURLRef outAppURL)")},

	{"LSFindApplicationForInfo", (PyCFunction)Launch_LSFindApplicationForInfo, 1,

	 PyDoc_STR("(OSType inCreator, CFStringRef inBundleID, CFStringRef inName) -> (FSRef outAppRef, CFURLRef outAppURL)")},

	{"LSCanRefAcceptItem", (PyCFunction)Launch_LSCanRefAcceptItem, 1,

	 PyDoc_STR("(FSRef inItemFSRef, FSRef inTargetRef, LSRolesMask inRoleMask, LSAcceptanceFlags inFlags) -> (Boolean outAcceptsItem)")},

	{"LSCanURLAcceptURL", (PyCFunction)Launch_LSCanURLAcceptURL, 1,

	 PyDoc_STR("(CFURLRef inItemURL, CFURLRef inTargetURL, LSRolesMask inRoleMask, LSAcceptanceFlags inFlags) -> (Boolean outAcceptsItem)")},

	{"LSOpenFSRef", (PyCFunction)Launch_LSOpenFSRef, 1,

	 PyDoc_STR("(FSRef inRef) -> (FSRef outLaunchedRef)")},

	{"LSOpenCFURLRef", (PyCFunction)Launch_LSOpenCFURLRef, 1,

	 PyDoc_STR("(CFURLRef inURL) -> (CFURLRef outLaunchedURL)")},

	{NULL, NULL, 0}

};

void init_Launch(void)

{

	PyObject *m;

	PyObject *d;

	m = Py_InitModule("_Launch", Launch_methods);

	d = PyModule_GetDict(m);

	Launch_Error = PyMac_GetOSErrException();

	if (Launch_Error == NULL ||

	    PyDict_SetItemString(d, "Error", Launch_Error) != 0)

		return;

}

/* ======================= End module _Launch ======================= */