Sora/Data/Settings/Settings.swift


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

import SwiftUI

class Settings: ObservableObject {
    #if DEBUG
        @AppStorage("detailViewType") var detailViewType: BooruPostFileType = .sample
    #else
        @AppStorage("detailViewType") var detailViewType: BooruPostFileType = .original
    #endif
    @AppStorage("thumbnailType") var thumbnailType: BooruPostFileType = .preview
    @AppStorage("searchSuggestions") var searchSuggestions: Bool = false
    @AppStorage("columns") var columns: Int = 2
    @AppStorage("blurNSFWThumbnails") var blurNSFWThumbnails: Bool = true
    @AppStorage("showNSFWPosts") var showNSFWPosts: Bool = false
    @AppStorage("bookmarks") private var bookmarksData: Data = .init()
    @AppStorage("preferredBooru") var preferredBooru: BooruProvider = .yandere

    var bookmarks: [Bookmark] {
        get {
            if let bookmarks = try? JSONDecoder().decode([Bookmark].self, from: bookmarksData) {
                return bookmarks
            }

            return []
        }

        set {
            if let data = try? JSONEncoder().encode(newValue) {
                bookmarksData = data
            }
        }
    }

    func resetToDefaults() {
        #if DEBUG
            detailViewType = .preview
        #else
            detailViewType = .original
        #endif
        thumbnailType = .preview
        searchSuggestions = false
        columns = 2
        blurNSFWThumbnails = true
        showNSFWPosts = false
    }

    func addBookmark(provider: BooruProvider, tags: [String]) {
        var currentBookmarks = bookmarks

        currentBookmarks.append(Bookmark(provider: provider, tags: tags.map { $0.lowercased() }))

        bookmarks = currentBookmarks
    }

    func removeBookmark(at index: IndexSet) {
        var currentBookmarks = bookmarks

        currentBookmarks.remove(atOffsets: index)

        bookmarks = currentBookmarks
    }

    func removeBookmark(withTags tags: [String]) {
        var currentBookmarks = bookmarks

        currentBookmarks.removeAll { bookmark in
            bookmark.tags.contains(where: tags.contains)
        }

        bookmarks = currentBookmarks
    }

    func removeBookmark(withID: UUID) {
        var currentBookmarks = bookmarks

        currentBookmarks.removeAll { bookmark in
            bookmark.id == withID
        }

        bookmarks = currentBookmarks
    }
}