X-Git-Url: https://irasnyder.com/gitweb/?a=blobdiff_plain;f=app%2Fmodels%2Fvideo.rb;h=fd5d22f130a57ee3383a4dbed076a42362959ad6;hb=50e2ce05003eb4a56d1daa8e030a2f477aeb59bc;hp=4ecf31b53c8891e61d2e94658ebd53886d65b8d5;hpb=0acc9549b65f2c3cd1840f39e31f2177a98bc55e;p=cs356-p2-videostore.git

diff --git a/app/models/video.rb b/app/models/video.rb
index 4ecf31b..fd5d22f 100644
--- a/app/models/video.rb
+++ b/app/models/video.rb
@@ -1,6 +1,44 @@
-class Video < ActiveRecord::Base
-  belongs_to :rentable
+class Video < Rentable
+  has_many :video_genres
+  has_many :video_medias
 
   validates_presence_of :director
-  validates_presence_of :genre
+  validates_presence_of :video_genre_id
+  validates_presence_of :video_media_id
+
+  def genre
+    VideoGenre.find_by_id(video_genre_id)
+  end
+
+  def media
+    VideoMedia.find_by_id(video_media_id)
+  end
+
+  def calculated_price
+    # FIXME: generate this based on day of week, newrelease
+    day_of_week = Time.now.to_date.wday
+    base_fee = VideoPolicy.find_by_day(day_of_week).fee
+
+    # Check for newrelease
+    newrelease_fee = newrelease ? VideoPolicy.find_by_day(8).fee : 0.00
+
+    return base_fee + newrelease_fee
+  end
+
+  def due_date
+    # FIXME: generate this based on the day of week, newrelease
+    # NOTE: a Date.wday will tell you the day of week (0-6, meaning Sunday-Saturday)
+    day_of_week = Time.now.to_date.wday
+    base_period = VideoPolicy.find_by_day(day_of_week).period
+    newrelease_period = newrelease ? VideoPolicy.find_by_day(8).period : 0
+
+    period = base_period + newrelease_period
+    return Time.now.advance(:days => period).to_date
+  end
+
+  protected
+  def validate
+    errors.add(:video_genre_id, "does not exist in the database") if genre.nil?
+    errors.add(:video_media_id, "does not exist in the database") if media.nil?
+  end
 end