From b7f792a7d7e5f4db8b1e76a7313ad9ed2be50073 Mon Sep 17 00:00:00 2001
From: Justin Mazzocchi <2831158+jzzocc@users.noreply.github.com>
Date: Sun, 7 Feb 2021 09:55:56 -0800
Subject: [PATCH] Show visibility levels on reblog button

---
 Views/UIKit/Content Views/StatusView.swift | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/Views/UIKit/Content Views/StatusView.swift b/Views/UIKit/Content Views/StatusView.swift
index ad5695c..5862bff 100644
--- a/Views/UIKit/Content Views/StatusView.swift	
+++ b/Views/UIKit/Content Views/StatusView.swift	
@@ -683,9 +683,23 @@ private extension StatusView {
     }
 
     func setButtonImages(scale: UIImage.SymbolScale) {
+        let visibility = statusConfiguration.viewModel.visibility
+        let reblogSystemImageName: String
+
+        if statusConfiguration.viewModel.configuration.isContextParent {
+            reblogSystemImageName = "arrow.2.squarepath"
+        } else {
+            switch visibility {
+            case .public, .unlisted:
+                reblogSystemImageName = "arrow.2.squarepath"
+            default:
+                reblogSystemImageName = visibility.systemImageName
+            }
+        }
+
         replyButton.setImage(UIImage(systemName: "bubble.right",
                                      withConfiguration: UIImage.SymbolConfiguration(scale: scale)), for: .normal)
-        reblogButton.setImage(UIImage(systemName: "arrow.2.squarepath",
+        reblogButton.setImage(UIImage(systemName: reblogSystemImageName,
                                       withConfiguration: UIImage.SymbolConfiguration(scale: scale)), for: .normal)
         favoriteButton.setImage(UIImage(systemName: statusConfiguration.viewModel.favorited ? "star.fill" : "star",
                                         withConfiguration: UIImage.SymbolConfiguration(scale: scale)), for: .normal)