Add validator

2 files changed, 26 insertions, 0 deletions

diff --git a/app/validators/cron_validator.rb b/app/validators/cron_validator.rb
new file mode 100644
index 00000000000..ad70e0897ba
--- /dev/null
+++ b/app/validators/cron_validator.rb
@@ -0,0 +1,16 @@
+# CronValidator
+#
+# Custom validator for Cron.
+class CronValidator < ActiveModel::EachValidator
+  def validate_each(record, attribute, value)
+    cron_parser = Ci::CronParser.new(record.cron, record.cron_time_zone)
+    is_valid_cron, is_valid_cron_time_zone = cron_parser.validation
+    next_time = cron_parser.next_time_from(Time.now)
+
+    if !is_valid_cron
+      record.errors.add(:cron, " is invalid syntax")
+    elsif !is_valid_cron_time_zone
+      record.errors.add(:cron_time_zone, " is invalid timezone")
+    end
+  end
+end
diff --git a/app/validators/ref_validator.rb b/app/validators/ref_validator.rb
new file mode 100644
index 00000000000..2024255a770
--- /dev/null
+++ b/app/validators/ref_validator.rb
@@ -0,0 +1,10 @@
+# RefValidator
+#
+# Custom validator for Ref.
+class RefValidator < ActiveModel::EachValidator
+  def validate_each(record, attribute, value)
+    unless record.project.repository.branch_exists?(value)
+      record.errors.add(attribute, " does not exist")
+    end
+  end
+end