swagger-to-rbs 0.1.1 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 54cc8e6fb12e53ddc9dfbbafaa1628cdb1059fcfef1eaadaf1457a8ee113eaf9
-  data.tar.gz: 063253b30a32967fe2a20766885ffb32da17a2b98bc0a86846f55425b7db80b9
+  metadata.gz: f8ac0ad5697156071c8422296d55ea3eb34bc39e8359b8eb3b6d796a2be4d298
+  data.tar.gz: 2b0172153de41682a387d9909b0db5bec00cbf4f0708b4b5cdf6728289ca8d16
 SHA512:
-  metadata.gz: e5923cfe74e6ebd320e3cf9695d8f11186cb5470fabaf301ae57d2ee1fcd129012401332728e60edfed772c3de658f503e1f64c530b2f82d191d4f2afc6324fe
-  data.tar.gz: 733c15163c84675309f9850f503bc1dffdb251148c9de14ea3e494d5406315e6f743d49d575d71c9df6a1f05d6b2463446fbf91424d11e8ba8e4e6d32e976ca0
+  metadata.gz: ae3b412fa2e72e7af7d66740b68ac9ecd107a479719e80354cfb1a6896e2fa693688c65017c6c9bb9dd4492c01171d4c6ee9a5cdbe9687078abcac036e58f83e
+  data.tar.gz: 458ed2649c08a5019d2491887996869b3d6f7d1f701193aff13264a0aec782035ff10a049609d2a8ab15d32973ce50c8d10c3db979c9b9b37bf93367bfc40590

data/README.md

@@ -1,4 +1,10 @@
-### Usage
+## Install
+
+```
+gem install swagger-to-rbs
+```
+
+## Usage
 
 Dowload swagger spec
 
