MediaTypeFilter.java

/*******************************************************************************
 * Copyright (c) 2012-2017 Codenvy, S.A.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *   Codenvy, S.A. - initial API and implementation
 *******************************************************************************/
package org.eclipse.che.api.vfs.search;

import org.apache.tika.config.TikaConfig;
import org.apache.tika.exception.TikaException;
import org.apache.tika.metadata.Metadata;
import org.apache.tika.mime.MediaType;
import org.eclipse.che.api.core.ForbiddenException;
import org.eclipse.che.api.core.ServerException;
import org.eclipse.che.api.vfs.VirtualFile;
import org.eclipse.che.api.vfs.VirtualFileFilter;

import java.io.IOException;
import java.io.InputStream;
import java.util.Set;

import static com.google.common.collect.Sets.newHashSet;

/**
 * Filter based on media type of the file.
 * The filter includes in result files with media type different from the specified types in the set {@link MediaTypeFilter#excludedMediaTypes}
 * Note: if media type can not be detected a file will be not include in result as well.
 *
 * @author Valeriy Svydenko
 * @author Roman Nikitenko
 */
public class MediaTypeFilter implements VirtualFileFilter {
    private final Set<MediaType> excludedMediaTypes;
    private final Set<String> excludedTypes;

    public MediaTypeFilter() {
        this.excludedMediaTypes = newHashSet(MediaType.APPLICATION_ZIP, MediaType.OCTET_STREAM);
        this.excludedTypes = newHashSet("video", "audio", "image");
    }

    @Override
    public boolean accept(VirtualFile file) {
        try (InputStream content = file.getContent()) {
            TikaConfig tikaConfig = new TikaConfig();
            MediaType mimeType = tikaConfig.getDetector().detect(content, new Metadata());
            if (excludedMediaTypes.contains(mimeType) || excludedTypes.contains(mimeType.getType())) {
                return true;
            }
            return false;
        } catch (TikaException | ForbiddenException | ServerException | IOException e) {
            return true;
        }
    }
}