/*#io
Directory ioDoc(
                docCopyright("Steve Dekorte", 2002)
                docLicense("BSD revised")
                docDescription("""The Directory object supports accessing filesystem directories. A note on paths; 
if a path begins with a "/" it's the root, 
if it beings with a "./" it's the launch path, 
if not specified, "./" is assumed.""")
			 docCredits("""Cygwin code by Mike Austin. WIN32 code by Daniel Vollmer.""")
			 docCategory("FileSystem")
                */

#include "IoDirectory.h"
#include "IoState.h"
#include "IoNumber.h"
#include "IoList.h"
#include "IoFile.h"
#include <sys/stat.h>

#if !defined(_MSC_VER) && !defined(__SYMBIAN32__)
#include <unistd.h> /* ok, this isn't ANSI */
#endif

#if defined(_MSC_VER) && !defined(__SYMBIAN32__)
#include <direct.h>
#define getcwd _getcwd
#endif

#if defined(__SYMBIAN32__)
static char* getcwd(char* buf, int size) { return 0; }
#endif

#ifndef _WIN32

#include <dirent.h>
#include <sys/file.h>
#include <unistd.h>
#define MKDIR mkdir

#else

#include <windows.h>
#define S_IRGRP 0
#define S_IXGRP 0
#define S_IROTH 0
#define S_IXOTH 0
#define S_IRWXU 0

#define DT_DIR 0x01
#define MKDIR mkdir_win32

struct dirent {
    char d_name[MAX_PATH];
    unsigned char d_type;
};

typedef struct {
    WIN32_FIND_DATA wfd;
    HANDLE hFind;
    struct dirent de;
    unsigned char valid;
} DIR;

static DIR *opendir(char *pSpec)
{
	 DIR *pDir = malloc(sizeof *pDir);
	 char *longer_string = malloc((strlen(pSpec) + 3) * sizeof *longer_string);
	
	 strcpy(longer_string, pSpec);
	 strcat(longer_string, "/*");
	 pDir->hFind = FindFirstFile(longer_string, &pDir->wfd);
	 free(longer_string);
	 pDir->valid = pDir->hFind != INVALID_HANDLE_VALUE;
	 
	 if (!pDir->valid)
	 {
		 DWORD err = GetLastError();
		 if (err == ERROR_PATH_NOT_FOUND)
		 {
			 free(pDir);
			 return (DIR*)0;
		 }
	 }
	 
	 return pDir;
}

static void closedir(DIR * pDir)
{
    if (pDir->hFind != INVALID_HANDLE_VALUE)
    {
        FindClose(pDir->hFind);
    }
    
    free(pDir);
}

