Build page. Status methods for build

4 files changed, 43 insertions, 8 deletions

diff --git a/lib/build.rb b/lib/build.rb
index b6a4377..c004425 100644
--- a/lib/build.rb
+++ b/lib/build.rb
@@ -1,3 +1,31 @@
 class Build < ActiveRecord::Base
   belongs_to :project
+
+  def failed?
+    status == 'fail'
+  end
+
+  def success?
+    status == 'success'
+  end
+
+  def running?
+    status == 'running'
+  end
+
+  def success!
+    update_status 'success'
+  end
+
+  def fail!
+    update_status 'fail'
+  end
+
+  def running!
+    update_status 'running'
+  end
+
+  def update_status status
+    update_attributes(status: status)
+  end
 end
diff --git a/lib/helper.rb b/lib/helper.rb
index f4b05ad..9660f21 100644
--- a/lib/helper.rb
+++ b/lib/helper.rb
@@ -10,11 +10,21 @@ module Helper
   end
 
   def build_status_class build
-    if build.status == 'success'
+    if build.success?
       'label-success'
-    elsif build.status == 'fail'
+    elsif build.failed?
       'label-important'
     else
+      'label-inverse'
+    end
+  end
+
+  def build_status_alert_class build
+    if build.success?
+      'alert-success'
+    elsif build.failed?
+      'alert-error'
+    else
       ''
     end
   end
diff --git a/lib/project.rb b/lib/project.rb
index 962b3c5..727246e 100644
--- a/lib/project.rb
+++ b/lib/project.rb
@@ -10,7 +10,7 @@ class Project < ActiveRecord::Base
   def register_build opts={}
     default_opts = {
       project_id: self.id,
-      status: 'runing'
+      status: 'running'
     }
 
     allowed_opts = {}
diff --git a/lib/runner.rb b/lib/runner.rb
index 557b596..bb76a2a 100644
--- a/lib/runner.rb
+++ b/lib/runner.rb
@@ -21,15 +21,12 @@ class Runner
     Dir.chdir(path) do
       commands.each_line do |line|
         unless command(line, path)
-          build.update_attributes(status: 'fail')
+          build.failed!
         end
       end
     end
 
-    build.update_attributes(
-      trace: @output,
-      status: 'success'
-    )
+    build.success!
   end
 
   def command(cmd, path)