Project

General

Profile

1
package eu.dnetlib.repo.manager.service;
2

    
3
import com.fasterxml.jackson.databind.ObjectMapper;
4
import eu.dnetlib.domain.data.PiwikInfo;
5
import eu.dnetlib.domain.data.Repository;
6
import eu.dnetlib.repo.manager.domain.OrderByField;
7
import eu.dnetlib.repo.manager.domain.OrderByType;
8
import eu.dnetlib.repo.manager.domain.RepositoryServiceException;
9
import org.apache.commons.codec.digest.DigestUtils;
10
import org.apache.log4j.Logger;
11
import org.springframework.beans.factory.annotation.Autowired;
12
import org.springframework.beans.factory.annotation.Qualifier;
13
import org.springframework.beans.factory.annotation.Value;
14
import org.springframework.dao.EmptyResultDataAccessException;
15
import org.springframework.http.HttpStatus;
16
import org.springframework.http.ResponseEntity;
17
import org.springframework.jdbc.core.JdbcTemplate;
18
import org.springframework.jdbc.core.RowMapper;
19
import org.springframework.security.access.prepost.PreAuthorize;
20
import org.springframework.security.core.context.SecurityContextHolder;
21
import org.springframework.stereotype.Service;
22

    
23
import javax.sql.DataSource;
24
import java.io.IOException;
25
import java.io.UnsupportedEncodingException;
26
import java.net.URL;
27
import java.net.URLEncoder;
28
import java.sql.Types;
29
import java.util.List;
30
import java.util.Map;
31

    
32
@Service("piwikService")
33
public class PiWikServiceImpl implements PiWikService {
34

    
35
    @Autowired
36
    private DataSource dataSource;
37

    
38

    
39
    @Value("${services.repomanager.analyticsURL}")
40
    private String analyticsURL;
41

    
42

    
43
    @Autowired
44
    private RepositoryService repositoryService;
45

    
46
    @Autowired
47
    @Qualifier("emailUtils")
48
    private EmailUtils emailUtils;
49

    
50
    private static final Logger LOGGER = Logger
51
            .getLogger(PiWikServiceImpl.class);
52

    
53
    private final static String GET_PIWIK_SITE = "select repositoryid, siteid, authenticationtoken, creationdate, requestorname, requestoremail, validated, validationdate, comment, repositoryname, country from piwik_site where repositoryid = ?;";
54

    
55
    private final static String INSERT_PIWIK_INFO = "insert into piwik_site (repositoryid, siteid, creationdate, requestorname, requestoremail, validated, repositoryname, country, authenticationtoken) values (?, ?, now(), ?, ?, ?, ?, ?, ?)";
56

    
57
    private final static String GET_PIWIK_SITES = "select  repositoryid, siteid, authenticationtoken, creationdate, requestorname, requestoremail, validated, validationdate, comment, repositoryname, country from piwik_site ";
58

    
59
    private final static String GET_PIWIK_SITES_TOTAL = "select count(*) as totals from piwik_site ";
60

    
61
    private final static String APPROVE_PIWIK_SITE = "update piwik_site set validated=true, validationdate=now() where repositoryid = ?;";
62

    
63
    private RowMapper<PiwikInfo> piwikRowMapper = (rs, i) -> new PiwikInfo(rs.getString("repositoryid"), getOpenaireId(rs.getString("repositoryid")), rs.getString("repositoryname"), rs.getString("country"),
64
            rs.getString("siteid"), rs.getString("authenticationtoken"), rs.getTimestamp("creationdate"), rs.getString("requestorname"), rs.getString("requestoremail"),
65
            rs.getBoolean("validated"), rs.getTimestamp("validationdate"), rs.getString("comment"));
66

    
67

    
68
    @Override
69
    public PiwikInfo getPiwikSiteForRepo(String repositoryId) {
70
        try{
71
            return new JdbcTemplate(dataSource).queryForObject(GET_PIWIK_SITE, new String[]{repositoryId}, new int[]{Types.VARCHAR}, piwikRowMapper);
72
        }catch (EmptyResultDataAccessException e){
73
            return null;
74
        }
75
    }
76

    
77
    @Override
78
    @PreAuthorize("hasRole('ROLE_ADMIN') or hasRole('ROLE_PROVIDE_ADMIN') or (hasRole('ROLE_USER') and #piwikInfo.requestorEmail == authentication.userInfo.email)")
79
    public PiwikInfo savePiwikInfo(PiwikInfo piwikInfo) {
80
        JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);
81
        jdbcTemplate.update(INSERT_PIWIK_INFO, new Object[]{piwikInfo.getRepositoryId(), piwikInfo.getSiteId(), piwikInfo.getRequestorName(),
82
                        piwikInfo.getRequestorEmail(), piwikInfo.isValidated(), piwikInfo.getRepositoryName(), piwikInfo.getCountry(), piwikInfo.getAuthenticationToken()},
83
                new int[]{Types.VARCHAR, Types.VARCHAR, Types.VARCHAR, Types.VARCHAR, Types.BOOLEAN, Types.VARCHAR, Types.VARCHAR, Types.VARCHAR});
84
        return piwikInfo;
85
    }
