gapic-generator 0.1.7 → 0.2.0

data/lib/gapic/presenters/file_presenter.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+# Copyright 2018 Google LLC
+#
+# 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
+#
+#     https://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.
+
+require "gapic/helpers/namespace_helper"
+
+module Gapic
+  module Presenters
+    ##
+    # A presenter for proto files.
+    #
+    class FilePresenter
+      include Gapic::Helpers::NamespaceHelper
+
+      # @param file [Gapic::Schema::File] the file to present
+      def initialize api, file
+        @api = api
+        @file = file
+      end
+
+      def name
+        @file.name
+      end
+
+      def address
+        @file.address
+      end
+
+      def namespace
+        return @file.ruby_package if @file.ruby_package.present?
+        ruby_namespace_for_address address
+      end
+
+      def messages
+        @messages ||= @file.messages.map { |m| MessagePresenter.new @api, m }
+      end
+
+      def enums
+        @enums ||= @file.enums.map { |e| EnumPresenter.new e }
+      end
+
+      def docs_file_path
+        @file.name.gsub ".proto", ".rb"
+      end
+    end
+  end
+end

data/lib/gapic/presenters/gem_presenter.rb

@@ -0,0 +1,170 @@
+# frozen_string_literal: true
+
+# Copyright 2019 Google LLC
+#
+# 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
+#
+#     https://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.
+
+require "gapic/helpers/filepath_helper"
+require "gapic/helpers/namespace_helper"
+
+module Gapic
+  module Presenters
+    ##
+    # A presenter for gems.
+    #
+    class GemPresenter
+      include Gapic::Helpers::FilepathHelper
+      include Gapic::Helpers::NamespaceHelper
+
+      def initialize api
+        @api = api
+      end
+
+      def packages
+        @packages ||= begin
+          packages = @api.generate_files.map(&:package).uniq.sort
+          packages.map { |p| PackagePresenter.new @api, p }.delete_if(&:empty?)
+        end
+      end
+
+      def services
+        @services ||= begin
+          files = @api.generate_files
+          files.map(&:services).flatten.map { |s| ServicePresenter.new @api, s }
+        end
+      end
+
+      def proto_files
+        @proto_files ||= begin
+          files = @api.files
+          files = files.reject { |f| blacklist_protos.include? f.name }
+          files = files.reject { |f| f.messages.empty? && f.enums.empty? }
+          files.map { |f| FilePresenter.new @api, f }
+        end
+      end
+
+      def address
+        name.split("-").map(&:camelize)
+      end
+
+      def name
+        gem_config :name
+      end
+
+      def namespace
+        gem_config(:namespace) ||
+          fix_namespace(@api, name.split("-").map(&:camelize).join("::"))
+      end
+
+      def title
+        gem_config(:title) ||
+          namespace.split("::").join(" ")
+      end
+
+      def version
+        gem_config(:version) ||
+          "0.0.1"
+      end
+
+      def version_require
+        ruby_file_path @api, version_name_full
+      end
+
+      def version_file_path
+        "#{version_require}.rb"
+      end
+
+      def version_name_full
+        "#{namespace}::VERSION"
+      end
+
+      def authors
+        gem_config(:authors) ||
+          ["Google LLC"]
+      end
+
+      def email
+        gem_config(:email) ||
+          "googleapis-packages@google.com"
+      end
+
+      def description
+        gem_config(:description) ||
+          "#{name} is the official client library for the #{title} API."
+      end
+
+      def summary
+        gem_config(:summary) ||
+          "API Client library for the #{title} API"
+      end
+
+      def homepage
+        gem_config(:homepage) ||
+          "https://github.com/googleapis/googleapis"
+      end
+
+      def env_prefix
+        (gem_config(:env_prefix) || name.split("-").last).upcase
+      end
+
+      def iam_dependency?
+        @api.files.map(&:name).any? { |f| f.start_with? "google/iam/v1/" }
+      end
+
+      def library_documentation_url
+        gem_config(:library_documentation_url) || "https://googleapis.dev/ruby/#{name}/latest"
+      end
+
+      def product_documentation_url
+        gem_config :product_documentation_url
+      end
+
+      def api_id
+        gem_config :api_id
+      end
+
+      def free_tier?
+        gem_config(:free_tier) ? true : false
+      end
+
+      def yard_strict?
+        gem_config(:yard_strict) ? true : false
+      end
+
+      def entrypoint_require
+        packages.first.package_require
+      end
+
+      private
+
+      def gem_config key
+        return unless @api.configuration[:gem]
+
+        @api.configuration[:gem][key]
+      end
+
+      def blacklist_protos
+        blacklist = gem_config :blacklist
+
+        return default_blacklist_protos if blacklist.nil?
+        return default_blacklist_protos if blacklist[:protos].nil?
+
+        default_blacklist_protos[:protos]
+      end
+
+      def default_blacklist_protos
+        ["google/api/http.proto", "google/protobuf/descriptor.proto"]
+      end
+    end
+  end
+end

