1 files changed, 33 insertions, 16 deletions

diff --git a/src/mixins/comment.pug b/src/mixins/comment.pug
index 2ec076f..3af1f53 100644
--- a/src/mixins/comment.pug
+++ b/src/mixins/comment.pug
@@ -1,24 +1,41 @@
 include ../utils
+
+mixin infoContainer(data)
+  div.comment-info-container
+    div.info-item #{fmtnum(data.ups)} ↑
+    div.info-item u/#{data.author} #{data.is_submitter ? '(OP)' : ''}
+    if data.collapsed_reason_code == "DELETED"
+      div.info-item
+        a(href=`https://undelete.pullpush.io${data.permalink}`) search on undelete
+    div.info-item #{timeDifference(Date.now(), data.created * 1000)}
+
+-
+  function hasReplies(data) {
+    return data.replies && data.replies.data && data.replies.data.children && data.replies.data.children.length > 0;
+  }
+
 mixin comment(com, isfirst)
   - var data = com.data
   - var kind = com.kind
+  - var hasReplyData = hasReplies(data)
+  
   if kind == "more"
-    div(class=`${isfirst?'first':''}`)
+    div(class=`${isfirst ? 'first' : ''}`)
       div.more #{data.count} more comments
   else
-    div(class=`comment ${isfirst?'first':''}`)
-      div.comment-info-container
-        div.info-item #{fmtnum(data.ups)} ↑ 
-        div.info-item u/#{data.author} #{data.is_submitter?'(OP)':''}
-        if data.collapsed_reason_code == "DELETED"
-          div.info-item 
-            a(href=`https://undelete.pullpush.io${data.permalink}`) search on undelete
-        div.info-item #{timeDifference(Date.now(), data.created * 1000)}
-      div.comment-body 
-        != data.body_html
-      div.replies
-      if data.replies
-        if data.replies.data
-          if data.replies.data.children
+    div(class=`comment ${isfirst ? 'first' : ''}`)
+      if hasReplyData
+        details(id=`${data.id}` open="")
+          summary.expand-comments
+            +infoContainer(data)
+          div.comment-body
+            != data.body_html
+    
+          div.replies
             each reply in data.replies.data.children
-              +comment(reply,false)
+              +comment(reply, false)
+    
+      else
+        +infoContainer(data)
+        div.comment-body
+          != data.body_html