[CI] Updates generate code task for automation

Author: picandocodigo

.ci/make.sh

@@ -63,14 +63,8 @@ case $CMD in
         TASK_ARGS=("$VERSION" "$output_folder")
         ;;
     codegen)
-        if [ -v $VERSION ]; then
-            echo -e "\033[31;1mTARGET: codegen -> missing version parameter\033[0m"
-            exit 1
-        fi
-        echo -e "\033[36;1mTARGET: codegen API v$VERSION\033[0m"
         TASK=codegen
-        # VERSION is BRANCH here for now
-        TASK_ARGS=("$VERSION" "$codegen_folder")
+        TASK_ARGS=()
         ;;
     docsgen)
         if [ -v $VERSION ]; then
@@ -154,10 +148,6 @@ if [[ "$CMD" == "assemble" ]]; then
 	  fi
 fi
 
-if [[ "$CMD" == "codegen" ]]; then
-    echo "TODO"
-fi
-
 if [[ "$CMD" == "docsgen" ]]; then
     echo "TODO"
 fi

Rakefile

@@ -130,8 +130,3 @@ task :release do
     puts '-' * 80
   end
 end
-
-desc 'Generate API code'
-task :generate do
-  sh "cd #{CURRENT_PATH.join('elasticsearch-api/utils')} && bundle exec thor code:generate"
-end

rake_tasks/elasticsearch_tasks.rake

@@ -82,10 +82,10 @@ namespace :elasticsearch do
   end
 
   desc 'Download artifacts (tests and REST spec) for currently running cluster'
-  task :download_artifacts do
+  task :download_artifacts, :version do |_, args|
     json_filename = CURRENT_PATH.join('tmp/artifacts.json')
 
-    unless (version_number = ENV['STACK_VERSION'])
+    unless (version_number = args[:version] || ENV['STACK_VERSION'])
       # Get version number and build hash of running cluster:
       version_number = cluster_info['number']
       build_hash = cluster_info['build_hash']

rake_tasks/unified_release_tasks.rake

@@ -57,6 +57,15 @@ namespace :unified_release do
        "zip -r #{@zip_filename}.zip * " \
   end
 
+  desc 'Generate API code'
+  task :codegen do
+    version = YAML.load_file(File.expand_path(__dir__ + '/../.ci/test-matrix.yml'))['STACK_VERSION'].first
+    Rake::Task['elasticsearch:download_artifacts'].invoke(version)
+    sh "cd #{CURRENT_PATH.join('elasticsearch-api/utils')} \
+          && BUNDLE_GEMFILE=`pwd`/Gemfile \
+          && bundle exec thor code:generate"
+  end
+
   desc <<-DESC
   Update Rubygems versions in version.rb and *.gemspec files