gstreamer/subprojects/gst-plugins-bad/ext/gs/gstgscommon.cpp

/* GStreamer
 * Copyright (C) 2020 Julien Isorce <jisorce@oblong.com>
 *
 * gstgscommon.h:
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library 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
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
 * Boston, MA 02110-1301, USA.
 */

#include "gstgscommon.h"

#include "google/cloud/storage/oauth2/compute_engine_credentials.h"

namespace gcs = google::cloud::storage;

namespace {

#if !GLIB_CHECK_VERSION(2, 62, 0)
static inline gchar* g_date_time_format_iso8601(GDateTime* datetime) {
  GString* outstr = NULL;
  gchar* main_date = NULL;
  gint64 offset;

  // Main date and time.
  main_date = g_date_time_format(datetime, "%Y-%m-%dT%H:%M:%S");
  outstr = g_string_new(main_date);
  g_free(main_date);

  // Timezone. Format it as `%:::z` unless the offset is zero, in which case
  // we can simply use `Z`.
  offset = g_date_time_get_utc_offset(datetime);

  if (offset == 0) {
    g_string_append_c(outstr, 'Z');
  } else {
    gchar* time_zone = g_date_time_format(datetime, "%:::z");
    g_string_append(outstr, time_zone);
    g_free(time_zone);
  }

  return g_string_free(outstr, FALSE);
}
#endif

}  // namespace

std::unique_ptr<google::cloud::storage::Client> gst_gs_create_client(
    const gchar* service_account_email,
    GError** error) {
  if (service_account_email) {
    // Meant to be used from a container running in the Cloud.

    google::cloud::StatusOr<std::shared_ptr<gcs::oauth2::Credentials>> creds(
        std::make_shared<gcs::oauth2::ComputeEngineCredentials<>>(
            service_account_email));
    if (!creds) {
      g_set_error(error, GST_RESOURCE_ERROR, GST_RESOURCE_ERROR_NOT_AUTHORIZED,
                  "Could not retrieve credentials for the given service "
                  "account %s (%s)",
                  service_account_email, creds.status().message().c_str());
      return nullptr;
    }

    gcs::ClientOptions client_options(std::move(creds.value()));
    return std::make_unique<gcs::Client>(client_options,
                                         gcs::StrictIdempotencyPolicy());
  }
  // Default account. This is meant to retrieve the credentials automatically
  // using diffrent methods.
  google::cloud::StatusOr<gcs::ClientOptions> client_options =
      gcs::ClientOptions::CreateDefaultClientOptions();

  if (!client_options) {
    g_set_error(error, GST_RESOURCE_ERROR, GST_RESOURCE_ERROR_NOT_AUTHORIZED,
                "Could not create default client options (%s)",
                client_options.status().message().c_str());
    return nullptr;
  }
  return std::make_unique<gcs::Client>(client_options.value(),
                                       gcs::StrictIdempotencyPolicy());
}

gboolean gst_gs_get_buffer_date(GstBuffer* buffer,
                                GDateTime* start_date,
                                gchar** buffer_date_str_ptr) {
  gchar* buffer_date_str = NULL;
  GstClockTime buffer_timestamp = GST_CLOCK_TIME_NONE;
  GTimeSpan buffer_timespan = 0;

  if (!buffer || !start_date)
    return FALSE;

  buffer_timestamp = GST_BUFFER_PTS(buffer);

  // GTimeSpan is in micro seconds.
  buffer_timespan = GST_TIME_AS_USECONDS(buffer_timestamp);

  GDateTime* buffer_date = g_date_time_add(start_date, buffer_timespan);
  if (!buffer_date)
    return FALSE;

  buffer_date_str = g_date_time_format_iso8601(buffer_date);
  g_date_time_unref(buffer_date);

  if (!buffer_date_str)
    return FALSE;

  if (buffer_date_str_ptr)
    *buffer_date_str_ptr = buffer_date_str;

  return TRUE;
}