Java.libraries.retrofit2.play26.ApiClient.mustache Maven / Gradle / Ivy
package {{invokerPackage}};
import java.io.File;
import java.io.IOException;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.*;
import com.fasterxml.jackson.databind.ObjectMapper;
import retrofit2.Converter;
import retrofit2.Retrofit;
import retrofit2.converter.scalars.ScalarsConverterFactory;
import retrofit2.converter.jackson.JacksonConverterFactory;
{{#openApiNullable}}
import org.openapitools.jackson.nullable.JsonNullableModule;
{{/openApiNullable}}
import play.libs.Json;
import play.libs.ws.WSClient;
import {{invokerPackage}}.Play26CallAdapterFactory;
import {{invokerPackage}}.Play26CallFactory;
import okhttp3.Interceptor;
import okhttp3.ResponseBody;
import {{invokerPackage}}.auth.ApiKeyAuth;
import {{invokerPackage}}.auth.Authentication;
/**
* API client
*/
public class ApiClient {
/** Underlying HTTP-client */
private WSClient wsClient;
/** Creates HTTP call instances */
private Play26CallFactory callFactory;
/** Create {@link java.util.concurrent.CompletionStage} instances from HTTP calls */
private Play26CallAdapterFactory callAdapterFactory;
/** Supported auths */
private Map authentications;
/** API base path */
private String basePath = "{{{basePath}}}";
/** Default ObjectMapper */
private ObjectMapper defaultMapper;
public ApiClient(WSClient wsClient) {
this();
this.wsClient = wsClient;
}
public ApiClient() {
// Setup authentications (key: authentication name, value: authentication).
authentications = new HashMap<>();{{#authMethods}}{{#isBasicBasic}}
// authentications.put("{{name}}", new HttpBasicAuth());{{/isBasicBasic}}{{#isApiKey}}
authentications.put("{{name}}", new ApiKeyAuth({{#isKeyInHeader}}"header"{{/isKeyInHeader}}{{#isKeyInQuery}}"query"{{/isKeyInQuery}}{{#isKeyInCookie}}"cookie"{{/isKeyInCookie}}, "{{keyParamName}}"));{{/isApiKey}}{{#isOAuth}}
// authentications.put("{{name}}", new OAuth());{{/isOAuth}}{{/authMethods}}
// Prevent the authentications from being modified.
authentications = Collections.unmodifiableMap(authentications);
}
/**
* Creates a retrofit2 client for given API interface
*/
public S createService(Class serviceClass) {
if(!basePath.endsWith("/")) {
basePath = basePath + "/";
}
Map extraHeaders = new HashMap<>();
Map extraCookies = new HashMap<>();
List extraQueryParams = new ArrayList<>();
for (String authName : authentications.keySet()) {
Authentication auth = authentications.get(authName);
if (auth == null) throw new RuntimeException("Authentication undefined: " + authName);
auth.applyToParams(extraQueryParams, extraHeaders, extraCookies);
}
if (callFactory == null) {
callFactory = new Play26CallFactory(wsClient, extraHeaders, extraCookies, extraQueryParams);
}
if (callAdapterFactory == null) {
callAdapterFactory = new Play26CallAdapterFactory();
}
if (defaultMapper == null) {
defaultMapper = Json.mapper();
{{#openApiNullable}}
JsonNullableModule jnm = new JsonNullableModule();
defaultMapper.registerModule(jnm);
{{/openApiNullable}}
}
return new Retrofit.Builder()
.baseUrl(basePath)
.addConverterFactory(new FileConverter())
.addConverterFactory(ScalarsConverterFactory.create())
.addConverterFactory(JacksonConverterFactory.create(defaultMapper))
.callFactory(callFactory)
.addCallAdapterFactory(callAdapterFactory)
.build()
.create(serviceClass);
}
/**
* Helper method to set API base path
*/
public ApiClient setBasePath(String basePath) {
this.basePath = basePath;
return this;
}
/**
* Get authentications (key: authentication name, value: authentication).
*/
public Map getAuthentications() {
return authentications;
}
/**
* Get authentication for the given name.
*
* @param authName The authentication name
* @return The authentication, null if not found
*/
public Authentication getAuthentication(String authName) {
return authentications.get(authName);
}
/**
* Helper method to set API key value for the first API key authentication.
*/
public ApiClient setApiKey(String apiKey) {
for (Authentication auth : authentications.values()) {
if (auth instanceof ApiKeyAuth) {
((ApiKeyAuth) auth).setApiKey(apiKey);
return this;
}
}
throw new RuntimeException("No API key authentication configured!");
}
/**
* Helper method to set API key prefix for the first API key authentication.
*/
public ApiClient setApiKeyPrefix(String apiKeyPrefix) {
for (Authentication auth : authentications.values()) {
if (auth instanceof ApiKeyAuth) {
((ApiKeyAuth) auth).setApiKeyPrefix(apiKeyPrefix);
return this;
}
}
throw new RuntimeException("No API key authentication configured!");
}
/**
* Helper method to set HTTP call instances factory
*/
public ApiClient setCallFactory(Play26CallFactory callFactory) {
this.callFactory = callFactory;
return this;
}
/**
* Helper method to set {@link java.util.concurrent.CompletionStage} instances factory
*/
public ApiClient setCallAdapterFactory(Play26CallAdapterFactory callAdapterFactory) {
this.callAdapterFactory = callAdapterFactory;
return this;
}
/**
* Helper method to set Jackson's {@link ObjectMapper}
*/
public ApiClient setObjectMapper(ObjectMapper mapper) {
this.defaultMapper = mapper;
return this;
}
static class FileConverter extends Converter.Factory {
@Override
public Converter responseBodyConverter(Type type,
Annotation[] annotations, Retrofit retrofit) {
if (!File.class.getTypeName().equals(type.getTypeName())) {
return null;
}
return new Converter() {
@Override
public File convert(ResponseBody value) throws IOException {
Path path = Files.createTempFile("retrofit-file", ".tmp");
Files.write(path, value.bytes());
return path.toFile();
}
};
}
}
}