001 package ezvcard.types; 002 003 import java.io.File; 004 import java.io.IOException; 005 import java.io.InputStream; 006 import java.util.List; 007 008 import ezvcard.VCardSubTypes; 009 import ezvcard.io.SkipMeException; 010 import ezvcard.parameters.ImageTypeParameter; 011 import ezvcard.util.DataUri; 012 import ezvcard.util.HCardElement; 013 014 /* 015 Copyright (c) 2012, Michael Angstadt 016 All rights reserved. 017 018 Redistribution and use in source and binary forms, with or without 019 modification, are permitted provided that the following conditions are met: 020 021 1. Redistributions of source code must retain the above copyright notice, this 022 list of conditions and the following disclaimer. 023 2. Redistributions in binary form must reproduce the above copyright notice, 024 this list of conditions and the following disclaimer in the documentation 025 and/or other materials provided with the distribution. 026 027 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND 028 ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 029 WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 030 DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR 031 ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 032 (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 033 LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 034 ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 035 (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 036 SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 037 038 The views and conclusions contained in the software and documentation are those 039 of the authors and should not be interpreted as representing official policies, 040 either expressed or implied, of the FreeBSD Project. 041 */ 042 043 /** 044 * A company logo. 045 * 046 * <p> 047 * <b>Adding a logo</b> 048 * </p> 049 * 050 * <pre> 051 * VCard vcard = new VCard(); 052 * 053 * //URL 054 * LogoType logo = new LogoType("http://www.company.com/logo.png", ImageTypeParameter.PNG); 055 * vcard.addLogo(logo); 056 * 057 * //binary data 058 * byte data[] = ... 059 * logo = new LogoType(data, ImageTypeParameter.PNG); 060 * vcard.addLogo(logo); 061 * 062 * //if "ImageTypeParameter" does not have the pre-defined constant that you need, then create a new instance 063 * //arg 1: the value of the 2.1/3.0 TYPE parameter 064 * //arg 2: the value to use for the 4.0 MEDIATYPE parameter and for 4.0 data URIs 065 * //arg 3: the file extension of the data type (optional) 066 * ImageKeyTypeParameter param = new ImageTypeParameter("bmp", "image/x-ms-bmp", "bmp"); 067 * logo = new LogoType("http://www.company.com/logo.bmp", param); 068 * vcard.addLogo(logo); 069 * </pre> 070 * 071 * <p> 072 * <b>Getting the logos</b> 073 * </p> 074 * 075 * <pre> 076 * VCard vcard = ... 077 * 078 * int fileCount = 0; 079 * for (LogoType logo : vcard.getLogos()){ 080 * //the logo will have either a URL or a binary data 081 * if (logo.getData() == null){ 082 * System.out.println("Logo URL: " + logo.getUrl()); 083 * } else { 084 * ImageTypeParameter type = logo.getContentType(); 085 * 086 * if (type == null) { 087 * //the vCard may not have any content type data associated with the logo 088 * System.out.println("Saving a logo file..."); 089 * } else { 090 * System.out.println("Saving a \"" + type.getMediaType() + "\" file..."); 091 * } 092 * 093 * String folder; 094 * if (type == ImageTypeParameter.PNG){ //it is safe to use "==" instead of "equals()" 095 * folder = "png-files"; 096 * } else { 097 * folder = "image-files"; 098 * } 099 * 100 * byte data[] = logo.getData(); 101 * String filename = "logo" + fileCount; 102 * if (type != null && type.getExtension() != null){ 103 * filename += "." + type.getExtension(); 104 * } 105 * OutputStream out = new FileOutputStream(new File(folder, filename)); 106 * out.write(data); 107 * out.close(); 108 * fileCount++; 109 * } 110 * } 111 * </pre> 112 * 113 * <p> 114 * vCard property name: LOGO 115 * </p> 116 * <p> 117 * vCard versions: 2.1, 3.0, 4.0 118 * </p> 119 * @author Michael Angstadt 120 */ 121 public class LogoType extends BinaryType<ImageTypeParameter> { 122 public static final String NAME = "LOGO"; 123 124 public LogoType() { 125 super(NAME); 126 } 127 128 /** 129 * @param url the URL to the logo 130 * @param type the content type (e.g. PNG) 131 */ 132 public LogoType(String url, ImageTypeParameter type) { 133 super(NAME, url, type); 134 } 135 136 /** 137 * @param data the binary data of the logo 138 * @param type the content type (e.g. PNG) 139 */ 140 public LogoType(byte[] data, ImageTypeParameter type) { 141 super(NAME, data, type); 142 } 143 144 /** 145 * @param in an input stream to the binary data (will be closed) 146 * @param type the content type (e.g. PNG) 147 * @throws IOException if there's a problem reading from the input stream 148 */ 149 public LogoType(InputStream in, ImageTypeParameter type) throws IOException { 150 super(NAME, in, type); 151 } 152 153 /** 154 * @param file the image file 155 * @param type the content type (e.g. PNG) 156 * @throws IOException if there's a problem reading from the file 157 */ 158 public LogoType(File file, ImageTypeParameter type) throws IOException { 159 super(NAME, file, type); 160 } 161 162 /** 163 * Gets the language that the address is written in. 164 * @return the language or null if not set 165 * @see VCardSubTypes#getLanguage 166 */ 167 public String getLanguage() { 168 return subTypes.getLanguage(); 169 } 170 171 /** 172 * Sets the language that the address is written in. 173 * @param language the language or null to remove 174 * @see VCardSubTypes#setLanguage 175 */ 176 public void setLanguage(String language) { 177 subTypes.setLanguage(language); 178 } 179 180 @Override 181 protected void doUnmarshalHtml(HCardElement element, List<String> warnings) { 182 String elementName = element.tagName(); 183 if ("img".equals(elementName)) { 184 String src = element.absUrl("src"); 185 if (src.length() > 0) { 186 try { 187 DataUri uri = new DataUri(src); 188 ImageTypeParameter mediaType = buildMediaTypeObj(uri.getContentType()); 189 setData(uri.getData(), mediaType); 190 } catch (IllegalArgumentException e) { 191 //TODO create buildTypeObjFromExtension() method 192 setUrl(src, null); 193 } 194 } else { 195 throw new SkipMeException("<img> tag does not have a \"src\" attribute."); 196 } 197 } else { 198 super.doUnmarshalHtml(element, warnings); 199 } 200 } 201 202 @Override 203 protected ImageTypeParameter buildTypeObj(String type) { 204 ImageTypeParameter param = ImageTypeParameter.valueOf(type); 205 if (param == null) { 206 param = new ImageTypeParameter(type, "image/" + type, null); 207 } 208 return param; 209 } 210 211 @Override 212 protected ImageTypeParameter buildMediaTypeObj(String mediaType) { 213 ImageTypeParameter p = ImageTypeParameter.findByMediaType(mediaType); 214 if (p == null) { 215 int slashPos = mediaType.indexOf('/'); 216 String type; 217 if (slashPos == -1 || slashPos < mediaType.length() - 1) { 218 type = ""; 219 } else { 220 type = mediaType.substring(slashPos + 1); 221 } 222 p = new ImageTypeParameter(type, mediaType, null); 223 } 224 return p; 225 } 226 }