err.no Git - sope/blob - xmlrpc_call/XmlRpcClientTool.m

   1 /*
   2   Copyright (C) 2004-2005 Helge Hess
   3
   4   This file is part of SOPE.
   5
   6   SOPE is free software; you can redistribute it and/or modify it under
   7   the terms of the GNU Lesser General Public License as published by the
   8   Free Software Foundation; either version 2, or (at your option) any
   9   later version.
  10
  11   SOPE is distributed in the hope that it will be useful, but WITHOUT ANY
  12   WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13   FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
  14   License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public
  17   License along with SOPE; see the file COPYING.  If not, write to the
  18   Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
  19   02111-1307, USA.
  20 */
  21
  22 #include "XmlRpcClientTool.h"
  23 #include "HandleCredentialsClient.h"
  24 #include <NGStreams/NGLocalSocketAddress.h>
  25 #include <NGExtensions/NSString+Ext.h>
  26 #include "common.h"
  27
  28 #define EXIT_FAIL -1
  29
  30 @interface NSObject(Printing)
  31 - (void)printWithTool:(XmlRpcClientTool *)_tool;
  32 @end
  33
  34 @implementation XmlRpcClientTool
  35
  36 /* initialization */
  37
  38 - (id)init {
  39   return [self initWithArguments:nil];
  40 }
  41
  42 - (id)initWithArguments:(NSArray *)_arguments {
  43   if ((self = [super init])) {
  44     NSUserDefaults *ud;
  45     NSString *s;
  46     int argc;
  47
  48     argc = [_arguments count];
  49     if(argc == 1) {
  50       [self help:[_arguments objectAtIndex:0]];
  51       return nil;
  52     }
  53
  54     s = [_arguments objectAtIndex:1];
  55     if (![self initXmlRpcClientWithStringURL:s]) {
  56       printf("Error initializing the XML-RPC client\n");
  57       [self release];
  58       return nil;
  59     }
  60
  61     if (argc > 2) {
  62       [self initMethodCall:_arguments];
  63     }
  64     else {
  65       self->methodName = @"system.listMethods";
  66       self->parameters = nil;
  67     }
  68
  69     ud = [NSUserDefaults standardUserDefaults];
  70     if ([ud boolForKey:@"forceauth"]) {
  71       [self->client setUserName:[ud stringForKey:@"login"]];
  72       [self->client setPassword:[ud stringForKey:@"password"]];
  73     }
  74     else {
  75       [self->client setDefLogin:[ud stringForKey:@"login"]];
  76       [self->client setDefPassword:[ud stringForKey:@"password"]];
  77     }
  78   }
  79   return self;
  80 }
  81
  82 - (BOOL)initXmlRpcClientWithStringURL:(NSString *)_url {
  83   NSURL *url = nil;
  84
  85   if (![_url isAbsoluteURL]) {
  86     /* make a raw, Unix domain socket connection */
  87     NGLocalSocketAddress *addr;
  88
  89     addr = [NGLocalSocketAddress addressWithPath:_url];
  90     self->client = [[HandleCredentialsClient alloc] initWithRawAddress:addr];
  91     return YES;
  92   }
  93
  94   if ((url = [NSURL URLWithString:_url]) != nil) {
  95 #if 0
  96     if ((uri = [url path]) == nil)
  97       uri = @"/RPC2";
  98 #endif
  99
 100     self->client = [[HandleCredentialsClient alloc] initWithURL:url];
 101     return YES;
 102
 103   }
 104   else {
 105     printf("Invalid URL\n");
 106     return NO;
 107   }
 108 }
 109
 110 - (void)initMethodCall:(NSArray *)_arguments {
 111   self->methodName = [_arguments objectAtIndex:2];
 112
 113   if ([_arguments count] > 2) {
 114     NSRange range = NSMakeRange(3, [_arguments count] - 3);
 115     self->parameters = [[_arguments subarrayWithRange:range] retain];
 116   }
 117 }
 118
 119 - (void)dealloc {
 120   [self->client      release];
 121   [self->methodName  release];
 122   [self->parameters  release];
 123   [super dealloc];
 124 }
 125
 126 /* printing */
 127
 128 - (void)printElement:(id)element {
 129   [element printWithTool:self];
 130 }
 131
 132 /* printing objects */
 133
 134 - (void)printDictionary:(NSDictionary *)_dict {
 135   NSEnumerator *dictEnum;
 136   NSString     *dictKey;
 137
 138   dictEnum = [_dict keyEnumerator];
 139   while((dictKey = [dictEnum nextObject]) != nil) {
 140     printf("%s=", [dictKey cString]);
 141     [self printElement:[_dict objectForKey:dictKey]];
 142   }
 143 }
 144
 145 - (void)printArray:(NSArray *)_array {
 146   NSEnumerator *arrayEnum;
 147   id arrayElem;
 148
 149   arrayEnum = [_array objectEnumerator];
 150   while((arrayElem = [arrayEnum nextObject]) != nil) {
 151     [self printElement:arrayElem];
 152   }
 153 }
 154
 155 - (void)help:(NSString *)pn {
 156   fprintf(stderr,
 157           "usage:    %s <url> [<method-name>] [<arg1>,...]\n"
 158           "  sample: %s http://localhost:20000/RPC2 bc 1 2\n",
 159           [pn cString], [pn cString]);
 160 }
 161
 162 /* running */
 163
 164 - (int)run {
 165   int  exitCode  = 0;
 166   int  loopCount = 0;
 167   id   result;
 168
 169   if (self->client == nil) {
 170     NSLog(@"missing XML-RPC client object ...");
 171     return EXIT_FAIL;
 172   }
 173
 174   do {
 175     result = [self->client
 176                   invokeMethodNamed:self->methodName
 177                   parameters:self->parameters];
 178     loopCount++;
 179   }
 180   while ((result == nil) && loopCount < 20);
 181
 182   if (result == nil) {
 183     NSLog(@"call failed, no result (looped %i times) ?!", loopCount);
 184     return EXIT_FAIL;
 185   }
 186
 187   if ([[NSUserDefaults standardUserDefaults] boolForKey:@"plist"]) {
 188     NSString *s;
 189
 190     s = [result description];
 191     printf("%s\n", [s cString]);
 192   }
 193   else
 194     [self printElement:result];
 195
 196   if ([result isKindOfClass:[NSException class]]) {
 197     exitCode = 255;
 198   }
 199
 200   return exitCode;
 201 }
 202
 203 @end /* XmlRpcClientTool */