Tusker/Pachyderm/Client.swift

//
//  Client.swift
//  Pachyderm
//
//  Created by Shadowfacts on 9/8/18.
//  Copyright © 2018 Shadowfacts. All rights reserved.
//

import Foundation

/**
 The base Mastodon API client.
 */
public class Client {
    
    public typealias Callback<Result: Decodable> = (Response<Result>) -> Void
    
    let baseURL: URL
    let session: URLSession
    
    public var accessToken: String?
    
    public var appID: String?
    public var clientID: String?
    public var clientSecret: String?
    
    public var timeoutInterval: TimeInterval = 60
    
    lazy var decoder: JSONDecoder = {
        let decoder = JSONDecoder()
        let formatter = DateFormatter()
        formatter.dateFormat = "yyyy-MM-dd'T'HH:mm:ss.SZ"
        formatter.timeZone = TimeZone(abbreviation: "UTC")
        formatter.locale = Locale(identifier: "en_US_POSIX")
        decoder.dateDecodingStrategy = .formatted(formatter)
        return decoder
    }()
    
    public init(baseURL: URL, accessToken: String? = nil, session: URLSession = .shared) {
        self.baseURL = baseURL
        self.accessToken = accessToken
        self.session = session
    }
    
    public func run<Result>(_ request: Request<Result>, completion: @escaping Callback<Result>) {
        guard let request = createURLRequest(request: request) else {
            completion(.failure(Error.invalidRequest))
            return
        }
        
        let task = session.dataTask(with: request) { data, response, error in
            if let error = error {
                completion(.failure(error))
                return
            }
            guard let data = data,
                let response = response as? HTTPURLResponse else {
                    completion(.failure(Error.invalidResponse))
                    return
            }
            guard response.statusCode == 200 else {
                let mastodonError = try? self.decoder.decode(MastodonError.self, from: data)
                let error = mastodonError.flatMap { Error.mastodonError($0.description) } ?? Error.unknownError
                completion(.failure(error))
                return
            }
            guard let result = try? self.decoder.decode(Result.self, from: data) else {
                completion(.failure(Error.invalidModel))
                return
            }
            if var result = result as? ClientModel {
                result.client = self
            } else if var result = result as? [ClientModel] {
                result.client = self
            }
            let pagination = response.allHeaderFields["Link"].flatMap { $0 as? String }.flatMap(Pagination.init)
            
            completion(.success(result, pagination))
        }
        task.resume()
    }
    
    func createURLRequest<Result>(request: Request<Result>) -> URLRequest? {
        guard var components = URLComponents(url: baseURL, resolvingAgainstBaseURL: true) else { return nil }
        components.path = request.path
        components.queryItems = request.queryParameters.queryItems
        guard let url = components.url else { return nil }
        var urlRequest = URLRequest(url: url, timeoutInterval: timeoutInterval)
        urlRequest.httpMethod = request.method.name
        urlRequest.httpBody = request.body.data
        urlRequest.setValue(request.body.mimeType, forHTTPHeaderField: "Content-Type")
        if let accessToken = accessToken {
            urlRequest.setValue("Bearer \(accessToken)", forHTTPHeaderField: "Authorization")
        }
        return urlRequest
    }
    
    // MARK: - Authorization
    public func registerApp(name: String, redirectURI: String, scopes: [Scope], website: URL? = nil, completion: @escaping Callback<RegisteredApplication>) {
        let request = Request<RegisteredApplication>(method: .post, path: "/api/v1/apps", body: .parameters([
                "client_name" => name,
                "redirect_uris" => redirectURI,
                "scopes" => scopes.scopeString,
                "website" => website?.absoluteString
            ]))
        run(request) { result in
            defer { completion(result) }
            guard case let .success(application, _) = result else { return }
            
            self.appID = application.id
            self.clientID = application.clientID
            self.clientSecret = application.clientSecret
        }
    }
    
    public func getAccessToken(authorizationCode: String, redirectURI: String, completion: @escaping Callback<LoginSettings>) {
        let request = Request<LoginSettings>(method: .post, path: "/oauth/token", body: .parameters([
                "client_id" => clientID,
                "client_secret" => clientSecret,
                "grant_type" => "authorization_code",
                "code" => authorizationCode,
                "redirect_uri" => redirectURI
            ]))
        run(request) { result in
            defer { completion(result) }
            guard case let .success(loginSettings, _) = result else { return }
            
            self.accessToken = loginSettings.accessToken
        }
    }
    
