Subscribe/Unsubscribe

Classes/Issues/GithubClient+Issues.swift

@@ -168,7 +168,8 @@ extension GithubClient {
                         viewerCanAdminister: canAdmin,
                         defaultBranch: repository.defaultBranchRef?.name ?? "master",
                         fileChanges: issueType.fileChanges,
-                        mergeModel: issueType.mergeModel(availableTypes: availableMergeTypes)
+                        mergeModel: issueType.mergeModel(availableTypes: availableMergeTypes),
+                        subscriptionState: issueType.subscriptionState
                     )
 
                     DispatchQueue.main.async {
@@ -297,6 +298,37 @@ extension GithubClient {
         }
     }
 
+    func setSubscription(
+        previous: IssueResult,
+        subscribe: Bool,
+        completion: ((Result<SubscriptionState?>) -> Void)? = nil
+        ) {
+
+        let state: SubscriptionState = subscribe
+            ? .subscribed
+            : .unsubscribed
+
+        let optimisticResult = previous.updated(subscriptionState: state)
+
+        let cache = self.cache
+        cache.set(value: optimisticResult)
+
+        let mutation = UpdateSubscriptionMutation(subscribable_Id: previous.id, subscription_state: state)
+
+        client.mutate(mutation, result: { data in
+            data.updateSubscription?.subscribable
+        }, completion: { result in
+            switch result {
+            case .success(let subscribable):
+                completion?(.success(subscribable.viewerSubscription))
+            case .failure(let error):
+                cache.set(value: previous)
+                completion?(.error(error))
+                Squawk.show(error: error)
+            }
+        })
+    }
+
     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 subscriptionState: SubscriptionState {
+        return viewerSubscription ?? .unsubscribed
+    }
+
     var pullRequest: Bool {
         return false
     }

Classes/Issues/IssueManagingContextController.swift

@@ -84,10 +84,12 @@ final class IssueManagingContextController: NSObject, ContextMenuDelegate {
         case lock
         case reopen
         case close
+        case subscribe
+        case unsubscribe
     }
 
     var actions: [Action] {
-        if case .none = permissions { return [] }
+
         guard let result = self.result else { return [] }
 
         var actions = [Action]()
@@ -97,21 +99,30 @@ final class IssueManagingContextController: NSObject, ContextMenuDelegate {
             if result.pullRequest {
                 actions.append(.reviewers)
             }
+        }
+
+        if result.subscriptionState == .subscribed {
+            actions.append(.unsubscribe)
+        } else {
+            actions.append(.subscribe)
+        }
+
+        if case .collaborator = permissions {
             if result.labels.locked {
                 actions.append(.unlock)
             } else {
                 actions.append(.lock)
             }
         }
-
-        switch result.labels.status.status {
-        case .closed:
-            actions.append(.reopen)
-        case .open:
-            actions.append(.close)
-        case .merged: break
+        if permissions == .collaborator || permissions == .author {
+            switch result.labels.status.status {
+            case .closed:
+                actions.append(.reopen)
+            case .open:
+                actions.append(.close)
+            case .merged: break
+            }
         }
-
         return actions
     }
 
@@ -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, subscribe: doSubscribe)
+        Haptic.triggerNotification(.success)
+    }
     func close(_ doClose: Bool) {
         guard let previous = result else { return }
         delegate?.willMutateModel(from: self)

Classes/Issues/IssueResult.swift

@@ -31,6 +31,7 @@ struct IssueResult: Cachable {
     let defaultBranch: String
     let fileChanges: FileChanges?
     let mergeModel: IssueMergeModel?
+    let subscriptionState: SubscriptionState
 
     var timelineViewModels: [ListDiffable] {
         return timelinePages.reduce([], { $0 + $1.viewModels })
@@ -77,7 +78,8 @@ struct IssueResult: Cachable {
         hasIssuesEnabled: Bool? = nil,
         viewerCanAdminister: Bool? = nil,
         defaultBranch: String? = nil,
-        mergeModel: IssueMergeModel? = nil
+        mergeModel: IssueMergeModel? = nil,
+        subscriptionState: SubscriptionState? = nil
         ) -> IssueResult {
         return IssueResult(
             id: id ?? self.id,
@@ -95,7 +97,8 @@ struct IssueResult: Cachable {
             viewerCanAdminister: viewerCanAdminister ?? self.viewerCanAdminister,
             defaultBranch: defaultBranch ?? self.defaultBranch,
             fileChanges: fileChanges,
-            mergeModel: mergeModel ?? self.mergeModel
+            mergeModel: mergeModel ?? self.mergeModel,
+            subscriptionState: subscriptionState ?? self.subscriptionState
         )
     }
 
@@ -119,7 +122,8 @@ struct IssueResult: Cachable {
             viewerCanAdminister: self.viewerCanAdminister,
             defaultBranch: self.defaultBranch,
             fileChanges: self.fileChanges,
-            mergeModel: self.mergeModel
+            mergeModel: self.mergeModel,
+            subscriptionState: self.subscriptionState
         )
     }
 
@@ -143,7 +147,8 @@ struct IssueResult: Cachable {
             viewerCanAdminister: self.viewerCanAdminister,
             defaultBranch: self.defaultBranch,
             fileChanges: self.fileChanges,
-            mergeModel: self.mergeModel
+            mergeModel: self.mergeModel,
+            subscriptionState: self.subscriptionState
         )
     }
 

Classes/Issues/IssueType.swift

@@ -34,6 +34,8 @@ protocol IssueType {
     var headPaging: HeadPaging { get }
     var viewerCanUpdate: Bool { get }
     var fileChanges: FileChanges? { get }
+    var subscriptionState: SubscriptionState { get }
+
 
     var reviewRequestModel: IssueAssigneesModel? { get }
     func mergeModel(availableTypes: [IssueMergeType]) -> IssueMergeModel?

Classes/Issues/PullRequest+IssueType.swift

@@ -12,6 +12,10 @@ import StyledTextKit
 
 extension IssueOrPullRequestQuery.Data.Repository.IssueOrPullRequest.AsPullRequest: IssueType {
 
+    var subscriptionState: SubscriptionState {
+        return viewerSubscription ?? .unsubscribed
+    }
+
     var pullRequest: Bool {
         return true
     }

Classes/Views/Constants.swift

@@ -52,6 +52,8 @@ 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: "")
         static let overview = NSLocalizedString("Overview", comment: "")
     }
 }

FreetimeTests/SplitViewTests.swift

@@ -114,9 +114,7 @@ class SplitViewTests: XCTestCase {
             client: GithubClient(userSession: nil),
             repo: RepositoryDetails(
                 owner: "Foo",
-                name: "Bar",
-                defaultBranch: "Baz",
-                hasIssuesEnabled: false
+                name: "Bar"
         ))
 
         let detail2 = IssuesViewController(

gql/Fragments.graphql

@@ -1,3 +1,9 @@
+fragment subscribableFields on Subscribable {
+  id
+  viewerCanSubscribe
+  viewerSubscription
+}
+
 fragment reactionFields on Reactable {
   viewerCanReact
   reactionGroups {

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 {

gql/UpdateSubscription.graphql

@@ -0,0 +1,7 @@
+mutation UpdateSubscription($subscribable_Id: ID!, $subscription_state: SubscriptionState!) {
+  updateSubscription(input:{subscribableId: $subscribable_Id, state: $subscription_state}) {
+    subscribable{
+      ...subscribableFields
+    }
+  }
+}