From f0cddf7e42786370b00ba89b9dbd6d731ad96020 Mon Sep 17 00:00:00 2001
From: BBaoVanC <bbaovanc@bbaovanc.com>
Date: Sun, 10 Oct 2021 17:28:05 -0500
Subject: [PATCH] Add icon to drafts

Fix #23
---
 layouts/_default/single.html  | 5 ++++-
 layouts/_default/summary.html | 1 +
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/layouts/_default/single.html b/layouts/_default/single.html
index f4e18ca..c22c4a1 100644
--- a/layouts/_default/single.html
+++ b/layouts/_default/single.html
@@ -1,6 +1,9 @@
 {{ define "main" }}
 <div class="post-title">
-    <h1>{{ .Title | markdownify }}</h1>
+    <h1>
+        {{ .Title | markdownify }}
+        {{ if .Draft }}{{ partial "icon.html" "edit" }}{{ end }}
+    </h1>
 </div>
 
 {{ partial "post-metadata.html" . }}
diff --git a/layouts/_default/summary.html b/layouts/_default/summary.html
index ec2161d..895980f 100644
--- a/layouts/_default/summary.html
+++ b/layouts/_default/summary.html
@@ -2,6 +2,7 @@
     <h2 class="post-title">
         {{ with .Parent.Params.icon }}{{ partial "icon.html" . }}{{ end }}
         <a href="{{ .Permalink }}">{{ .Title | markdownify }}</a>
+        {{ if .Draft }}{{ partial "icon.html" "edit" }}{{ end }}
     </h2>
 
     {{ partial "post-metadata.html" . }}