io.camunda.connector.gsheets.operation.GoogleSheetOperation Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of connector-google-sheets Show documentation
Camunda Google Sheets Connector
The newest version!
/*
 * Copyright Camunda Services GmbH and/or licensed to Camunda Services GmbH
 * under one or more contributor license agreements. Licensed under a proprietary license.
 * See the License.txt file for more information. You may not use this file
 * except in compliance with the proprietary license.
 */
package io.camunda.connector.gsheets.operation;

import com.google.api.services.sheets.v4.Sheets;
import com.google.api.services.sheets.v4.model.BatchUpdateSpreadsheetRequest;
import com.google.api.services.sheets.v4.model.BatchUpdateSpreadsheetResponse;
import com.google.api.services.sheets.v4.model.ValueRange;
import io.camunda.connector.gsheets.supplier.GoogleSheetsServiceSupplier;
import io.camunda.google.model.Authentication;
import java.io.IOException;
import java.util.List;

public abstract class GoogleSheetOperation {

  protected static final String SPREADSHEET_ID_FIELD = "spreadsheetId";
  protected static final String REPLIES_FIELD = "replies";
  protected static final String SPREADSHEET_URL_FIELD = "spreadsheetUrl";
  protected static final String VALUE_INPUT_OPTION = "USER_ENTERED";

  public abstract Object execute(final Authentication auth);

  protected final BatchUpdateSpreadsheetResponse batchUpdate(
      final Authentication auth, final String spreadsheetId, BatchUpdateSpreadsheetRequest request)
      throws IOException {
    Sheets service = GoogleSheetsServiceSupplier.getGoogleSheetsService(auth);

    return service
        .spreadsheets()
        .batchUpdate(spreadsheetId, request)
        .setFields(buildFields(SPREADSHEET_ID_FIELD, REPLIES_FIELD))
        .execute();
  }

  protected final void update(
      final Authentication auth,
      final String spreadsheetId,
      final String range,
      final ValueRange valueRange)
      throws IOException {
    Sheets service = GoogleSheetsServiceSupplier.getGoogleSheetsService(auth);

    service
        .spreadsheets()
        .values()
        .update(spreadsheetId, range, valueRange)
        .setValueInputOption(VALUE_INPUT_OPTION)
        .execute();
  }

  protected final void append(
      final Authentication auth,
      final String spreadsheetId,
      final String range,
      final ValueRange valueRange)
      throws IOException {
    Sheets service = GoogleSheetsServiceSupplier.getGoogleSheetsService(auth);

    service
        .spreadsheets()
        .values()
        .append(spreadsheetId, range, valueRange)
        .setValueInputOption(VALUE_INPUT_OPTION)
        .execute();
  }

  protected final List> get(
      final Authentication auth, final String spreadsheetId, final String range)
      throws IOException {
    Sheets service = GoogleSheetsServiceSupplier.getGoogleSheetsService(auth);

    ValueRange result = service.spreadsheets().values().get(spreadsheetId, range).execute();

    return result.getValues();
  }

  protected final String buildRange(String worksheetName, String range) {
    if (worksheetName != null) {
      return worksheetName + "!" + range;
    }

    return range;
  }

  protected final String buildRowRange(Integer rowIndex) {
    return rowIndex + ":" + rowIndex;
  }

  protected String buildFields(String... fields) {
    return String.join(",", fields);
  }
}