[DOCS] Updates DOCS generator

Author: picandocodigo

rake_tasks/doc_generator.rake

@@ -17,6 +17,7 @@
 
 require 'json'
 require 'fileutils'
+require 'logger'
 
 namespace :docs do
   SRC_FILE          = "#{__dir__}/docs/parsed_alternative_report.json".freeze
@@ -29,26 +30,33 @@ namespace :docs do
     FileUtils.remove_dir(TARGET_DIR)
     Dir.mkdir(TARGET_DIR)
 
-    # Only select the files in the EXAMPLES_TO_PARSE array and that are not
-    # console result examples
-    entries = json_data.select do |d|
-      EXAMPLES_TO_PARSE.include? d['source_location']['file']
-    end.select { |d| d['lang'] == 'console' }
-
-    entries.each do |entry|
+    entries = json_data.select { |d| d['lang'] == 'console' }
+    start_time = Time.now.to_i
+    entries.each_with_index do |entry, index|
+      percentage = index * 100 / entries.length
+      hourglass = index.even? ? '⌛ ' : '⏳ '
+      print "\r" + ("\e[A\e[K" * 2) if index > 0
+      puts "📝 Generating file #{index + 1} of #{entries.length} - #{percentage}% complete"
+      puts hourglass + '▩' * (percentage / 2) + '⬚' * (50 - percentage / 2) + ' ' + hourglass
       generate_docs(entry)
     end
+    puts "Finished generating #{entries.length} files in #{Time.now.to_i - start_time} seconds"
   end
 
   def json_data
     JSON.parse(File.read(SRC_FILE))
   end
 
   def generate_docs(entry)
-    file_name = "#{entry['digest']}.asciidoc"
-    api = entry['parsed_source'].first['api']
-    code = build_client_query(api, entry)
-    write_file(code, file_name)
+    require 'elasticsearch'
+
+    filename = "#{entry['digest']}.asciidoc"
+    unless entry['parsed_source'].empty?
+      api = entry['parsed_source'].first['api']
+      code = build_client_query(api, entry)
+      TestDocs::perform(code, filename)
+      write_file(code, filename)
+    end
   end
 
   def self.build_client_query(api, entry)
@@ -61,8 +69,12 @@ namespace :docs do
       request_body << show_parameters(params) if params
       body = entry&.[]('body')
       request_body << show_body(body) if body
-      request_body = request_body.compact.join(",\n")
-      code = "response = client.#{api}(\n#{request_body}\n)\nputs response"
+      request_body = request_body.compact.join(",\n").gsub('null', 'nil')
+      code = if api.include? '_internal'
+               "response = client.perform_request('#{entry['method']}', '#{api}', #{request_body})"
+             else
+               "response = client.#{api}(\n#{request_body}\n)\nputs response"
+             end
       client_query << format_code(code)
     end
     client_query.join("\n\n")
@@ -73,8 +85,12 @@ namespace :docs do
     File.open('temp.rb', 'w') do |f|
       f.puts code
     end
-    # Format code:
-    system("rubocop --config #{__dir__}/docs_rubocop_config.yml --format autogenconf -a ./temp.rb")
+    # Format code with Rubocop
+    require 'rubocop'
+    options = "--config #{__dir__}/docs_rubocop_config.yml -o /dev/null -a ./temp.rb".split
+    cli = RuboCop::CLI.new
+    cli.run(options)
+
     # Read it back
     template = File.read('./temp.rb')
     File.delete('./temp.rb')
@@ -120,3 +136,32 @@ namespace :docs do
     end
   end
 end
+
+#
+# Test module to run the generated code
+#
+module TestDocs
+  @formatter = -> (_, d, _, msg) { "[#{d}] : #{msg}" }
+
+  def self.perform(code, filename)
+    # Eval the example code, but remove printing out the response
+    response = eval(code.gsub('puts response', ''))
+    if response.status == 200
+      logger = Logger.new('log/200-ok.log')
+      logger.formatter = -> (_, _, _, msg) { "#{msg} " }
+      logger.info(filename)
+    end
+  rescue Elastic::Transport::Transport::Error => e
+    logger = Logger.new('log/docs-generation-elasticsearch.log')
+    logger.formatter = @formatter
+    logger.info("Located in #{filename}: #{e.message}\n")
+  rescue ArgumentError, NoMethodError, TypeError => e
+    logger = Logger.new('log/docs-generation-client.log')
+    logger.formatter = @formatter
+    logger.info("Located in #{filename}: #{e.message}\n")
+  end
+
+  def self.client
+    @client ||= Elasticsearch::Client.new(trace: false, log: false)
+  end
+end