    // MARK: - Self
    public func getSelfAccount(completion: @escaping Callback<Account>) {
        let request = Request<Account>(method: .get, path: "/api/v1/accounts/verify_credentials")
        run(request, completion: completion)
    }
    
    public func getFavourites(completion: @escaping Callback<[Status]>) {
        let request = Request<[Status]>(method: .get, path: "/api/v1/favourites")
        run(request, completion: completion)
    }
    
    public func getRelationships(accounts: [Account]? = nil, completion: @escaping Callback<[Relationship]>) {
        let request = Request<[Relationship]>(method: .get, path: "/api/v1/accounts/relationships", queryParameters: "id" => accounts?.map { $0.id })
        run(request, completion: completion)
    }
    
    public func getInstance(completion: @escaping Callback<Instance>) {
        let request = Request<Instance>(method: .get, path: "/api/v1/instance")
        run(request, completion: completion)
    }
    
    public func getCustomEmoji(completion: @escaping Callback<[Emoji]>) {
        let request = Request<[Emoji]>(method: .get, path: "/api/v1/custom_emojis")
        run(request, completion: completion)
    }
    
    // MARK: - Accounts
    public func getAccount(id: String, completion: @escaping Callback<Account>) {
        let request = Request<Account>(method: .get, path: "/api/v1/accounts/\(id)")
        run(request, completion: completion)
    }
    
    public func searchForAccount(query: String, limit: Int? = nil, following: Bool? = nil, completion: @escaping Callback<[Account]>) {
        let request = Request<[Account]>(method: .get, path: "/api/v1/accounts/search", queryParameters: [
                "q" => query,
                "limit" => limit,
                "following" => following
            ])
        run(request, completion: completion)
    }
    
    // MARK: - Blocks
    public func getBlocks(completion: @escaping Callback<[Account]>) {
        let request = Request<[Account]>(method: .get, path: "/api/v1/blocks")
        run(request, completion: completion)
    }
    
    public func getDomainBlocks(completion: @escaping Callback<[String]>) {
        let request = Request<[String]>(method: .get, path: "api/v1/domain_blocks")
        run(request, completion: completion)
    }
    
    public func block(domain: String, completion: @escaping Callback<Empty>) {
        let request = Request<Empty>(method: .post, path: "/api/v1/domain_blocks", body: .parameters([
                "domain" => domain
            ]))
        run(request, completion: completion)
    }
    
    public func unblock(domain: String, completion: @escaping Callback<Empty>) {
        let request = Request<Empty>(method: .delete, path: "/api/v1/domain_blocks", body: .parameters([
                "domain" => domain
            ]))
        run(request, completion: completion)
    }
    
    // MARK: - Filters
    public func getFilters(completion: @escaping Callback<[Filter]>) {
        let request = Request<[Filter]>(method: .get, path: "/api/v1/filters")
        run(request, completion: completion)
    }
    
    public func createFilter(phrase: String, context: [Filter.Context], irreversible: Bool? = nil, wholeWord: Bool? = nil, expiresAt: Date? = nil, completion: @escaping Callback<Filter>) {
        let request = Request<Filter>(method: .post, path: "/api/v1/filters", body: .parameters([
                "phrase" => phrase,
                "irreversible" => irreversible,
                "whole_word" => wholeWord,
                "expires_at" => expiresAt
            ] + "context" => context.contextStrings))
        run(request, completion: completion)
    }
    
    public func getFilter(id: String, completion: @escaping Callback<Filter>) {
        let request = Request<Filter>(method: .get, path: "/api/v1/filters/\(id)")
        run(request, completion: completion)
    }
    
    // MARK: - Follows
    public func getFollowRequests(range: RequestRange = .default, completion: @escaping Callback<[Account]>) {
        var request = Request<[Account]>(method: .get, path: "/api/v1/follow_requests")
        request.range = range
        run(request, completion: completion)
    }
    
    public func getFollowSuggestions(completion: @escaping Callback<[Account]>) {
        let request = Request<[Account]>(method: .get, path: "/api/v1/suggestions")
        run(request, completion: completion)
    }
    
