/*
* LinShare is an open source filesharing software, part of the LinPKI software
* suite, developed by Linagora.
*
* Copyright (C) 2016 LINAGORA
*
* This program is free software: you can redistribute it and/or modify it under
* the terms of the GNU Affero General Public License as published by the Free
* Software Foundation, either version 3 of the License, or (at your option) any
* later version, provided you comply with the Additional Terms applicable for
* LinShare software by Linagora pursuant to Section 7 of the GNU Affero General
* Public License, subsections (b), (c), and (e), pursuant to which you must
* notably (i) retain the display of the “LinShare™” trademark/logo at the top
* of the interface window, the display of the “You are using the Open Source
* and free version of LinShare™, powered by Linagora © 2009–2015. Contribute to
* Linshare R&D by subscribing to an Enterprise offer!” infobox and in the
* e-mails sent with the Program, (ii) retain all hypertext links between
* LinShare and linshare.org, between linagora.com and Linagora, and (iii)
* refrain from infringing Linagora intellectual property rights over its
* trademarks and commercial brands. Other Additional Terms apply, see
* <http://www.linagora.com/licenses/> for more details.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
* details.
*
* You should have received a copy of the GNU Affero General Public License and
* its applicable Additional Terms for LinShare along with this program. If not,
* see <http://www.gnu.org/licenses/> for the GNU Affero General Public License
* version 3 and <http://www.linagora.com/licenses/> for the Additional Terms
* applicable to LinShare software.
*/
package org.linagora.linshare.core.batches.impl;
import java.io.InputStream;
import java.util.List;
import org.linagora.linshare.core.dao.FileDataStore;
import org.linagora.linshare.core.dao.MimeTypeMagicNumberDao;
import org.linagora.linshare.core.domain.constants.FileMetaDataKind;
import org.linagora.linshare.core.domain.entities.Account;
import org.linagora.linshare.core.domain.entities.Document;
import org.linagora.linshare.core.domain.objects.FileMetaData;
import org.linagora.linshare.core.exception.BatchBusinessException;
import org.linagora.linshare.core.exception.BusinessException;
import org.linagora.linshare.core.job.quartz.BatchResultContext;
import org.linagora.linshare.core.job.quartz.Context;
import org.linagora.linshare.core.repository.AccountRepository;
import org.linagora.linshare.core.repository.DocumentRepository;
public class ComputeDocumentMimeTypeBatchImpl extends GenericBatchImpl {
private final DocumentRepository documentRepository;
private final FileDataStore fileDataStore;
private final MimeTypeMagicNumberDao mimeTypeMagicNumberDao;
public ComputeDocumentMimeTypeBatchImpl(
AccountRepository<Account> accountRepository,
DocumentRepository documentRepository,
MimeTypeMagicNumberDao mimeTypeMagicNumberDao,
FileDataStore fileDataStore) {
super(accountRepository);
this.documentRepository = documentRepository;
this.fileDataStore = fileDataStore;
this.mimeTypeMagicNumberDao = mimeTypeMagicNumberDao;
}
@Override
public List<String> getAll() {
logger.info(getClass().toString() + " job starting ...");
List<String> entries = documentRepository.findAllDocumentWithMimeTypeCheckEnabled();
logger.info(entries.size()
+ " document(s) have been found, mime type will be check again for each file.");
return entries;
}
@Override
public Context execute(String identifier, long total, long position)
throws BatchBusinessException, BusinessException {
Document resource = documentRepository.findByUuid(identifier);
logInfo(total, position,
"processing document : " + resource.getRepresentation());
Context context = new BatchResultContext<Document>(resource);
context.setProcessed(false);
FileMetaData metadata = new FileMetaData(FileMetaDataKind.DATA, resource);
if (fileDataStore.exists(metadata)) {
try (InputStream stream = fileDataStore.get(metadata)) {
if (stream != null) {
String type = mimeTypeMagicNumberDao.getMimeType(stream);
resource.setType(type);
resource.setCheckMimeType(false);
documentRepository.update(resource);
context.setProcessed(true);
}
} catch (Exception e) {
logger.error(e.getMessage(), e);
}
}
return context;
}
@Override
public void notify(Context context, long total, long position) {
@SuppressWarnings("unchecked")
BatchResultContext<Document> cc = (BatchResultContext<Document>) context;
if (cc.getProcessed()) {
Document entry = cc.getResource();
logInfo(total, position, "Document mime type was updated {}.", entry.getRepresentation());
}
}
@Override
public void notifyError(BatchBusinessException exception,
String identifier, long total, long position) {
@SuppressWarnings("unchecked")
BatchResultContext<Document> context = (BatchResultContext<Document>) exception
.getContext();
Document entry = context.getResource();
logError(total, position, "Document mime type was not updated, batch has failed : {}",
entry.getRepresentation(), exception);
}
@Override
public void terminate(List<String> all, long errors, long unhandled_errors,
long total, long processed) {
long success = total - errors - unhandled_errors;
logger.info(success + " document(s) have been checked.");
if (errors > 0) {
logger.error(errors + " document(s) failed to be checked.");
}
if (unhandled_errors > 0) {
logger.error(unhandled_errors
+ " document(s) failed to be checked (unhandled error).");
}
logger.info(getClass().toString() + " job terminated.");
}
}