Change saved hashtags/instances to be per-account

See #16

Tusker.xcodeproj/project.pbxproj

@@ -156,10 +156,9 @@
 		D67C57B221E28FAD00C3118B /* ComposeStatusReplyView.xib in Resources */ = {isa = PBXBuildFile; fileRef = D67C57B121E28FAD00C3118B /* ComposeStatusReplyView.xib */; };
 		D67C57B421E2910700C3118B /* ComposeStatusReplyView.swift in Sources */ = {isa = PBXBuildFile; fileRef = D67C57B321E2910700C3118B /* ComposeStatusReplyView.swift */; };
 		D68FEC4F232C5BC300C84F23 /* SegmentedPageViewController.swift in Sources */ = {isa = PBXBuildFile; fileRef = D68FEC4E232C5BC300C84F23 /* SegmentedPageViewController.swift */; };
-		D6945C2F23AC47C3005C403C /* SavedHashtagsManager.swift in Sources */ = {isa = PBXBuildFile; fileRef = D6945C2E23AC47C3005C403C /* SavedHashtagsManager.swift */; };
+		D6945C2F23AC47C3005C403C /* SavedDataManager.swift in Sources */ = {isa = PBXBuildFile; fileRef = D6945C2E23AC47C3005C403C /* SavedDataManager.swift */; };
 		D6945C3223AC4D36005C403C /* HashtagTimelineViewController.swift in Sources */ = {isa = PBXBuildFile; fileRef = D6945C3123AC4D36005C403C /* HashtagTimelineViewController.swift */; };
 		D6945C3423AC6431005C403C /* AddSavedHashtagViewController.swift in Sources */ = {isa = PBXBuildFile; fileRef = D6945C3323AC6431005C403C /* AddSavedHashtagViewController.swift */; };
-		D6945C3623AC6C09005C403C /* SavedInstancesManager.swift in Sources */ = {isa = PBXBuildFile; fileRef = D6945C3523AC6C09005C403C /* SavedInstancesManager.swift */; };
 		D6945C3823AC739F005C403C /* InstanceTimelineViewController.swift in Sources */ = {isa = PBXBuildFile; fileRef = D6945C3723AC739F005C403C /* InstanceTimelineViewController.swift */; };
 		D6945C3A23AC75E2005C403C /* FindInstanceViewController.swift in Sources */ = {isa = PBXBuildFile; fileRef = D6945C3923AC75E2005C403C /* FindInstanceViewController.swift */; };
 		D6A3BC7723218E1300FD64D5 /* TimelineSegment.swift in Sources */ = {isa = PBXBuildFile; fileRef = D6A3BC7323218C6E00FD64D5 /* TimelineSegment.swift */; };
@@ -432,10 +431,9 @@
 		D67C57B121E28FAD00C3118B /* ComposeStatusReplyView.xib */ = {isa = PBXFileReference; lastKnownFileType = file.xib; path = ComposeStatusReplyView.xib; sourceTree = "<group>"; };
 		D67C57B321E2910700C3118B /* ComposeStatusReplyView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ComposeStatusReplyView.swift; sourceTree = "<group>"; };
 		D68FEC4E232C5BC300C84F23 /* SegmentedPageViewController.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = SegmentedPageViewController.swift; sourceTree = "<group>"; };
-		D6945C2E23AC47C3005C403C /* SavedHashtagsManager.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = SavedHashtagsManager.swift; sourceTree = "<group>"; };
+		D6945C2E23AC47C3005C403C /* SavedDataManager.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = SavedDataManager.swift; sourceTree = "<group>"; };
 		D6945C3123AC4D36005C403C /* HashtagTimelineViewController.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = HashtagTimelineViewController.swift; sourceTree = "<group>"; };
 		D6945C3323AC6431005C403C /* AddSavedHashtagViewController.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = AddSavedHashtagViewController.swift; sourceTree = "<group>"; };
-		D6945C3523AC6C09005C403C /* SavedInstancesManager.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = SavedInstancesManager.swift; sourceTree = "<group>"; };
 		D6945C3723AC739F005C403C /* InstanceTimelineViewController.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = InstanceTimelineViewController.swift; sourceTree = "<group>"; };
 		D6945C3923AC75E2005C403C /* FindInstanceViewController.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; name = FindInstanceViewController.swift; path = Tusker/Screens/FindInstanceViewController.swift; sourceTree = SOURCE_ROOT; };
 		D6A3BC7323218C6E00FD64D5 /* TimelineSegment.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = TimelineSegment.swift; sourceTree = "<group>"; };
@@ -1193,8 +1191,7 @@
 				D6AC956623C4347E008C9946 /* SceneDelegate.swift */,
 				D64D0AAC2128D88B005A6F37 /* LocalData.swift */,
 				D627FF75217E923E00CC0648 /* DraftsManager.swift */,
