00001 /* 00002 * Copyright 2010 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_ADD_INSTRUMENTATION_FILTER_H_ 00020 #define NET_INSTAWEB_REWRITER_PUBLIC_ADD_INSTRUMENTATION_FILTER_H_ 00021 00022 #include "net/instaweb/htmlparse/public/empty_html_filter.h" 00023 #include "net/instaweb/util/public/basictypes.h" 00024 #include "net/instaweb/util/public/string.h" 00025 00026 namespace net_instaweb { 00027 00028 class HtmlElement; 00029 class RewriteDriver; 00030 class Statistics; 00031 class Variable; 00032 00034 class AddInstrumentationFilter : public EmptyHtmlFilter { 00035 public: 00036 static const char kLoadTag[]; 00037 static const char kUnloadTag[]; 00038 static GoogleString* kUnloadScriptFormatXhtml; 00039 static GoogleString* kTailScriptFormatXhtml; 00040 00042 static const char kInstrumentationScriptAddedCount[]; 00043 00044 explicit AddInstrumentationFilter(RewriteDriver* driver); 00045 virtual ~AddInstrumentationFilter(); 00046 00047 static void Initialize(Statistics* statistics); 00048 static void Terminate(); 00049 00050 virtual void StartDocument(); 00051 virtual void StartElement(HtmlElement* element); 00052 virtual void EndElement(HtmlElement* element); 00053 virtual const char* Name() const { return "AddInstrumentation"; } 00054 00055 protected: 00057 Variable* instrumentation_script_added_count_; 00058 00059 private: 00060 bool IsXhtml(); 00061 00064 void AddScriptNode(HtmlElement* element, const GoogleString& script_format, 00065 const GoogleString& tag_name, bool is_xhtml); 00066 00067 RewriteDriver* driver_; 00068 bool found_head_; 00069 bool use_cdata_hack_; 00070 00071 DISALLOW_COPY_AND_ASSIGN(AddInstrumentationFilter); 00072 }; 00073 00074 } 00075 00076 #endif ///< NET_INSTAWEB_REWRITER_PUBLIC_ADD_INSTRUMENTATION_FILTER_H_