/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.apache.shindig.gadgets.rewrite;
import org.apache.shindig.common.uri.Uri;
import org.apache.shindig.gadgets.GadgetContext;
import org.apache.shindig.gadgets.GadgetException;
import org.apache.shindig.gadgets.GadgetSpecFactory;
import org.apache.shindig.gadgets.http.HttpRequest;
import org.apache.shindig.gadgets.spec.GadgetSpec;
import com.google.common.collect.ImmutableSet;
import com.google.inject.Inject;
import com.google.inject.Singleton;
import com.google.inject.name.Named;
import java.util.Collections;
import java.util.Set;
/**
* Factory for content rewriter features
*/
@Singleton
public class ContentRewriterFeatureFactory {
private final GadgetSpecFactory specFactory;
private final String includeUrls;
private final String excludeUrls;
private final String expires;
private final Set<String> includeTags;
private final boolean onlyAllowExcludes;
private final ContentRewriterFeature defaultFeature;
@Inject
public ContentRewriterFeatureFactory(
GadgetSpecFactory specFactory,
@Named("shindig.content-rewrite.include-urls")String includeUrls,
@Named("shindig.content-rewrite.exclude-urls")String excludeUrls,
@Named("shindig.content-rewrite.expires")String expires,
@Named("shindig.content-rewrite.include-tags")String includeTags,
@Named("shindig.content-rewrite.only-allow-excludes")String onlyAllowExcludes) {
this.specFactory = specFactory;
this.includeUrls = includeUrls;
this.excludeUrls = excludeUrls;
this.expires = expires;
this.onlyAllowExcludes = Boolean.parseBoolean(onlyAllowExcludes);
ImmutableSet.Builder<String> includeTagsBuilder = ImmutableSet.builder();
for (String s : includeTags.trim().toLowerCase().split("\\s*,\\s*")) {
if (s != null && s.length() > 0) {
includeTagsBuilder.add(s);
}
}
this.includeTags = includeTagsBuilder.build();
defaultFeature = new ContentRewriterFeature(null, includeUrls, excludeUrls, expires,
this.includeTags, this.onlyAllowExcludes);
}
public ContentRewriterFeature getDefault() {
return defaultFeature;
}
public ContentRewriterFeature get(HttpRequest request) {
final Uri gadgetUri = request.getGadget();
GadgetSpec spec;
if (gadgetUri != null) {
try {
GadgetContext context = new GadgetContext() {
@Override
public Uri getUrl() {
return gadgetUri;
}
};
spec = specFactory.getGadgetSpec(context);
if (spec != null) {
return get(spec);
}
} catch (GadgetException ge) {
return defaultFeature;
}
}
return defaultFeature;
}
public ContentRewriterFeature get(GadgetSpec spec) {
ContentRewriterFeature rewriterFeature =
(ContentRewriterFeature)spec.getAttribute("content-rewriter");
if (rewriterFeature != null) return rewriterFeature;
rewriterFeature
= new ContentRewriterFeature(spec, includeUrls, excludeUrls, expires, includeTags, onlyAllowExcludes);
spec.setAttribute("content-rewriter", rewriterFeature);
return rewriterFeature;
}
/**
* Create a rewriter feature that allows all URIs to be rewritten.
*/
public ContentRewriterFeature createRewriteAllFeature(int ttl) {
return new ContentRewriterFeature(null,
".*", "", (ttl == -1) ? "HTTP" : Integer.toString(ttl),
Collections.<String>emptySet(), false);
}
}