001package ezvcard.io.scribe;
002
003import ezvcard.io.CannotParseException;
004import ezvcard.io.ParseContext;
005import ezvcard.io.html.HCardElement;
006import ezvcard.parameter.ImageType;
007import ezvcard.property.ImageProperty;
008import ezvcard.util.DataUri;
009
010/*
011 Copyright (c) 2012-2023, Michael Angstadt
012 All rights reserved.
013
014 Redistribution and use in source and binary forms, with or without
015 modification, are permitted provided that the following conditions are met: 
016
017 1. Redistributions of source code must retain the above copyright notice, this
018 list of conditions and the following disclaimer. 
019 2. Redistributions in binary form must reproduce the above copyright notice,
020 this list of conditions and the following disclaimer in the documentation
021 and/or other materials provided with the distribution. 
022
023 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
024 ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
025 WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
026 DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
027 ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
028 (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
029 LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
030 ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
031 (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
032 SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
033 */
034
035/**
036 * Marshals properties that contain images.
037 * @param <T> the property class
038 * @author Michael Angstadt
039 */
040public abstract class ImagePropertyScribe<T extends ImageProperty> extends BinaryPropertyScribe<T, ImageType> {
041        public ImagePropertyScribe(Class<T> clazz, String propertyName) {
042                super(clazz, propertyName);
043        }
044
045        @Override
046        protected ImageType _mediaTypeFromTypeParameter(String type) {
047                return ImageType.get(type, null, null);
048        }
049
050        @Override
051        protected ImageType _mediaTypeFromMediaTypeParameter(String mediaType) {
052                return ImageType.get(null, mediaType, null);
053        }
054
055        @Override
056        protected ImageType _mediaTypeFromFileExtension(String extension) {
057                return ImageType.find(null, null, extension);
058        }
059
060        @Override
061        protected T _parseHtml(HCardElement element, ParseContext context) {
062                String elementName = element.tagName();
063                if (!"img".equals(elementName)) {
064                        return super._parseHtml(element, context);
065                }
066
067                String src = element.absUrl("src");
068                if (src.isEmpty()) {
069                        throw new CannotParseException(13);
070                }
071
072                try {
073                        DataUri uri = DataUri.parse(src);
074                        ImageType mediaType = _mediaTypeFromMediaTypeParameter(uri.getContentType());
075                        return _newInstance(uri.getData(), mediaType);
076                } catch (IllegalArgumentException e) {
077                        //not a data URI
078                        String extension = getFileExtension(src);
079                        ImageType mediaType = (extension == null) ? null : _mediaTypeFromFileExtension(extension);
080                        return _newInstance(src, mediaType);
081                }
082        }
083}