From 0d79c02440b5594d4be378574d9033e319504fe5 Mon Sep 17 00:00:00 2001
From: Sergey Yarkov <serzh.yarkov@gmail.com>
Date: Mon, 17 Aug 2020 12:48:05 +0300
Subject: [PATCH] amp fixes

---
 gatsby-browser.js | 2 +-
 gatsby-ssr.js     | 8 ++++++--
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/gatsby-browser.js b/gatsby-browser.js
index 2acf0ab..2a3a6d9 100644
--- a/gatsby-browser.js
+++ b/gatsby-browser.js
@@ -31,7 +31,7 @@ export const shouldUpdateScroll = ({ routerProps: { location } }) => {
 
 export const wrapPageElement = ({ element, props }) => {
   const { location: { pathname } } = props
-  pathname.includes('/amp/') ? props.amp = true : props.amp = false
+  pathname.match(/article\/.*\/amp/) !== null ? props.amp = true : props.amp = false
 
   return <Layout {...props}>{element}</Layout>
 }
diff --git a/gatsby-ssr.js b/gatsby-ssr.js
index 8ba0fdb..a33376c 100644
--- a/gatsby-ssr.js
+++ b/gatsby-ssr.js
@@ -39,13 +39,17 @@ const OnDark = () => {
   )
 }
 
-export const onRenderBody = ({ setPreBodyComponents }) => {
+export const onRenderBody = ({ pathname, setPreBodyComponents }) => {
+  if (pathname.match(/article\/.*\/amp/) !== null) {
+    return
+  }
+  
   setPreBodyComponents(<OnDark key="isDarkTheme" />)
 }
 
 export const wrapPageElement = ({ element, props }) => {
   const { location: { pathname } } = props
-  pathname.includes('/amp/') ? props.amp = true : props.amp = false
+  pathname.match(/article\/.*\/amp/) !== null ? props.amp = true : props.amp = false
 
   return <Layout {...props}>{element}</Layout>
 }