lhj-tools 0.1.73 → 0.1.74

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ec999ed3fcb4f5fff34b2dc483b5932c77bfd34c59cee78dfcebd7ea1a10fc54
-  data.tar.gz: c0a438db4402ef344ddf83f3acd4f3bca228c9335aa8e8d240bab02ba4745269
+  metadata.gz: f86764f14d88348e59221a4e339765a6ec4488a815d179a710d15e7c415471a1
+  data.tar.gz: 9f5635a3e149ce527aea1c298fa7a39316c446e75aaed941d23c6010ce0896de
 SHA512:
-  metadata.gz: 9201f7d556c51665acb83d63cbe4978eb6507d99d00b1da5ab92fa064557298242845fec4484d2cf327fc41a6bf1663c3648573627ecb3098338d5c591aa1d04
-  data.tar.gz: 217cc7ee797ee78ab6a074715bf346fc3e044779e2553a20a1b813a63239a3e72611d17e439477912415ba600325ed4f1d597446e213833c6146651419a22a67
+  metadata.gz: 144bfb6d35d9190b81ed1715dd518c2b707971815a3412f279f673faffb4c6dd36768c05c86da1249d9eed7328620a0588fe62a3753b249fd9fd8625726bb0d3
+  data.tar.gz: 6d875c726247374420c5b2856614283a817078e073517a3dc5849b7a9258567eefffc9df76f3d54ec58836f647ef922ea6169c5fe9211d53630ab39a2fe78570

data/lib/lhj/command/yapi.rb

@@ -13,6 +13,9 @@ module Lhj
       self.summary = '通过yapi接口生成请求'
       self.description = '执行`lhj api`生成接口模型'
 
+      API_INTERFACE_URL = 'api/interface/get?id='.freeze
+      API_PROJECT_URL = 'api/project/get?id='.freeze
+
       def self.options
         [
           %w[--id api的id],
@@ -39,7 +42,7 @@ module Lhj
         @config_model_name = 'Result'
         @model_default_suffix = 'Model'
         @type_trans = {}
-        @http_url = ''
+        @base_url = 'http://yapi.xx.com'
         super
       end
 
@@ -49,13 +52,16 @@ module Lhj
 
       def process(config_file)
         load_config(config_file)
-        res_body = req_api_model
+        res_body = get_interface_api_model
         unless res_body['errcode'].to_i.zero?
           puts '请执行 lhj yapi-init 初始化配置'.red
           return
         end
         return unless res_body && !res_body.empty?
 
+        # get project info
+        project_id = res_body['data']['project_id']
+        project_info = get_project_info(project_id)
         # 1.print request result
         result_model_name = print_res_body_model(res_body)
         # 2.print request json body
@@ -63,7 +69,7 @@ module Lhj
         # 3.print request param
         print_req_query(res_body['data'])
         # 4.print request method
-        service_code = print_http_method(res_body['data'], result_model_name, param_model_name)
+        service_code = print_http_method(res_body['data'], project_info, result_model_name, param_model_name)
         # 5.save to file
         file_map = save_to_file(service_code) if @save
         # 6.push to git
@@ -128,8 +134,12 @@ module Lhj
         Lhj::Dingtalk.post_message_robot(robot_url, '生成代码', output)
       end
 
-      def url_str
-        "#{@http_url}#{api_id}"
+      def interface_url_str
+        "#{@base_url}/#{API_INTERFACE_URL}#{api_id}"
+      end
+
+      def project_url_str(project_id)
+        "#{@base_url}/#{API_PROJECT_URL}#{project_id}"
       end
 
       def yml_file
@@ -141,7 +151,7 @@ module Lhj
         config.each do |k, v|
           @http_headers << "#{k}=#{v}" if k.eql?('__wpkreporterwid_') || k.eql?('_yapi_token') || k.eql?('_yapi_uid')
         end
-        @http_url = config['url']
+        @base_url = config['base_url']
         @config_id = config['id']
         @config_model_pre = config['model_pre']
         @config_model_suffix = config['model_suffix']
@@ -169,8 +179,8 @@ module Lhj
         @config_model_suffix || @model_default_suffix
       end
 
-      def req_api_model
-        uri = URI.parse(url_str)
+      def get_interface_api_model
+        uri = URI.parse(interface_url_str)
         req = Net::HTTP::Get.new(uri)
         req['Cookie'] = @http_headers.join('; ')
         res = Net::HTTP.start(uri.hostname, uri.port) do |http|
@@ -182,6 +192,17 @@ module Lhj
         res_json
       end
 
+      def get_project_info(project_id)
+        url = project_url_str(project_id)
+        uri = URI.parse(url)
+        req = Net::HTTP::Get.new(uri)
+        req['Cookie'] = @http_headers.join('; ')
+        res = Net::HTTP.start(uri.hostname, uri.port) do |http|
+          http.request(req)
+        end
+        JSON.parse(res.body)
+      end
+
       def print_res_body_model(res_json)
         res_body = fetch_res_boy(res_json)
         return unless res_body
@@ -401,7 +422,7 @@ module Lhj
         end
       end
 
-      def print_http_method(data, result_model_name, param_model_name)
+      def print_http_method(data, project_info, result_model_name, param_model_name)
         return unless data
 
         path = data['path']
@@ -421,7 +442,7 @@ module Lhj
           yapi_vars = { title: data['title'],
                         desc: data['desc'],
                         username: data['username'],
-                        path: path,
+                        path: "#{project_info['data']['basepath']}#{path}",
                         path_key: path_key,
                         path_name: path_name,
                         result_model_name: result_model_name,

data/lib/lhj/tools/version.rb

@@ -2,6 +2,6 @@
 
 module Lhj
   module Tools
-    VERSION = "0.1.73"
+    VERSION = "0.1.74"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lhj-tools
 version: !ruby/object:Gem::Version
-  version: 0.1.73
+  version: 0.1.74
 platform: ruby
 authors:
 - lihaijian
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-12-09 00:00:00.000000000 Z
+date: 2022-12-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: xcodeproj