    public func followRemote(acct: String, completion: @escaping Callback<Account>) {
        let request = Request<Account>(method: .post, path: "/api/v1/follows", body: .parameters(["uri" => acct]))
        run(request, completion: completion)
    }
    
    // MARK: - Lists
    public func getLists(completion: @escaping Callback<[List]>) {
        let request = Request<[List]>(method: .get, path: "/api/v1/lists")
        run(request, completion: completion)
    }
    
    public func getList(id: String, completion: @escaping Callback<List>) {
        let request = Request<List>(method: .get, path: "/api/v1/lists/\(id)")
        run(request, completion: completion)
    }
    
    public func createList(title: String, completion: @escaping Callback<List>) {
        let request = Request<List>(method: .post, path: "/api/v1/lists", body: .parameters(["title" => title]))
        run(request, completion: completion)
    }
    
    // MARK: - Media
    public func upload(attachment: FormAttachment, description: String? = nil, focus: (Float, Float)? = nil, completion: @escaping Callback<Attachment>) {
        let request = Request<Attachment>(method: .post, path: "/api/v1/media", body: .formData([
                "description" => description,
                "focus" => focus
            ], attachment))
        run(request, completion: completion)
    }
    
    // MARK: - Mutes
    public func getMutes(range: RequestRange, completion: @escaping Callback<[Account]>) {
        var request = Request<[Account]>(method: .get, path: "/api/v1/mutes")
        request.range = range
        run(request, completion: completion)
    }
    
    // MARK: - Notifications
    public func getNotifications(range: RequestRange = .default, completion: @escaping Callback<[Notification]>) {
        var request = Request<[Notification]>(method: .get, path: "/api/v1/notifications")
        request.range = range
        run(request, completion: completion)
    }
    
    public func clearNotifications(completion: @escaping Callback<Empty>) {
        let request = Request<Empty>(method: .post, path: "/api/v1/notifications/clear")
        run(request, completion: completion)
    }
    
    // MARK: - Reports
    public func getReports(completion: @escaping Callback<[Report]>) {
        let request = Request<[Report]>(method: .get, path: "/api/v1/reports")
        run(request, completion: completion)
    }
    
    public func report(account: Account, statuses: [Status], comment: String, completion: @escaping Callback<Report>) {
        let request = Request<Report>(method: .post, path: "/api/v1/reports", body: .parameters([
                "account_id" => account.id,
                "comment" => comment
            ] + "status_ids" => statuses.map { $0.id }))
        run(request, completion: completion)
    }
    
    // MARK: - Search
    public func search(query: String, resolve: Bool? = nil, completion: @escaping Callback<SearchResults>) {
        let request = Request<SearchResults>(method: .get, path: "/api/v2/search", queryParameters: [
                "q" => query,
                "resolve" => resolve
            ])
        run(request, completion: completion)
    }
    
    // MARK: - Statuses
    public func getStatus(id: String, completion: @escaping Callback<Status>) {
        let request = Request<Status>(method: .get, path: "/api/v1/statuses/\(id)")
        run(request, completion: completion)
    }
    
    public func createStatus(text: String,
                             inReplyTo: Status? = nil,
                             media: [Attachment]? = nil,
                             sensitive: Bool? = nil,
                             spoilerText: String? = nil,
                             visiblity: Status.Visibility? = nil,
                             language: String? = nil,
                             completion: @escaping Callback<Status>) {
        let request = Request<Status>(method: .post, path: "/api/v1/statuses", body: .parameters([
                "status" => text,
                "in_reply_to_id" => inReplyTo?.id,
                "sensitive" => sensitive,
                "spoiler_text" => spoilerText,
                "visibility" => visiblity?.rawValue,
                "language" => language
            ] + "media_ids" => media?.map { $0.id }))
        run(request, completion: completion)
    }
    
    // MARK: - Timelines
    public func getStatuses(timeline: Timeline, range: RequestRange = .default, completion: @escaping Callback<[Status]>) {
        let request = timeline.request(range: range)
        run(request, completion: completion)
    }
    
}

extension Client {
    public enum Error: Swift.Error {
        case unknownError
        case invalidRequest
        case invalidResponse
        case invalidModel
        case mastodonError(String)
        
    }
}