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_SIMPLE_TEXT_FILTER_H_ 00020 #define NET_INSTAWEB_REWRITER_PUBLIC_SIMPLE_TEXT_FILTER_H_ 00021 00022 #include "net/instaweb/htmlparse/public/html_element.h" 00023 #include "net/instaweb/rewriter/public/output_resource_kind.h" 00024 #include "net/instaweb/rewriter/public/resource.h" 00025 #include "net/instaweb/rewriter/public/resource_manager.h" 00026 #include "net/instaweb/rewriter/public/resource_slot.h" 00027 #include "net/instaweb/rewriter/public/rewrite_filter.h" 00028 #include "net/instaweb/rewriter/public/single_rewrite_context.h" 00029 #include "net/instaweb/util/public/basictypes.h" 00030 #include "net/instaweb/util/public/ref_counted_ptr.h" 00031 #include "net/instaweb/util/public/string.h" 00032 #include "net/instaweb/util/public/string_util.h" 00033 00034 namespace net_instaweb { 00035 00036 class RewriteContext; 00037 class RewriteDriver; 00038 00047 class SimpleTextFilter : public RewriteFilter { 00048 public: 00049 class Rewriter : public RefCounted<Rewriter> { 00050 public: 00051 Rewriter() {} 00052 virtual bool RewriteText(const StringPiece& url, 00053 const StringPiece& in, 00054 GoogleString* out, 00055 ResourceManager* resource_manager) = 0; 00056 virtual HtmlElement::Attribute* FindResourceAttribute( 00057 HtmlElement* element) = 0; 00058 00059 virtual OutputResourceKind kind() const = 0; 00060 virtual const char* id() const = 0; 00061 virtual const char* name() const = 0; 00062 00064 virtual bool OptimizationOnly() const { return true; } 00065 00066 protected: 00067 REFCOUNT_FRIEND_DECLARATION(Rewriter); 00068 virtual ~Rewriter(); 00069 00070 private: 00071 DISALLOW_COPY_AND_ASSIGN(Rewriter); 00072 }; 00073 00074 typedef RefCountedPtr<Rewriter> RewriterPtr; 00075 00076 class Context : public SingleRewriteContext { 00077 public: 00078 Context(const RewriterPtr& rewriter, RewriteDriver* driver, 00079 RewriteContext* parent); 00080 virtual ~Context(); 00081 virtual void RewriteSingle( 00082 const ResourcePtr& input, const OutputResourcePtr& output); 00083 00084 protected: 00085 virtual const char* id() const { return rewriter_->id(); } 00086 virtual OutputResourceKind kind() const { return rewriter_->kind(); } 00087 virtual bool OptimizationOnly() const { 00088 return rewriter_->OptimizationOnly(); 00089 } 00090 00091 private: 00092 RewriterPtr rewriter_; 00093 00094 DISALLOW_COPY_AND_ASSIGN(Context); 00095 }; 00096 00097 SimpleTextFilter(Rewriter* rewriter, RewriteDriver* driver); 00098 virtual ~SimpleTextFilter(); 00099 00100 virtual void StartDocumentImpl() {} 00101 virtual void EndElementImpl(HtmlElement* element) {} 00102 virtual void StartElementImpl(HtmlElement* element); 00103 00104 virtual RewriteContext* MakeRewriteContext(); 00105 virtual RewriteContext* MakeNestedRewriteContext( 00106 RewriteContext* parent, const ResourceSlotPtr& slot); 00107 00108 protected: 00109 virtual const char* id() const { return rewriter_->id(); } 00110 virtual const char* Name() const { return rewriter_->name(); } 00111 virtual bool ComputeOnTheFly() const { 00112 return rewriter_->kind() == kOnTheFlyResource; 00113 } 00114 00115 private: 00116 RewriterPtr rewriter_; 00117 00118 DISALLOW_COPY_AND_ASSIGN(SimpleTextFilter); 00119 }; 00120 00121 } 00122 00123 #endif ///< NET_INSTAWEB_REWRITER_PUBLIC_SIMPLE_TEXT_FILTER_H_