CartoDB/cartodb20

View on GitHub
lib/carto/connector/provider.rb

Summary

Maintainability
A
3 hrs
Test Coverage
# Base class for Connector Providers
#
# This is an abstract class; concrete classes derived from this one
# must implement these methods:
#
# * `copy_table(schema_name:, table_name:, limits:)`
# * `list_tables(limits:)`
# * `check_connection`
# * `remote_data_updated?`
# * `table_name`
# * `required_parameters`
# * `optional_parameters`
# * `features_information`
#
module Carto
  class Connector
    class Provider
      # Provider identifier (internal name, used in APIs, etc)
      def self.provider_id
        must_be_defined_in_derived_class
      end

      # Human-readable name of the provider
      def self.friendly_name
        must_be_defined_in_derived_class
      end

      # This means that the provider is publicly announced (so it is accessible through UI, visible in lists of
      # providers, etc.) A provider may be available or not (see Connector.limits) independently of its public status,
      # so that a public provider may not be available for all users, and non-public providers may be available to
      # some users.
      def self.public?
        # Providers are public by default
        true
      end

      def initialize(parameters: {}, user: nil, logger: nil, previous_last_modified: nil)
        @params = Parameters.new(parameters, required: required_parameters + [:provider], optional: optional_parameters)
        @user = user
        @logger = logger
        # previous_last_modified is the time at which external data had been modified at the previous synchronization
        @previous_last_modified = previous_last_modified
      end

      def errors(only_for: nil)
        @params.errors(only_for: only_for)
      end

      def valid?
        errors.empty?
      end

      def validate!(only: nil)
        errors = self.errors(only_for: only)
        raise InvalidParametersError.new(message: errors * "\n") if errors.present?
      end

      def copy_table(schema_name:, table_name:, limits:)
        must_be_defined_in_derived_class schema_name: schema_name, table_name: table_name, limits: limits
      end

      def list_tables(limits:)
        must_be_defined_in_derived_class limits: limits
      end

      def list_projects
        must_be_defined_in_derived_class
      end

      def list_project_datasets(project_id)
        must_be_defined_in_derived_class
      end

      def list_project_dataset_tables(project_id, dataset_id)
        must_be_defined_in_derived_class
      end

      def check_connection
        must_be_defined_in_derived_class
      end

      def dry_run
        must_be_defined_in_derived_class
      end

      def remote_data_updated?
        # By default connectors don't check for updated data
        true
      end

      def last_modified
        # By default connectors don't support last modified time
        nil
      end

      # Name of the table to be imported
      def table_name
        must_be_defined_in_derived_class
      end

      # Parameters required by this connector provider
      def required_parameters
        must_be_defined_in_derived_class
      end

      # Optional parameters accepted by this connector provider
      def optional_parameters
        must_be_defined_in_derived_class
      end

      # Parameters accepted by this connector provider
      def accepted_parameters
        required_parameters + optional_parameters
      end

      def self.information
        # For convenience we'll use instance methods to provide the information
        # en each class. Otherwise all the information needed by such methods
        # would have to be defined in class methods too.
        test_provider = new(parameters: {})
        {
          features: test_provider.features_information,
          parameters: test_provider.parameters_information
        }
      end

      def features_information
        must_be_defined_in_derived_class
      end

      def parameters_information
        # TODO: add templates with parameter descriptions
        info = {}
        required_parameters.each do |name|
          # TODO: description = load template for parameter name of @provider.name
          info[name.to_s] = {
            required: true
          }
        end
        optional_parameters.each do |name|
          # TODO: description = load template for parameter name of @provider.name
          info[name.to_s] = {
            required: false
          }
        end
        info
      end

      def provider_id
        self.class.provider_id
      end

      METADATA_KEYS = {
        id: :provider_id,
        name: :friendly_name,
        public: :public?
      }
      class <<self
        def metadata(options)
          options.each do |key, value|
            method = METADATA_KEYS[key]
            raise "Invalid Provider metadata key: #{key.inspect}" unless method
            define_singleton_method(method) { value.freeze }
          end
        end
        def optional_parameters(params)
          define_method(:optional_parameters) { params.freeze }
        end
        def required_parameters(params)
          define_method(:required_parameters) { params.freeze }
        end
      end

      def log(message, truncate = true)
        @logger.append message, truncate if @logger
      end

      private

      def must_be_defined_in_derived_class(*_)
        raise NotImplementedError, "Method \"#{caller_locations(1, 1)[0].label}\" must be defined in derived class"
      end

      def self.must_be_defined_in_derived_class(*_)
        raise NotImplementedError, "Class method \"#{caller_locations(1, 1)[0].label}\" must be defined in derived class"
      end
    end

  end
end