feat: Development commit

5 files changed, 216 insertions, 0 deletions

diff --git a/Sora/Data/Moebooru/MoebooruManager.swift b/Sora/Data/Moebooru/MoebooruManager.swift
new file mode 100644
index 0000000..fd8d337
--- /dev/null
+++ b/Sora/Data/Moebooru/MoebooruManager.swift
@@ -0,0 +1,114 @@
+import SwiftUI
+
+@MainActor
+class MoebooruManager: ObservableObject {
+    @Published var posts: [MoebooruPost] = []
+    @Published var allTags: [MoebooruTag] = []
+    @Published var isLoading: Bool = false
+    @Published var currentPage: Int = 1
+    @Published var searchText = ""
+    @Published var endOfData: Bool = false
+    #if os(macOS)
+        @Published var selectedPost: MoebooruPost?
+    #endif
+    private var currentTask: Task<Void, Never>?
+    var tags: [String] {
+        if searchText.isEmpty {
+            return []
+        }
+
+        return searchText
+            .split(separator: " ")
+            .map { $0.trimmingCharacters(in: .whitespacesAndNewlines) }
+            .filter { !$0.isEmpty }
+    }
+
+    init() {
+        fetchAllTags()
+    }
+
+    func fetchPosts(page: Int = 1, limit: Int = 100, tags: [String] = [], replace: Bool = false) async {
+        guard !isLoading else { return }
+
+        currentTask?.cancel()
+
+        currentTask = Task {
+            isLoading = true
+
+            defer { isLoading = false }
+
+            if replace {
+                self.posts = []
+                self.currentPage = 1
+            }
+
+            guard let url = URL(string: "https://yande.re/post.xml?page=\(page)&limit=\(limit)&tags=\(tags.joined(separator: "+"))") else { return }
+
+            do {
+                let (data, _) = try await URLSession.shared.data(from: url)
+
+                if Task.isCancelled { return }
+
+                DispatchQueue.main.async {
+                    let newPosts = Array(Set(MoebooruPostXMLParser().parse(data: data))).sorted { $0.id > $1.id }
+
+                    if newPosts == [] {
+                        self.endOfData = true
+                    } else {
+                        self.posts += newPosts
+                    }
+                }
+            } catch {
+                if (error as? URLError)?.code != .cancelled {
+                    #if DEBUG
+                        print(error)
+                    #endif
+                }
+            }
+        }
+    }
+
+    func performSearch() {
+        Task {
+            await fetchPosts(
+                page: 1,
+                tags: tags,
+                replace: true
+            )
+        }
+    }
+
+    func loadNextPage() {
+        guard !isLoading else { return }
+
+        Task {
+            await fetchPosts(page: currentPage + 1, tags: tags)
+
+            DispatchQueue.main.async {
+                self.currentPage += 1
+            }
+        }
+    }
+
+    func fetchAllTags(limit: Int = 100_000) {
+        Task {
+            guard let url = URL(string: "https://yande.re/tag.xml?limit=\(limit)") else { return }
+
+            do {
+                let (data, _) = try await URLSession.shared.data(from: url)
+
+                if Task.isCancelled { return }
+
+                DispatchQueue.main.async {
+                    self.allTags = (MoebooruTagXMLParser().parse(data: data)).sorted { $0.count > $1.count }
+                }
+            } catch {
+                if (error as? URLError)?.code != .cancelled {
+                    #if DEBUG
+                        print(error)
+                    #endif
+                }
+            }
+        }
+    }
+}
diff --git a/Sora/Data/Moebooru/MoebooruPost.swift b/Sora/Data/Moebooru/MoebooruPost.swift
new file mode 100644
index 0000000..df332ec
--- /dev/null
+++ b/Sora/Data/Moebooru/MoebooruPost.swift
@@ -0,0 +1,16 @@
+import Foundation
+
+struct MoebooruPost: Identifiable, Hashable {
+    let id: Int
+    let tags: [String]
+    let createdAt: Date
+    let author: String
+    let source: URL?
+    let score: Int
+    let fileURL: URL?
+    let previewURL: URL?
+    let sampleURL: URL?
+    let jpegURL: URL?
+    let width: Int
+    let height: Int
+}
diff --git a/Sora/Data/Moebooru/MoebooruPostXMLParser.swift b/Sora/Data/Moebooru/MoebooruPostXMLParser.swift
new file mode 100644
index 0000000..418a9c6
--- /dev/null
+++ b/Sora/Data/Moebooru/MoebooruPostXMLParser.swift
@@ -0,0 +1,42 @@
+import Foundation
+
+class MoebooruPostXMLParser: NSObject, XMLParserDelegate {
+    private var posts: [MoebooruPost] = []
+
+    func parse(data: Data) -> [MoebooruPost] {
+        let parser = XMLParser(data: data)
+
+        parser.delegate = self
+        parser.parse()
+
+        return posts
+    }
+
+    func parser(_: XMLParser, didStartElement elementName: String,
+                namespaceURI _: String?, qualifiedName _: String?,
+                attributes attributeDict: [String: String])
+    {
+        if elementName == "post",
+           let id = Int(attributeDict["id"] ?? ""),
+           let createdAtTimestamp = TimeInterval(attributeDict["created_at"] ?? ""),
+           let score = Int(attributeDict["score"] ?? ""),
+           let width = Int(attributeDict["width"] ?? ""),
+           let height = Int(attributeDict["height"] ?? "")
+        {
+            posts.append(MoebooruPost(
+                id: id,
+                tags: attributeDict["tags"]?.components(separatedBy: " ") ?? [],
+                createdAt: Date(timeIntervalSince1970: createdAtTimestamp),
+                author: attributeDict["author"] ?? "",
+                source: URL(string: attributeDict["source"] ?? ""),
+                score: score,
+                fileURL: URL(string: attributeDict["file_url"] ?? ""),
+                previewURL: URL(string: attributeDict["preview_url"] ?? ""),
+                sampleURL: URL(string: attributeDict["sample_url"] ?? ""),
+                jpegURL: URL(string: attributeDict["jpeg_url"] ?? ""),
+                width: width,
+                height: height
+            ))
+        }
+    }
+}
diff --git a/Sora/Data/Moebooru/MoebooruTag.swift b/Sora/Data/Moebooru/MoebooruTag.swift
new file mode 100644
index 0000000..cc783f0
--- /dev/null
+++ b/Sora/Data/Moebooru/MoebooruTag.swift
@@ -0,0 +1,9 @@
+import Foundation
+
+struct MoebooruTag: Identifiable, Hashable {
+    let id: Int
+    let name: String
+    let count: Int
+    let type: Int
+    let ambiguous: Bool
+}
diff --git a/Sora/Data/Moebooru/MoebooruTagXMLParser.swift b/Sora/Data/Moebooru/MoebooruTagXMLParser.swift
new file mode 100644
index 0000000..7ea2e8e
--- /dev/null
+++ b/Sora/Data/Moebooru/MoebooruTagXMLParser.swift
@@ -0,0 +1,35 @@
+import Foundation
+
+class MoebooruTagXMLParser: NSObject, XMLParserDelegate {
+    private var tags: [MoebooruTag] = []
+
+    func parse(data: Data) -> [MoebooruTag] {
+        let parser = XMLParser(data: data)
+
+        parser.delegate = self
+        parser.parse()
+
+        return tags
+    }
+
+    func parser(_: XMLParser, didStartElement elementName: String,
+                namespaceURI _: String?, qualifiedName _: String?,
+                attributes attributeDict: [String: String])
+    {
+        if elementName == "tag" {
+            if let id = Int(attributeDict["id"] ?? ""),
+               let count = Int(attributeDict["count"] ?? ""),
+               let type = Int(attributeDict["type"] ?? ""),
+               let ambiguous = Bool(attributeDict["ambiguous"] ?? "false")
+            {
+                tags.append(MoebooruTag(
+                    id: id,
+                    name: attributeDict["name"] ?? "",
+                    count: count,
+                    type: type,
+                    ambiguous: ambiguous
+                ))
+            }
+        }
+    }
+}