51 lines
2.0 KiB
Kotlin
51 lines
2.0 KiB
Kotlin
package org.mariotaku.yandex
|
|
|
|
import org.mariotaku.restfu.RestAPIFactory
|
|
import org.mariotaku.restfu.RestConverter
|
|
import org.mariotaku.restfu.RestMethod
|
|
import org.mariotaku.restfu.RestRequest
|
|
import org.mariotaku.restfu.http.Endpoint
|
|
import org.mariotaku.restfu.http.RestHttpClient
|
|
import org.mariotaku.restfu.http.ValueMap
|
|
import org.mariotaku.restfu.logansqaure.LoganSquareConverterFactory
|
|
|
|
|
|
class YandexAPIFactory(apiKey: String, endpoint: String) {
|
|
private val factory: RestAPIFactory<YandexException> = RestAPIFactory<YandexException>()
|
|
|
|
init {
|
|
factory.setEndpoint(Endpoint(endpoint))
|
|
factory.setExceptionFactory { cause, _, _ ->
|
|
cause?.let { YandexException(it) }
|
|
?: YandexException()
|
|
}
|
|
factory.setRestConverterFactory(LoganSquareConverterFactory())
|
|
factory.setRestRequestFactory(object : RestRequest.DefaultFactory<YandexException?>() {
|
|
override fun create(restMethod: RestMethod<YandexException?>,
|
|
factory: RestConverter.Factory<YandexException?>, valuePool: ValueMap?): RestRequest {
|
|
val method = restMethod.method
|
|
val path = restMethod.path
|
|
val headers = restMethod.getHeaders(valuePool)
|
|
val queries = restMethod.getQueries(valuePool)
|
|
val params = restMethod.getParams(factory, valuePool)
|
|
val rawValue = restMethod.rawValue
|
|
val bodyType = restMethod.bodyType
|
|
val extras = restMethod.extras
|
|
queries.add("key", apiKey)
|
|
return RestRequest(method.value, method.allowBody, path, headers, queries,
|
|
params, rawValue, bodyType, extras)
|
|
}
|
|
})
|
|
}
|
|
|
|
fun setHttpClient(restClient: RestHttpClient): YandexAPIFactory {
|
|
factory.setHttpClient(restClient)
|
|
return this
|
|
}
|
|
|
|
fun build(): YandexAPI {
|
|
return factory.build(YandexAPI::class.java)
|
|
}
|
|
}
|
|
|