static struct dirent *readdir(DIR *pDir)
{
    if (pDir->valid)
    {
        strcpy(pDir->de.d_name, pDir->wfd.cFileName);
        pDir->de.d_type = (pDir->wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0 ? DT_DIR : 0;
        pDir->valid = FindNextFile(pDir->hFind, &pDir->wfd);
        return &pDir->de;
    }
    
    return NULL;
}

typedef int mode_t_win32;

int mkdir_win32(const char *path, mode_t_win32 mode)
{
    /* returns zero on sucess */
    LPCTSTR lpPathName = path;
    LPSECURITY_ATTRIBUTES lpSecurityAttributes = NULL;
    return (CreateDirectory(lpPathName, lpSecurityAttributes) != 0);
}

int chdir(const char *path)
{
    LPCTSTR lpPathName = path;
    return SetCurrentDirectory(lpPathName) ? 1 : -1;
}

#endif

int isDirectory(struct dirent *dp, char *path)
{
    #ifdef DT_UNKNOWN
    if (dp->d_type != DT_UNKNOWN)
    {
        return (dp->d_type == DT_DIR);
    }
    else
    #endif
    {
        struct stat st;
        /*fstat( dp->d_fd, &st );*/
        stat(path, &st);
        return ( (st.st_mode & S_IFMT) == S_IFDIR );
    }
}

#define DATA(self) ((IoDirectoryData *)IoObject_dataPointer(self))

IoTag *IoDirectory_tag(void *state)
{
    IoTag *tag = IoTag_newWithName_("Directory");
    tag->state = state;
    tag->cloneFunc = (TagCloneFunc *)IoDirectory_rawClone;
    tag->freeFunc = (TagFreeFunc *)IoDirectory_free;
    tag->markFunc = (TagMarkFunc *)IoDirectory_mark;
    return tag;
}

IoDirectory *IoDirectory_proto(void *state)
{
    IoObject *self = IoObject_new(state);
    self->tag = IoDirectory_tag(state);
    
    IoObject_setDataPointer_(self, calloc(1, sizeof(IoDirectoryData)));
    DATA(self)->path = IOSYMBOL(".");
    
    IoState_registerProtoWithFunc_((IoState *)state, self, IoDirectory_proto);
    
    {
        IoMethodTable methodTable[] = {
        {"setPath", IoDirectory_setPath},
        {"path", IoDirectory_path},
        {"name", IoDirectory_name},
        {"exists", IoDirectory_exists},
        {"items", IoDirectory_items},
        {"at", IoDirectory_at},
        {"size", IoDirectory_size},
        {"create", IoDirectory_create},
        {"createSubdirectory", IoDirectory_createSubdirectory},
        {"currentWorkingDirectory", IoDirectory_currentWorkingDirectory},
        {"setCurrentWorkingDirectory", IoDirectory_setCurrentWorkingDirectory},
        {NULL, NULL},
        };
        
        IoObject_addMethodTable_(self, methodTable);
    }
    return self;
}

IoDirectory *IoDirectory_rawClone(IoDirectory *proto) 
{ 
    IoObject *self = IoObject_rawClonePrimitive(proto);
    IoObject_setDataPointer_(self, cpalloc(IoObject_dataPointer(proto), sizeof(IoDirectoryData)));
    return self; 
}

IoDirectory *IoDirectory_new(void *state)
{
    IoDirectory *proto = IoState_protoWithInitFunction_((IoState *)state, IoDirectory_proto);
    return IOCLONE(proto);
}

// ----------------------------------------------------------- 

IoDirectory *IoDirectory_newWithPath_(void *state, IoSymbol *path)
{
    IoDirectory *self = IoDirectory_new(state);
    DATA(self)->path = IOREF(path);
    return self;
}

IoDirectory *IoDirectory_cloneWithPath_(IoDirectory *self, IoSymbol *path)
{
    IoDirectory *d = IOCLONE(self);
    DATA(d)->path = IOREF(path);
    return d;
}

void IoDirectory_free(IoDirectory *self) 
{ 
    free(IoObject_dataPointer(self)); 
}

void IoDirectory_mark(IoDirectory *self)
{ 
    IoObject_shouldMark((IoObject *)DATA(self)->path); 
}

// ----------------------------------------------------------- 

IoObject *IoDirectory_path(IoDirectory *self, IoObject *locals, IoMessage *m)
{ 
    /*#io
    docSlot("path", "Returns the directory path. The default path is '.'.")
    */
    
    return DATA(self)->path; 
}

IoObject *IoDirectory_setPath(IoDirectory *self, IoObject *locals, IoMessage *m)
{
    /*#io
    docSlot("setPath(aString)", "Sets the directory path. Returns self. ")
    */
    
    DATA(self)->path = IOREF(IoMessage_locals_symbolArgAt_(m, locals, 0));
    return self;
}

IoObject *IoDirectory_name(IoDirectory *self, IoObject *locals, IoMessage *m)
{ 
    /*#io
    docSlot("name", "Returns the receiver's last path component.  ")
    */
    
    return IoSeq_lastPathComponent(DATA(self)->path, locals, m); 
}

IoObject *IoDirectory_itemForDirent_(IoDirectory *self, struct dirent *dp)
{
    IoSymbol *pathString;
    int isDir;
    ByteArray *ba = ByteArray_clone(IoSeq_rawByteArray(DATA(self)->path));
    
    if (ByteArray_size(ba) && !IsPathSeparator(ByteArray_at_(ba, ByteArray_size(ba) - 1)))
    { 
        ByteArray_appendCString_(ba, IO_PATH_SEPARATOR); 
    }
    
    ByteArray_appendCString_(ba, dp->d_name);
    pathString = IoState_symbolWithByteArray_copy_(IOSTATE, ba, 0);
    
    isDir = isDirectory(dp, CSTRING(pathString));
    
    if (isDir) 
    { 
        return IoDirectory_newWithPath_(IOSTATE, pathString); 
    }
    
    return IoFile_newWithPath_(IOSTATE, pathString);
}

IoObject *IoDirectory_exists(IoDirectory *self, IoObject *locals, IoMessage *m)
{
    /*#io
    docSlot("exists(optionalPath)", 
            "Returns true if the Directory path exists, and false otherwise. 
If optionalPath string is provided, it tests the existance of that path instead. ")
    */
    
    IoSymbol *path = DATA(self)->path;
    DIR *dirp;
    
    if (IoMessage_argCount(m) > 0) 
    {
        path = IoMessage_locals_symbolArgAt_(m, locals, 0);
    }
    
    dirp = opendir(CSTRING(path));
    
    if (!dirp) 
    {
        return IOFALSE(self);
    }
    
    (void)closedir(dirp);
    return IOTRUE(self);
}

IoObject *IoDirectory_items(IoDirectory *self, IoObject *locals, IoMessage *m)
{
    /*#io
    docSlot("items", 
            "Returns a list object containing File and Directory objects 
for the files and directories of the receiver's path. ")
    */
    
    IoList *items = IoList_new(IOSTATE);
    IoDirectoryData *data = DATA(self);
    DIR *dirp = opendir(CSTRING(data->path));
    struct dirent *dp;
    
    if (!dirp)
    {
        IoState_error_(IOSTATE, m, "Unable to open directory %s", CSTRING(DATA(self)->path));
    }
    
    while ((dp = readdir(dirp)) != NULL)
    { 
        IoList_rawAppend_(items, IoDirectory_itemForDirent_(self, dp)); 
    }
    
    (void)closedir(dirp);
    return items;
}

IoObject *IoDirectory_justFullPath(IoDirectory *self, IoSymbol *name)
{
    ByteArray *fullPath = ByteArray_clone(IoSeq_rawByteArray(DATA(self)->path));
    ByteArray_appendPathCString_(fullPath, CSTRING(name));
    return IoState_symbolWithByteArray_copy_(IOSTATE, fullPath, 0);
}

IoObject *IoDirectory_justAt(IoDirectory *self, IoSymbol *name)
{
    IoState *state = IOSTATE;
    IoSymbol *fullPath = IoDirectory_justFullPath(self, name);
    struct stat st;
    
    if (stat(CSTRING(fullPath), &st) == -1)
    { 
        return IONIL(self); 
    }
    
    if ((st.st_mode & S_IFMT) == S_IFDIR)
    { 
        return IoDirectory_newWithPath_(state, fullPath); 
    }
    else
    { 
        return IoFile_newWithPath_(state, fullPath); 
    }
    
    return IONIL(self);
}

IoObject *IoDirectory_at(IoDirectory *self, IoObject *locals, IoMessage *m)
{
    /*#io
    docSlot("at(aString)", 
            "Returns a File or Directory object matching the name specified 
by aString or Nil if no such file or directory exists. ")
    */
    
    IoSymbol *name = IoMessage_locals_symbolArgAt_(m, locals, 0);
    return IoDirectory_justAt(self, name);
    /*
     IoObject *item = IoDirectory_justAt(self, name);
     if (ISNIL(item))
     {
         IoState_error_(IOSTATE, m, "Unable to open path %s", CSTRING(IoDirectory_justFullPath(self, name)));
     }
     return item;
     */
}

/*
 IoObject *IoDirectory_atPut(IoDirectory *self, IoObject *locals, IoMessage *m)
 {
     IoSymbol *name = IoMessage_locals_symbolArgAt_(m, locals, 0);
     IoObject *item = IoDirectory_justAt(self, name);
     if (ISNIL(item))
     {
         IoState_error_(IOSTATE,  m, "Unable to open path %s", CSTRING(IoDirectory_justFullPath(self, name)));
     }
     return item;
 }
 */

/*
 IoObject *IoDirectory_itemNamed(IoDirectory *self, IoObject *locals, IoMessage *m)
 {
     IoSymbol *itemName = IoMessage_locals_symbolArgAt_(m, locals, 0);
     char *name = CSTRING(itemName);
     DIR *dirp = opendir(CSTRING(DATA(self)->path));
     struct dirent *dp;
     if (!dirp)
     {
         IoState_error_(IOSTATE, m, "Unable to open directory %s", CSTRING(DATA(self)->path));
     }
     
     while ((dp = readdir(dirp)) != NULL)
     {
         if (strcmp(dp->d_name, name) == 0)
         {
             IoObject *item = IoDirectory_itemForDirent_(self, dp);
             (void)closedir(dirp);
             return item;
         }
     }
     (void)closedir(dirp);
     return IONIL(self);
 }
 */

IoObject *IoDirectory_createSubdirectory(IoDirectory *self, IoObject *locals, IoMessage *m)
{
    /*#io
    docSlot("createSubdirectory(name)", 
            "Create a subdirectory with the specified name.")
    */
    
    IoState *state = IOSTATE;
    IoSymbol *subfolderName = IoMessage_locals_symbolArgAt_(m, locals, 0);
    IoObject *currentItem = IoDirectory_justAt(self, subfolderName);
    
    if (ISDIRECTORY(currentItem)) 
    {
        return currentItem;
    }
    
    if (ISFILE(currentItem))
    {
        IoState_error_(IOSTATE, m, "Attempt to create directory %s on top of existing file", 
                                   CSTRING(subfolderName));
    }
    else
    {
        IoSymbol *fullPath = IoDirectory_justFullPath(self, subfolderName);
        
        MKDIR(CSTRING(fullPath), S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH);
        return IoDirectory_newWithPath_(state, fullPath);
    }
    
    return IONIL(self);
}


IoObject *IoDirectory_create(IoDirectory *self, IoObject *locals, IoMessage *m)
{
    /*#io
    docSlot("create", 
            "Create the directory if it doesn't exist.")
    */
    
    int r = MKDIR(CSTRING(DATA(self)->path), S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH);
    return IOBOOL(self, r == 0);
}

IoObject *IoDirectory_size(IoDirectory *self, IoObject *locals, IoMessage *m)
{
    /*#io
    docSlot("size", 
            "Returns a Number containing the number of file and directory 
object at the receiver's path. ")
    */
    
    int count = 0;
    DIR *dirp = opendir(CSTRING(DATA(self)->path));
    struct dirent *dp;
    
    if (!dirp)
    {
        IoState_error_(IOSTATE, m, "Unable to open directory %s", CSTRING(DATA(self)->path));
    }
    
    while ((dp = readdir(dirp)) != NULL)
    { 
        count ++; 
    }
    
    (void)closedir(dirp);
    return IONUMBER((double)count);
}


/* -------------------------------- */

ByteArray *IoDirectory_CurrentWorkingDirectoryAsByteArray(void)
{
#if defined(sparc) || defined(__sparc)
    char *buf = getcwd(NULL, FILENAME_MAX + 1);
#else
    char *buf = NULL;
    buf = (char *)getcwd(buf, 1024);
#endif /* sparc || _sparc */
    
    if (!buf) 
    {
        return ByteArray_newWithCString_(".");
    }
    else
    {
        ByteArray *ba =  ByteArray_newWithData_size_((unsigned char *)buf, strlen(buf));
	//free(buf);
        /*free(buf); OSX get cwd man page says we should free this, but MallocDebug doesn't like it */
        return ba;
    }
}

int IoDirectory_SetCurrentWorkingDirectory(const char *path)
{
	return chdir(path);
}

IoObject *IoDirectory_currentWorkingDirectory(IoFile *self, IoObject *locals, IoMessage *m)
{
    /*#io
    docSlot("currentWorkingDirectory", 
            "Returns the current working directory path.")
    */
    
    return IoState_symbolWithByteArray_copy_(IOSTATE,
                                             IoDirectory_CurrentWorkingDirectoryAsByteArray(), 0);
}

/*
 int IoDirectory_SetCurrentWorkingDirectory(char *p)
 { 
     return chdir(p); 
 }
 */

IoObject *IoDirectory_setCurrentWorkingDirectory(IoFile *self, IoObject *locals, IoMessage *m)
{
    /*#io
    docSlot("setCurrentWorkingDirectory(pathString)", 
            "Set's the current working directory path. 
Returns true on success or false on error.")
    */
    
    IoSymbol *path = IoMessage_locals_symbolArgAt_(m, locals, 0);
	
    if(chdir(CSTRING(path)) == -1) 
    {
	    return IOSUCCESS(self);
    }
    
    return IOFAILURE(self);
}