@@ -11,7 +17,7 @@ curl https://petstore3.swagger.io/api/v3/openapi.json > swagger.json
 Generate http client based on [Httparty gem](https://github.com/jnunemaker/httparty)
 
 ```
-ruby lib/index.rb MyApi
+swagger-to-rbs generate --name PetApi --spec swagger.json
 ```
 
 Result:
@@ -24,16 +30,20 @@ class PetApi
   base_uri "/api/v3"
   #...
 
-  def getPetById(petId, params, options = {})
-    self.class.get("/pet/#{petId}", params: params.to_json, headers: { 'Content-Type' => 'application/json' }.merge(options))
+  def initialize
+    self.class.headers({ 'Content-Type' => 'application/json' })
+  end
+
+  def getPetById(petId, options = {})
+    self.class.get("/pet/#{petId}", options)
   end
 
   def updatePet(body, options = {})
-    self.class.put("/pet", body: body.to_json, headers: { 'Content-Type' => 'application/json' }.merge(options))
+    self.class.put("/pet", { body: body.to_json }.merge(options))
   end
 
   def addPet(body, options = {})
-    self.class.post("/pet", body: body.to_json, headers: { 'Content-Type' => 'application/json' }.merge(options))
+    self.class.post("/pet", { body: body.to_json }.merge(options))
   end
   #...
 end
@@ -44,7 +54,7 @@ end
 Authorization example:
 
 ```
-api = MyApi.new
+api = PetApi.new
 
 MyApi.headers(Authorization: "Bearer #{access_token}")
 MyApi.base_uri("http://otherurl.test/api/v1)

data/lib/swagger2_rbs/cli.rb

@@ -22,6 +22,19 @@ module Swagger2Rbs
       File.write("#{file_name}.rbs", ::Swagger2Rbs.generate_rbs(name, data.dup))
     end
 
+    desc 'genearte-yaml', 'generate yaml file'
+    option :name, desc: 'Name'
+    option :spec, desc: 'Swagger file path'
+    def generate_yaml
+      @name = options[:name]
+      @swagger_path = options[:spec]
+      @debug = options[:debug]
+      swagger_spec = JSON.parse(File.read(@swagger_path))
+
+      file_name = to_underscore(name)
+      File.write("#{file_name}.yaml", ::Swagger2Rbs.swagger_to_rest_api_yaml(swagger_spec))
+    end
+
     private
     def fetch_data
       if swagger_path

data/lib/swagger2_rbs/hash_helper.rb

@@ -0,0 +1,41 @@
+module Swagger2Rbs
+  class HashHelper
+
+    # https://github.com/rails/rails/blob/master/activesupport/lib/active_support/core_ext/hash/keys.rb#L129
+    def self.deep_transform_keys_in_object!(object, &block)
+      case object
+      when Hash
+        object.keys.each do |key|
+          value = object.delete(key)
+          object[yield(key)] = deep_transform_keys_in_object!(value, &block)
+        end
+        object
+      when Array
+        object.map! { |e| deep_transform_keys_in_object!(e, &block) }
+      else
+        object
+      end
+    end
+
+    def self.set_value(hash, key, value)
+      arr = key.split(".")
+      last_key = arr.pop()
+      hash.dig(*arr)[last_key] = value
+      hash
+    rescue => e
+      hash
+    end
+
+    def self.walk(hash, &block)
+      hash.each do |k, v|
+        if v.is_a?(Hash)
+          walk(v) do |k2, v2|
+            yield "#{k}.#{k2}", v2
+          end
+        else
+          yield k, v
+        end
+      end
+    end
+  end
+end

data/lib/swagger2_rbs/rest_endpoint.rb

@@ -1,8 +1,10 @@
 require 'slugify'
+require_relative 'rest_endpoint_typed'
 
 module Swagger2Rbs
   class RestEndpoint
     attr_reader :path, :method, :props
+    include RestEndpointTyped
 
     def initialize(path, method, props)
       @path = path
@@ -13,15 +15,32 @@ module Swagger2Rbs
     def to_h
       {
         path: path_with_parameters,
-        method: method,
-        parameters: parameters,
+        http_method: method,
         parameters_for_method: parameters_for_method,
-        parameters_typed: parameters_typed,
+        typed_parameters_for_method: typed_parameters_for_method,
+        has_body: body?,
         method_name: method_name,
-        body: body,
-        body_typed: body_typed,
         response_typed: response_typed,
       }
+    rescue => e
+      raise e, "Context: #{path} #{method} Message: #{e.message}"
+    end
+
+    def body?
+      body && !body.empty?
+    end
+
+    def to_yaml
+      {
+        path: path,
+        method: method,
+        path_parameters: parameters,
+        method_name: method_name,
+        body: body,
+        response: response_typed,
+      }
+    rescue => e
+      raise e, "Context: #{path} #{method} Message: #{e.message}"
     end
 
     def method_name
@@ -35,39 +54,7 @@ module Swagger2Rbs
     def parameters
       return [] unless props["parameters"]
 
-      props["parameters"].select{|it| it["in"] == "path"}.map{|it| it["name"]}
-    end
-
-    def response_typed
-      schema = resolve_all_of(@props.dig("responses", "200", "content", "application/json", "schema"))
-      schema_to_typed(schema, {})
-    end
-
-    def parameters_typed
-      return nil if method != "get"
-      parameters.map{|it| "String #{it}" }
-      .push("?Hash[untyped, untyped] options")
-      .join(", ")
-    end
-
-    def body_typed
-      return nil if method != "post"
-
-      return "Hash[String, untyped]" unless body
-      return "Hash[String, untyped]" if body.empty?
-
-    "{" + body.map{ |k, v| to_typed(k, v) }.join(", ") + "}" + " body, ?Hash[untyped, untyped] options"
-    end
-
-    def to_typed(k, v)
-      return "#{k}: #{v&.capitalize}" unless v.is_a?(Array) || v.is_a?(Hash)
-      return "#{k}: {#{v.map{ |k2, v2| to_typed(k2, v2) }.join(", ")}}" if v.is_a?(Hash)
-
-      if v[0]&.is_a?(Hash)
-        "#{k}: Array[{" + v[0].map{ |k, v| "#{k}: #{v&.capitalize}" }.join(", ") + "}]"
-      else
-        "#{k}: Array[#{v[0]&.capitalize}]"
-      end
+      props["parameters"]&.select{|it| it["in"] == "path"}&.map{|it| it["name"]}
     end
 
     def body
@@ -78,26 +65,12 @@ module Swagger2Rbs
     end
 
     def parameters_for_method
-      return parameters.push("options = {}").join(", ") if (method == "get")
+      return parameters.push("options = {}").join(", ") if method == "get"
 
-      parameters.push("body").push("options = {}").join(", ")
-    end
-
-    def schema_to_typed(schema, memo = {})
-      return nil unless schema
-
-      schema["properties"]&.reduce(memo)do |memo, (k,v)|
-        if v["type"] == "object"
-          memo.merge({k => schema_to_typed(v, {})})
-        elsif v["type"] == "array"
-          if v.dig("items", "type") == "object"
-            memo.merge({k => [schema_to_typed(v["items"], {})] })
-          else
-            memo.merge({k => [v.dig("items", "type")] })
-          end
-        else
-          memo.merge({k => v["type"] })
-        end
+      if body&.empty?
+        parameters.push("options = {}").join(", ")
+      else
+        parameters.push("body").push("options = {}").join(", ")
       end
     end
 

data/lib/swagger2_rbs/rest_endpoint_typed.rb

@@ -0,0 +1,74 @@
+module Swagger2Rbs
+
+  module RestEndpointTyped
+    def response_typed
+      schema = resolve_all_of(@props.dig("responses", "200", "content", "application/json", "schema"))
+      schema_to_typed(schema, {})
+    end
+
+    def typed_parameters_for_method
+      options_typed = "?Hash[untyped, untyped] options"
+      return "(#{options_typed})" unless body && parameters
+      return "(#{options_typed})" if body.empty? && parameters.empty?
+
+      typed_parameters = parameters&.map{|it| "String #{it}" } || []
+
+      typed_parameters.push("#{body_typed} body") if body?
+
+      "(#{typed_parameters.push(options_typed).join(', ')})"
+    end
+
+    def body_typed
+      return nil unless body?
+      typed = (body.is_a?(Array) ? body[0] : body)&.map{ |k, v| to_typed(k, v) }.join(', ')
+      if body.is_a?(Array)
+        "Array[{ #{typed} }]"
+      else
+        "{ #{typed} }"
+      end
+    end
+
+    def type_case(str)
+      case str
+      when "boolean"
+        "bool"
+      else
+        str&.capitalize
+      end
+    end
+
+    def to_typed(k, v)
+      return "#{k}: #{type_case(v)}" unless v.is_a?(Array) || v.is_a?(Hash)
+      return "#{k}: {#{v.map{ |k2, v2| to_typed(k2, v2) }.join(", ")}}" if v.is_a?(Hash)
+
+      if v[0]&.is_a?(Hash)
+        "#{k}: Array[{" + v[0].map{ |k, v| to_typed(k, v) }.join(", ") + "}]"
+      else
+        "#{k}: Array[#{type_case(v[0])}]"
+      end
+    end
+
+    def schema_to_typed(schema, memo = {})
+      return nil unless schema
+
+      properties = schema["type"]["array"] ? schema["items"]["properties"] : schema["properties"]
+
+      result = properties&.reduce(memo)do |memo, (k,v)|
+        if v["type"] == "object"
+          memo.merge({k => schema_to_typed(v, {})})
+        elsif v["type"] == "array"
+          if v.dig("items", "type") == "object"
+            memo.merge({k => [schema_to_typed(v["items"], {})] })
+          else
+            memo.merge({k => [v.dig("items", "type")] })
+          end
+        else
+          memo.merge({k => v["type"] })
+        end
+      end
+      return [result] if schema["type"]["array"]
+
+      result
+    end
+  end
+end

data/lib/swagger2_rbs.rb

@@ -1,21 +1,48 @@
 require 'json'
 require 'erb'
 require_relative 'swagger2_rbs/rest_endpoint'
+require_relative 'swagger2_rbs/hash_helper'
 
 module Swagger2Rbs
 
-  def self.swagger_to_rest_api(swagger_spec)
+  def self.resolve_ref(hash, key, value)
+    ref_key = value.gsub("#/", "").split("/")
+    data = hash.dig(*ref_key)
+    update_key = key.split(".").reject{|k| k == "$ref"}.join(".")
+
+    [update_key, data]
+  end
+
+  def self.resolve_all_ref(swagger_spec)
+    new_swagger_spec = swagger_spec.dup
+    HashHelper.walk(swagger_spec) do |key, value|
+      if key.split(".").last == "$ref"
+        update_key, data = resolve_ref(swagger_spec, key, value)
+        HashHelper.set_value(new_swagger_spec, update_key, data)
+      end
+    end
+    new_swagger_spec
+  end
+
+  def self.swagger_to_rest_api(swagger_spec, parse_method = :to_h)
     result = []
-    swagger_spec["paths"].each do |path, data|
+    resolve_all_ref(swagger_spec)["paths"].each do |path, data|
       data.each do |method, props|
         rest_data = RestEndpoint.new(path, method, props)
-        result << rest_data.to_h
+        result << rest_data.send(parse_method)
       end
     end
 
     { base_uri: swagger_spec["servers"].first["url"], endpoints: result }
   end
 
+  def self.swagger_to_rest_api_yaml(swagger_spec)
+    response = swagger_to_rest_api(swagger_spec, :to_yaml)
+    YAML.dump(
+      HashHelper.deep_transform_keys_in_object!(response, &:to_s)
+    ).gsub("---\n", "")
+  end
+
   def self.rest_api_all(spec)
     result = []
     spec[:endpoints].each do |endpoint|

data/lib/templates/http_client.rb.erb

@@ -1,5 +1,5 @@
-
-require 'httparty'
+# require 'httparty'
+require 'json'
 
 class <%= @module_name %>
   include HTTParty
@@ -11,13 +11,13 @@ class <%= @module_name %>
   end
   <%- @data[:endpoints].each do |endpoint| -%>
 
-  <%- if endpoint[:method] == 'get' -%>
+  <%- unless endpoint[:has_body] -%>
   def <%= endpoint[:method_name] %>(<%= endpoint[:parameters_for_method] %>)
-    self.class.<%= endpoint[:method] %>("<%= endpoint[:path] %>", options)
+    self.class.<%= endpoint[:http_method] %>("<%= endpoint[:path] %>", options)
   end
   <%- else -%>
   def <%= endpoint[:method_name] %>(<%= endpoint[:parameters_for_method] %>)
-    self.class.<%= endpoint[:method] %>("<%= endpoint[:path] %>", { body: body.to_json }.merge(options))
+    self.class.<%= endpoint[:http_method] %>("<%= endpoint[:path] %>", { body: body.to_json }.merge(options))
   end
   <%- end -%>
   <%- end -%>

data/lib/templates/http_client.rbs.erb

@@ -1,16 +1,25 @@
-
 # Classes
 
+module HTTParty
+  class Response
+    def body: -> untyped
+    def success?: -> bool
+  end
+end
+
 class <%= @module_name %>
   include HTTParty
 
+  def self.headers: (untyped headers) -> void
+  def self.base_uri: (String uri) -> void
+  def self.get: (String path, untyped options) -> void
+  def self.post: (String path, untyped options) -> void
+  def self.put: (String path, untyped options) -> void
+  def self.delete: (String path, untyped options) -> void
+
   def initialize: -> void
   <%- @data[:endpoints].each do |endpoint| -%>
 
-  <%- if endpoint[:method] == 'get' -%>
-  def <%= endpoint[:method_name] %>(<%= endpoint[:parameters_typed] %>) -> HTTParty::Response
-  <%- else -%>
-  def <%= endpoint[:method_name] %>(<%= endpoint[:body_typed] %>) -> HTTParty::Response
-  <%- end -%>
+  def <%= endpoint[:method_name] %>: <%= endpoint[:typed_parameters_for_method] %> -> HTTParty::Response
   <%- end -%>
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: swagger-to-rbs
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.4.0
 platform: ruby
 authors:
 - Miguel Savignano
@@ -14,14 +14,14 @@ dependencies:
   name: thor
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
@@ -50,7 +50,9 @@ files:
 - lib/index.rb
 - lib/swagger2_rbs.rb
 - lib/swagger2_rbs/cli.rb
+- lib/swagger2_rbs/hash_helper.rb
 - lib/swagger2_rbs/rest_endpoint.rb
+- lib/swagger2_rbs/rest_endpoint_typed.rb
 - lib/templates/http_client.rb.erb
 - lib/templates/http_client.rbs.erb
 homepage: https://github.com/MiguelSavignano/swagger-to-rbs