From 32f3b643b009bda6e4c8bb654aa9d6a855967451 Mon Sep 17 00:00:00 2001
From: Angel Perez <iangel.p93@gmail.com>
Date: Wed, 4 Oct 2017 17:25:51 -0400
Subject: [PATCH] Create 'poll_question_answer_videos' table and model

---
 app/models/poll/question/answer.rb               |  1 +
 app/models/poll/question/answer/video.rb         | 16 ++++++++++++++++
 ...4210108_create_poll_question_answer_videos.rb | 11 +++++++++++
 db/schema.rb                                     | 11 ++++++++++-
 4 files changed, 38 insertions(+), 1 deletion(-)
 create mode 100644 app/models/poll/question/answer/video.rb
 create mode 100644 db/migrate/20171004210108_create_poll_question_answer_videos.rb

diff --git a/app/models/poll/question/answer.rb b/app/models/poll/question/answer.rb
index 3f8f4a172..cb43e2d44 100644
--- a/app/models/poll/question/answer.rb
+++ b/app/models/poll/question/answer.rb
@@ -1,5 +1,6 @@
 class Poll::Question::Answer < ActiveRecord::Base
   belongs_to :question, class_name: 'Poll::Question', foreign_key: 'question_id'
+  has_many :videos, class_name: 'Poll::Question::Answer::Video'
 
   validates :title, presence: true
 
diff --git a/app/models/poll/question/answer/video.rb b/app/models/poll/question/answer/video.rb
new file mode 100644
index 000000000..3d214af96
--- /dev/null
+++ b/app/models/poll/question/answer/video.rb
@@ -0,0 +1,16 @@
+class Poll::Question::Answer::Video < ActiveRecord::Base
+  belongs_to :answer, class_name: 'Poll::Question::Answer', foreign_key: 'answer_id'
+
+  VIMEO_REGEX = /vimeo.*(staffpicks\/|channels\/|videos\/|video\/|\/)([^#\&\?]*).*/
+  YOUTUBE_REGEX = /youtu.*(be\/|v\/|u\/\w\/|embed\/|watch\?v=|\&v=)([^#\&\?]*).*/
+
+  validates :title, presence: true
+  validate :valid_url?
+
+  def valid_url?
+    return if url.blank?
+    return if url.match(VIMEO_REGEX)
+    return if url.match(YOUTUBE_REGEX)
+    errors.add(:url, :invalid)
+  end
+end
diff --git a/db/migrate/20171004210108_create_poll_question_answer_videos.rb b/db/migrate/20171004210108_create_poll_question_answer_videos.rb
new file mode 100644
index 000000000..7cce33b29
--- /dev/null
+++ b/db/migrate/20171004210108_create_poll_question_answer_videos.rb
@@ -0,0 +1,11 @@
+class CreatePollQuestionAnswerVideos < ActiveRecord::Migration
+  def change
+    create_table :poll_question_answer_videos do |t|
+      t.string :title
+      t.string :url
+      t.integer :answer_id, index: true
+    end
+
+    add_foreign_key :poll_question_answer_videos, :poll_question_answers, column: :answer_id
+  end
+end
diff --git a/db/schema.rb b/db/schema.rb
index dbbd3ec46..22e362a1f 100644
--- a/db/schema.rb
+++ b/db/schema.rb
@@ -11,7 +11,7 @@
 #
 # It's strongly recommended that you check this file into your version control system.
 
-ActiveRecord::Schema.define(version: 20171004151553) do
+ActiveRecord::Schema.define(version: 20171004210108) do
 
   # These are extensions that must be enabled in order to support this database
   enable_extension "plpgsql"
@@ -668,6 +668,14 @@ ActiveRecord::Schema.define(version: 20171004151553) do
   add_index "poll_partial_results", ["origin"], name: "index_poll_partial_results_on_origin", using: :btree
   add_index "poll_partial_results", ["question_id"], name: "index_poll_partial_results_on_question_id", using: :btree
 
+  create_table "poll_question_answer_videos", force: :cascade do |t|
+    t.string  "title"
+    t.string  "url"
+    t.integer "answer_id"
+  end
+
+  add_index "poll_question_answer_videos", ["answer_id"], name: "index_poll_question_answer_videos_on_answer_id", using: :btree
+
   create_table "poll_question_answers", force: :cascade do |t|
     t.string  "title"
     t.text    "description"
@@ -1152,6 +1160,7 @@ ActiveRecord::Schema.define(version: 20171004151553) do
   add_foreign_key "poll_partial_results", "poll_officer_assignments", column: "officer_assignment_id"
   add_foreign_key "poll_partial_results", "poll_questions", column: "question_id"
   add_foreign_key "poll_partial_results", "users", column: "author_id"
+  add_foreign_key "poll_question_answer_videos", "poll_question_answers", column: "answer_id"
   add_foreign_key "poll_question_answers", "poll_questions", column: "question_id"
   add_foreign_key "poll_questions", "polls"
   add_foreign_key "poll_questions", "proposals"