require 'net/https' require 'uri' require 'json' require 'yaml' require 'lhj/command/yapi/formatters/service' require 'lhj/helper/dingtalk_config' require 'lhj/helper/erb_template_helper' module Lhj class Command # generate model from yapi class Yapi < Command self.summary = '通过yapi接口生成请求' self.description = '更新 ~/.lhj/yapi.yml 文件配置后执行`lhj api`生成接口模型' def self.options [ %w[--id api的id], %w[--model-pre 模型的前缀], %w[--save 保存生成文件] ] end def initialize(argv) @id = argv.option('id') @model_pre_name = argv.option('model-pre') @language = argv.option('lan', 'oc') @save = argv.flag?('save', false) @sync = argv.flag?('sync', false) @notify = argv.flag?('notify', false) @debug = argv.flag?('debug', false) @sync = true if @notify @save = true if @sync @http_headers = [] @config_id = '' @config_model_pre = 'ML' @model_default_suffix = 'Model' @type_trans = {} @config_model_names = [] @model_names = [] @http_url = '' super end def handle process(yml_file) if File.exist?(yml_file) end def process(config_file) load_config(config_file) res_body = req_api_model unless res_body['errcode'].to_i.zero? puts '请执行 lhj yapi-init 初始化配置'.red return end return unless res_body && !res_body.empty? # 1.print request result result_model_name = print_res_body_model(res_body) # 2.print request json body param_model_name = print_req_body_model(res_body) # 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) # 5.save to file file_map = save_to_file(service_code) if @save # 6.push to git push_to_git if @sync # 7.notify robot notify_robot(file_map) if @notify end def puts_h(str) puts str.magenta @h_file_array ||= [] @h_file_array << str end def puts_h_red(str) puts str.red @h_file_array ||= [] @h_file_array << str end def puts_m(str) puts str.blue @m_file_array ||= [] @m_file_array << str end def sub_folder_name return @sub_folder_name if @sub_folder_name time = Time.now @sub_folder_name ||= time.strftime('%Y%m%d%H%M%S') @sub_folder_name end def save_to_file(service_code) @model_names = [] model_name = @config_model_names.frist file_name = gen_model_name('') unless File.exist?(File.expand_path(sub_folder_name, '.')) FileUtils.mkdir_p(File.expand_path(sub_folder_name, '.')) end h_file = File.join('.', sub_folder_name, "#{file_name}.h") m_file = File.join('.', sub_folder_name, "#{file_name}.m") service_file = File.join('.', sub_folder_name, "#{model_pre}#{model_name}Service.m") File.write(h_file, @h_file_array.join("\n")) if @h_file_array.count.positive? File.write(m_file, @m_file_array.join("\n")) if @m_file_array.count.positive? File.write(service_file, service_code) if service_code puts "\n\n生成文件成功!所在路径:\n#{File.expand_path(h_file)} \n#{File.expand_path(m_file)}".green { h_file: h_file, m_file: m_file, s_file: service_file } end def push_to_git Actions.sh('git checkout master', log: false) Actions.sh('git add .', log: false) Actions.sh("git commit -m 'generate yapi code'", log: false) Actions.sh('git push', log: false) end def notify_robot(template_vars) robot_url = 'https://oapi.dingtalk.com/robot/send?access_token=fe879fd3e7a3b5e59d5719b2384845b7884901919be5a78fe443cbf777869807' template = Lhj::ErbTemplateHelper.load('oc_code_notify') output = Lhj::ErbTemplateHelper.render(template, template_vars, '-') Lhj::Dingtalk.post_message_robot(robot_url, '生成代码', output) end def url_str "#{@http_url}#{api_id}" end def yml_file File.join(Lhj::Config.instance.home_dir, 'yapi.yml') end def load_config(file) config = YAML.load_file(file) 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'] @config_id = config['id'] @config_model_pre = config['model_pre'] @config_model_suffix = config['model_suffix'] @config_model_names = config['model_names'] @type_trans = config['type_trans'] end def api_id @id || @config_id.to_s end def model_pre @model_pre_name || @config_model_pre end def model_suffix @config_model_suffix || @model_default_suffix end def req_api_model uri = URI.parse(url_str) 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 res_json = JSON.parse(res.body) puts res.body unless res_json['errcode'].to_i.zero? puts res.body if @debug res_json end def print_res_body_model(res_json) res_body = fetch_res_boy(res_json) return unless res_body puts "\n<===============打印返回数据模型-Begin=====================>\n".green models = [] handle_model(res_body) do |model| models << model end case @language when 'oc' print_models(models) print_models_impl(models) when 'java' print_models_for_java(models) end puts "\n<===============打印返回数据模型-End=====================>\n".green models.last[:name] end def print_req_body_model(res_json) req_body = fetch_req_body(res_json) return unless req_body puts "\n<===============打印请求模型-Begin=====================>\n".green models = [] handle_model(req_body) do |model| models << model end case @language when 'oc' print_models(models) print_models_impl(models) when 'java' print_models_for_java(models) end puts "\n<===============打印请求模型-End=====================>\n".green models.last[:name] end def fetch_res_boy(res_json) return if !res_json || !res_json['data'] || !res_json['data']['res_body'] res_body = JSON.parse(res_json['data']['res_body']) return if !res_body['properties'] || !res_body['properties']['detailMsg'] result = res_body['properties']['detailMsg'] return unless result['type'] == 'object' || result['type'] == 'array' result['name'] = gen_model_name('') result end def fetch_req_body(res_json) return if !res_json || !res_json['data'] || !res_json['data']['req_body_other'] result = JSON.parse(res_json['data']['req_body_other']) result['name'] = gen_model_name('') result end def gen_model_name(name) n = name.gsub(/vo|model|list/i, '').gsub(/(.*)s$/, '\1').gsub(/^\w/) { $&.upcase } if n.length <= 0 n = @config_model_names.detect { |c| @model_names.none? { |na| na.gsub(/#{model_pre}(.*)Model/, '\1').eql?(c) } } end model_name = "#{model_pre}#{n}#{model_suffix}" @model_names << model_name model_name end def handle_model(model, &block) p_type = model['type'] p_name = model['name'] p_properties = model['properties'] p_model = { name: p_name } properties = [] case p_type when 'object' p_properties.each do |k, v| c_type = @type_trans[v['type']] || v['type'] c_model = { key: k, type: c_type, description: v['description'], default: '' } if v['type'].eql?('object') || v['type'].eql?('array') o = v['items'] || v o['name'] = gen_model_name(k) if v['type'].eql?('array') && v['items']['type'].eql?('string') c_model[:type_name] = 'NSString' else c_model[:type_name] = o['name'] handle_model(o, &block) end end properties << c_model end p_model[:properties] = properties block[p_model] if block_given? when 'array' t = model['items'] t['name'] = p_name handle_model(t, &block) end end def print_models(models) models.each do |model| model_name = model[:name] || '' model_properties = model[:properties] puts_h "@interface #{model_name} : NSObject" model_properties.each do |m| print_model(m) end puts_h "@end\n\n\n" end end def print_models_for_java(models) models.each do |model| model_name = model[:name] || '' model_properties = model[:properties] puts_h "public class #{model_name} extends BaseModel implements BusinessEvent.ProductDataCollect {" model_properties.each do |m| print_model_for_java(m) end puts_h "}\n\n\n" end end def print_models_impl(models) models.each do |model| puts_m "@implementation #{model[:name]}" str = model[:properties].filter { |p| p[:type].eql?('array') && !p[:type_name].eql?('NSString') }.map { |p| "@\"#{p[:key]}\": #{p[:type_name]}.class" }.join(', ') if str&.length&.positive? puts_m '+(NSDictionary *)modelContainerPropertyGenericClass {' puts_m " return @{#{str}};" puts_m '}' end puts_m "@end\n" puts "\n\n" end end def print_model(m) key = m[:key] type_name = m[:type_name] type = m[:type] des = m[:description] || '' des.gsub!(/\n/, ' ') default = m[:default] puts_h "///#{des} #{default}" case type when 'integer' puts_h "@property (nonatomic, assign) NSInteger #{key};" when 'string' puts_h "@property (nonatomic, copy) NSString *#{key};" when 'number' puts_h "@property (nonatomic, strong) NSNumber *#{key};" when 'float' puts_h "@property (nonatomic, assign) CGFloat #{key};" when 'double' puts_h "@property (nonatomic, assign) double #{key};" when 'boolean' puts_h "@property (nonatomic, assign) BOOL #{key};" when 'object' puts_h "@property (nonatomic, strong) #{type_name} *#{key};" when 'array' puts_h "@property (nonatomic, strong) NSArray<#{type_name} *> *#{key};" else puts_h "@property (nonatomic, copy) NSString *#{key};" end end def print_model_for_java(m) key = m[:key] type_name = m[:type_name] type = m[:type] des = m[:description] || '' des.gsub!(/\n/, ' ') default = m[:default] case type when 'integer' puts_h "int #{key};//#{des} #{default}" when 'string' puts_h "String #{key};//#{des} #{default}" when 'number' puts_h "double #{key};//#{des} #{default}" when 'float' puts_h "double #{key};//#{des} #{default}" when 'double' puts_h "double #{key};//#{des} #{default}" when 'boolean' puts_h "boolean #{key};//#{des} #{default}" when 'object' puts_h "#{type_name} #{key};//#{des} #{default}" when 'array' puts_h "#{type_name}[] #{key};//#{des} #{default}" else puts_h "String #{key};//#{des} #{default}" end end # @param [Object] data def print_req_query(data) return unless data && data['req_query'] case @language when 'oc' puts_h '@interface MLParamModel : NSObject' data['req_query'].each do |h| des = h['desc'] puts_h "///#{des} #{h['example']}" puts_h "@property (nonatomic, copy) NSString *#{h['name']};" end puts_h '@end' puts "\n\n" when 'java' end end def print_http_method(data, result_model_name, param_model_name) return unless data path = data['path'] path_name = path.split('/').map { |s| s.gsub(/[^A-Za-z0-9]/, '').gsub(/^\w/) { $&.upcase } }.join('') if path path_key = "k#{path_name}URL" mth = data['method'] mth = 'JSON' if data['req_body_is_json_schema'] case @language when 'oc' puts "\n<===============方法调用=====================>\n".green model_temp = Lhj::ErbTemplateHelper.load('oc_code_service_inner') model_temp_result = Lhj::ErbTemplateHelper.render(model_temp, { result_model_name: result_model_name }, '-') yapi_temp = Lhj::ErbTemplateHelper.load('oc_code_service') yapi_vars = { title: data['title'], desc: data['desc'], username: data['username'], path: path, path_key: path_key, path_name: path_name, result_model_name: result_model_name, param_model_name: param_model_name || 'MLParamModel', mth: mth, model_template: model_temp_result } yapi_temp_result = Lhj::ErbTemplateHelper.render(yapi_temp, yapi_vars, '-') puts yapi_temp_result.green return yapi_temp_result when 'java' end end end end end