Smack/smack-experimental/src/main/java/org/jivesoftware/smackx/file_metadata/element/FileMetadataElement.java

313 lines
10 KiB
Java

/**
*
* Copyright 2020 Paul Schaub
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.jivesoftware.smackx.file_metadata.element;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.jivesoftware.smack.packet.ExtensionElement;
import org.jivesoftware.smack.packet.XmlEnvironment;
import org.jivesoftware.smack.util.CollectionUtil;
import org.jivesoftware.smack.util.EqualsUtil;
import org.jivesoftware.smack.util.HashCode;
import org.jivesoftware.smack.util.StringUtils;
import org.jivesoftware.smack.util.XmlStringBuilder;
import org.jivesoftware.smackx.hashes.HashManager;
import org.jivesoftware.smackx.hashes.element.HashElement;
import org.jivesoftware.smackx.thumbnails.element.ThumbnailElement;
/**
* File metadata element as defined in XEP-0446: File Metadata Element.
* This element is used in a generic way to provide information about files, e.g. during file sharing.
*/
public final class FileMetadataElement implements ExtensionElement {
public static final String ELEMENT = "file";
public static final String NAMESPACE = "urn:xmpp:file:metadata:0";
public static final String ELEM_DATE = "date";
public static final String ELEM_HEIGHT = "height";
public static final String ELEM_WIDTH = "width";
public static final String ELEM_DESC = "desc";
public static final String ELEM_LENGTH = "length";
public static final String ELEM_MEDIA_TYPE = "media-type";
public static final String ELEM_NAME = "name";
public static final String ELEM_SIZE = "size";
private final Date date;
private final Integer height;
private final Integer width;
private final Map<String, String> descriptions;
private final Map<HashManager.ALGORITHM, HashElement> hashElements;
private final Long length;
private final String mediaType;
private final String name;
private final Long size;
private final List<ThumbnailElement> thumbnails;
private FileMetadataElement(Date date, Integer height, Integer width, Map<String, String> descriptions,
Map<HashManager.ALGORITHM, HashElement> hashElements, Long length,
String mediaType, String name, Long size,
List<ThumbnailElement> thumbnails) {
this.date = date;
this.height = height;
this.width = width;
this.descriptions = CollectionUtil.cloneAndSeal(descriptions);
this.hashElements = CollectionUtil.cloneAndSeal(hashElements);
this.length = length;
this.mediaType = mediaType;
this.name = name;
this.size = size;
this.thumbnails = CollectionUtil.cloneAndSeal(thumbnails);
}
@Override
public XmlStringBuilder toXML(XmlEnvironment xmlEnvironment) {
XmlStringBuilder sb = new XmlStringBuilder(this)
.rightAngleBracket()
.optElement(ELEM_DATE, date)
.optElement(ELEM_HEIGHT, height)
.optElement(ELEM_WIDTH, width);
for (String key : descriptions.keySet()) {
sb.halfOpenElement(ELEM_DESC)
.optXmlLangAttribute(key)
.rightAngleBracket()
.append(descriptions.get(key))
.closeElement(ELEM_DESC);
}
sb.append(hashElements.values())
.optElement(ELEM_LENGTH, length != null ? Long.toString(length) : null)
.optElement(ELEM_MEDIA_TYPE, mediaType)
.optElement(ELEM_NAME, name)
.optElement(ELEM_SIZE, size != null ? Long.toString(size) : null)
.append(thumbnails);
return sb.closeElement(this);
}
@Override
public String getNamespace() {
return NAMESPACE;
}
@Override
public String getElementName() {
return ELEMENT;
}
public Date getDate() {
return date;
}
public Integer getHeight() {
return height;
}
public Integer getWidth() {
return width;
}
public Map<String, String> getDescriptions() {
return Collections.unmodifiableMap(descriptions);
}
public String getDescription() {
return getDescription(getLanguage());
}
public String getDescription(String lang) {
return descriptions.get(lang != null ? lang : "");
}
public Map<HashManager.ALGORITHM, HashElement> getHashElements() {
return Collections.unmodifiableMap(hashElements);
}
public HashElement getHashElement(HashManager.ALGORITHM algorithm) {
return hashElements.get(algorithm);
}
public Long getLength() {
return length;
}
public String getMediaType() {
return mediaType;
}
/**
* Return the name of the file.
*
* @return escaped name
*/
public String getName() {
if (name == null) {
return null;
}
try {
return URLEncoder.encode(name, "UTF-8");
} catch (UnsupportedEncodingException e) {
throw new AssertionError(e); // UTF-8 MUST be supported
}
}
public String getRawName() {
return name;
}
public Long getSize() {
return size;
}
public List<ThumbnailElement> getThumbnails() {
return Collections.unmodifiableList(thumbnails);
}
@Override
public int hashCode() {
return HashCode.builder()
.append(getElementName())
.append(getNamespace())
.append(getDate())
.append(getDescriptions())
.append(getHeight())
.append(getWidth())
.append(getHashElements())
.append(getLength())
.append(getMediaType())
.append(getRawName())
.append(getSize())
.append(getThumbnails())
.build();
}
@Override
public boolean equals(Object other) {
return EqualsUtil.equals(this, other, (equalsBuilder, o) -> equalsBuilder
.append(getElementName(), o.getElementName())
.append(getNamespace(), o.getNamespace())
.append(getDate(), o.getDate())
.append(getDescriptions(), o.getDescriptions())
.append(getHeight(), o.getHeight())
.append(getWidth(), o.getWidth())
.append(getHashElements(), o.getHashElements())
.append(getLength(), o.getLength())
.append(getMediaType(), o.getMediaType())
.append(getRawName(), o.getRawName())
.append(getSize(), o.getSize())
.append(getThumbnails(), o.getThumbnails()));
}
public static Builder builder() {
return new Builder();
}
public static class Builder {
private Date date;
private Integer height;
private Integer width;
private Map<String, String> descriptions = new HashMap<>();
private Map<HashManager.ALGORITHM, HashElement> hashElements = new HashMap<>();
private Long length;
private String mediaType;
private String name;
private Long size;
private List<ThumbnailElement> thumbnails = new ArrayList<>();
public Builder setModificationDate(Date date) {
this.date = date;
return this;
}
public Builder setDimensions(int width, int height) {
return setHeight(height).setWidth(width);
}
public Builder setHeight(int height) {
if (height <= 0) {
throw new IllegalArgumentException("Height must be a positive number");
}
this.height = height;
return this;
}
public Builder setWidth(int width) {
if (width <= 0) {
throw new IllegalArgumentException("Width must be a positive number");
}
this.width = width;
return this;
}
public Builder addDescription(String description) {
return addDescription(description, null);
}
public Builder addDescription(String description, String language) {
this.descriptions.put(language != null ? language : "", StringUtils.requireNotNullNorEmpty(description, "Description MUST NOT be null nor empty"));
return this;
}
public Builder addHash(HashElement hashElement) {
hashElements.put(hashElement.getAlgorithm(), hashElement);
return this;
}
public Builder setLength(long length) {
if (length < 0) {
throw new IllegalArgumentException("Length cannot be negative.");
}
this.length = length;
return this;
}
public Builder setMediaType(String mediaType) {
this.mediaType = StringUtils.requireNotNullNorEmpty(mediaType, "Media-Type MUST NOT be null nor empty");
return this;
}
public Builder setName(String name) {
this.name = StringUtils.requireNotNullNorEmpty(name, "Name MUST NOT be null nor empty");
return this;
}
public Builder setSize(long size) {
if (size < 0) {
throw new IllegalArgumentException("Size MUST NOT be negative.");
}
this.size = size;
return this;
}
public Builder addThumbnail(ThumbnailElement thumbnail) {
thumbnails.add(thumbnail);
return this;
}
public FileMetadataElement build() {
return new FileMetadataElement(date, height, width, descriptions, hashElements, length,
mediaType, name, size, thumbnails);
}
}
}