Project

General

Profile

1
package eu.dnetlib.enabling.datasources;
2

    
3
import java.io.IOException;
4
import java.sql.Array;
5
import java.sql.Date;
6
import java.sql.SQLException;
7
import java.sql.Timestamp;
8
import java.util.*;
9
import java.util.stream.Collectors;
10

    
11
import eu.dnetlib.enabling.datasources.common.*;
12
import eu.dnetlib.miscutils.datetime.DateUtils;
13
import org.apache.commons.lang.BooleanUtils;
14
import org.apache.commons.lang.StringUtils;
15
import org.apache.commons.logging.Log;
16
import org.apache.commons.logging.LogFactory;
17
import org.dom4j.Document;
18
import org.dom4j.DocumentException;
19
import org.dom4j.Element;
20
import org.dom4j.io.SAXReader;
21
import org.springframework.core.io.ClassPathResource;
22
import org.springframework.core.io.Resource;
23

    
24
public class DatasourceFunctions {
25

    
26
    private static final Log log = LogFactory.getLog(DatasourceFunctions.class);
27

    
28
    private static final Resource baseDsProfile = new ClassPathResource("/eu/dnetlib/enabling/datasources/templates/datasource_base.xml");
29

    
30
    public static SimpleDatasource mapToSimpleDs(final Map<String, Object> map) {
31
        final SimpleDatasource ds = new SimpleDatasource();
32
        ds.setId(castObject(map.get("id"), String.class));
33
        ds.setName(castObject(map.get("name"), String.class));
34
        ds.setOrigId(castObject(map.get("id"), String.class));
35
        ds.setTypology(castObject(map.get("typology"), String.class));
36
        ds.setValid(true);
37
        try {
38
            final Array arr = castObject(map.get("apis"), Array.class);
39

    
40
            for (Object o : (Object[]) (arr).getArray()) {
41
                log.info(o);
42
            }
43
            if (arr != null && ((Object[]) arr.getArray()).length > 0) {
44
                ds.setApis(Arrays.stream((Object[]) arr.getArray())
45
                        .filter(Objects::nonNull)
46
                        .map(Object::toString)
47
                        .collect(Collectors.toSet()));
48
            }
49
        } catch (final SQLException e) {
50
            log.error("Error parsing array (apis)", e);
51
            throw new RuntimeException("Error parsing array (apis)", e);
52
        }
53

    
54
        return ds;
55
    }
56

    
57
    public static Datasource<Organization<?>, Identity> mapToDatasource(final Map<String, Object> map) {
58
        final Datasource<Organization<?>, Identity> ds = new Datasource<>();
59
        ds.setId(castObject(map.get("id"), String.class));
60
        ds.setOfficialname(castObject(map.get("officialname"), String.class));
61
        ds.setEnglishname(castObject(map.get("englishname"), String.class));
62
        ds.setWebsiteurl(castObject(map.get("websiteurl"), String.class));
63
        ds.setLogourl(castObject(map.get("logourl"), String.class));
64
        ds.setContactemail(castObject(map.get("contactemail"), String.class));
65
        ds.setLatitude(castObject(map.get("latitude"), Double.class));
66
        ds.setLongitude(castObject(map.get("longitude"), Double.class));
67
        ds.setTimezone(castObject(map.get("timezone"), String.class));
68
        ds.setNamespaceprefix(castObject(map.get("namespaceprefix"), String.class));
69
        ds.setLanguages(castObject(map.get("languages"), String.class));
70
        ds.setOd_contenttypes(castObject(map.get("od_contenttypes"), String.class));
71
        ds.setCollectedfrom(castObject(map.get("collectedfrom"), String.class));
72
        ds.setDateofvalidation(castObject(map.get("dateofvalidation"), Date.class));
73
        ds.setTypology(castObject(map.get("typology"), String.class));
74
        ds.setProvenanceaction(castObject(map.get("provenanceaction"), String.class));
75
        ds.setDateofcollection(castObject(map.get("dateofcollection"), Date.class));
76
        ds.setPlatform(castObject(map.get("platform"), String.class));
77
        ds.setActivationId(castObject(map.get("activationId"), String.class));
78
        ds.setDescription(castObject(map.get("description"), String.class));
79
        ds.setReleasestartdate(castObject(map.get("releasestartdate"), Date.class));
80
        ds.setReleaseenddate(castObject(map.get("releaseenddate"), Date.class));
81
        ds.setMissionstatementurl(castObject(map.get("missionstatementurl"), String.class));
82
        ds.setDataprovider(castObject(map.get("dataprovider"), Boolean.class));
83
        ds.setServiceprovider(castObject(map.get("serviceprovider"), Boolean.class));
84
        ds.setDatabaseaccesstype(castObject(map.get("databaseaccesstype"), String.class));
85
        ds.setDatauploadtype(castObject(map.get("datauploadtype"), String.class));
86
        ds.setDatabaseaccessrestriction(castObject(map.get("databaseaccessrestriction"), String.class));
87
        ds.setDatauploadrestriction(castObject(map.get("datauploadrestriction"), String.class));
88
        ds.setVersioning(castObject(map.get("versioning"), Boolean.class));
89
        ds.setCitationguidelineurl(castObject(map.get("citationguidelineurl"), String.class));
90
        ds.setQualitymanagementkind(castObject(map.get("qualitymanagementkind"), String.class));
91
        ds.setPidsystems(castObject(map.get("pidsystems"), String.class));
92
        ds.setCertificates(castObject(map.get("certificates"), String.class));
93
        ds.setAggregator(castObject(map.get("aggregator"), String.class));
94
        ds.setIssn(castObject(map.get("issn"), String.class));
95
        ds.setEissn(castObject(map.get("eissn"), String.class));
96
        ds.setLissn(castObject(map.get("lissn"), String.class));
97
        ds.setRegisteredby(castObject(map.get("registeredby"), String.class));
98
        ds.setSubjects(castObject(map.get("subjects"), String.class));
99
        ds.setManaged(castObject(map.get("managed"), Boolean.class));
100
        return ds;
101
    }
102

    
103
    public static Identity mapToDsIdentity(final Map<String, Object> map) {
104
        final Identity id = new Identity();
105
        id.setPid(castObject(map.get("pid"), String.class));
106
        id.setIssuertype(castObject(map.get("issuertype"), String.class));
107
        return id;
108
    }
109

    
110
    public static Organization<Datasource<?, ?>> mapToDsOrganization(final Map<String, Object> map) {
111
        final Organization<Datasource<?, ?>> org = new Organization<>();
112
        org.setId(castObject(map.get("id"), String.class));
113
        org.setLegalshortname(castObject(map.get("legalshortname"), String.class));
114
        org.setLegalname(castObject(map.get("legalname"), String.class));
115
        org.setWebsiteurl(castObject(map.get("websiteurl"), String.class));
116
        org.setLogourl(castObject(map.get("logourl"), String.class));
117
        org.setCountry(castObject(map.get("country"), String.class));
118
        org.setCollectedfrom(castObject(map.get("collectedfrom"), String.class));
119
        org.setDateofcollection(castObject(map.get("dateofcollection"), Date.class));
120
        org.setProvenanceaction(castObject(map.get("provenanceaction"), String.class));
121
        return org;
122
    }
123

    
124
    public static SearchApisEntry mapToSearchApisEntry(final Map<String, Object> map) {
125
        final SearchApisEntry a = new SearchApisEntry();
126
        a.setId(castObject(map.get("id"), String.class));
127
        a.setCompliance(castObject(map.get("compliance"), String.class));
128
        a.setActive(castObject(map.get("active"), Boolean.class));
129
        a.setRepoId(castObject(map.get("dsId"), String.class));
130
        a.setRepoName(castObject(map.get("name"), String.class));
131
        a.setRepoCountry(castObject(map.get("country"), String.class));
132
        a.setRepoPrefix(castObject(map.get("prefix"), String.class));
133
        a.setAggrDate(castObject(map.get("aggrDate"), String.class));
134
        a.setAggrTotal(castObject(map.get("aggrTotal"), Integer.class));
135
        a.setProtocol(castObject(map.get("protocol"), String.class));
136
        a.setAlternativeName(castObject(map.get("alternativeName"), String.class));
137
        a.setRepoOrganization(castObject(map.get("organization"), String.class));
138
        return a;
139
    }
140

    
141
    public static Api<ApiParam> mapToApi(final Map<String, Object> map) {
142
        final Api<ApiParam> a = new Api<>();
143
        a.setId(castObject(map.get("id"), String.class));
144
        a.setProtocol(castObject(map.get("protocol"), String.class));
145
        a.setDatasource(castObject(map.get("datasource"), String.class));
146
        a.setContentdescription(castObject(map.get("contentdescription"), String.class));
147
        a.setActive(castObject(map.get("active"), Boolean.class));
148
        a.setRemovable(castObject(map.get("removable"), Boolean.class));
149
        a.setTypology(castObject(map.get("typology"), String.class));
150
        a.setCompatibility(castObject(map.get("compatibility"), String.class));
151
        a.setCompatibilityOverrided(castObject(map.get("isCompatibilityOverrided"), Boolean.class));
152
        a.setMetadataIdentifierPath(castObject(map.get("metadataIdentifierPath"), String.class));
153
        a.setLastCollectionTotal(castObject(map.get("lastCollectionTotal"), Integer.class));
154
        a.setLastCollectionDate(castObject(map.get("lastCollectionDate"), Timestamp.class));
155
        a.setLastCollectionMdid(castObject(map.get("lastCollectionMdid"), String.class));
156
        a.setLastAggregationTotal(castObject(map.get("lastAggregationTotal"), Integer.class));
157
        a.setLastAggregationDate(castObject(map.get("lastAggregationDate"), Timestamp.class));
158
        a.setLastAggregationMdid(castObject(map.get("lastAggregationMdid"), String.class));
159
        a.setLastDownloadTotal(castObject(map.get("lastDownloadTotal"), Integer.class));
160
        a.setLastDownloadDate(castObject(map.get("lastDownloadDate"), Timestamp.class));
161
        a.setLastDownloadObjid(castObject(map.get("lastDownloadObjid"), String.class));
162
        a.setLastValidationJob(castObject(map.get("lastValidationJob"), String.class));
163
        a.setBaseurl(castObject(map.get("baseUrl"), String.class));
164
        try {
165
            final Array arr = castObject(map.get("params"), Array.class);
166
            if (arr != null) {
167
                a.setApiParams(Arrays.stream((Object[]) arr.getArray())
168
                        .filter(Objects::nonNull)
169
                        .map(Object::toString)
170
                        .map(s -> {
171
                            final ApiParam p = new ApiParamImpl();
172
                            p.setParam(StringUtils.substringBefore(s, "="));
173
                            p.setValue(StringUtils.substringAfter(s, "="));
174
                            return p;
175
                        })
176
                        .collect(Collectors.toSet()));
177
            }
178
        } catch (final SQLException e) {
179
            log.error("Error parsing array params", e);
180
            throw new RuntimeException("Error parsing array params", e);
181
        }
182

    
183
        return a;
184
    }
185

    
186
    public static String dsToProfile(final Datasource<Organization<?>, Identity> ds, final List<Api<ApiParam>> apis, final String profId)
187
            throws DocumentException, IOException {
188
        final Document doc = new SAXReader().read(baseDsProfile.getInputStream());
189

    
190
        setValue(doc, "//DATASOURCE_TYPE", ds.getPlatform());
191
        setValue(doc, "//DATASOURCE_ORIGINAL_ID", ds.getId());
192
        setValue(doc, "//TYPOLOGY", ds.getTypology());
193
        setValue(doc, "//OFFICIAL_NAME", ds.getOfficialname());
194
        setValue(doc, "//ENGLISH_NAME", ds.getEnglishname());
195
        setValue(doc, "//ICON_URI", ds.getLogourl());
196
        setValue(doc, "//COUNTRY", ds.getOrganizations().stream().map(Organization::getCountry).findFirst().orElse(""));
197
        setValue(doc, "//LONGITUDE", ds.getLongitude());
198
        setValue(doc, "//LATITUDE", ds.getLatitude());
199
        setValue(doc, "//TIMEZONE", ds.getTimezone());
200
        setValue(doc, "//REPOSITORY_WEBPAGE", ds.getWebsiteurl());
201
        setValue(doc, "//REPOSITORY_INSTITUTION", ds.getOrganizations().stream().map(Organization::getLegalname).findFirst().orElse(""));
202
        setValue(doc, "//ADMIN_INFO", ds.getContactemail());
203
        setValue(doc, "//REGISTERED_BY", ds.getRegisteredby());
204
        setValue(doc, "//LAST_UPDATE/@value", DateUtils.now_ISO8601());
205

    
206
        final Element ef = (Element) doc.selectSingleNode("//EXTRA_FIELDS");
207
        addExtraField(ef, "OpenAireDataSourceId", ds.getId());
208
        addExtraField(ef, "NamespacePrefix", ds.getNamespaceprefix());
209
        addExtraField(ef, "VERIFIED", "NO");
210
        addExtraField(ef, "aggregatorName", ds.getAggregator());
211
        addExtraField(ef, "dateOfValidation", ds.getDateofvalidation());
212
        addExtraField(ef, "dateOfCollection", ds.getDateofcollection());
213
        addExtraField(ef, "mergeHomonyms", "NO");
214
        addExtraField(ef, "ACTID", ds.getActivationId());
215

    
216
        if (apis != null) {
217
            final Element ifaces = (Element) doc.selectSingleNode("//INTERFACES");
218
            apis.forEach(api -> addInterface(ifaces, api));
219
        }
220
        return doc.asXML();
221
    }
222

    
223
    private static void setValue(final Document doc, final String xpath, final Object value) {
224
        if (value != null) {
225
            doc.selectSingleNode(xpath).setText(value.toString());
226
        }
227
    }
228

    
229
    private static void addInterface(final Element ifaces, final Api<?> api) {
230

    
231
        final Element ifc = ifaces.addElement("INTERFACE");
232
        ifc.addAttribute("id", api.getId());
233
        ifc.addAttribute("label", String.format("%s (%s)",
234
                StringUtils.defaultIfBlank(api.getTypology(), "-"),
235
                StringUtils.defaultIfBlank(api.getCompatibility(), "-")));
236
        ifc.addAttribute("typology", StringUtils.defaultIfBlank(api.getTypology(), ""));
237
        ifc.addAttribute("active", "" + BooleanUtils.toBooleanDefaultIfNull(api.getActive(), false));
238
        ifc.addAttribute("compliance", StringUtils.defaultIfBlank(api.getCompatibility(), ""));
239
        ifc.addAttribute("contentDescription", StringUtils.defaultIfBlank(api.getContentdescription(), ""));
240
        ifc.addAttribute("removable", "" + BooleanUtils.toBooleanDefaultIfNull(api.getRemovable(), false));
241

    
242
        final Element ap = ifc.addElement("ACCESS_PROTOCOL");
243
        api.getApiParams().forEach(p -> ap.addAttribute(p.getParam(), p.getValue()));
244
        ap.setText(StringUtils.defaultIfBlank(api.getProtocol(), ""));
245
        ifc.addElement("BASE_URL").setText(StringUtils.defaultIfBlank(api.getBaseurl(), ""));
246

    
247
        if (api.isCompatibilityOverrided()) {
248
            addInterfaceExtraField(ifc, "overriding_compliance", true);
249
        }
250

    
251
        addInterfaceExtraField(ifc, "metadata_identifier_path", api.getMetadataIdentifierPath());
252
        addInterfaceExtraField(ifc, "last_collection_date", api.getLastCollectionDate());
253
        addInterfaceExtraField(ifc, "last_collection_mdId", api.getLastCollectionMdid());
254
        addInterfaceExtraField(ifc, "last_collection_total", api.getLastCollectionTotal());
255
        addInterfaceExtraField(ifc, "last_aggregation_date", api.getLastAggregationDate());
256
        addInterfaceExtraField(ifc, "last_aggregation_mdId", api.getLastAggregationMdid());
257
        addInterfaceExtraField(ifc, "last_aggregation_total", api.getLastAggregationTotal());
258
        addInterfaceExtraField(ifc, "last_download_date", api.getLastDownloadDate());
259
        addInterfaceExtraField(ifc, "last_download_objId", api.getLastDownloadObjid());
260
        addInterfaceExtraField(ifc, "last_download_total", api.getLastDownloadTotal());
261
    }
262

    
263
    private static void addExtraField(final Element parent, final String field, final Object value) {
264
        if (value != null && StringUtils.isNotBlank(value.toString())) {
265
            final Element f = parent.addElement("FIELD");
266
            f.addElement("key").setText(field);
267
            f.addElement("value").setText(value.toString());
268
        }
269
    }
270

    
271
    private static void addInterfaceExtraField(final Element parent, final String field, final Object value) {
272
        if (value != null && StringUtils.isNotBlank(value.toString())) {
273
            final Element f = parent.addElement("INTERFACE_EXTRA_FIELD");
274
            f.addAttribute("name", field);
275
            f.setText(value.toString());
276
        }
277
    }
278

    
279
    public static Map<String, Object> dsToMap(final Datasource<Organization<?>, Identity> ds) {
280
        final Map<String, Object> map = new HashMap<>();
281

    
282
        map.put("id", ds.getId());
283
        map.put("officialname", ds.getOfficialname());
284
        map.put("englishname", ds.getEnglishname());
285
        map.put("websiteurl", ds.getWebsiteurl());
286
        map.put("logourl", ds.getLogourl());
287
        map.put("contactemail", ds.getContactemail());
288
        map.put("latitude", ds.getLatitude());
289
        map.put("longitude", ds.getLongitude());
290
        map.put("timezone", ds.getTimezone());
291
        map.put("namespaceprefix", ds.getNamespaceprefix());
292
        map.put("languages", ds.getLanguages());
293
        map.put("od_contenttypes", ds.getOd_contenttypes());
294
        map.put("collectedfrom", ds.getCollectedfrom());
295
        map.put("dateofvalidation", ds.getDateofvalidation());
296
        map.put("typology", ds.getTypology());
297
        map.put("provenanceaction", ds.getProvenanceaction());
298
        map.put("platform", ds.getPlatform());
299
        map.put("activationid", ds.getActivationId());
300
        map.put("description", ds.getDescription());
301
        map.put("releasestartdate", ds.getReleasestartdate());
302
        map.put("releaseenddate", ds.getReleaseenddate());
303
        map.put("missionstatementurl", ds.getMissionstatementurl());
304
        map.put("dataprovider", ds.isDataprovider());
305
        map.put("serviceprovider", ds.isServiceprovider());
306
        map.put("databaseaccesstype", ds.getDatabaseaccesstype());
307
        map.put("datauploadtype", ds.getDatauploadtype());
308
        map.put("databaseaccessrestriction", ds.getDatabaseaccessrestriction());
309
        map.put("datauploadrestriction", ds.getDatauploadrestriction());
310
        map.put("versioning", ds.isVersioning());
311
        map.put("citationguidelineurl", ds.getCitationguidelineurl());
312
        map.put("qualitymanagementkind", ds.getQualitymanagementkind());
313
        map.put("pidsystems", ds.getPidsystems());
314
        map.put("certificates", ds.getCertificates());
315
        map.put("aggregator", ds.getAggregator());
316
        map.put("issn", ds.getIssn());
317
        map.put("eissn", ds.getEissn());
318
        map.put("lissn", ds.getLissn());
319
        map.put("registeredby", ds.getRegisteredby());
320
        map.put("subjects", ds.getSubjects());
321
        map.put("managed", ds.isManaged());
322

    
323
        return map;
324
    }
325

    
326
    public static Map<String, Object> orgToMap(final String dsId, final Organization<?> org) {
327
        final Map<String, Object> map = new HashMap<>();
328
        map.put("dsId", dsId);
329
        map.put("orgId", org.getId());
330
        map.put("legalname", org.getLegalname());
331
        map.put("legalshortname", org.getLegalshortname());
332
        map.put("websiteurl", org.getWebsiteurl());
333
        map.put("logourl", org.getLogourl());
334
        map.put("country", org.getCountry());
335
        return map;
336
    }
337

    
338
    public static Map<String, Object> apiToMap(final Api<?> api) {
339
        final Map<String, Object> map = new HashMap<>();
340
        map.put("apiId", api.getId());
341
        map.put("protocol", api.getProtocol());
342
        map.put("baseUrl", api.getBaseurl());
343
        map.put("dsId", api.getDatasource());
344
        map.put("contentDescription", api.getContentdescription());
345
        map.put("typology", api.getTypology());
346
        map.put("compatibility", api.getCompatibility());
347
        map.put("metadataIdentifierPath", api.getMetadataIdentifierPath());
348
        // The other fields are not required in the INSERT operation
349
        return map;
350
    }
351

    
352
    @SuppressWarnings("unchecked")
353
    private static <T> T castObject(final Object o, final Class<T> clazz) {
354
        if (o == null) {
355
            return null;
356
        }
357
        if (clazz.isInstance(o)) {
358
            return (T) o;
359
        }
360
        throw new IllegalArgumentException("Type not managed: " + clazz.getSimpleName());
361
    }
362

    
363
}
(1-1/5)