Subscribe/Unsubscribe

Classes/Issues/GithubClient+Issues.swift

@@ -163,6 +163,7 @@ extension GithubClient {
                         milestone: milestoneModel,
                         targetBranch: targetBranchModel,
                         timelinePages: [newPage] + (prependResult?.timelinePages ?? []),
+                        viewerIsSubscribed: issueType.viewerSubscribed,
                         viewerCanUpdate: issueType.viewerCanUpdate,
                         hasIssuesEnabled: repository.hasIssuesEnabled,
                         viewerCanAdminister: canAdmin,
@@ -297,6 +298,25 @@ extension GithubClient {
         }
     }
 
+
+    func setSubscription(
+        previous: IssueResult,
+        subscribed: Bool,
+        completion: ((Result<Bool>) -> Void)? = nil
+    ) {
+
+        let cache = self.cache
+
+        client.send(V3SubscribeThreadRequest(id: previous.id, ignore: subscribed)) { result in
+            switch result {
+            case .success:
+                completion?(.success(true))
+            case .failure(let error):
+                cache.set(value: previous)
+            }
+        }
+    }
+
     enum CollaboratorPermission: String {
         case admin
         case write

Classes/Issues/Issue+IssueType.swift

@@ -11,6 +11,10 @@ import IGListKit
 
 extension IssueOrPullRequestQuery.Data.Repository.IssueOrPullRequest.AsIssue: IssueType {
 
+    var viewerSubscribed: Bool {
+        return viewerSubscription?.rawValue != "IGNORED"
+    }
+
     var pullRequest: Bool {
         return false
     }

Classes/Issues/IssueManagingContextController.swift

@@ -84,6 +84,8 @@ final class IssueManagingContextController: NSObject, ContextMenuDelegate {
         case lock
         case reopen
         case close
+        case subscribe
+        case unsubscribe
     }
 
     var actions: [Action] {
@@ -97,6 +99,15 @@ final class IssueManagingContextController: NSObject, ContextMenuDelegate {
             if result.pullRequest {
                 actions.append(.reviewers)
             }
+        }
+
+        if result.viewerIsSubscribed {
+            actions.append(.unsubscribe)
+        } else {
+            actions.append(.subscribe)
+        }
+
+        if case .collaborator = permissions {
             if result.labels.locked {
                 actions.append(.unlock)
             } else {
@@ -144,13 +155,19 @@ final class IssueManagingContextController: NSObject, ContextMenuDelegate {
         case .close:
             title = Constants.Strings.close
             iconName = "x"
+        case .subscribe:
+            title = Constants.Strings.subscribe
+            iconName = "unmute"
+        case .unsubscribe:
+            title = Constants.Strings.unsubscribe
+            iconName = "mute"
         }
 
         // Lock always has the divider above it assuming you're a collaborator.
         // If you aren't a collaborator (Lock does not show), close has the divider above it.
         let separator: Bool
         switch action {
-        case .lock, .unlock: separator = true
+        case .subscribe, .unsubscribe: separator = true
         case .reopen, .close: separator = permissions != .collaborator
         default: separator = false
         }
@@ -186,6 +203,8 @@ final class IssueManagingContextController: NSObject, ContextMenuDelegate {
             case .lock: strongSelf.lock(true)
             case .reopen: strongSelf.close(false)
             case .close: strongSelf.close(true)
+            case .subscribe: strongSelf.subscribe(true)
+            case .unsubscribe: strongSelf.subscribe(false)
             }
         }
     }
@@ -271,6 +290,12 @@ final class IssueManagingContextController: NSObject, ContextMenuDelegate {
         )
     }
 
+    func subscribe(_ doSubscribe: Bool) {
+        guard let previous = result else { return }
+        delegate?.willMutateModel(from: self)
+        client.setSubscription(previous: previous, subscribed: doSubscribe)
+        Haptic.triggerNotification(.success)
+    }
     func close(_ doClose: Bool) {
         guard let previous = result else { return }
         delegate?.willMutateModel(from: self)

Classes/Issues/IssueResult.swift

@@ -25,6 +25,7 @@ struct IssueResult: Cachable {
     let targetBranch: IssueTargetBranchModel?
     // end optionals
     let timelinePages: [IssueTimelinePage]
+    let viewerIsSubscribed: Bool
     let viewerCanUpdate: Bool
     let hasIssuesEnabled: Bool
     let viewerCanAdminister: Bool
@@ -73,6 +74,7 @@ struct IssueResult: Cachable {
         labels: IssueLabelsModel? = nil,
         assignee: IssueAssigneesModel? = nil,
         timelinePages: [IssueTimelinePage]? = nil,
+        viewerIsSubscribed: Bool? = nil,
         viewerCanUpdate: Bool? = nil,
         hasIssuesEnabled: Bool? = nil,
         viewerCanAdminister: Bool? = nil,
@@ -90,6 +92,7 @@ struct IssueResult: Cachable {
             milestone: self.milestone,
             targetBranch: self.targetBranch,
             timelinePages: timelinePages ?? self.timelinePages,
+            viewerIsSubscribed: viewerIsSubscribed ?? self.viewerIsSubscribed,
             viewerCanUpdate: viewerCanUpdate ?? self.viewerCanUpdate,
             hasIssuesEnabled: hasIssuesEnabled ?? self.hasIssuesEnabled,
             viewerCanAdminister: viewerCanAdminister ?? self.viewerCanAdminister,
@@ -114,6 +117,7 @@ struct IssueResult: Cachable {
             milestone: milestone,
             targetBranch: self.targetBranch,
             timelinePages: timelinePages ?? self.timelinePages,
+            viewerIsSubscribed: self.viewerIsSubscribed,
             viewerCanUpdate: self.viewerCanUpdate,
             hasIssuesEnabled: self.hasIssuesEnabled,
             viewerCanAdminister: self.viewerCanAdminister,
@@ -138,6 +142,7 @@ struct IssueResult: Cachable {
             milestone: self.milestone,
             targetBranch: self.targetBranch,
             timelinePages: timelinePages ?? self.timelinePages,
+            viewerIsSubscribed: self.viewerIsSubscribed,
             viewerCanUpdate: self.viewerCanUpdate,
             hasIssuesEnabled: self.hasIssuesEnabled,
             viewerCanAdminister: self.viewerCanAdminister,

Classes/Issues/IssueType.swift

@@ -32,6 +32,7 @@ protocol IssueType {
     var targetBranch: String? { get }
     var locked: Bool { get }
     var headPaging: HeadPaging { get }
+    var viewerSubscribed: Bool { get }
     var viewerCanUpdate: Bool { get }
     var fileChanges: FileChanges? { get }
 

Classes/Issues/PullRequest+IssueType.swift

@@ -12,6 +12,10 @@ import StyledTextKit
 
 extension IssueOrPullRequestQuery.Data.Repository.IssueOrPullRequest.AsPullRequest: IssueType {
 
+    var viewerSubscribed: Bool {
+        return viewerSubscription?.rawValue != "IGNORED"
+    }
+
     var pullRequest: Bool {
         return true
     }

Classes/Views/Constants.swift

@@ -52,5 +52,7 @@ enum Constants {
         static let reviewers = NSLocalizedString("Reviewers", comment: "")
         static let clear = NSLocalizedString("Clear", comment: "")
         static let preview = NSLocalizedString("Preview", comment: "")
+        static let subscribe = NSLocalizedString("Subscribe", comment: "")
+        static let unsubscribe = NSLocalizedString("Unsubscribe", comment: "")
     }
 }

gql/IssueOrPullRequest.graphql

@@ -17,6 +17,7 @@ query IssueOrPullRequest($owner: String!, $repo: String!, $number: Int!, $page_s
     }
     issueOrPullRequest(number: $number) {
       ... on Issue {
+	viewerSubscription
         timeline(last: $page_size, before: $before) {
           pageInfo{...headPaging}
           nodes {
@@ -182,6 +183,7 @@ query IssueOrPullRequest($owner: String!, $repo: String!, $number: Int!, $page_s
         title
       }
       ... on PullRequest {
+	viewerSubscription
         timeline(last: $page_size, before: $before) {
           pageInfo{...headPaging}
           nodes {