Search in sources :

Example 1 with VfsComponentContext

use of org.apache.commons.vfs2.provider.VfsComponentContext in project hop by apache.

the class S3NFileNameParser method parseUri.

@Override
public FileName parseUri(VfsComponentContext context, FileName base, String uri) throws FileSystemException {
    StringBuilder name = new StringBuilder();
    String scheme = UriParser.extractScheme(uri, name);
    UriParser.canonicalizePath(name, 0, name.length(), this);
    // Normalize separators in the path
    UriParser.fixSeparators(name);
    // Normalise the path
    FileType fileType = UriParser.normalisePath(name);
    // Extract bucket name
    final String bucketName = UriParser.extractFirstElement(name);
    return new S3NFileName(scheme, bucketName, name.toString(), fileType);
}
Also used : FileType(org.apache.commons.vfs2.FileType)

Example 2 with VfsComponentContext

use of org.apache.commons.vfs2.provider.VfsComponentContext in project hop by apache.

the class S3FileNameParserTest method testParseUri.

@Test
public void testParseUri() throws Exception {
    VfsComponentContext context = mock(VfsComponentContext.class);
    FileName fileName = mock(FileName.class);
    String uri = "s3://bucket/file";
    FileName noBaseFile = parser.parseUri(context, null, uri);
    assertNotNull(noBaseFile);
    assertEquals("bucket", ((S3FileName) noBaseFile).getBucketId());
    FileName withBaseFile = parser.parseUri(context, fileName, uri);
    assertNotNull(withBaseFile);
    assertEquals("bucket", ((S3FileName) withBaseFile).getBucketId());
    // assumption is that the whole URL is valid until it comes time to resolve to S3 objects
    uri = "s3://s3/bucket/file";
    withBaseFile = parser.parseUri(context, fileName, uri);
    assertEquals("s3", ((S3FileName) withBaseFile).getBucketId());
}
Also used : VfsComponentContext(org.apache.commons.vfs2.provider.VfsComponentContext) S3FileName(org.apache.hop.vfs.s3.s3.vfs.S3FileName) FileName(org.apache.commons.vfs2.FileName) Test(org.junit.Test)

Example 3 with VfsComponentContext

use of org.apache.commons.vfs2.provider.VfsComponentContext in project hop by apache.

the class DropboxFileNameParser method parseUri.

@Override
public FileName parseUri(VfsComponentContext context, FileName base, String uri) throws FileSystemException {
    StringBuilder name = new StringBuilder();
    String scheme = UriParser.extractScheme(context.getFileSystemManager().getSchemes(), uri, name);
    UriParser.canonicalizePath(name, 0, name.length(), this);
    // Normalize separators in the path
    UriParser.fixSeparators(name);
    // Normalise the path
    FileType fileType = UriParser.normalisePath(name);
    return new DropboxFileName(scheme, name.toString(), fileType);
}
Also used : FileType(org.apache.commons.vfs2.FileType)

Example 4 with VfsComponentContext

use of org.apache.commons.vfs2.provider.VfsComponentContext in project hop by apache.

the class S3FileObjectTest method setUp.

