xpcom/tests/RegFactory.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* ***** BEGIN LICENSE BLOCK *****
   3  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   4  *
   5  * The contents of this file are subject to the Mozilla Public License Version
   6  * 1.1 (the "License"); you may not use this file except in compliance with
   7  * the License. You may obtain a copy of the License at
   8  * http://www.mozilla.org/MPL/
   9  *
  10  * Software distributed under the License is distributed on an "AS IS" basis,
  11  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  12  * for the specific language governing rights and limitations under the
  13  * License.
  14  *
  15  * The Original Code is mozilla.org code.
  16  *
  17  * The Initial Developer of the Original Code is
  18  * Netscape Communications Corporation.
  19  * Portions created by the Initial Developer are Copyright (C) 1998
  20  * the Initial Developer. All Rights Reserved.
  21  *
  22  * Contributor(s):
  23  *
  24  * Alternatively, the contents of this file may be used under the terms of
  25  * either of the GNU General Public License Version 2 or later (the "GPL"),
  26  * or the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  27  * in which case the provisions of the GPL or the LGPL are applicable instead
  28  * of those above. If you wish to allow use of your version of this file only
  29  * under the terms of either the GPL or the LGPL, and not to allow others to
  30  * use your version of this file under the terms of the MPL, indicate your
  31  * decision by deleting the provisions above and replace them with the notice
  32  * and other provisions required by the GPL or the LGPL. If you do not delete
  33  * the provisions above, a recipient may use your version of this file under
  34  * the terms of any one of the MPL, the GPL or the LGPL.
  35  *
  36  * ***** END LICENSE BLOCK ***** */
  37
  38 #include <iostream.h>
  39 #include "plstr.h"
  40 #include "prlink.h"
  41 #include "nsIComponentRegistrar.h"
  42 #include "nsIComponentManager.h"
  43 #include "nsIServiceManager.h"
  44 #include "nsILocalFile.h"
  45 #include "nsCOMPtr.h"
  46 #include "nsString.h"
  47
  48 static PRBool gUnreg = PR_FALSE;
  49
  50 void print_err(nsresult err)
  51 {
  52   switch (err) {
  53   case NS_ERROR_FACTORY_NOT_LOADED:
  54     cerr << "Factory not loaded";
  55     break;
  56   case NS_NOINTERFACE:
  57     cerr << "No Interface";
  58     break;
  59   case NS_ERROR_NULL_POINTER:
  60     cerr << "Null pointer";
  61     break;
  62   case NS_ERROR_OUT_OF_MEMORY:
  63     cerr << "Out of memory";
  64     break;
  65   default:
  66     cerr << hex << err << dec;
  67   }
  68 }
  69
  70 nsresult Register(nsIComponentRegistrar* registrar, const char *path)
  71 {
  72   nsCOMPtr<nsILocalFile> file;
  73   nsresult rv =
  74     NS_NewLocalFile(
  75       NS_ConvertUTF8toUTF16(path),
  76       PR_TRUE,
  77       getter_AddRefs(file));
  78   if (NS_FAILED(rv)) return rv;
  79   rv = registrar->AutoRegister(file);
  80   return rv;
  81 }
  82
  83 nsresult Unregister(const char *path)
  84 {
  85   /* NEEDS IMPLEMENTATION */
  86 #if 0
  87     nsresult res = nsComponentManager::AutoUnregisterComponent(path);
  88   return res;
  89 #else
  90   return NS_ERROR_FAILURE;
  91 #endif
  92 }
  93
  94 int ProcessArgs(nsIComponentRegistrar* registrar, int argc, char *argv[])
  95 {
  96   int i = 1;
  97   nsresult res;
  98
  99   while (i < argc) {
 100     if (argv[i][0] == '-') {
 101       int j;
 102       for (j = 1; argv[i][j] != '\0'; j++) {
 103         switch (argv[i][j]) {
 104         case 'u':
 105           gUnreg = PR_TRUE;
 106           break;
 107         default:
 108           cerr << "Unknown option '" << argv[i][j] << "'\n";
 109         }
 110       }
 111       i++;
 112     } else {
 113       if (gUnreg) {
 114         res = Unregister(argv[i]);
 115         if (NS_SUCCEEDED(res)) {
 116           cout << "Successfully unregistered: " << argv[i] << "\n";
 117         } else {
 118           cerr << "Unregister failed (";
 119           print_err(res);
 120           cerr << "): " << argv[i] << "\n";
 121         }
 122       } else {
 123         res = Register(registrar, argv[i]);
 124         if (NS_SUCCEEDED(res)) {
 125           cout << "Successfully registered: " << argv[i] << "\n";
 126         } else {
 127           cerr << "Register failed (";
 128           print_err(res);
 129           cerr << "): " << argv[i] << "\n";
 130         }
 131       }
 132       i++;
 133     }
 134   }
 135   return 0;
 136 }
 137
 138 int main(int argc, char *argv[])
 139 {
 140   int ret = 0;
 141   nsresult rv;
 142   {
 143     nsCOMPtr<nsIServiceManager> servMan;
 144     rv = NS_InitXPCOM2(getter_AddRefs(servMan), nsnull, nsnull);
 145     if (NS_FAILED(rv)) return -1;
 146     nsCOMPtr<nsIComponentRegistrar> registrar = do_QueryInterface(servMan);
 147     NS_ASSERTION(registrar, "Null nsIComponentRegistrar");
 148
 149     /* With no arguments, RegFactory will autoregister */
 150     if (argc <= 1)
 151     {
 152       rv = registrar->AutoRegister(nsnull);
 153       ret = (NS_FAILED(rv)) ? -1 : 0;
 154     }
 155     else
 156       ret = ProcessArgs(registrar, argc, argv);
 157   } // this scopes the nsCOMPtrs
 158   // no nsCOMPtrs are allowed to be alive when you call NS_ShutdownXPCOM
 159   rv = NS_ShutdownXPCOM( NULL );
 160   NS_ASSERTION(NS_SUCCEEDED(rv), "NS_ShutdownXPCOM failed");
 161   return ret;
 162 }