cpputools/source/regsingleton/regsingleton.cxx

   1 /*************************************************************************
   2  *
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * Copyright 2008 by Sun Microsystems, Inc.
   6  *
   7  * OpenOffice.org - a multi-platform office productivity suite
   8  *
   9  * $RCSfile: regsingleton.cxx,v $
  10  * $Revision: 1.5 $
  11  *
  12  * This file is part of OpenOffice.org.
  13  *
  14  * OpenOffice.org is free software: you can redistribute it and/or modify
  15  * it under the terms of the GNU Lesser General Public License version 3
  16  * only, as published by the Free Software Foundation.
  17  *
  18  * OpenOffice.org is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  * GNU Lesser General Public License version 3 for more details
  22  * (a copy is included in the LICENSE file that accompanied this code).
  23  *
  24  * You should have received a copy of the GNU Lesser General Public License
  25  * version 3 along with OpenOffice.org.  If not, see
  26  * <http://www.openoffice.org/license.html>
  27  * for a copy of the LGPLv3 License.
  28  *
  29  ************************************************************************/
  30
  31 #include <stdio.h>
  32
  33 #include "sal/main.h"
  34 #include <osl/diagnose.h>
  35 #include <osl/file.h>
  36
  37 #include <cppuhelper/bootstrap.hxx>
  38
  39 #include <com/sun/star/registry/XSimpleRegistry.hpp>
  40
  41 #define OUSTR(x) ::rtl::OUString( RTL_CONSTASCII_USTRINGPARAM(x) )
  42
  43
  44 using namespace ::rtl;
  45 using namespace ::com::sun::star;
  46 using namespace ::com::sun::star::uno;
  47
  48 static void print_options() SAL_THROW( () )
  49 {
  50     printf(
  51         "\nusage: regsingleton [-r|-ra] registry_file singleton_name[=service_name] ...\n\n"
  52         "Inserts a singleton entry into rdb.\n"
  53         "Option -r revokes given entries, -ra revokes all entries.\n" );
  54 }
  55
  56 //==================================================================================================
  57 SAL_IMPLEMENT_MAIN_WITH_ARGS(argc, argv)
  58 {
  59     if (argc < 3)
  60     {
  61         print_options();
  62         return 1;
  63     }
  64
  65     bool insert_entry = true;
  66     bool remove_all = false;
  67     int nPos = 1;
  68     if ('-' == argv[ nPos ][ 0 ] && 'r' == argv[ nPos ][ 1 ])
  69     {
  70         if ('a' == argv[ nPos ][ 2 ] && '\0' == argv[ nPos ][ 3 ])
  71         {
  72             remove_all = true;
  73         }
  74         else if ('\0' != argv[ nPos ][ 2 ])
  75         {
  76             print_options();
  77             return 1;
  78         }
  79         insert_entry = false;
  80         ++nPos;
  81     }
  82
  83     OUString sys_path( OUString::createFromAscii( argv[ nPos ] ) );
  84     OUString file_url;
  85     oslFileError rc = osl_getFileURLFromSystemPath( sys_path.pData, &file_url.pData );
  86     if (osl_File_E_None != rc)
  87     {
  88         fprintf( stderr, "\nerror: cannot make file url out of %s\n", argv[ nPos ] );
  89         return 1;
  90     }
  91     ++nPos;
  92
  93     try
  94     {
  95         Reference< registry::XSimpleRegistry > xSimReg( ::cppu::createSimpleRegistry() );
  96         xSimReg->open( file_url, sal_False, sal_True );
  97         Reference< registry::XRegistryKey > xRoot( xSimReg->getRootKey() );
  98
  99         if (remove_all)
 100         {
 101             try
 102             {
 103                 xRoot->deleteKey( OUSTR("SINGLETONS") );
 104             }
 105             catch (registry::InvalidRegistryException & exc)
 106             {
 107                 OString cstr_msg(
 108                     OUStringToOString( exc.Message, RTL_TEXTENCODING_ASCII_US ) );
 109                 fprintf(
 110                     stderr, "\nwarning: removing all singletons failed: %s\n",
 111                     cstr_msg.getStr() );
 112             }
 113         }
 114         else
 115         {
 116             Reference< registry::XRegistryKey > xKey( xRoot->openKey( OUSTR("SINGLETONS") ) );
 117             if (! xKey.is())
 118                 xKey = xRoot->createKey( OUSTR("SINGLETONS") );
 119
 120             for ( ; nPos < argc; ++nPos )
 121             {
 122                 OUString singleton( OUString::createFromAscii( argv[ nPos ] ) );
 123                 OUString service;
 124                 sal_Int32 eq = singleton.indexOf( '=' );
 125                 if (eq >= 0)
 126                 {
 127                     service = singleton.copy( eq +1 );
 128                     singleton = singleton.copy( 0, eq );
 129                 }
 130
 131                 if (insert_entry)
 132                 {
 133                     if (service.getLength())
 134                     {
 135                         Reference< registry::XRegistryKey > xEntry( xKey->openKey( singleton ) );
 136                         if (! xEntry.is())
 137                             xEntry = xKey->createKey( singleton );
 138                         xEntry->setStringValue( service );
 139                     }
 140                     else
 141                     {
 142                         OString entry( OUStringToOString( singleton, RTL_TEXTENCODING_ASCII_US ) );
 143                         fprintf(
 144                             stderr, "\nwarning: no service name given for singleton %s!\n",
 145                             entry.getStr() );
 146                     }
 147                 }
 148                 else
 149                 {
 150                     try
 151                     {
 152                         xKey->deleteKey( singleton );
 153                     }
 154                     catch (registry::InvalidRegistryException & exc)
 155                     {
 156                         OString cstr_singleton(
 157                             OUStringToOString( singleton, RTL_TEXTENCODING_ASCII_US ) );
 158                         OString cstr_msg(
 159                             OUStringToOString( exc.Message, RTL_TEXTENCODING_ASCII_US ) );
 160                         fprintf(
 161                             stderr, "\nwarning: singleton %s is not registered: %s\n",
 162                             cstr_singleton.getStr(), cstr_msg.getStr() );
 163                     }
 164                 }
 165             }
 166         }
 167
 168         return 0;
 169     }
 170     catch (Exception & rExc)
 171     {
 172         OString msg( OUStringToOString( rExc.Message, RTL_TEXTENCODING_ASCII_US ) );
 173         fprintf( stderr, "\nerror: %s\n", msg.getStr() );
 174         return 1;
 175     }
 176 }