From ce61b831b1e417a5f34e06b897a06778b1fc2715 Mon Sep 17 00:00:00 2001
From: Amio <amio.cn@gmail.com>
Date: Sun, 2 Jun 2019 14:20:35 +0800
Subject: [PATCH] Setup homepage style switcher

---
 components/home-intro.js |  3 +--
 pages/index.js           | 13 +++++++++++--
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/components/home-intro.js b/components/home-intro.js
index de87864..01dfab6 100644
--- a/components/home-intro.js
+++ b/components/home-intro.js
@@ -1,8 +1,7 @@
 import icons from 'badgen-icons'
 
-export default ({ isFlat, children }) => (
+export default ({ isFlat }) => (
   <div>
-    <h3>Usage</h3>
     <pre>{ explainCode(isFlat) }</pre>
 
     <h3 id='colors'>Available color names</h3>
diff --git a/pages/index.js b/pages/index.js
index df0b376..84cda63 100644
--- a/pages/index.js
+++ b/pages/index.js
@@ -1,4 +1,4 @@
-import { useState } from 'react'
+import { useState, useEffect } from 'react'
 import BadgeExamples from '../components/badge-examples.js'
 import BadgenTitle from '../components/badgen-title.jsx'
 import Intro from '../components/home-intro.js'
@@ -7,10 +7,19 @@ import examples from '../static/.gen/badges.json'
 
 const Index = () => {
   const [ tab, setTab ] = useState('live')
+  const [ host, setHost ] = useState('host')
   const badges = examples[tab]
 
+  useEffect(() => {
+    const forceHost = new URL(window.location).searchParams.get('host')
+    const autoHost = window.location.host === 'flat.badgen.net'
+      ? 'https://flat.badgen.net'
+      : 'https://badgen.net'
+    setHost((forceHost || autoHost) + '/')
+  })
+
   return <>
-    <BadgenTitle host='https://badgen.net' />
+    <BadgenTitle host={host} />
     <div className='docs'>
       <Intro />
       <h2 style={{ textAlign: 'center' }}>Badge Gallery</h2>