-				D6945C2E23AC47C3005C403C /* SavedHashtagsManager.swift */,
+				D6945C2E23AC47C3005C403C /* SavedDataManager.swift */,
-				D6945C3523AC6C09005C403C /* SavedInstancesManager.swift */,
 				D6028B9A2150811100F223B9 /* MastodonCache.swift */,
 				D6C693EE216192C2007D6A6D /* TuskerNavigationDelegate.swift */,
 				D6F1F84E2193B9BE00F5FE67 /* Caching */,
@@ -1641,7 +1638,6 @@
 				D68FEC4F232C5BC300C84F23 /* SegmentedPageViewController.swift in Sources */,
 				0450531F22B0097E00100BA2 /* Timline+UI.swift in Sources */,
 				D667E5F52135BCD50057A976 /* ConversationTableViewController.swift in Sources */,
-				D6945C3623AC6C09005C403C /* SavedInstancesManager.swift in Sources */,
 				D6C7D27D22B6EBF800071952 /* AttachmentsContainerView.swift in Sources */,
 				D620483823D38190008A63EF /* StatusContentTextView.swift in Sources */,
 				D6F953F021251A2900CF0F2B /* MastodonController.swift in Sources */,
@@ -1697,7 +1693,7 @@
 				0427033822B30F5F000D31B6 /* BehaviorPrefsView.swift in Sources */,
 				D627943923A553B600D38C68 /* UnbookmarkStatusActivity.swift in Sources */,
 				D64D0AAD2128D88B005A6F37 /* LocalData.swift in Sources */,
-				D6945C2F23AC47C3005C403C /* SavedHashtagsManager.swift in Sources */,
+				D6945C2F23AC47C3005C403C /* SavedDataManager.swift in Sources */,
 				D6C94D892139E6EC00CB5196 /* AttachmentView.swift in Sources */,
 				D6B053AE23BD322B00A066FA /* AssetPickerSheetContainerViewController.swift in Sources */,
 				D6C693EF216192C2007D6A6D /* TuskerNavigationDelegate.swift in Sources */,

Tusker/SavedDataManager.swift

