yaml migrator: only section

1 files changed, 9 insertions, 5 deletions

diff --git a/db/migrate/20150601043227_migrate_jobs_to_yaml.rb b/db/migrate/20150601043231_migrate_jobs_to_yaml.rb
index 56570e2..1fa4cf7 100644
--- a/db/migrate/20150601043227_migrate_jobs_to_yaml.rb
+++ b/db/migrate/20150601043231_migrate_jobs_to_yaml.rb
@@ -53,11 +53,15 @@ class MigrateJobsToYaml < ActiveRecord::Migration
           tags: job["tags"] && job["tags"].split(",").map(&:strip)
         }
 
-        except = build_except_param(parse_boolean_value(job["build_branches"]), parse_boolean_value(job["build_tags"]))
-        except = except + skip_refs
-
-        if except.any?
-          config[job["name"].to_s][:except] = except
+        if job["refs"].present?
+          config[job["name"].to_s][:only] = job["refs"].split(",").map(&:strip)
+        else
+          except = build_except_param(parse_boolean_value(job["build_branches"]), parse_boolean_value(job["build_tags"]))
+          except = except + skip_refs
+
+          if except.any?
+            config[job["name"].to_s][:except] = except
+          end
         end
       end