xpcom/tests/nsIFileEnumerator.cpp

   1 #include "nsILocalFile.h"
   2 #include "nsStringGlue.h"
   3
   4 #include <stdio.h>
   5 #include "nsIComponentRegistrar.h"
   6 #include "nsIComponentManager.h"
   7 #include "nsIServiceManager.h"
   8 #include "nsMemory.h"
   9 #include "nsISimpleEnumerator.h"
  10 #include "nsCOMPtr.h"
  11
  12 PRBool LoopInDir(nsILocalFile* file)
  13 {
  14     nsresult rv;
  15     nsCOMPtr<nsISimpleEnumerator> entries;
  16     rv = file->GetDirectoryEntries(getter_AddRefs(entries));
  17     if(NS_FAILED(rv) || !entries)
  18         return PR_FALSE;
  19
  20     PRBool hasMore;
  21     while(NS_SUCCEEDED(entries->HasMoreElements(&hasMore)) && hasMore)
  22     {
  23         nsCOMPtr<nsISupports> sup;
  24         entries->GetNext(getter_AddRefs(sup));
  25         if(!sup)
  26             return PR_FALSE;
  27
  28         nsCOMPtr<nsILocalFile> file = do_QueryInterface(sup);
  29         if(!file)
  30             return PR_FALSE;
  31
  32         nsCAutoString name;
  33         if(NS_FAILED(file->GetNativeLeafName(name)))
  34             return PR_FALSE;
  35
  36         PRBool isDir;
  37         printf("%s\n", name.get());
  38         rv = file->IsDirectory(&isDir);
  39         if (NS_FAILED(rv))
  40                 {
  41                         printf("IsDirectory Failed!!!\n");
  42                                 return PR_FALSE;
  43                 }
  44
  45                 if (isDir == PR_TRUE)
  46         {
  47            nsCOMPtr<nsILocalFile> lfile = do_QueryInterface(file);
  48            LoopInDir(lfile);
  49         }
  50     }
  51     return PR_TRUE;
  52 }
  53
  54
  55 int
  56 main(int argc, char* argv[])
  57 {
  58     nsresult rv;
  59     {
  60         nsCOMPtr<nsILocalFile> topDir;
  61
  62         nsCOMPtr<nsIServiceManager> servMan;
  63         rv = NS_InitXPCOM2(getter_AddRefs(servMan), nsnull, nsnull);
  64         if (NS_FAILED(rv)) return -1;
  65         nsCOMPtr<nsIComponentRegistrar> registrar = do_QueryInterface(servMan);
  66         NS_ASSERTION(registrar, "Null nsIComponentRegistrar");
  67         if (registrar)
  68             registrar->AutoRegister(nsnull);
  69
  70         if (argc > 1 && argv[1] != nsnull)
  71         {
  72             char* pathStr = argv[1];
  73             NS_NewNativeLocalFile(nsDependentCString(pathStr), PR_FALSE, getter_AddRefs(topDir));
  74         }
  75
  76         if (!topDir)
  77         {
  78            printf("No Top Dir\n");
  79            return -1;
  80         }
  81         PRInt32 startTime = PR_IntervalNow();
  82
  83         LoopInDir(topDir);
  84
  85         PRInt32 endTime = PR_IntervalNow();
  86
  87         printf("\nTime: %d\n", PR_IntervalToMilliseconds(endTime - startTime));
  88     } // this scopes the nsCOMPtrs
  89     // no nsCOMPtrs are allowed to be alive when you call NS_ShutdownXPCOM
  90     rv = NS_ShutdownXPCOM(nsnull);
  91     NS_ASSERTION(NS_SUCCEEDED(rv), "NS_ShutdownXPCOM failed");
  92     return 0;
  93 }