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)
  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
  66         if (argc > 1 && argv[1] != nsnull)
  67         {
  68             char* pathStr = argv[1];
  69             NS_NewNativeLocalFile(nsDependentCString(pathStr), PR_FALSE, getter_AddRefs(topDir));
  70         }
  71
  72         if (!topDir)
  73         {
  74            printf("No Top Dir\n");
  75            return -1;
  76         }
  77         PRInt32 startTime = PR_IntervalNow();
  78
  79         LoopInDir(topDir);
  80
  81         PRInt32 endTime = PR_IntervalNow();
  82
  83         printf("\nTime: %d\n", PR_IntervalToMilliseconds(endTime - startTime));
  84     } // this scopes the nsCOMPtrs
  85     // no nsCOMPtrs are allowed to be alive when you call NS_ShutdownXPCOM
  86     rv = NS_ShutdownXPCOM(nsnull);
  87     NS_ASSERTION(NS_SUCCEEDED(rv), "NS_ShutdownXPCOM failed");
  88     return 0;
  89 }