00001 /* 00002 * Copyright 2011 Google Inc. 00003 * 00004 * Licensed under the Apache License, Version 2.0 (the "License"); 00005 * you may not use this file except in compliance with the License. 00006 * You may obtain a copy of the License at 00007 * 00008 * http:///www.apache.org/licenses/LICENSE-2.0 00009 * 00010 * Unless required by applicable law or agreed to in writing, software 00011 * distributed under the License is distributed on an "AS IS" BASIS, 00012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 00013 * See the License for the specific language governing permissions and 00014 * limitations under the License. 00015 */ 00016 00018 00019 #ifndef NET_INSTAWEB_REWRITER_PUBLIC_DOMAIN_REWRITE_FILTER_H_ 00020 #define NET_INSTAWEB_REWRITER_PUBLIC_DOMAIN_REWRITE_FILTER_H_ 00021 00022 #include "net/instaweb/rewriter/public/common_filter.h" 00023 #include "net/instaweb/rewriter/public/resource_tag_scanner.h" 00024 #include "net/instaweb/util/public/basictypes.h" 00025 #include "net/instaweb/util/public/string.h" 00026 #include "net/instaweb/util/public/string_util.h" 00027 00028 namespace net_instaweb { 00029 00030 class HtmlElement; 00031 class GoogleUrl; 00032 class RewriteDriver; 00033 class Statistics; 00034 class Variable; 00035 00040 class DomainRewriteFilter : public CommonFilter { 00041 public: 00042 DomainRewriteFilter(RewriteDriver* rewrite_driver, Statistics* stats); 00043 ~DomainRewriteFilter(); 00044 static void Initialize(Statistics* statistics); 00045 virtual void StartDocumentImpl(); 00046 virtual void StartElementImpl(HtmlElement* element); 00047 virtual void EndElementImpl(HtmlElement* element) {} 00048 00049 virtual const char* Name() const { return "DomainRewrite"; } 00050 00051 enum RewriteResult { 00052 kRewroteDomain, 00053 kDomainUnchanged, 00054 kFail, 00055 }; 00056 00062 RewriteResult Rewrite(const StringPiece& input_url, 00063 const GoogleUrl& base_url, 00064 bool apply_sharding, 00065 GoogleString* output_url); 00066 00067 private: 00068 ResourceTagScanner tag_scanner_; 00070 Variable* rewrite_count_; 00071 00072 DISALLOW_COPY_AND_ASSIGN(DomainRewriteFilter); 00073 }; 00074 00075 } 00076 00077 #endif ///< NET_INSTAWEB_REWRITER_PUBLIC_DOMAIN_REWRITE_FILTER_H_