Add makePostContent(fromPlainText

Author: kean

Sources/CoreAPI/WordPressComRestApi.swift

@@ -309,7 +309,7 @@ open class WordPressComRestApi: NSObject {
         return "\(String(describing: oAuthToken)),\(String(describing: userAgent))".hashValue
     }
 
-    private func requestBuilder(URLString: String) throws -> HTTPRequestBuilder {
+    func requestBuilder(URLString: String) throws -> HTTPRequestBuilder {
         guard let url = URL(string: URLString, relativeTo: baseURL) else {
             throw URLError(.badURL)
         }
@@ -414,9 +414,9 @@ open class WordPressComRestApi: NSObject {
         return await perform(request: builder, fulfilling: progress, decoder: decoder)
     }
 
-    private func perform<T>(
+    func perform<T>(
         request: HTTPRequestBuilder,
-        fulfilling progress: Progress?,
+        fulfilling progress: Progress? = nil,
         decoder: @escaping (Data) throws -> T,
         taskCreated: ((Int) -> Void)? = nil,
         session: URLSession? = nil

Sources/WordPressKit/Services/JetpackAIServiceRemote.swift

@@ -24,4 +24,55 @@ public final class JetpackAIServiceRemote: SiteServiceRemoteWordPressComREST {
         }
         return text
     }
+
+    /// - parameter token: Token retrieved using ``JetpackAIServiceRemote/getAuthorizationToken``.
+    public func makePostContent(fromPlainText plainText: String, token: String) async throws -> String {
+        let path = path(forEndpoint: "jetpack-ai-query", withVersion: ._2_0)
+        let request = JetpackAIQueryRequest(messages: [
+            .init(role: "jetpack-ai", context: .init(type: "voice-to-content-simple-draft", content: plainText))
+        ], feature: "voice-to-content", stream: false)
+        let builder = try wordPressComRestApi.requestBuilder(URLString: path)
+            .method(.post)
+            .headers(["Authorization": "Bearer \(token)"])
+            .body(json: request, jsonEncoder: JSONEncoder())
+        let result = await wordPressComRestApi.perform(request: builder) { data in
+            try JSONDecoder().decode(JetpackAIQueryResponse.self, from: data)
+        }
+        let response = try result.get().body
+        guard let content = response.choices.first?.message.content else {
+            throw URLError(.unknown)
+        }
+        return content
+    }
+}
+
+private struct JetpackAIQueryRequest: Encodable {
+    let messages: [Message]
+    let feature: String
+    let stream: Bool
+
+    struct Message: Encodable {
+        let role: String
+        let context: Context
+    }
+
+    struct Context: Codable {
+        let type: String
+        let content: String
+    }
+}
+
+private struct JetpackAIQueryResponse: Decodable {
+    let model: String?
+    let choices: [Choice]
+
+    struct Choice: Codable {
+        let index: Int
+        let message: Message
+    }
+
+    struct Message: Codable {
+        let role: String?
+        let content: String
+    }
 }