data/lib/gapic/presenters/message_presenter.rb

@@ -0,0 +1,73 @@
+# frozen_string_literal: true
+
+# Copyright 2018 Google LLC
+#
+# 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
+#
+#     https://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.
+
+require "active_support/inflector"
+
+require "gapic/helpers/namespace_helper"
+
+module Gapic
+  module Presenters
+    ##
+    # A presenter for proto messages.
+    #
+    class MessagePresenter
+      include Gapic::Helpers::NamespaceHelper
+
+      def initialize api, message
+        @api = api
+        @message = message
+      end
+
+      def name
+        @message.name
+      end
+
+      def doc_types
+        type_name_full
+      end
+
+      def doc_description
+        @message.docs_leading_comments
+      end
+
+      def default_value
+        "{}"
+      end
+
+      def type_name_full
+        message_ruby_type @message
+      end
+
+      def fields
+        @fields = @message.fields.map { |f| FieldPresenter.new @api, @message, f }
+      end
+
+      def nested_enums
+        @nested_enums ||= @message.nested_enums.map { |e| EnumPresenter.new e }
+      end
+
+      def nested_messages
+        @nested_messages ||= @message.nested_messages.map { |m| MessagePresenter.new @api, m }
+      end
+
+      protected
+
+      def message_ruby_type message
+        ruby_namespace @api, message.address.join(".")
+      end
+    end
+  end
+end

data/lib/gapic/presenters/method_presenter.rb