@Before
public void setUp() throws Exception {
    s3ServiceMock = mock(AmazonS3.class);
    S3Object s3Object = new S3Object();
    s3Object.setKey(OBJECT_NAME);
    s3Object.setBucketName(BUCKET_NAME);
    filename = new S3FileName(SCHEME, BUCKET_NAME, BUCKET_NAME, FileType.FOLDER);
    S3FileName rootFileName = new S3FileName(SCHEME, "", "", FileType.FOLDER);
    S3HopProperty s3HopProperty = mock(S3HopProperty.class);
    when(s3HopProperty.getPartSize()).thenReturn("5MB");
    S3FileSystem fileSystem = new S3FileSystem(rootFileName, new FileSystemOptions(), new StorageUnitConverter(), s3HopProperty);
    fileSystemSpy = spy(fileSystem);
    VfsComponentContext context = mock(VfsComponentContext.class);
    final DefaultFileSystemManager fsm = new DefaultFileSystemManager();
    FilesCache cache = mock(FilesCache.class);
    fsm.setFilesCache(cache);
    fsm.setCacheStrategy(CacheStrategy.ON_RESOLVE);
    when(context.getFileSystemManager()).thenReturn(fsm);
    fileSystemSpy.setContext(context);
    S3FileObject s3FileObject = new S3FileObject(filename, fileSystemSpy);
    s3FileObjectBucketSpy = spy(s3FileObject);
    s3FileObjectFileSpy = spy(new S3FileObject(new S3FileName(SCHEME, BUCKET_NAME, BUCKET_NAME + "/" + origKey, FileType.IMAGINARY), fileSystemSpy));
    S3FileObject s3FileObjectRoot = new S3FileObject(rootFileName, fileSystemSpy);
    s3FileObjectSpyRoot = spy(s3FileObjectRoot);
    // specify the behaviour of S3 Service
    // when( s3ServiceMock.getBucket( BUCKET_NAME ) ).thenReturn( testBucket );
    when(s3ServiceMock.getObject(BUCKET_NAME, OBJECT_NAME)).thenReturn(s3Object);
    when(s3ServiceMock.getObject(BUCKET_NAME, OBJECT_NAME)).thenReturn(s3Object);
    when(s3ServiceMock.listBuckets()).thenReturn(createBuckets());
    when(s3ServiceMock.doesBucketExistV2(BUCKET_NAME)).thenReturn(true);
    childObjectListing = mock(ObjectListing.class);
    when(childObjectListing.getObjectSummaries()).thenReturn(createObjectSummaries(0)).thenReturn(new ArrayList<>());
    when(childObjectListing.getCommonPrefixes()).thenReturn(new ArrayList<>()).thenReturn(createCommonPrefixes(3));
    when(childObjectListing.isTruncated()).thenReturn(true).thenReturn(false);
    when(s3ServiceMock.listObjects(any(ListObjectsRequest.class))).thenReturn(childObjectListing);
    when(s3ServiceMock.listObjects(anyString(), anyString())).thenReturn(childObjectListing);
    when(s3ServiceMock.listNextBatchOfObjects(any(ObjectListing.class))).thenReturn(childObjectListing);
    s3ObjectMock = mock(S3Object.class);
    s3ObjectInputStream = mock(S3ObjectInputStream.class);
    s3ObjectMetadata = mock(ObjectMetadata.class);
    when(s3ObjectMock.getObjectContent()).thenReturn(s3ObjectInputStream);
    when(s3ServiceMock.getObjectMetadata(anyString(), anyString())).thenReturn(s3ObjectMetadata);
    when(s3ObjectMetadata.getContentLength()).thenReturn(contentLength);
    when(s3ObjectMetadata.getLastModified()).thenReturn(testDate);
    when(s3ServiceMock.getObject(anyString(), anyString())).thenReturn(s3ObjectMock);
    when(fileSystemSpy.getS3Client()).thenReturn(s3ServiceMock);
}
Also used : AmazonS3(com.amazonaws.services.s3.AmazonS3) StorageUnitConverter(org.apache.hop.core.util.StorageUnitConverter) ArrayList(java.util.ArrayList) S3FileSystem(org.apache.hop.vfs.s3.s3.vfs.S3FileSystem) S3FileObject(org.apache.hop.vfs.s3.s3.vfs.S3FileObject) S3FileName(org.apache.hop.vfs.s3.s3.vfs.S3FileName) VfsComponentContext(org.apache.commons.vfs2.provider.VfsComponentContext) DefaultFileSystemManager(org.apache.commons.vfs2.impl.DefaultFileSystemManager) S3HopProperty(org.apache.hop.vfs.s3.s3common.S3HopProperty) Before(org.junit.Before)

Example 5 with VfsComponentContext

use of org.apache.commons.vfs2.provider.VfsComponentContext in project commons-vfs by apache.

the class LayeredFileNameParser method parseUri.

/**
 * Parses the base and name into a FileName.
 *
 * @param context The component context.
 * @param baseFileName The base FileName.
 * @param fileName name The target file name.
 * @return The constructed FileName.
 * @throws FileSystemException if an error occurs.
 */
@Override
public FileName parseUri(final VfsComponentContext context, final FileName baseFileName, final String fileName) throws FileSystemException {
    final StringBuilder name = new StringBuilder();
    // Extract the scheme
    final String scheme = UriParser.extractScheme(context.getFileSystemManager().getSchemes(), fileName, name);
    // Extract the Layered file URI
    final String rootUriName = extractRootName(name);
    FileName rootUri = null;
    if (rootUriName != null) {
        rootUri = context.parseURI(rootUriName);
    }
    // Decode and normalise the path
    UriParser.canonicalizePath(name, 0, name.length(), this);
    UriParser.fixSeparators(name);
    final FileType fileType = UriParser.normalisePath(name);
    final String path = name.toString();
    return new LayeredFileName(scheme, rootUri, path, fileType);
}
Also used : FileType(org.apache.commons.vfs2.FileType) FileName(org.apache.commons.vfs2.FileName)

Aggregations

FileType (org.apache.commons.vfs2.FileType)15 VfsComponentContext (org.apache.commons.vfs2.provider.VfsComponentContext)9 FileName (org.apache.commons.vfs2.FileName)6 Test (org.junit.Test)5 AmazonS3 (com.amazonaws.services.s3.AmazonS3)4 ArrayList (java.util.ArrayList)4 DefaultFileSystemManager (org.apache.commons.vfs2.impl.DefaultFileSystemManager)4 Before (org.junit.Before)4 FileSystemException (org.apache.commons.vfs2.FileSystemException)3 ListObjectsRequest (com.amazonaws.services.s3.model.ListObjectsRequest)2 ObjectListing (com.amazonaws.services.s3.model.ObjectListing)2 ObjectMetadata (com.amazonaws.services.s3.model.ObjectMetadata)2 S3Object (com.amazonaws.services.s3.model.S3Object)2 S3ObjectInputStream (com.amazonaws.services.s3.model.S3ObjectInputStream)2 FileSystemOptions (org.apache.commons.vfs2.FileSystemOptions)2 FilesCache (org.apache.commons.vfs2.FilesCache)2 S3FileName (org.apache.hop.vfs.s3.s3.vfs.S3FileName)2 S3NFileName (org.apache.hop.vfs.s3.s3n.vfs.S3NFileName)2 FileSystemManager (org.apache.commons.vfs2.FileSystemManager)1 Cryptor (org.apache.commons.vfs2.util.Cryptor)1