001/*
002 * SonarQube
003 * Copyright (C) 2009-2017 SonarSource SA
004 * mailto:info AT sonarsource DOT com
005 *
006 * This program is free software; you can redistribute it and/or
007 * modify it under the terms of the GNU Lesser General Public
008 * License as published by the Free Software Foundation; either
009 * version 3 of the License, or (at your option) any later version.
010 *
011 * This program is distributed in the hope that it will be useful,
012 * but WITHOUT ANY WARRANTY; without even the implied warranty of
013 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
014 * Lesser General Public License for more details.
015 *
016 * You should have received a copy of the GNU Lesser General Public License
017 * along with this program; if not, write to the Free Software Foundation,
018 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
019 */
020package org.sonar.api.batch.fs.internal;
021
022import java.util.Locale;
023import org.sonar.api.batch.fs.FileSystem;
024import org.sonar.api.batch.fs.InputFile;
025
026/**
027 * @since 6.3
028 */
029public class FileExtensionPredicate extends AbstractFilePredicate {
030
031  private final String extension;
032
033  public FileExtensionPredicate(String extension) {
034    this.extension = lowercase(extension);
035  }
036
037  @Override
038  public boolean apply(InputFile inputFile) {
039    return extension.equals(getExtension(inputFile));
040  }
041
042  @Override
043  public Iterable<InputFile> get(FileSystem.Index index) {
044    return index.getFilesByExtension(extension);
045  }
046
047  public static String getExtension(InputFile inputFile) {
048    return getExtension(FilenamePredicate.getFilename(inputFile));
049  }
050
051  static String getExtension(String name) {
052    int index = name.lastIndexOf('.');
053    if (index < 0) {
054      return "";
055    }
056    return lowercase(name.substring(index + 1));
057  }
058
059  private static String lowercase(String extension) {
060    return extension.toLowerCase(Locale.ENGLISH);
061  }
062}