@@ -0,0 +1,113 @@
+//
+//  SavedDataManager.swift
+//  Tusker
+//
+//  Created by Shadowfacts on 12/19/19.
+//  Copyright © 2019 Shadowfacts. All rights reserved.
+//
+
+import Foundation
+import Pachyderm
+
+class SavedDataManager: Codable {
+    private(set) static var shared: SavedDataManager = load()
+    
+    private static var documentsDirectory = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask).first!
+    private static var archiveURL = SavedDataManager.documentsDirectory.appendingPathComponent("saved_data").appendingPathExtension("plist")
+    
+    static func save() {
+        DispatchQueue.global(qos: .utility).async {
+            let encoder = PropertyListEncoder()
+            let data = try? encoder.encode(shared)
+            try? data?.write(to: archiveURL, options: .noFileProtection)
+        }
+    }
+    
+    static func load() -> SavedDataManager {
+        let decoder = PropertyListDecoder()
+        if let data = try? Data(contentsOf: archiveURL),
+            let savedHashtagsManager = try? decoder.decode(Self.self, from: data) {
+            return savedHashtagsManager
+        }
+        return SavedDataManager()
+    }
+    
+    private init() {}
+    
+    private var savedHashtags: [String: [Hashtag]] = [:] {
+        didSet {
+            SavedDataManager.save()
+            NotificationCenter.default.post(name: .savedHashtagsChanged, object: nil)
+        }
+    }
+
+    private var savedInstances: [String: [URL]] = [:] {
+        didSet {
+            SavedDataManager.save()
+            NotificationCenter.default.post(name: .savedInstancesChanged, object: nil)
+        }
+    }
+    
+    func sortedHashtags(for account: LocalData.UserAccountInfo) -> [Hashtag] {
+        if let hashtags = savedHashtags[account.id] {
+            return hashtags.sorted(by: { $0.name < $1.name })
+        } else {
+            return []
+        }
+    }
+    
+    func isSaved(hashtag: Hashtag, for account: LocalData.UserAccountInfo) -> Bool {
+        return savedHashtags[account.id]?.contains(hashtag) ?? false
+    }
+    
+    func add(hashtag: Hashtag, for account: LocalData.UserAccountInfo) {
+        if isSaved(hashtag: hashtag, for: account) {
+            return
+        }
+        if var saved = savedHashtags[account.id] {
+            saved.append(hashtag)
+            savedHashtags[account.id] = saved
+        } else {
+            savedHashtags[account.id] = [hashtag]
+        }
+    }
+    
+    func remove(hashtag: Hashtag, for account: LocalData.UserAccountInfo) {
+        guard isSaved(hashtag: hashtag, for: account) else { return }
+        if var saved = savedHashtags[account.id] {
+            saved.removeAll(where: { $0.name == hashtag.name })
+            savedHashtags[account.id] = saved
+        }
+    }
+    
+    func savedInstances(for account: LocalData.UserAccountInfo) -> [URL] {
+        return savedInstances[account.id] ?? []
+    }
+    
+    func isSaved(instance url: URL, for account: LocalData.UserAccountInfo) -> Bool {
+        return savedInstances[account.id]?.contains(url) ?? false
+    }
+    
+    func add(instance url: URL, for account: LocalData.UserAccountInfo) {
+        if isSaved(instance: url, for: account) { return }
+        if var saved = savedInstances[account.id] {
+            saved.append(url)
+            savedInstances[account.id] = saved
+        } else {
+            savedInstances[account.id] = [url]
+        }
+    }
+    
+    func remove(instance url: URL, for account: LocalData.UserAccountInfo) {
+        guard isSaved(instance: url, for: account) else { return }
+        if var saved = savedInstances[account.id] {
+            saved.removeAll(where: { $0 == url })
+            savedInstances[account.id] = saved
+        }
+    }
+}
+
+extension Foundation.Notification.Name {
+    static let savedHashtagsChanged = Notification.Name("savedHashtagsChanged")
+    static let savedInstancesChanged = Notification.Name("savedInstancesChanged")
+}

Tusker/SavedHashtagsManager.swift

@@ -1,65 +0,0 @@
-//
-//  SavedHashtagsManager.swift
-//  Tusker
-//
-//  Created by Shadowfacts on 12/19/19.
-//  Copyright © 2019 Shadowfacts. All rights reserved.
-//
-
-import Foundation
-import Pachyderm
-
-class SavedHashtagsManager: Codable {
-    private(set) static var shared: SavedHashtagsManager = load()
-    
-    private static var documentsDirectory = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask).first!
-    private static var archiveURL = SavedHashtagsManager.documentsDirectory.appendingPathComponent("saved_hashtags").appendingPathExtension("plist")
-    
-    static func save() {
-        DispatchQueue.global(qos: .utility).async {
-            let encoder = PropertyListEncoder()
-            let data = try? encoder.encode(shared)
-            try? data?.write(to: archiveURL, options: .noFileProtection)
-        }
-    }
-    
-    static func load() -> SavedHashtagsManager {
-        let decoder = PropertyListDecoder()
-        if let data = try? Data(contentsOf: archiveURL),
-            let savedHashtagsManager = try? decoder.decode(Self.self, from: data) {
-            return savedHashtagsManager
-        }
-        return SavedHashtagsManager()
-    }
-    
-    private init() {}
-    
-    private var savedHashtags: [Hashtag] = []
-    var sorted: [Hashtag] {
-        return savedHashtags.sorted(by: { $0.name < $1.name })
-    }
-    
-    func isSaved(_ hashtag: Hashtag) -> Bool {
-        return savedHashtags.contains(hashtag)
-    }
-    
-    func add(_ hashtag: Hashtag) {
-        if isSaved(hashtag) {
-            return
-        }
-        savedHashtags.append(hashtag)
-        SavedHashtagsManager.save()
-        NotificationCenter.default.post(name: .savedHashtagsChanged, object: nil)
-    }
-    
-    func remove(_ hashtag: Hashtag) {
-        guard isSaved(hashtag) else { return }
-        savedHashtags.removeAll(where: { $0.name == hashtag.name })
-        SavedHashtagsManager.save()
-        NotificationCenter.default.post(name: .savedHashtagsChanged, object: nil)
-    }
-}
-
-extension Foundation.Notification.Name {
-    static let savedHashtagsChanged = Notification.Name("savedHashtagsChanged")
-}

