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
    @Autowired
43
    private RepositoryService repositoryService;
44

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

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

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

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

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

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

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

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

    
66

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

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

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

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

    
114
    }
115

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

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

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

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

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

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

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

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

    
195
            savePiwikInfo(piwikInfo);
196

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

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

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

    
219

    
220
}
(10-10/20)