1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254
/* * $Id: MimeHeaders.java,v 1.4 2004/04/02 01:24:17 ofung Exp $ * $Revision: 1.4 $ * $Date: 2004/04/02 01:24:17 $ */ /* * Copyright 2005 Sun Microsystems, Inc. All rights reserved. * SUN PROPRIETARY/CONFIDENTIAL. Use is subject to license terms. */ package javax.xml.soap; import java.util.Iterator; import java.util.Vector; /** * A container for <code>MimeHeader</code> objects, which represent * the MIME headers present in a MIME part of a message. * * <p>This class is used primarily when an application wants to * retrieve specific attachments based on certain MIME headers and * values. This class will most likely be used by implementations of * <code>AttachmentPart</code> and other MIME dependent parts of the SAAJ * API. * @see SOAPMessage#getAttachments * @see AttachmentPart */ public class MimeHeaders { private Vector headers; /** * Constructs a default <code>MimeHeaders</code> object initialized with * an empty <code>Vector</code> object. */ public MimeHeaders() { headers = new Vector(); } /** * Returns all of the values for the specified header as an array of * <code>String</code> objects. * * @param name the name of the header for which values will be returned * @return a <code>String</code> array with all of the values for the * specified header * @see #setHeader */ public String[] getHeader(String name) { Vector values = new Vector(); for(int i = 0; i < headers.size(); i++) { MimeHeader hdr = (MimeHeader) headers.elementAt(i); if (hdr.getName().equalsIgnoreCase(name) && hdr.getValue() != null) values.addElement(hdr.getValue()); } if (values.size() == 0) return null; String r[] = new String[values.size()]; values.copyInto(r); return r; } /** * Replaces the current value of the first header entry whose name matches * the given name with the given value, adding a new header if no existing header * name matches. This method also removes all matching headers after the first one. * <P> * Note that RFC822 headers can contain only US-ASCII characters. * * @param name a <code>String</code> with the name of the header for * which to search * @param value a <code>String</code> with the value that will replace the * current value of the specified header * * @exception IllegalArgumentException if there was a problem in the * mime header name or the value being set * @see #getHeader */ public void setHeader(String name, String value) { boolean found = false; if ((name == null) || name.equals("")) throw new IllegalArgumentException("Illegal MimeHeader name"); for(int i = 0; i < headers.size(); i++) { MimeHeader hdr = (MimeHeader) headers.elementAt(i); if (hdr.getName().equalsIgnoreCase(name)) { if (!found) { headers.setElementAt(new MimeHeader(hdr.getName(), value), i); found = true; } else headers.removeElementAt(i--); } } if (!found) addHeader(name, value); } /** * Adds a <code>MimeHeader</code> object with the specified name and value * to this <code>MimeHeaders</code> object's list of headers. * <P> * Note that RFC822 headers can contain only US-ASCII characters. * * @param name a <code>String</code> with the name of the header to * be added * @param value a <code>String</code> with the value of the header to * be added * * @exception IllegalArgumentException if there was a problem in the * mime header name or value being added */ public void addHeader(String name, String value) { if ((name == null) || name.equals("")) throw new IllegalArgumentException("Illegal MimeHeader name"); int pos = headers.size(); for(int i = pos - 1 ; i >= 0; i--) { MimeHeader hdr = (MimeHeader) headers.elementAt(i); if (hdr.getName().equalsIgnoreCase(name)) { headers.insertElementAt(new MimeHeader(name, value), i+1); return; } } headers.addElement(new MimeHeader(name, value)); } /** * Remove all <code>MimeHeader</code> objects whose name matches the * given name. * * @param name a <code>String</code> with the name of the header for * which to search */ public void removeHeader(String name) { for(int i = 0; i < headers.size(); i++) { MimeHeader hdr = (MimeHeader) headers.elementAt(i); if (hdr.getName().equalsIgnoreCase(name)) headers.removeElementAt(i--); } } /** * Removes all the header entries from this <code>MimeHeaders</code> object. */ public void removeAllHeaders() { headers.removeAllElements(); } /** * Returns all the <code>MimeHeader</code>s in this <code>MimeHeaders</code> object. * * @return an <code>Iterator</code> object over this <code>MimeHeaders</code> * object's list of <code>MimeHeader</code> objects */ public Iterator getAllHeaders() { return headers.iterator(); } class MatchingIterator implements Iterator { private boolean match; private Iterator iterator; private String[] names; private Object nextHeader; MatchingIterator(String[] names, boolean match) { this.match = match; this.names = names; this.iterator = headers.iterator(); } private Object nextMatch() { next: while (iterator.hasNext()) { MimeHeader hdr = (MimeHeader) iterator.next(); if (names == null) return match ? null : hdr; for(int i = 0; i < names.length; i++) if (hdr.getName().equalsIgnoreCase(names[i])) if (match) return hdr; else continue next; if (!match) return hdr; } return null; } public boolean hasNext() { if (nextHeader == null) nextHeader = nextMatch(); return nextHeader != null; } public Object next() { // hasNext should've prefetched the header for us, // return it. if (nextHeader != null) { Object ret = nextHeader; nextHeader = null; return ret; } if (hasNext()) return nextHeader; return null; } public void remove() { iterator.remove(); } } /** * Returns all the <code>MimeHeader</code> objects whose name matches * a name in the given array of names. * * @param names an array of <code>String</code> objects with the names * for which to search * @return an <code>Iterator</code> object over the <code>MimeHeader</code> * objects whose name matches one of the names in the given list */ public Iterator getMatchingHeaders(String[] names) { return new MatchingIterator(names, true); } /** * Returns all of the <code>MimeHeader</code> objects whose name does not * match a name in the given array of names. * * @param names an array of <code>String</code> objects with the names * for which to search * @return an <code>Iterator</code> object over the <code>MimeHeader</code> * objects whose name does not match one of the names in the given list */ public Iterator getNonMatchingHeaders(String[] names) { return new MatchingIterator(names, false); } }