diff --git a/app/models/build.js b/app/models/build.js
index 07f20bf2..5f021229 100644
--- a/app/models/build.js
+++ b/app/models/build.js
@@ -4,8 +4,22 @@ import Ember from 'ember';
import Model from 'travis/models/model';
import DurationCalculations from 'travis/utils/duration-calculations';
import DS from 'ember-data';
+import Config from 'travis/config/environment';
-export default Model.extend(DurationCalculations, {
+var Build;
+
+if (Config.useV3API) {
+ Build = DS.Model.extend({
+ branch: DS.belongsTo('branch', { async: false, inverse: 'builds' }),
+ branchName: Ember.computed.alias('branch.name')
+ });
+} else {
+ Build = DS.Model.extend({
+ branchName: Ember.computed.alias('commit.branch')
+ });
+}
+
+Build.reopen({
ajax: Ember.inject.service(),
state: DS.attr(),
number: DS.attr('number'),
@@ -18,7 +32,6 @@ export default Model.extend(DurationCalculations, {
pullRequestTitle: DS.attr(),
pullRequestNumber: DS.attr('number'),
eventType: DS.attr('string'),
- branch: DS.belongsTo('branch', { async: false, inverse: 'builds' }),
repo: DS.belongsTo('repo', { async: true }),
commit: DS.belongsTo('commit', { async: false }),
jobs: DS.hasMany('job', { async: true }),
@@ -124,4 +137,7 @@ export default Model.extend(DurationCalculations, {
return moment(finishedAt).format('lll');
}
}.property('finishedAt')
+
});
+
+export default Build;
diff --git a/app/models/job.js b/app/models/job.js
index 5f2686a4..5472f222 100644
--- a/app/models/job.js
+++ b/app/models/job.js
@@ -24,6 +24,7 @@ export default Model.extend(DurationCalculations, {
build: DS.belongsTo('build', { async: true }),
commit: DS.belongsTo('commit', { async: true }),
branch: Ember.computed.alias('build.branch'),
+ branchName: Ember.computed.alias('build.branchName'),
log: function() {
this.set('isLogAccessed', true);
diff --git a/app/templates/components/build-header.hbs b/app/templates/components/build-header.hbs
index 15a1fc88..2d35e48c 100644
--- a/app/templates/components/build-header.hbs
+++ b/app/templates/components/build-header.hbs
@@ -5,7 +5,7 @@
Pull Request #{{item.pullRequestNumber}}
{{item.pullRequestTitle}}
{{else}}
- {{item.branch.name}}
+ {{item.branchName}}
{{format-message commit.subject repo=item.repo}}
{{/if}}