D-Net

package eu.dnetlib.openaire.exporter.model.project;

import java.io.IOException;

import java.io.StringReader;

import java.io.StringWriter;

import java.util.Arrays;

import java.util.List;

import javax.persistence.*;

import com.fasterxml.jackson.annotation.JsonIgnore;

import com.google.gson.Gson;

import io.swagger.annotations.ApiModel;

import org.hibernate.annotations.Type;

import org.supercsv.cellprocessor.Optional;

import org.supercsv.cellprocessor.ift.CellProcessor;

import org.supercsv.cellprocessor.ift.StringCellProcessor;

import org.supercsv.io.CsvBeanReader;

import org.supercsv.io.CsvBeanWriter;

import org.supercsv.io.ICsvBeanReader;

import org.supercsv.io.ICsvBeanWriter;

import org.supercsv.prefs.CsvPreference;

import org.supercsv.util.CsvContext;

/**

 * Created by claudio on 04/07/2017.

 */

@Entity

@Table(name = "project_details")

@ApiModel(value = "Project details model", description = "provides project details")

public class ProjectDetails {

	@Transient

	@JsonIgnore

	private transient static final String[] FIELDS = Arrays.stream(ProjectDetails.class.getDeclaredFields())

			.map(f -> f.getName())

			.filter(s -> !s.equals("FIELDS"))

			.filter(s -> !s.startsWith("optional"))

			.toArray(String[]::new);

	@Id

	@Column(name = "projectid")

	private String projectId;

	private String acronym;

	private String code;

	@Transient

	@JsonIgnore

	private transient String optional1;

	@Transient

	@JsonIgnore

	private transient String optional2;

	private String jsonextrainfo;

	@Type(type = "eu.dnetlib.openaire.exporter.model.GenericArrayUserType")

	@Column(name = "fundingpath", columnDefinition = "text[]")

	private String[] fundingPath;

	public ProjectDetails() {

	}

	public String getProjectId() {

		return projectId;

	}

	public void setProjectId(final String projectId) {

		this.projectId = projectId;

	}

	public String getAcronym() {

		return acronym;

	}

	public void setAcronym(final String acronym) {

		this.acronym = acronym;

	}

	public String getCode() {

		return code;

	}

	public void setCode(final String code) {

		this.code = code;

	}

	public String getOptional1() {

		return optional1;

	}

	public void setOptional1(final String optional1) {

		this.optional1 = optional1;

	}

	public String getOptional2() {

		return optional2;

	}

	public void setOptional2(final String optional2) {

		this.optional2 = optional2;

	}

	public String getJsonextrainfo() {

		return jsonextrainfo;

	}

	public void setJsonextrainfo(final String jsonextrainfo) {

		this.jsonextrainfo = jsonextrainfo;

	}

	public String[] getFundingPath() {

		return fundingPath;

	}

	public void setFundingPath(final String[] fundingPath) {

		this.fundingPath = fundingPath;

	}

	public void setFundingPath(final List<String> fundingPath) {

		if(fundingPath != null && !fundingPath.isEmpty()) {

			this.fundingPath = fundingPath.toArray(new String[fundingPath.size()]);

		}

	}

	public static ProjectDetails fromJson(final String json) {

		return new Gson().fromJson(json, ProjectDetails.class);

	}

	public static ProjectDetails fromCSV(final String csv) throws IOException {

		ICsvBeanReader beanReader = null;

		try {

			beanReader = new CsvBeanReader(new StringReader(csv), CsvPreference.STANDARD_PREFERENCE);

			return beanReader.read(ProjectDetails.class, FIELDS, getProcessors(new StringCellProcessor() {

				@Override

				public Object execute(final Object value, final CsvContext context) {

					return new Gson().fromJson(value.toString(), List.class);

				}

			}));

		} finally {

			if (beanReader != null) {

				beanReader.close();

			}

		}

	}

	/**

	 * Sets up the processors used for the examples. There are 10 CSV columns, so 10 processors are defined. Empty

	 * columns are read as null (hence the NotNull() for mandatory columns).

	 *

	 * @return the cell processors

	 */

	private static CellProcessor[] getProcessors(final CellProcessor fundingPathProcessor) {

		return new CellProcessor[] {

				new Optional(), // projectId

				new Optional(), // acronym

				new Optional(), // code

				new Optional(), // jsonextrainfo

				fundingPathProcessor

		};

	}

	public String asJson() {

		return new Gson().toJson(this) + '\n';

	}

	public String asCSV() throws IOException {

		final StringWriter sb = new StringWriter();

		try (ICsvBeanWriter beanWriter = new CsvBeanWriter(sb, CsvPreference.STANDARD_PREFERENCE)) {

			beanWriter.write(this, FIELDS, getProcessors(new StringCellProcessor() {

				@Override

				public Object execute(final Object value, final CsvContext context) {

					return new Gson().toJson(value);

				}

			}));

			beanWriter.flush();

		}

		return sb.toString();

	}

}