From 944dbb749b988b3f23fc53497a0d3395cb112101 Mon Sep 17 00:00:00 2001
From: Noah Petherbridge <root@kirsle.net>
Date: Fri, 1 Sep 2023 22:27:18 -0700
Subject: [PATCH] Private photos: see who shares back

---
 pkg/controller/photo/private.go  |  1 +
 pkg/models/private_photo.go      | 30 ++++++++++++++++++++++++++++++
 web/templates/photo/private.html | 17 ++++++++++++++++-
 3 files changed, 47 insertions(+), 1 deletion(-)

diff --git a/pkg/controller/photo/private.go b/pkg/controller/photo/private.go
index 7c58bd8..f6434af 100644
--- a/pkg/controller/photo/private.go
+++ b/pkg/controller/photo/private.go
@@ -45,6 +45,7 @@ func Private() http.HandlerFunc {
 		var vars = map[string]interface{}{
 			"IsGrantee":    isGrantee,
 			"CountGrantee": models.CountPrivateGrantee(currentUser.ID),
+			"GranteeMap":   models.MapPrivatePhotoGrantee(currentUser, users),
 			"Users":        users,
 			"Pager":        pager,
 		}
diff --git a/pkg/models/private_photo.go b/pkg/models/private_photo.go
index b332b26..a743452 100644
--- a/pkg/models/private_photo.go
+++ b/pkg/models/private_photo.go
@@ -226,3 +226,33 @@ func (pb *PrivatePhoto) Save() error {
 	result := DB.Save(pb)
 	return result.Error
 }
+
+// PrivateGranteeMap maps user IDs to whether they have granted you their private photos.
+type PrivateGranteeMap map[uint64]bool
+
+// MapShyAccounts looks up a set of user IDs in bulk and returns a PrivateGranteeMap suitable for templates.
+func MapPrivatePhotoGrantee(currentUser *User, users []*User) PrivateGranteeMap {
+	var (
+		usermap = PrivateGranteeMap{}
+	)
+
+	var (
+		matched = []*PrivatePhoto{}
+		result  = DB.Model(&PrivatePhoto{}).
+			Where("target_user_id = ?", currentUser.ID).
+			Find(&matched)
+	)
+
+	if result.Error == nil {
+		for _, row := range matched {
+			usermap[row.SourceUserID] = true
+		}
+	}
+
+	return usermap
+}
+
+// Get a user from the PrivateGranteeMap.
+func (um PrivateGranteeMap) Get(id uint64) bool {
+	return um[id]
+}
diff --git a/web/templates/photo/private.html b/web/templates/photo/private.html
index 23a9e12..c423d29 100644
--- a/web/templates/photo/private.html
+++ b/web/templates/photo/private.html
@@ -91,7 +91,7 @@
                                 <p class="title is-4">
                                     <a href="/u/{{.Username}}" class="has-text-dark">{{.NameOrUsername}}</a>
                                 </p>
-                                <p class="subtitle is-6">
+                                <p class="subtitle is-6 mb-1">
                                     <span class="icon"><i class="fa fa-user"></i></span>
                                     <a href="/u/{{.Username}}">{{.Username}}</a>
                                     {{if not .Certified}}
@@ -108,6 +108,21 @@
                                     </span>
                                     {{end}}
                                 </p>
+
+                                <!-- Indicator if they are sharing back -->
+                                <div>
+                                    {{if $Root.GranteeMap.Get .ID}}
+                                    <span class="has-text-success">
+                                        <i class="fa fa-check mr-1"></i>
+                                        Sharing with me too
+                                    </span>
+                                    {{else}}
+                                    <span class="has-text-danger">
+                                        <i class="fa fa-xmark mr-1"></i>
+                                        Not sharing with me
+                                    </span>
+                                    {{end}}
+                                </div>
                             </div>
                         </div>
                     </div>