Add JetpackAIServiceRemote/transcribeAudio

Author: kean

Sources/CoreAPI/HTTPRequestBuilder.swift

@@ -68,6 +68,13 @@ final class HTTPRequestBuilder {
             .append(query: urlComponents.queryItems ?? [])
     }
 
+    func headers(_ headers: [String: String]) -> Self {
+        for (key, value) in headers {
+            self.headers[key] = value
+        }
+        return self
+    }
+
     func header(name: String, value: String?) -> Self {
         headers[name] = value
         return self

Sources/CoreAPI/WordPressComRestApi.swift

@@ -449,7 +449,8 @@ open class WordPressComRestApi: NSObject {
 
     public func upload(
         URLString: String,
-        parameters: [String: AnyObject]?,
+        parameters: [String: AnyObject]? = nil,
+        httpHeaders: [String: String]? = nil,
         fileParts: [FilePart],
         requestEnqueued: RequestEnqueuedBlock? = nil,
         fulfilling progress: Progress? = nil
@@ -462,6 +463,7 @@ open class WordPressComRestApi: NSObject {
             builder = try requestBuilder(URLString: URLString)
                 .method(.post)
                 .body(form: form)
+                .headers(httpHeaders ?? [:])
         } catch {
             return .failure(.requestEncodingFailure(underlyingError: error))
         }

Sources/WordPressKit/Services/JetpackAIServiceRemote.swift

@@ -2,12 +2,26 @@ import Foundation
 
 public final class JetpackAIServiceRemote: SiteServiceRemoteWordPressComREST {
     /// Returns short-lived JWT token (lifetime is in minutes).
-    public func getJWT() async throws -> String {
+    public func getAuthorizationToken() async throws -> String {
         struct Response: Decodable {
             let token: String
         }
         let path = path(forEndpoint: "sites/\(siteID)/jetpack-openai-query/jwt", withVersion: ._2_0)
         let response = await wordPressComRestApi.perform(.post, URLString: path, type: Response.self)
         return try response.get().body.token
     }
+
+    /// - parameter token: Token retrieved using ``JetpackAIServiceRemote/getAuthorizationToken``.
+    public func transcribeAudio(from fileURL: URL, token: String) async throws -> String {
+        let path = path(forEndpoint: "jetpack-ai-transcription?feature=voice-to-content", withVersion: ._2_0)
+        let file = FilePart(parameterName: "audio_file", url: fileURL, fileName: "voice_recording", mimeType: "audio/m4a")
+        let result = await wordPressComRestApi.upload(URLString: path, httpHeaders: [
+            "Authorization": "Bearer \(token)"
+        ], fileParts: [file])
+        guard let body = try result.get().body as? [String: Any],
+              let text = body["text"] as? String else {
+            throw URLError(.unknown)
+        }
+        return text
+    }
 }