From 95bde60dc4a0a3920c9a9d663b3ff26d030722c9 Mon Sep 17 00:00:00 2001
From: Artem Sapegin <artem@sapegin.ru>
Date: Tue, 14 Apr 2020 16:35:51 +0200
Subject: [PATCH] feat: Add prod URL to Demo component

---
 package.json            |  2 +-
 src/components/Demo.jsx | 12 ++++++++++--
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/package.json b/package.json
index 5658931..8e27b94 100644
--- a/package.json
+++ b/package.json
@@ -24,7 +24,7 @@
   "scripts": {
     "build": "babel --delete-dir-on-start --ignore '**/*.spec.js' --env-name cjs -d dist/ src/",
     "prepare": "npm run build",
-    "test": "echo \"Error: no test specified\" && exit 1",
+    "test": "true",
     "posttest": "npm run format",
     "slides:start": "mdx-deck ./examples/example.mdx",
     "slides:build": "mdx-deck build ./examples/example.mdx",
diff --git a/src/components/Demo.jsx b/src/components/Demo.jsx
index 3cc5340..773fc51 100644
--- a/src/components/Demo.jsx
+++ b/src/components/Demo.jsx
@@ -1,8 +1,9 @@
 /* @jsx jsx */
 import * as React from "react";
+import PropTypes from "prop-types";
 import { jsx, Box } from "theme-ui";
 
-export default ({ src }) => {
+function Demo({ src, prodSrc }) {
   return (
     <Box sx={{ position: "relative" }}>
       <Box
@@ -20,9 +21,16 @@ export default ({ src }) => {
         }}
       />
       <iframe
-        src={src}
+        src={prodSrc && process.env.NODE_ENV === "production" ? prodSrc : src}
         style={{ border: "none", width: "100vw", height: "100vh" }}
       />
     </Box>
   );
+}
+
+Demo.propTypes = {
+  src: PropTypes.string.isRequired,
+  prodSrc: PropTypes.string,
 };
+
+export default Demo;