Tusker/SavedInstancesManager.swift

@@ -1,61 +0,0 @@
-//
-//  SavedInstancesManager.swift
-//  Tusker
-//
-//  Created by Shadowfacts on 12/19/19.
-//  Copyright © 2019 Shadowfacts. All rights reserved.
-//
-
-import Foundation
-
-class SavedInstanceManager: Codable {
-    private(set) static var shared: SavedInstanceManager = load()
-    
-    private static var documentsDirectory = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask).first!
-    private static var archiveURL = SavedInstanceManager.documentsDirectory.appendingPathComponent("saved_instances").appendingPathExtension("plist")
-    
-    static func save() {
-        DispatchQueue.global(qos: .utility).async {
-            let encoder = PropertyListEncoder()
-            let data = try? encoder.encode(shared)
-            try? data?.write(to: archiveURL, options: .noFileProtection)
-        }
-    }
-    
-    static func load() -> SavedInstanceManager {
-        let decoder = PropertyListDecoder()
-        if let data = try? Data(contentsOf: archiveURL),
-            let savedInstanceManager = try? decoder.decode(Self.self, from: data) {
-            return savedInstanceManager
-        }
-        return SavedInstanceManager()
-    }
-    
-    private init() {}
-    
-    private(set) var savedInstances: [URL] = []
-    
-    func isSaved(_ url: URL) -> Bool {
-        return savedInstances.contains(url)
-    }
-    
-    func add(_ url: URL) {
-        if isSaved(url) {
-            return
-        }
-        savedInstances.append(url)
-        SavedInstanceManager.save()
-        NotificationCenter.default.post(name: .savedInstancesChanged, object: nil)
-    }
-    
-    func remove(_ url: URL) {
-        guard isSaved(url) else { return }
-        savedInstances.removeAll(where: { $0 == url })
-        SavedInstanceManager.save()
-        NotificationCenter.default.post(name: .savedInstancesChanged, object: nil)
-    }
-}
-
-extension Notification.Name {
-    static let savedInstancesChanged = Notification.Name("savedInstancesChanged")
-}

Tusker/Screens/Explore/AddSavedHashtagViewController.swift

@@ -52,7 +52,7 @@ class AddSavedHashtagViewController: SearchResultsViewController {
 
 extension AddSavedHashtagViewController: SearchResultsViewControllerDelegate {
     func selectedSearchResult(hashtag: Hashtag) {
-        SavedHashtagsManager.shared.add(hashtag)
+        SavedDataManager.shared.add(hashtag: hashtag, for: mastodonController.accountInfo!)
         dismiss(animated: true)
     }
 }

Tusker/Screens/Explore/ExploreViewController.swift

@@ -81,12 +81,14 @@ class ExploreViewController: EnhancedTableViewController {
         })
         dataSource.exploreController = self
         
+        let account = mastodonController.accountInfo!
+        
         var snapshot = NSDiffableDataSourceSnapshot<Section, Item>()
         snapshot.appendSections([.bookmarks, .lists, .savedHashtags, .savedInstances])
         snapshot.appendItems([.bookmarks], toSection: .bookmarks)
         snapshot.appendItems([.addList], toSection: .lists)
