/*---------------------------------------------------------------------------*
  Project:      Sample demo program for NAND library
  File:         directory.c
  Programmer:   HIRATSU Daisuke

  Copyright (C) 2006 Nintendo.  All rights reserved.

  These coded instructions, statements, and computer programs contain
  proprietary information of Nintendo of America Inc. and/or Nintendo
  Company Ltd., and are protected by Federal copyright law.  They may
  not be disclosed to third parties or copied or duplicated in any form,
  in whole or in part, without the prior written consent of Nintendo.

  $Log: directory.c,v $
  Revision 1.9  07/19/2006 13:02:56  hiratsu
  Now, this demo can distinguish between file and directory.

  Revision 1.8  06/08/2006 11:12:05  hiratsu
  Buffer allocation for NANDReadDir was wrong.  Fixed.
  Now printDir() prints extra information, such as permission.

  Revision 1.7  06/05/2006 13:10:30  hiratsu
  Removed NANDFinalize().
  Now this program outputs current (home) directory.

  Revision 1.6  05/03/2006 09:06:21  hiratsu
  Revised flow.

  Revision 1.5  2006/04/27 17:57:35  orest
  Added NANDFinalize() api.

  Revision 1.4  03/17/2006 04:39:05  hiratsu
  Adapted the source to real file system module.

  Revision 1.3  03/06/2006 12:05:34  hiratsu
  Removed finalize function.

  Revision 1.2  2006/03/06 09:59:08  kawaset
  Eliminated warnings.

  Revision 1.1  02/16/2006 13:01:08  hiratsu
  Initial check in.
 *---------------------------------------------------------------------------*/


#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <revolution.h>
#include <revolution/nand.h>

#include "util.h"

#define PERM (NAND_PERM_OWNER_READ | NAND_PERM_OWNER_WRITE)    // Owner can read/write
#define ATTR 0x00                                              // No attributes.

static void printDir  (const char *path);
static void createDir (const char *path);
static void deleteDir (const char *path);
static void changeDir (const char *path);
static void printCurrentDir(void);
static void createFile(const char *path, u8 perm);


static void printDir(const char *path)
{
    u32 num   = 0;
    u32 idx   = 0;
    u32 i     = 0;
    char *buf = 0;

    NANDReadDir(path, NULL, &num);  // Get number of elements.
    buf = (char*)alloc32(num*(NAND_MAX_NAME+1));
    if(!buf)
    {
        OSReport("Failed to alloc memory.  Cannot print directory.\n");
        return;
    }

    NANDReadDir(path, buf, &num);   // Get directory list.
    for(i=0; i<num; ++i)
    {
        char name[NAND_MAX_NAME+1]="";
        char absPath[NAND_MAX_PATH]="";
        int n = 0;
        NANDStatus stat;
        u8 type = 0x00;
        while(buf[idx] != '\0')
        {
            name[n] = buf[idx];
            ++idx;
            ++n;
        }
        name[n]='\0';
        if(strcmp(path, "/")==0)
        {
            sprintf(absPath, "/%s", name);
        }
        else
        {
            sprintf(absPath, "%s/%s", path, name);
        }
        NANDGetType(absPath, &type);
        if(type==NAND_TYPE_FILE)
        {
            OSReport("-");
        }
        else if(type==NAND_TYPE_DIR)
        {
            OSReport("d");
        }
        else
        {
            OSReport("?");
        }
        NANDGetStatus(absPath, &stat);
        stat.permission & NAND_PERM_OWNER_READ  ? OSReport("r") : OSReport("-");
        stat.permission & NAND_PERM_OWNER_WRITE ? OSReport("w") : OSReport("-");
        stat.permission & NAND_PERM_GROUP_READ  ? OSReport("r") : OSReport("-");
        stat.permission & NAND_PERM_GROUP_WRITE ? OSReport("w") : OSReport("-");
        stat.permission & NAND_PERM_OTHER_READ  ? OSReport("r") : OSReport("-");
        stat.permission & NAND_PERM_OTHER_WRITE ? OSReport("w") : OSReport("-");
        OSReport(" %02x %08x %04x %s\n", stat.attribute, stat.ownerId, stat.groupId, name);
        ++idx;
    }

    free32(buf);
}


static void createDir(const char *path)
{
    s32 result = NANDCreateDir(path, PERM, ATTR);
    if(result != NAND_RESULT_OK)
    {
        printErrMsg(result);
        OSHalt("NANDCreateDir() failed.");
    }
}


static void deleteDir(const char *path)
{
    s32 result = NANDDelete(path);
    if(result != NAND_RESULT_OK)
    {
        printErrMsg(result);
        OSHalt("NANDDelete() failed.");
    }
}


static void changeDir(const char *path)
{
    s32 result = NANDChangeDir(path);
    if(result != NAND_RESULT_OK)
    {
        printErrMsg(result);
        OSHalt("NANDChangeDir() failed.");
    }
}


static void printCurrentDir(void)
{
    char path[NAND_MAX_PATH]="";
    s32 result = NANDGetCurrentDir(path);
    if(result == NAND_RESULT_OK)
    {
        OSReport("%s\n", path);
    }
    else
    {
        printErrMsg(result);
        OSHalt("NANDGetCurrentDir() failed.\n");
    }
}


static void createFile(const char *path, const u8 perm)
{
    s32 result = NANDCreate(path, perm, ATTR);
    if(result != NAND_RESULT_OK)
    {
        printErrMsg(result);
        OSHalt("NANDCreate() failed.");
    }
}


int main(void)
{
    s32 ret = NAND_RESULT_FATAL_ERROR;

    initializeHeap();

    ret = NANDInit();
    if(ret != NAND_RESULT_OK)
    {
        printErrMsg(ret);
        OSReport("NANDInit() failed.\n");
        return EXIT_FAILURE;
    }

    OSReport("--- Current directory ---\n");
    printCurrentDir();
    OSReport("\n");

    OSReport("--- Under / directory ---\n");
    printDir("/");
    OSReport("\n");

    changeDir("/tmp");
    createDir("dir");
    OSReport("--- Under /tmp directory ---\n");
    printDir("/tmp");
    OSReport("\n");

    changeDir("/tmp/dir");
    createDir("subdir0");
    createDir("subdir1");
    createDir("subdir2");
    createFile("file0", NAND_PERM_OWNER_READ | NAND_PERM_OWNER_WRITE);
    createFile("file1", NAND_PERM_OWNER_READ | NAND_PERM_OWNER_WRITE | NAND_PERM_GROUP_READ);
    createFile("file2", NAND_PERM_OWNER_READ | NAND_PERM_OWNER_WRITE | NAND_PERM_GROUP_READ | NAND_PERM_OTHER_READ);
    OSReport("--- Under /tmp/dir directory ---\n");
    printDir(".");
    OSReport("\n");

    changeDir("/tmp");
    deleteDir("dir");      // Please note that subdirectories and files under specified directory will be deleted at once.

    OSReport("--- Under /tmp directory ---\n");
    printDir(".");
    OSReport("\n");

    finalizeHeap();

    OSHalt("sample demo program for NAND library end.\n");
    return EXIT_SUCCESS;  // Never reach here
}