add amp vary param

maciejmackowiak · Jan 30, 2020 · 5b7deb1 · 5b7deb1
1 parent cd6ee3f
commit 5b7deb1
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/widget-cache.php b/widget-cache.php
@@ -109,7 +109,7 @@ function __construct()
                         ),
                         "amp" => array(
                             &$this,
-                            'is_amp_endpoint'
+                            'amp_vary_param'
                         )
                     );
                 }
@@ -580,6 +580,13 @@ function get_user_agent()
         return $_SERVER ['HTTP_USER_AGENT'];
     }
 
+    function amp_vary_param(){
+        if(function_exists('is_amp_endpoint') && is_amp_endpoint()){
+            return 'amp';
+        } else {
+            return 'non-amp';
+        }
+    }
     function get_current_category()
     {
         if (is_single()) {