BrBETA_dnet-hadoop/dhp-workflows/dhp-graph-mapper/src/main/java/eu/dnetlib/dhp/oa/graph/dump/APIClient.java

package eu.dnetlib.dhp.oa.graph.dump;

import java.io.*;
import java.io.IOException;
import java.nio.charset.StandardCharsets;

import org.apache.commons.io.IOUtils;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.FSDataInputStream;
import org.apache.hadoop.fs.FileSystem;
import org.apache.hadoop.fs.Path;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpPut;

import org.apache.http.entity.ContentType;
import org.apache.http.entity.InputStreamEntity;
import org.apache.http.entity.StringEntity;

import org.apache.http.entity.mime.MultipartEntityBuilder;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.util.EntityUtils;

import com.google.gson.Gson;

import eu.dnetlib.dhp.oa.graph.dump.zenodo.ZenodoModel;

public class APIClient implements Serializable {

	String urlString;
	String bucket;

	String deposition_id;
	String access_token;

	public String getUrlString() {
		return urlString;
	}

	public void setUrlString(String urlString) {
		this.urlString = urlString;
	}

	public String getBucket() {
		return bucket;
	}

	public void setBucket(String bucket) {
		this.bucket = bucket;
	}

	public APIClient(String urlString, String access_token) throws IOException {

		this.urlString = urlString;
		this.access_token = access_token;
	}

	public int connect() throws IOException {

		String json = "{}";

		HttpClient client = new DefaultHttpClient();

		HttpPost post = new HttpPost(urlString);

		StringEntity input = new StringEntity(json);
		post.setEntity(input);
		post.addHeader("Content-Type", "application/json");
		post.setHeader("Authorization", "Bearer " + access_token);

		HttpResponse response = client.execute(post);

		json = EntityUtils.toString(response.getEntity());

		ZenodoModel newSubmission = new Gson().fromJson(json, ZenodoModel.class);
		this.bucket = newSubmission.getLinks().getBucket();
		this.deposition_id = newSubmission.getId();

		return response.getStatusLine().getStatusCode();


	}


	public int upload(File file, String file_name) throws IOException {
		HttpClient client = new DefaultHttpClient();

		HttpPut put = new HttpPut(bucket + "/" + file_name);
		put.setHeader("Authorization", "Bearer " + access_token);
		put.addHeader("Content-Type", "application/zip");
		HttpEntity data = MultipartEntityBuilder.create().addBinaryBody(file_name, file).build();
		put.setEntity(data);

		HttpResponse response = client.execute(put);

		return response.getStatusLine().getStatusCode();

	}

	public int sendMretadata(String metadata) throws IOException {

		HttpClient client = new DefaultHttpClient();
		HttpPut post = new HttpPut(urlString + "/" + deposition_id);
		post.setHeader("Authorization", "Bearer " + access_token);
		post.addHeader("Content-Type", "application/json");
		StringEntity entity = new StringEntity(metadata, StandardCharsets.UTF_8);
		post.setEntity(entity);

		HttpResponse response = client.execute(post);
		return response.getStatusLine().getStatusCode();


	}

	public int publish() throws IOException {
		HttpClient client = new DefaultHttpClient();
		HttpPost post = new HttpPost(urlString +"/"+ deposition_id +"/actions/publish") ;
		post.setHeader("Authorization", "Bearer " + access_token);

		HttpResponse response = client.execute(post);

		return response.getStatusLine().getStatusCode();
	}

}