package org.springframework.boot.origin;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URI;
import java.net.URL;
import java.nio.channels.ReadableByteChannel;
import org.springframework.core.io.Resource;
import org.springframework.core.io.WritableResource;
import org.springframework.util.Assert;
import org.springframework.util.ObjectUtils;
public class OriginTrackedResource implements Resource, OriginProvider {
private final Resource resource;
private final Origin origin;
OriginTrackedResource(Resource resource, Origin origin) {
Assert.notNull(resource, "Resource must not be null");
this.resource = resource;
this.origin = origin;
}
@Override
public InputStream getInputStream() throws IOException {
return getResource().getInputStream();
}
@Override
public boolean exists() {
return getResource().exists();
}
@Override
public boolean isReadable() {
return getResource().isReadable();
}
@Override
public boolean isOpen() {
return getResource().isOpen();
}
@Override
public boolean isFile() {
return getResource().isFile();
}
@Override
public URL getURL() throws IOException {
return getResource().getURL();
}
@Override
public URI getURI() throws IOException {
return getResource().getURI();
}
@Override
public File getFile() throws IOException {
return getResource().getFile();
}
@Override
public ReadableByteChannel readableChannel() throws IOException {
return getResource().readableChannel();
}
@Override
public long contentLength() throws IOException {
return getResource().contentLength();
}
@Override
public long lastModified() throws IOException {
return getResource().lastModified();
}
@Override
public Resource createRelative(String relativePath) throws IOException {
return getResource().createRelative(relativePath);
}
@Override
public String getFilename() {
return getResource().getFilename();
}
@Override
public String getDescription() {
return getResource().getDescription();
}
public Resource getResource() {
return this.resource;
}
@Override
public Origin getOrigin() {
return this.origin;
}
@Override
public boolean equals(Object obj) {
if (this == obj) {
return true;
}
if (obj == null || getClass() != obj.getClass()) {
return false;
}
OriginTrackedResource other = (OriginTrackedResource) obj;
return this.resource.equals(other) && ObjectUtils.nullSafeEquals(this.origin, other.origin);
}
@Override
public int hashCode() {
final int prime = 31;
int result = this.resource.hashCode();
result = prime * result + ObjectUtils.nullSafeHashCode(this.origin);
return result;
}
@Override
public String toString() {
return this.resource.toString();
}
public static OriginTrackedWritableResource of(WritableResource resource, Origin origin) {
return (OriginTrackedWritableResource) of((Resource) resource, origin);
}
public static OriginTrackedResource of(Resource resource, Origin origin) {
if (resource instanceof WritableResource) {
return new OriginTrackedWritableResource((WritableResource) resource, origin);
}
return new OriginTrackedResource(resource, origin);
}
public static class OriginTrackedWritableResource extends OriginTrackedResource implements WritableResource {
OriginTrackedWritableResource(WritableResource resource, Origin origin) {
super(resource, origin);
}
@Override
public WritableResource getResource() {
return (WritableResource) super.getResource();
}
@Override
public OutputStream getOutputStream() throws IOException {
return getResource().getOutputStream();
}
}
}