@@ -0,0 +1,298 @@
+# frozen_string_literal: true
+
+# Copyright 2018 Google LLC
+#
+# 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
+#
+#     https://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.
+
+require "active_support/inflector"
+require "gapic/path_template"
+require "gapic/helpers/namespace_helper"
+
+module Gapic
+  module Presenters
+    ##
+    # A presenter for rpc methods.
+    #
+    class MethodPresenter
+      include Gapic::Helpers::NamespaceHelper
+
+      def initialize api, method
+        @api = api
+        @method = method
+      end
+
+      def service
+        ServicePresenter.new @api, @method.parent
+      end
+
+      def name
+        ActiveSupport::Inflector.underscore @method.name
+      end
+
+      def kind
+        if client_streaming?
+          if server_streaming?
+            :bidi
+          else
+            :client
+          end
+        elsif server_streaming?
+          :server
+        else
+          :normal
+        end
+      end
+
+      def doc_description
+        @method.docs_leading_comments
+      end
+
+      def doc_response_type
+        ret = return_type
+        ret = "Gapic::Operation" if lro?
+        if server_streaming?
+          ret = "Enumerable<#{ret}>"
+        elsif paged?
+          paged_type = paged_response_type
+          paged_type = "Gapic::Operation" if paged_type == "Google::Longrunning::Operation"
+          ret = "Gapic::PagedEnumerable<#{paged_type}>"
+        end
+        ret
+      end
+
+      def arguments
+        arguments = @method.input.fields.reject(&:output_only?)
+        arguments.map { |arg| FieldPresenter.new @api, @method.input, arg }
+      end
+
+      def fields
+        @method.input.fields.map { |field| FieldPresenter.new @api, @method.input, field }
+      end
+
+      def fields_with_first_oneof
+        return fields if @method.input.oneof_decl.empty?
+
+        selected_fields = []
+        have_oneof = []
+
+        @method.input.fields.each do |field|
+          idx = field.oneof_index
+          selected_fields << field unless have_oneof.include? idx
+          have_oneof << idx
+        end
+
+        selected_fields.map { |field| FieldPresenter.new @api, @method.input, field }
+      end
+
+      def request_type
+        message_ruby_type @method.input
+      end
+
+      def return_type
+        message_ruby_type @method.output
+      end
+
+      def yields?
+        # Server streaming RCP calls are the only one that does not yield
+        !server_streaming?
+      end
+
+      def yield_doc_description
+        return "Register a callback to be run when an operation is done." if lro?
+
+        "Access the result along with the RPC operation"
+      end
+
+      def yield_params
+        if lro?
+          return [
+            OpenStruct.new(
+              name:      "operation",
+              doc_types: "Gapic::Operation"
+            )
+          ]
+        end
+        [
+          OpenStruct.new(
+            name:      "result",
+            doc_types: return_type
+          ),
+          OpenStruct.new(
+            name:      "operation",
+            doc_types: "GRPC::ActiveCall::Operation"
+          )
+        ]
+      end
+
+      # @api.incode samples and sample_configs are yaml configuration files such as
+      # speech_transcribe_sync_gcs.yaml
+      def samples
+        sample_configs = @api.incode_samples.select do |sample_config|
+          sample_config["service"] == @method.address[0...-1].join(".") &&
+            sample_config["rpc"] == @method.name
+        end
+        sample_configs.map { |sample_config| SamplePresenter.new @api, sample_config }
+      end
+
+      def lro?
+        return paged_response_type == "Google::Longrunning::Operation" if paged?
+
+        message_ruby_type(@method.output) == "Google::Longrunning::Operation"
+      end
+
+      def client_streaming?
+        @method.client_streaming
+      end
+
+      def server_streaming?
+        @method.server_streaming
+      end
+
+      def paged?
+        return false if server_streaming? # Cannot page a streaming response
+        paged_request?(@method.input) && paged_response?(@method.output)
+      end
+
+      def paged_response_type
+        return nil unless paged_response? @method.output
+
+        repeated_field = @method.output.fields.find do |f|
+          f.label == Google::Protobuf::FieldDescriptorProto::Label::LABEL_REPEATED &&
+            f.type == Google::Protobuf::FieldDescriptorProto::Type::TYPE_MESSAGE
+        end
+        message_ruby_type repeated_field.message
+      end
+
+      ##
+      #
+      # @return [Array<String>] The segment key names.
+      #
+      def routing_params
+        segments = Gapic::PathTemplate.parse method_path
+        segments.select { |s| s.is_a? Gapic::PathTemplate::Segment }.map(&:name)
+      end
+
+      def routing_params?
+        routing_params.any?
+      end
+
+      def grpc_service_config
+        if @api.grpc_service_config&.service_method_level_configs&.key?(service.grpc_full_name) &&
+           @api.grpc_service_config.service_method_level_configs[service.grpc_full_name]&.key?(grpc_method_name)
+          @api.grpc_service_config.service_method_level_configs[service.grpc_full_name][grpc_method_name]
+        end
+      end
+
+      def grpc_method_name
+        @method.name
+      end
+
+      protected
+
+      def message_ruby_type message
+        ruby_namespace @api, message.address.join(".")
+      end
+
+      def doc_types_for arg
+        if arg.message?
+          "#{message_ruby_type arg.message} | Hash"
+        elsif arg.enum?
+          # TODO: handle when arg message is nil and enum is the type
+          message_ruby_type arg.enum
+        else
+          case arg.type
+          when 1, 2                              then "Float"
+          when 3, 4, 5, 6, 7, 13, 15, 16, 17, 18 then "Integer"
+          when 9, 12                             then "String"
+          when 8                                 then "Boolean"
+          else
+            "Object"
+          end
+        end
+      end
+
+      def doc_desc_for arg
+        return nil if arg.docs.leading_comments.empty?
+
+        arg.docs.leading_comments
+      end
+
+      def default_value_for arg
+        if arg.message?
+          "{}"
+        elsif arg.enum?
+          # TODO: select the first non-0 enum value
+          # ":ENUM"
+          arg.enum.values.first
+        else
+          case arg.type
+          when 1, 2                              then "3.14"
+          when 3, 4, 5, 6, 7, 13, 15, 16, 17, 18 then "42"
+          when 9, 12                             then "\"hello world\""
+          when 8                                 then "true"
+          else
+            "Object"
+          end
+        end
+      end
+
+      def method_path
+        return "" if @method.http.nil?
+
+        method = [
+          @method.http.get, @method.http.post, @method.http.put,
+          @method.http.patch, @method.http.delete
+        ].find { |x| !x.empty? }
+        return method unless method.nil?
+
+        return @method.http.custom.path unless @method.http.custom.nil?
+      end
+
+      def paged_request? request
+        page_token = request.fields.find do |f|
+          f.name == "page_token" && f.type == Google::Protobuf::FieldDescriptorProto::Type::TYPE_STRING
+        end
+        return false if page_token.nil?
+
+        page_size_types = [
+          Google::Protobuf::FieldDescriptorProto::Type::TYPE_INT32,
+          Google::Protobuf::FieldDescriptorProto::Type::TYPE_INT64
+        ]
+        page_size = request.fields.find do |f|
+          f.name == "page_size" && page_size_types.include?(f.type)
+        end
+        return false if page_size.nil?
+
+        true
+      end
+
+      def paged_response? response
+        next_page_token = response.fields.find do |f|
+          f.name == "next_page_token" && f.type == Google::Protobuf::FieldDescriptorProto::Type::TYPE_STRING
+        end
+        return false if next_page_token.nil?
+
+        repeated_field = response.fields.find do |f|
+          f.label == Google::Protobuf::FieldDescriptorProto::Label::LABEL_REPEATED &&
+            f.type == Google::Protobuf::FieldDescriptorProto::Type::TYPE_MESSAGE
+        end
+        return false if repeated_field.nil?
+
+        # We want to make sure the first repeated field is also has the lowest field number,
+        # but the google-protobuf gem sorts fields by number, so we lose the original order.
+
+        true
+      end
+    end
+  end
+end