err.no Git - sope/blob - sope-mime/NGMime/NGMimeHeaderFieldGeneratorSet.m

   1 /*
   2   Copyright (C) 2000-2004 SKYRIX Software AG
   3
   4   This file is part of OpenGroupware.org.
   5
   6   OGo 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   OGo 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 OGo; 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 // $Id$
  22
  23 #include "NGMimeHeaderFieldGenerator.h"
  24 #include "NGMimeHeaderFields.h"
  25 #include "NGMimePartParser.h"
  26 #include "common.h"
  27
  28 @implementation NGMimeHeaderFieldGeneratorSet
  29
  30 + (int)version {
  31   return 2;
  32 }
  33
  34 static NGMimeHeaderFieldGeneratorSet *rfc822Set = nil;
  35
  36 + (id)headerFieldGenerator {
  37   return [[[self alloc] init] autorelease];
  38 }
  39
  40 + (id)headerFieldGeneratorSet {
  41   return [[[self alloc] init] autorelease];
  42 }
  43
  44 + (id)defaultRfc822HeaderFieldGeneratorSet {
  45   static NGMimeHeaderNames *Fields = NULL;
  46   id gen;
  47
  48   if (!Fields)
  49     Fields = (NGMimeHeaderNames *)[NGMimePartParser headerFieldNames];
  50
  51   if (rfc822Set)
  52     return rfc822Set;
  53
  54   rfc822Set = [[self alloc] init];
  55
  56   gen = [NGMimeContentDispositionHeaderFieldGenerator headerFieldGenerator];
  57   if (gen)
  58     [rfc822Set setGenerator:gen forField:Fields->contentDisposition];
  59
  60   if ((gen = [NGMimeContentLengthHeaderFieldGenerator headerFieldGenerator]))
  61     [rfc822Set setGenerator:gen forField:Fields->contentLength];
  62   if ((gen = [NGMimeContentTypeHeaderFieldGenerator headerFieldGenerator]))
  63     [rfc822Set setGenerator:gen forField:Fields->contentType];
  64   if ((gen = [NGMimeRFC822DateHeaderFieldGenerator headerFieldGenerator]))
  65     [rfc822Set setGenerator:gen forField:Fields->date];
  66
  67   if ((gen = [NGMimeAddressHeaderFieldGenerator headerFieldGenerator])) {
  68     [rfc822Set setGenerator:gen forField:@"resent-from"];
  69     [rfc822Set setGenerator:gen forField:@"resent-to"];
  70     [rfc822Set setGenerator:gen forField:Fields->to];
  71     [rfc822Set setGenerator:gen forField:Fields->cc];
  72     [rfc822Set setGenerator:gen forField:@"bcc"];
  73     [rfc822Set setGenerator:gen forField:Fields->from];
  74   }
  75
  76   if ((gen = [NGMimeStringHeaderFieldGenerator headerFieldGenerator]))
  77     [rfc822Set setDefaultGenerator:gen];
  78
  79   return rfc822Set;
  80 }
  81
  82 - (id)init {
  83   return [self initWithDefaultGenerator:nil];
  84 }
  85
  86 - (id)initWithDefaultGenerator:(id<NGMimeHeaderFieldGenerator>)_gen {
  87   if ((self = [super init])) {
  88     self->fieldNameToGenerate =
  89       [[NSMutableDictionary allocWithZone:[self zone]]
  90                             initWithCapacity:16];
  91     self->defaultGenerator = [_gen retain];
  92   }
  93   return self;
  94 }
  95
  96 - (void)dealloc {
  97   [self->fieldNameToGenerate release];
  98   [self->defaultGenerator    release];
  99   [super dealloc];
 100 }
 101
 102 /* accessors */
 103
 104 - (void)setGenerator:(id<NGMimeHeaderFieldGenerator>)_gen
 105   forField:(NSString *)_name
 106 {
 107   [self->fieldNameToGenerate setObject:_gen forKey:_name];
 108 }
 109
 110 - (void)setDefaultGenerator:(id<NGMimeHeaderFieldGenerator>)_gen {
 111   ASSIGN(self->defaultGenerator, _gen);
 112 }
 113
 114 - (id<NGMimeHeaderFieldGenerator>)_gen {
 115   return self->defaultGenerator;
 116 }
 117
 118 /* operation */
 119
 120 - (NSData *)generateDataForHeaderFieldNamed:(NSString *)_headerField
 121   value:(id)_value
 122 {
 123   NGMimeHeaderFieldGenerator *gen = nil;
 124
 125   if ((gen = [self->fieldNameToGenerate objectForKey:_headerField]) == nil)
 126     gen = (NGMimeHeaderFieldGenerator *)self->defaultGenerator;
 127
 128   if (gen == nil) {
 129     NSLog(@"WARNING(%s): no defaultGenerator is set", __PRETTY_FUNCTION__);
 130     return [NSData data];
 131   }
 132   return [gen generateDataForHeaderFieldNamed:_headerField
 133               value:_value];
 134 }
 135
 136 @end /* NGMimeHeaderFieldGeneratorSet */