86

    
87
    @Override
88
    public List<PiwikInfo> getPiwikSitesForRepos(OrderByField orderByField, OrderByType orderByType, int from, int quantity, String searchField) {
89
        try{
90
            String finalizedQuery = GET_PIWIK_SITES + " where ("+
91
                    " repositoryid ilike ? " +
92
                    " or siteid ilike ?" +
93
                    " or requestorname ilike ?" +
94
                    " or requestoremail ilike ?" +
95
                    " or comment ilike ?" +
96
                    " or repositoryname ilike ?"+
97
                    " or country ilike ?"
98
                    +") order by "+orderByField + " " + orderByType + " offset ? limit ?";
99

    
100
            return new JdbcTemplate(dataSource).query(finalizedQuery, preparedStatement -> {
101
                preparedStatement.setString(1,"%"+searchField+"%");
102
                preparedStatement.setString(2,"%"+searchField+"%");
103
                preparedStatement.setString(3,"%"+searchField+"%");
104
                preparedStatement.setString(4,"%"+searchField+"%");
105
                preparedStatement.setString(5,"%"+searchField+"%");
106
                preparedStatement.setString(6,"%"+searchField+"%");
107
                preparedStatement.setString(7,"%"+searchField+"%");
108
                preparedStatement.setInt(8,from);
109
                preparedStatement.setInt(9,quantity);
110
            }, piwikRowMapper);
111
        }catch (EmptyResultDataAccessException e){
112
            return null;
113
        }
114

    
115
    }
116

    
117
    @Override
118
    public int getPiwikSitesTotals(String searchField){
119
        try{
120
            String finalizedQuery = GET_PIWIK_SITES_TOTAL + " where ("+
121
                    " repositoryid ilike ? " +
122
                    " or siteid ilike ?" +
123
                    " or requestorname ilike ?" +
124
                    " or requestoremail ilike ?" +
125
                    " or comment ilike ?" +
126
                    " or repositoryname ilike ?"+
127
                    " or country ilike ?)";
128

    
129
            return new JdbcTemplate(dataSource).query(finalizedQuery, preparedStatement -> {
130
                preparedStatement.setString(1,"%"+searchField+"%");
131
                preparedStatement.setString(2,"%"+searchField+"%");
132
                preparedStatement.setString(3,"%"+searchField+"%");
133
                preparedStatement.setString(4,"%"+searchField+"%");
134
                preparedStatement.setString(5,"%"+searchField+"%");
135
                preparedStatement.setString(6,"%"+searchField+"%");
136
                preparedStatement.setString(7,"%"+searchField+"%");
137
            },rowMapper -> {
138
                rowMapper.next();
139
                return rowMapper.getInt("totals");
140
            });
141
        }catch (EmptyResultDataAccessException e){
142
            return 0;
143
        }
144
    }
145

    
146
    @Override
147
    @PreAuthorize("hasRole('ROLE_ADMIN') or hasRole('ROLE_PROVIDE_ADMIN')")
