S3FileHandler.java

/*
CloudTrail Viewer, is a Java desktop application for reading AWS CloudTrail logs
files.

Copyright (C) 2017  Mark P. Haskins

This program is free software: you can redistribute it and/or modify it under the
terms of the GNU General Public License as published by the Free Software Foundation,
either version 3 of the License, or (at your option) any later version.

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 General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

package io.haskins.java.cloudtrailviewer.controller.dialog.filechooser;

import com.amazonaws.services.s3.AmazonS3;
import com.amazonaws.services.s3.model.ListObjectsRequest;
import com.amazonaws.services.s3.model.ObjectListing;
import com.amazonaws.services.s3.model.S3ObjectSummary;
import io.haskins.java.cloudtrailviewer.model.aws.AwsAccount;
import io.haskins.java.cloudtrailviewer.model.observable.FileListModel;
import io.haskins.java.cloudtrailviewer.utils.AwsService;
import javafx.scene.control.ListView;

import java.util.List;

/**
 * Hanlding for navigating files in S3.
 *
 * Created by markhaskins on 27/01/2017.
 */
class S3FileHandler extends FileHandler {

    private AwsAccount currentAccount = null;
    private String prefix = "";

    private final AwsService awsService;

    S3FileHandler(
            ListView<FileListModel> listView, AwsAccount awsAccount,
            FileListControllerListener listener, AwsService awsService) {

        this.listView = listView;
        listView.setItems(data);

        this.fileListControllerListener = listener;
        this.currentAccount = awsAccount;

        this.awsService = awsService;

        setUpMouseListener();
        reloadContents();
    }

    public List<String> getSelectedItems() {

        addSelectedKeys();
        return this.selected_keys;
    }

    void handleDoubleClickEvent() {

        if (listView.getSelectionModel().getSelectedItem() != null) {

            String selected = listView.getSelectionModel().getSelectedItem().getName();

            if (selected.equalsIgnoreCase(MOVE_BACK)) {

                int lastSlash = prefix.lastIndexOf('/');
                String tmpPrefix = prefix.substring(0, lastSlash);

                prefix = "";
                if (tmpPrefix.contains("/")) {

                    lastSlash = tmpPrefix.lastIndexOf('/') + 1;
                    prefix = tmpPrefix.substring(0, lastSlash);
                }

                updateAccountPrefix(prefix);
                reloadContents();

            } else {

                int firstSlash = selected.indexOf('/');
                if (firstSlash == 0) {
                    selected = selected.substring(1, selected.length());
                }

                int lastSlash = selected.lastIndexOf('/') + 1;
                if (lastSlash == selected.length()) {

                    prefix = prefix + selected;
                    updateAccountPrefix(prefix);
                    reloadContents();
                } else {

                    addSelectedKeys();
                    updateAccountPrefix(prefix);

                    fileListControllerListener.selectionComplete();
                }
            }
        }
    }

    private void reloadContents() {

        listView.getItems().clear();

        ObjectListing objectListing = s3ListObjects(prefix, "/");

        // Add .. if not at root
        if (prefix.trim().length() != 0) {
            FileListModel model = new FileListModel(MOVE_BACK, MOVE_BACK, FileListModel.FILE_BACK);
            listView.getItems().add(model);
        }

        addDirectories(objectListing);
        addFileKeys(objectListing);
    }

    private void updateAccountPrefix(String newPrefix) {
        currentAccount.setPrefix(newPrefix);
    }


    private ObjectListing s3ListObjects(String pathPrefix, String delimiter) {

        ListObjectsRequest listObjectsRequest = new ListObjectsRequest();
        listObjectsRequest.setBucketName(currentAccount.getBucket());
        listObjectsRequest.setPrefix(pathPrefix);

        if (delimiter != null) {
            listObjectsRequest.setDelimiter(delimiter);
        }

        AmazonS3 s3Client = awsService.getS3Client(currentAccount);

        return s3Client.listObjects(listObjectsRequest);
    }

    private void addSelectedKeys() {

        selected_keys.clear();

        if (listView.getSelectionModel().getSelectedItems() != null &&
                !listView.getSelectionModel().getSelectedItems().isEmpty()) {

            String selected = prefix + listView.getSelectionModel().getSelectedItem().getName();

            if (isScanning && selected.endsWith("/")) {
                addFolderFiles(selected);

            } else {

                List<FileListModel> selectedItems = listView.getSelectionModel().getSelectedItems();
                for (FileListModel key : selectedItems) {
                    selected_keys.add(prefix + key.getName());
                }
            }
        }
    }

    private void addFolderFiles(String path) {

        AmazonS3 s3Client = awsService.getS3Client(currentAccount);

        ObjectListing current = s3Client.listObjects(currentAccount.getBucket(), path);
        List<S3ObjectSummary> objectSummaries = current.getObjectSummaries();

        for (final S3ObjectSummary objectSummary : objectSummaries) {
            String file = objectSummary.getKey();
            selected_keys.add(file);
        }

        while (current.isTruncated()) {

            current = s3Client.listNextBatchOfObjects(current);
            objectSummaries = current.getObjectSummaries();

            for (final S3ObjectSummary objectSummary : objectSummaries) {
                String file = objectSummary.getKey();
                selected_keys.add(file);
            }
        }
    }

    private void addDirectories(ObjectListing objectListing) {

        List<String> directories = objectListing.getCommonPrefixes();
        for (String directory : directories) {

            String dir = stripPrefix(directory);

            FileListModel model = new FileListModel(dir, dir, FileListModel.FILE_DIR);
            listView.getItems().add(model);
        }
    }

    private void addFileKeys(ObjectListing objectListing) {

        List<S3ObjectSummary> objectSummaries = objectListing.getObjectSummaries();
        for (final S3ObjectSummary objectSummary : objectSummaries) {

            String file = stripPrefix(objectSummary.getKey());

            FileListModel model = new FileListModel(file, file, FileListModel.FILE_DOC);
            listView.getItems().add(model);
        }
    }

    private String stripPrefix(String key) {

        String stripped = key;

        if (prefix.trim().length() > 0) {

            int prefixLength = prefix.length() - 1;
            stripped = key.substring(prefixLength, key.length());

            int firstSlash = stripped.indexOf('/');
            if (firstSlash == 0) {
                stripped = stripped.substring(1, stripped.length());
            }
        }

        return stripped;
    }
}