package com.walker.web.agent;
|
|
import com.walker.web.agent.impl.UserAgentFileParser;
|
|
import java.io.FileInputStream;
|
import java.io.FileNotFoundException;
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.io.InputStreamReader;
|
import java.util.Collection;
|
import java.util.Set;
|
import java.util.stream.Stream;
|
import java.util.zip.ZipEntry;
|
import java.util.zip.ZipInputStream;
|
|
import static java.nio.charset.StandardCharsets.UTF_8;
|
import static java.util.stream.Collectors.toSet;
|
|
public class UserAgentService {
|
|
// The version of the browscap file this bundle depends on
|
public static final int BUNDLED_BROWSCAP_VERSION = 6001002;
|
private String myZipFilePath;
|
private InputStream myZipFileStream;
|
|
public UserAgentService() {
|
// Default
|
}
|
|
/**
|
* Creates a user agent service based on the Browscap CSV file in the given ZIP file
|
* @param zipFilePath the zip file should contain the csv file. It will load the given zip file instead of the
|
* bundled one
|
*/
|
public UserAgentService(final String zipFilePath) {
|
myZipFilePath = zipFilePath;
|
}
|
|
/**
|
* Creates a user agent service based on the Browscap CSV file in the given ZIP InputStream
|
* @param zipFileStream the zip InputStream should contain the csv file. It will load the given zip InputStream
|
* instead of the bundled zip file
|
*/
|
public UserAgentService(final InputStream zipFileStream) {
|
myZipFileStream = zipFileStream;
|
}
|
|
/**
|
* Returns a parser based on the bundled BrowsCap version
|
* @return the user agent parser
|
*/
|
public UserAgentParser loadParser() throws IOException, ParseException {
|
|
// Use all default fields
|
final Set<BrowsCapField> defaultFields =
|
Stream.of(BrowsCapField.values()).filter(BrowsCapField::isDefault).collect(toSet());
|
|
return createParserWithFields(defaultFields);
|
}
|
|
/**
|
* Returns a parser based on the bundled BrowsCap version
|
* @param fields list
|
* @return the user agent parser
|
*/
|
public UserAgentParser loadParser(final Collection<BrowsCapField> fields) throws IOException, ParseException {
|
return createParserWithFields(fields);
|
}
|
|
private UserAgentParser createParserWithFields(final Collection<BrowsCapField> fields)
|
throws IOException, ParseException {
|
// http://browscap.org/version-number
|
try (final InputStream zipStream = getCsvFileStream();
|
final ZipInputStream zipIn = new ZipInputStream(zipStream)) {
|
// look for the first file that isn't a directory and a .csv
|
// that should be a BrowsCap .csv file
|
ZipEntry entry = null;
|
do {
|
entry = zipIn.getNextEntry();
|
} while (!(entry == null || entry.getName().endsWith(".csv")));
|
if (!(entry == null || entry.isDirectory())) {
|
return UserAgentFileParser.parse(new InputStreamReader(zipIn, UTF_8), fields);
|
} else {
|
throw new IOException(
|
"Unable to find the BrowsCap CSV file in the ZIP file");
|
}
|
}
|
}
|
|
/**
|
* Returns the bundled ZIP file name
|
* @return CSV file name
|
*/
|
public static String getBundledCsvFileName() {
|
return "browscap-" + BUNDLED_BROWSCAP_VERSION + ".zip";
|
}
|
|
/**
|
* Returns the InputStream to the CSV file. This is either the bundled ZIP file or the one passed in the
|
* constructor.
|
* @return
|
* @throws FileNotFoundException
|
*/
|
private InputStream getCsvFileStream() throws FileNotFoundException {
|
if (myZipFileStream == null) {
|
if (myZipFilePath == null) {
|
final String csvFileName = getBundledCsvFileName();
|
return getClass().getClassLoader().getResourceAsStream(csvFileName);
|
} else {
|
return new FileInputStream(myZipFilePath);
|
}
|
} else {
|
return myZipFileStream;
|
}
|
}
|
}
|