148
    public ResponseEntity<Object> approvePiwikSite(String repositoryId) {
149
        new JdbcTemplate(dataSource).update(APPROVE_PIWIK_SITE, new Object[] {repositoryId}, new int[] {Types.VARCHAR});
150
        return new ResponseEntity<>("OK",HttpStatus.OK);
151
    }
152

    
153
    @Override
154
    public String getOpenaireId(String repositoryId) {
155
        if (repositoryId != null && repositoryId.contains("::"))
156
            return repositoryId.split("::")[0] + "::" + DigestUtils.md5Hex(repositoryId.split("::")[1]);
157
        return null;
158
    }
159

    
160
    @Override
161
    @PreAuthorize("hasRole('ROLE_ADMIN') or hasRole('ROLE_PROVIDE_ADMIN')")
162
    public ResponseEntity<Object> markPiwikSiteAsValidated(String repositoryId) throws RepositoryServiceException {
163
        try {
164
            approvePiwikSite(repositoryId);
165

    
166
            PiwikInfo piwikInfo = getPiwikSiteForRepo(repositoryId);
167
            emailUtils.sendAdministratorMetricsEnabled(piwikInfo);
168
            emailUtils.sendUserMetricsEnabled(piwikInfo);
169

    
170
        } catch (EmptyResultDataAccessException e) {
171
            LOGGER.error("Error while approving piwik site: ", e);
172
            emailUtils.reportException(e);
173
            throw new RepositoryServiceException("General error", RepositoryServiceException.ErrorCode.GENERAL_ERROR);
174
        } catch (Exception e) {
175
            LOGGER.error("Error while sending email to administrator or user about the enabling of metrics", e);
176
            emailUtils.reportException(e);
177
        }
178
        return new ResponseEntity<>("OK",HttpStatus.OK);
179
    }
180

    
181
    @Override
182
    @PreAuthorize("hasRole('ROLE_ADMIN') or hasRole('ROLE_PROVIDE_ADMIN') or (hasRole('ROLE_USER') and #piwikInfo.requestorEmail == authentication.userInfo.email)")
183
    public PiwikInfo enableMetricsForRepository(String officialName,
184
                                                String repoWebsite,
185
                                                PiwikInfo piwikInfo) throws RepositoryServiceException {
186
        try {
187
            String URL = analyticsURL + "siteName=" + URLEncoder.encode(officialName, "UTF-8") + "&url="
188
                    + URLEncoder.encode(repoWebsite, "UTF-8");
189
            Map map = new ObjectMapper().readValue(new URL(URL), Map.class);
190
            String siteId = null;
191
            if(map.get("value")!=null) {
192
                siteId = map.get("value").toString();
193
            }
194
            piwikInfo.setSiteId(siteId);
195

    
196
            savePiwikInfo(piwikInfo);
197

    
198
            Repository repository = repositoryService.getRepositoryById(piwikInfo.getRepositoryId());
199
            repository.setPiwikInfo(piwikInfo);
200

    
201
            repositoryService.updateRepository(repository, SecurityContextHolder.getContext().getAuthentication());
202

    
203
            emailUtils.sendAdministratorRequestToEnableMetrics(piwikInfo);
204
            emailUtils.sendUserRequestToEnableMetrics(piwikInfo);
205
        } catch (UnsupportedEncodingException uee) {
206
            LOGGER.error("Error while creating piwikScript URL", uee);
207
            emailUtils.reportException(uee);
208
            throw new RepositoryServiceException("login.generalError", RepositoryServiceException.ErrorCode.GENERAL_ERROR);
209
        } catch (IOException ioe) {
210
            LOGGER.error("Error while creating piwik site", ioe);
211
            emailUtils.reportException(ioe);
212
            throw new RepositoryServiceException("login.generalError", RepositoryServiceException.ErrorCode.GENERAL_ERROR);
213
        } catch (Exception e) {
214
            LOGGER.error("Error while sending email to administrator or user about the request to enable metrics", e);
215
            emailUtils.reportException(e);
216
        }
217
        return piwikInfo;
218
    }
219

    
220

    
221
}
(10-10/20)