From 1aaad1d4e1e2401dcdf4047971b4c3ee9a67055e Mon Sep 17 00:00:00 2001
From: Vesa Luusua <vesa.luusua@gmail.com>
Date: Tue, 6 Nov 2018 11:50:06 +0200
Subject: [PATCH] ListingLink: add draft as a path variant

---
 src/components/ListingLink/ListingLink.js | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/src/components/ListingLink/ListingLink.js b/src/components/ListingLink/ListingLink.js
index b4cb66fc..e456778b 100644
--- a/src/components/ListingLink/ListingLink.js
+++ b/src/components/ListingLink/ListingLink.js
@@ -9,8 +9,12 @@
 import React from 'react';
 import { string, oneOfType, node } from 'prop-types';
 import { richText } from '../../util/richText';
-import { LISTING_STATE_PENDING_APPROVAL, propTypes } from '../../util/types';
-import { LISTING_PAGE_PENDING_APPROVAL_VARIANT, createSlug } from '../../util/urlHelpers';
+import { LISTING_STATE_DRAFT, LISTING_STATE_PENDING_APPROVAL, propTypes } from '../../util/types';
+import {
+  LISTING_PAGE_DRAFT_VARIANT,
+  LISTING_PAGE_PENDING_APPROVAL_VARIANT,
+  createSlug,
+} from '../../util/urlHelpers';
 import { NamedLink } from '../../components';
 
 import css from './ListingLink.css';
@@ -36,13 +40,19 @@ const ListingLink = props => {
   );
 
   const isPendingApproval = state === LISTING_STATE_PENDING_APPROVAL;
-  const linkProps = isPendingApproval
+  const isDraft = state === LISTING_STATE_DRAFT;
+  const variant = isPendingApproval
+    ? LISTING_PAGE_PENDING_APPROVAL_VARIANT
+    : isDraft
+      ? LISTING_PAGE_DRAFT_VARIANT
+      : null;
+  const linkProps = !!variant
     ? {
         name: 'ListingPageVariant',
         params: {
           id,
           slug,
-          variant: LISTING_PAGE_PENDING_APPROVAL_VARIANT,
+          variant,
         },
       }
     : {