|
|
@ -12,13 +12,13 @@ import java.nio.charset.Charset;
|
|
|
|
import net.yacy.cora.protocol.RequestHeader;
|
|
|
|
import net.yacy.cora.protocol.RequestHeader;
|
|
|
|
import net.yacy.interaction.AugmentHtmlStream;
|
|
|
|
import net.yacy.interaction.AugmentHtmlStream;
|
|
|
|
import net.yacy.kelondro.data.meta.DigestURI;
|
|
|
|
import net.yacy.kelondro.data.meta.DigestURI;
|
|
|
|
import net.yacy.search.Switchboard;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
public class AugmentedHtmlStream extends FilterOutputStream {
|
|
|
|
public class AugmentedHtmlStream extends FilterOutputStream {
|
|
|
|
private final Writer out;
|
|
|
|
private final Writer out;
|
|
|
|
private final ByteArrayOutputStream buffer;
|
|
|
|
private final ByteArrayOutputStream buffer;
|
|
|
|
private final Charset charset;
|
|
|
|
private final Charset charset;
|
|
|
|
private final DigestURI url;
|
|
|
|
private final DigestURI url;
|
|
|
|
|
|
|
|
private final String urls;
|
|
|
|
private final RequestHeader requestHeader;
|
|
|
|
private final RequestHeader requestHeader;
|
|
|
|
|
|
|
|
|
|
|
|
public AugmentedHtmlStream(OutputStream out, Charset charset, DigestURI url, RequestHeader requestHeader) {
|
|
|
|
public AugmentedHtmlStream(OutputStream out, Charset charset, DigestURI url, RequestHeader requestHeader) {
|
|
|
@ -27,6 +27,7 @@ public class AugmentedHtmlStream extends FilterOutputStream {
|
|
|
|
this.buffer = new ByteArrayOutputStream();
|
|
|
|
this.buffer = new ByteArrayOutputStream();
|
|
|
|
this.charset = charset;
|
|
|
|
this.charset = charset;
|
|
|
|
this.url = url;
|
|
|
|
this.url = url;
|
|
|
|
|
|
|
|
this.urls = this.url.toNormalform(false, true);
|
|
|
|
this.requestHeader = requestHeader;
|
|
|
|
this.requestHeader = requestHeader;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
@ -49,19 +50,10 @@ public class AugmentedHtmlStream extends FilterOutputStream {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
public StringBuffer process(StringBuffer data) {
|
|
|
|
public StringBuffer process(StringBuffer data) {
|
|
|
|
|
|
|
|
if (this.urls.contains("currentyacypeer/")) {
|
|
|
|
if (Switchboard.getSwitchboard().getConfigBool("proxyAugmentation", false) == true) {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (!this.url.toNormalform(false, true).contains("currentyacypeer/")) {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
return AugmentHtmlStream.process (data, this.charset, this.url, this.requestHeader);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
} else {
|
|
|
|
|
|
|
|
return data;
|
|
|
|
return data;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
} else {
|
|
|
|
} else {
|
|
|
|
return data;
|
|
|
|
return AugmentHtmlStream.process (data, this.charset, this.url, this.requestHeader);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|