palemoon27/xpcom/tests/nsIFileEnumerator.cpp

#include "nsIFile.h"
#include "nsStringGlue.h"

#include <stdio.h>
#include "nsIComponentRegistrar.h"
#include "nsIComponentManager.h"
#include "nsIServiceManager.h"
#include "nsMemory.h"
#include "nsISimpleEnumerator.h"
#include "nsCOMPtr.h"

static bool LoopInDir(nsIFile* aFile)
{
    nsresult rv;
    nsCOMPtr<nsISimpleEnumerator> entries;
    rv = aFile->GetDirectoryEntries(getter_AddRefs(entries));
    if(NS_FAILED(rv) || !entries)
        return false;
    
    bool hasMore;
    while(NS_SUCCEEDED(entries->HasMoreElements(&hasMore)) && hasMore)
    {
        nsCOMPtr<nsISupports> sup;
        entries->GetNext(getter_AddRefs(sup));
        if(!sup)
            return false;

        nsCOMPtr<nsIFile> nextFile = do_QueryInterface(sup);
        if (!nextFile)
            return false;
    
        nsAutoCString name;
        if(NS_FAILED(nextFile->GetNativeLeafName(name)))
            return false;
        
        bool isDir;
        printf("%s\n", name.get());
        rv = nextFile->IsDirectory(&isDir);
        if (NS_FAILED(rv))
		{
			printf("IsDirectory Failed!!!\n");
				return false;
		}

		if (isDir)
        {
           LoopInDir(nextFile);
        }
    }
    return true;
}


int
main(int argc, char* argv[])
{
    nsresult rv;
    {
        nsCOMPtr<nsIFile> topDir;

        nsCOMPtr<nsIServiceManager> servMan;
        rv = NS_InitXPCOM2(getter_AddRefs(servMan), nullptr, nullptr);
        if (NS_FAILED(rv)) return -1;

        if (argc > 1 && argv[1] != nullptr)
        {
            char* pathStr = argv[1];
            NS_NewNativeLocalFile(nsDependentCString(pathStr), false, getter_AddRefs(topDir));
        }
    
        if (!topDir)
        {
           printf("No Top Dir\n");
           return -1;
        }
        int32_t startTime = PR_IntervalNow();
    
        LoopInDir(topDir);
    
        int32_t endTime = PR_IntervalNow();
    
        printf("\nTime: %d\n", PR_IntervalToMilliseconds(endTime - startTime));
    } // this scopes the nsCOMPtrs
    // no nsCOMPtrs are allowed to be alive when you call NS_ShutdownXPCOM
    rv = NS_ShutdownXPCOM(nullptr);
    NS_ASSERTION(NS_SUCCEEDED(rv), "NS_ShutdownXPCOM failed");
    return 0;
}