-        snapshot.appendItems(SavedHashtagsManager.shared.sorted.map { .savedHashtag($0) } + [.addSavedHashtag], toSection: .savedHashtags)
+        snapshot.appendItems(SavedDataManager.shared.sortedHashtags(for: account).map { .savedHashtag($0) } + [.addSavedHashtag], toSection: .savedHashtags)
-        snapshot.appendItems(SavedInstanceManager.shared.savedInstances.map { .savedInstance($0) } + [.findInstance], toSection: .savedInstances)
+        snapshot.appendItems(SavedDataManager.shared.savedInstances(for: account).map { .savedInstance($0) } + [.findInstance], toSection: .savedInstances)
         // the initial, static items should not be displayed with an animation
         UIView.performWithoutAnimation {
             dataSource.apply(snapshot)
@@ -127,16 +129,18 @@ class ExploreViewController: EnhancedTableViewController {
     }
     
     @objc func savedHashtagsChanged() {
+        let account = mastodonController.accountInfo!
         var snapshot = dataSource.snapshot()
         snapshot.deleteItems(snapshot.itemIdentifiers(inSection: .savedHashtags))
-        snapshot.appendItems(SavedHashtagsManager.shared.sorted.map { .savedHashtag($0) } + [.addSavedHashtag], toSection: .savedHashtags)
+        snapshot.appendItems(SavedDataManager.shared.sortedHashtags(for: account).map { .savedHashtag($0) } + [.addSavedHashtag], toSection: .savedHashtags)
         dataSource.apply(snapshot)
     }
     
     @objc func savedInstancesChanged() {
+        let account = mastodonController.accountInfo!
         var snapshot = dataSource.snapshot()
         snapshot.deleteItems(snapshot.itemIdentifiers(inSection: .savedInstances))
-        snapshot.appendItems(SavedInstanceManager.shared.savedInstances.map { .savedInstance($0) } + [.findInstance], toSection: .savedInstances)
+        snapshot.appendItems(SavedDataManager.shared.savedInstances(for: account).map { .savedInstance($0) } + [.findInstance], toSection: .savedInstances)
         dataSource.apply(snapshot)
     }
     
@@ -163,11 +167,13 @@ class ExploreViewController: EnhancedTableViewController {
     }
     
     func removeSavedHashtag(_ hashtag: Hashtag) {
-        SavedHashtagsManager.shared.remove(hashtag)
+        let account = mastodonController.accountInfo!
+        SavedDataManager.shared.remove(hashtag: hashtag, for: account)
     }
     
     func removeSavedInstance(_ instanceURL: URL) {
-        SavedInstanceManager.shared.remove(instanceURL)
+        let account = mastodonController.accountInfo!
+        SavedDataManager.shared.remove(instance: instanceURL, for: account)
     }
     
     // MARK: - Table view delegate
@@ -217,11 +223,11 @@ class ExploreViewController: EnhancedTableViewController {
             present(navController, animated: true)
             
         case let .savedInstance(url):
-            show(InstanceTimelineViewController(for: url), sender: nil)
+            show(InstanceTimelineViewController(for: url, parentMastodonController: mastodonController), sender: nil)
             
         case .findInstance:
             tableView.selectRow(at: nil, animated: true, scrollPosition: .none)
-            let findController = FindInstanceViewController()
+            let findController = FindInstanceViewController(parentMastodonController: mastodonController)
             findController.instanceTimelineDelegate = self
             let navController = UINavigationController(rootViewController: findController)
             present(navController, animated: true)
@@ -348,7 +354,7 @@ extension ExploreViewController {
 extension ExploreViewController: InstanceTimelineViewControllerDelegate {
     func didSaveInstance(url: URL) {
         dismiss(animated: true) {
-            self.show(InstanceTimelineViewController(for: url), sender: nil)
+            self.show(InstanceTimelineViewController(for: url, parentMastodonController: self.mastodonController), sender: nil)
         }
     }
     

Tusker/Screens/FindInstanceViewController.swift

@@ -10,8 +10,20 @@ import UIKit
 
 class FindInstanceViewController: InstanceSelectorTableViewController {
 
+    weak var parentMastodonController: MastodonController?
+    
     var instanceTimelineDelegate: InstanceTimelineViewControllerDelegate?
     
+    init(parentMastodonController: MastodonController) {
+        self.parentMastodonController = parentMastodonController
+        
+        super.init()
+    }
+    
+    required init?(coder: NSCoder) {
+        fatalError("init(coder:) has not been implemented")
+    }
+    
     override func viewDidLoad() {
         super.viewDidLoad()
         
@@ -32,7 +44,7 @@ class FindInstanceViewController: InstanceSelectorTableViewController {
 
 extension FindInstanceViewController: InstanceSelectorTableViewControllerDelegate {
     func didSelectInstance(url: URL) {
-        let instanceTimelineController = InstanceTimelineViewController(for: url)
+        let instanceTimelineController = InstanceTimelineViewController(for: url, parentMastodonController: parentMastodonController!)
         instanceTimelineController.delegate = instanceTimelineDelegate
         show(instanceTimelineController, sender: self)
     }

Tusker/Screens/Onboarding/InstanceSelectorTableViewController.swift

@@ -55,7 +55,7 @@ class InstanceSelectorTableViewController: UITableViewController {
         tableView.rowHeight = UITableView.automaticDimension
         tableView.estimatedRowHeight = 120
         
-        dataSource = DataSource(tableView: tableView, cellProvider: { [weak self] (tableView, indexPath, item) -> UITableViewCell? in
+        dataSource = DataSource(tableView: tableView, cellProvider: { (tableView, indexPath, item) -> UITableViewCell? in
             switch item {
             case let .selected(instance):
                 let cell = tableView.dequeueReusableCell(withIdentifier: instanceCell, for: indexPath) as! InstanceTableViewCell

Tusker/Screens/Timeline/HashtagTimelineViewController.swift

@@ -15,7 +15,7 @@ class HashtagTimelineViewController: TimelineTableViewController {
     
     var toggleSaveButton: UIBarButtonItem!
     var toggleSaveButtonTitle: String {
-        if SavedHashtagsManager.shared.isSaved(hashtag) {
+        if SavedDataManager.shared.isSaved(hashtag: hashtag, for: mastodonController.accountInfo!) {
             return NSLocalizedString("Unsave", comment: "unsave hashtag button")
         } else {
             return NSLocalizedString("Save", comment: "save hashtag button")
@@ -48,10 +48,10 @@ class HashtagTimelineViewController: TimelineTableViewController {
     // MARK: - Interaction
     
     @objc func toggleSaveButtonPressed() {
-        if SavedHashtagsManager.shared.isSaved(hashtag) {
+        if SavedDataManager.shared.isSaved(hashtag: hashtag, for: mastodonController.accountInfo!) {
-            SavedHashtagsManager.shared.remove(hashtag)
+            SavedDataManager.shared.remove(hashtag: hashtag, for: mastodonController.accountInfo!)
         } else {
-            SavedHashtagsManager.shared.add(hashtag)
+            SavedDataManager.shared.add(hashtag: hashtag, for: mastodonController.accountInfo!)
         }
     }
 

Tusker/Screens/Timeline/InstanceTimelineViewController.swift

@@ -17,23 +17,29 @@ class InstanceTimelineViewController: TimelineTableViewController {
     
     weak var delegate: InstanceTimelineViewControllerDelegate?
 
+    weak var parentMastodonController: MastodonController?
+    
     let instanceURL: URL
+    let instanceMastodonController: MastodonController
     
     var toggleSaveButton: UIBarButtonItem!
     var toggleSaveButtonTitle: String {
-        if SavedInstanceManager.shared.isSaved(instanceURL) {
+        if SavedDataManager.shared.isSaved(instance: instanceURL, for: parentMastodonController!.accountInfo!) {
             return NSLocalizedString("Unsave", comment: "unsave instance button")
         } else {
             return NSLocalizedString("Save", comment: "save instance button")
         }
     }
     
-    init(for url: URL) {
+    init(for url: URL, parentMastodonController: MastodonController) {
-        self.instanceURL = url
+        self.parentMastodonController = parentMastodonController
         
-        let mastodonController = MastodonController(instanceURL: url)
+        self.instanceURL = url
 
-        super.init(for: .instance(instanceURL: url), mastodonController: mastodonController)
+        // the timeline VC only stores a weak reference to it, so we need to store a strong reference to make sure it's not released immediately
+        instanceMastodonController = MastodonController(instanceURL: url)
+
+        super.init(for: .instance(instanceURL: url), mastodonController: instanceMastodonController)
     }
     
     required init?(coder aDecoder: NSCoder) {
@@ -70,11 +76,11 @@ class InstanceTimelineViewController: TimelineTableViewController {
     
     // MARK: - Interaction
     @objc func toggleSaveButtonPressed() {
-        if SavedInstanceManager.shared.isSaved(instanceURL) {
+        if SavedDataManager.shared.isSaved(instance: instanceURL, for: parentMastodonController!.accountInfo!) {
-            SavedInstanceManager.shared.remove(instanceURL)
+            SavedDataManager.shared.remove(instance: instanceURL, for: parentMastodonController!.accountInfo!)
             delegate?.didUnsaveInstance(url: instanceURL)
         } else {
-            SavedInstanceManager.shared.add(instanceURL)
+            SavedDataManager.shared.add(instance: instanceURL, for: parentMastodonController!.accountInfo!)
             delegate?.didSaveInstance(url: instanceURL)
         }
     }