com.github.robozonky.integrations.stonky.Util Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of robozonky-integration-stonky Show documentation
There is a newer version: 5.4.0-beta-1
Show newest version
/*
 * Copyright 2019 The RoboZonky Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.github.robozonky.integrations.stonky;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.Collections;
import java.util.Optional;
import java.util.function.Function;
import java.util.function.Supplier;

import com.github.robozonky.internal.Defaults;
import com.github.robozonky.internal.util.UrlUtil;
import com.google.api.client.auth.oauth2.Credential;
import com.google.api.client.googleapis.javanet.GoogleNetHttpTransport;
import com.google.api.client.http.HttpTransport;
import com.google.api.client.json.JsonFactory;
import com.google.api.client.json.jackson2.JacksonFactory;
import com.google.api.services.drive.Drive;
import com.google.api.services.sheets.v4.Sheets;
import io.vavr.control.Try;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

public final class Util {

    static final JsonFactory JSON_FACTORY = JacksonFactory.getDefaultInstance();
    private static final Logger LOGGER = LogManager.getLogger(Util.class);
    private static final String APPLICATION_NAME = Defaults.ROBOZONKY_USER_AGENT;

    private Util() {
        // no instances
    }

    public static com.google.api.services.drive.model.File getFile(final Drive driveService,
                                                                   final String id) throws IOException {
        LOGGER.debug("Loading file: {}.", id);
        return driveService.files().get(id)
                .setFields("id,name,modifiedTime")
                .execute();
    }

    public static com.google.api.services.drive.model.File copyFile(final Drive driveService,
                                                                    final com.google.api.services.drive.model.File upstream,
                                                                    final com.google.api.services.drive.model.File parent,
                                                                    final String name) throws IOException {
        LOGGER.debug("Cloning master spreadsheet '{}', setting name: {}.", upstream.getId(), name);
        final com.google.api.services.drive.model.File f = new com.google.api.services.drive.model.File();
        f.setName(name);
        f.setParents(Collections.singletonList(parent.getId()));
        final com.google.api.services.drive.model.File result = driveService.files().copy(upstream.getId(), f)
                .setFields("id,name,modifiedTime")
                .execute();
        LOGGER.debug("Created a copy: {}.", result.getId());
        return result;
    }

    public static HttpTransport createTransport() {
        try {
            return GoogleNetHttpTransport.newTrustedTransport();
        } catch (final Exception ex) {
            throw new IllegalStateException(ex);
        }
    }

    public static Drive createDriveService(final Credential credential, final HttpTransport transport) {
        return new Drive.Builder(transport, JSON_FACTORY, credential)
                .setApplicationName(APPLICATION_NAME)
                .build();
    }

    public static Sheets createSheetsService(final Credential credential, final HttpTransport transport) {
        return new Sheets.Builder(transport, JSON_FACTORY, credential)
                .setApplicationName(APPLICATION_NAME)
                .build();
    }

    static  Function wrap(final ThrowingFunction function) {
        return s -> {
            try {
                return function.apply(s);
            } catch (final Exception ex) {
                throw new IllegalStateException("Function failed.", ex);
            }
        };
    }

    static  Supplier wrap(final ThrowingSupplier supplier) {
        return () -> {
            try {
                return supplier.get();
            } catch (final Exception ex) {
                throw new IllegalStateException("Supplier failed.", ex);
            }
        };
    }

    static Optional download(final URL url) {
        return Try.withResources(() -> UrlUtil.open(url))
                .of(Util::download)
                .getOrElseGet(t -> {
                    LOGGER.warn("Failed downloading file.", t);
                    return Optional.empty();
                });
    }

    static Optional download(final InputStream stream) {
        return Try.withResources(() -> stream)
                .of(s -> {
                    final Path p = Files.createTempFile("robozonky-", ".download");
                    s.transferTo(Files.newOutputStream(p));
                    return Optional.of(p.toFile());
                })
                .getOrElseGet(ex -> {
                    LOGGER.warn("Failed downloading file.", ex);
                    return Optional.empty();
                });
    }
}